gitolite & gitosis support

lib/gitlabhq/.directory

@@ -0,0 +1,5 @@
+[Dolphin]
+AdditionalInfoV2=Details_Size,Details_Date,CustomizedDetails
+Timestamp=2011,12,4,1,34,17
+Version=2
+ViewMode=1

lib/gitlabhq/git_host.rb

@@ -0,0 +1,18 @@
+require File.join(Rails.root, "lib", "gitlabhq", "gitolite")
+require File.join(Rails.root, "lib", "gitlabhq", "gitosis")
+
+module Gitlabhq
+  class GitHost
+    def self.system
+      if GIT_HOST["system"] == "gitolite"
+        Gitlabhq::Gitolite
+      else 
+        Gitlabhq::Gitosis
+      end
+    end
+
+    def self.admin_uri
+      GIT_HOST["admin_uri"]
+    end
+  end
+end

lib/gitlabhq/gitolite.rb

@@ -0,0 +1,80 @@
+require 'gitolite'
+require 'timeout'
+require 'fileutils'
+
+module Gitlabhq
+  class Gitolite
+    class AccessDenied < StandardError; end
+
+    def pull
+      # create tmp dir
+      @local_dir = File.join(Dir.tmpdir,"gitlabhq-gitolite-#{Time.now.to_i}")
+      Dir.mkdir @local_dir
+
+      `git clone #{GitHost.admin_uri} #{@local_dir}/gitolite`
+    end
+
+    def push
+      Dir.chdir(File.join(@local_dir, "gitolite"))
+      `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-gitolite.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 Gitolite::AccessDenied.new("gitolite timeout")
+    end
+
+    def destroy_project(project)
+      `sudo -u git rm -rf #{project.path_to_repo}`
+      
+      ga_repo = ::Gitolite::GitoliteAdmin.new(File.join(@local_dir,'gitolite'))
+      conf = ga_repo.config
+      conf.rm_repo(project.path)
+      ga_repo.save
+    end
+
+     #update or create
+    def update_keys(user, key)
+      File.open(File.join(@local_dir, 'gitolite/keydir',"#{user}.pub"), 'w') {|f| f.write(key.gsub(/\n/,'')) }
+    end
+
+    def delete_key(user)
+      File.unlink(File.join(@local_dir, 'gitolite/keydir',"#{user}.pub"))
+      `cd #{File.join(@local_dir,'gitolite')} ; git rm keydir/#{user}.pub`
+    end
+
+    # update or create
+    def update_project(repo_name, name_writers)
+      ga_repo = ::Gitolite::GitoliteAdmin.new(File.join(@local_dir,'gitolite'))
+      conf = ga_repo.config
+
+      repo = if conf.has_repo?(repo_name)
+               conf.get_repo(repo_name)
+             else 
+               ::Gitolite::Config::Repo.new(repo_name)
+             end
+
+      repo.add_permission("RW+", "", name_writers) unless name_writers.blank?
+      conf.add_repo(repo)
+
+      ga_repo.save
+    end
+  end
+end

lib/gitlabhq/gitosis.rb

@@ -0,0 +1,76 @@
+require 'inifile'
+require 'timeout'
+require 'fileutils'
+
+module Gitlabhq
+  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 #{GitHost.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)
+      `sudo -u git 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
+end