gitlabhq/lib/gitosis.rb

require 'inifile'
require 'timeout'
require 'fileutils'

class Gitosis
  class AccessDenied < StandardError; end

  def pull
    # create tmp dir
    @local_dir = File.join(Dir.tmpdir,"gitlabhq-gitosis-#{Time.now.to_i}")

    Dir.mkdir @local_dir

    `git clone #{GITOSIS['admin_uri']} #{@local_dir}/gitosis`
  end

  def push
    Dir.chdir(File.join(@local_dir, "gitosis"))
    `git add -A`
    `git commit -am "Gitlab"`
    `git push`
    Dir.chdir(Rails.root)

    FileUtils.rm_rf(@local_dir)
  end

  def configure
    status = Timeout::timeout(20) do
      File.open(File.join(Dir.tmpdir,"gitlabhq-gitosis.lock"), "w+") do |f|
        begin
          f.flock(File::LOCK_EX)

          pull
          yield(self)
          push

        ensure
          f.flock(File::LOCK_UN)
        end
      end
    end
  rescue Exception => ex
    raise Gitosis::AccessDenied.new("gitosis timeout")
  end

  def destroy_project(project)
    FileUtils.rm_rf(project.path_to_repo)
    
    conf = IniFile.new(File.join(@local_dir,'gitosis','gitosis.conf'))

    conf.delete_section("group #{project.path}")

    conf.write
  end

   #update or create
  def update_keys(user, key)
    File.open(File.join(@local_dir, 'gitosis/keydir',"#{user}.pub"), 'w') {|f| f.write(key.gsub(/\n/,'')) }
  end

  def delete_key(user)
    File.unlink(File.join(@local_dir, 'gitosis/keydir',"#{user}.pub"))
    `cd #{File.join(@local_dir,'gitosis')} ; git rm keydir/#{user}.pub`
  end

  #update or create
  def update_project(repo_name, name_writers)
    # write config file
    conf = IniFile.new(File.join(@local_dir,'gitosis','gitosis.conf'))

    conf["group #{repo_name}"]['writable'] = repo_name
    conf["group #{repo_name}"]['members'] = name_writers.join(' ')

    conf.write
  end
end