Sutto-gitauth-gh 0.0.2.0 → 0.0.3.0

data/lib/git_hub_api.rb

@@ -8,6 +8,8 @@ class GitHubApi
   
   class Error < StandardError; end
   
+  class SSHPublicKey < Struct.new(:name, :id, :key); end
+  
   base_uri 'https://github.com'
   
   attr_accessor :user, :api_key
@@ -36,6 +38,12 @@ class GitHubApi
     hash_get(results, "repository")
   end
   
+  def keys
+    keys = hash_get(get("/user/keys"), "public_keys")
+    return false if keys.nil?
+    keys.map { |k| SSHPublicKey.new(k["title"], k["id"], k["key"]) }
+  end
+  
   def repository(name, user = @user)
     results = get("repos/show/#{user}/#{name}")
     repo = hash_get(results, "repository")
@@ -49,8 +57,6 @@ class GitHubApi
   
   class Repository
     
-    class SSHPublicKey < Struct.new(:name, :id); end
-    
     attr_accessor :name, :api, :attributes, :owner
     
     def initialize(attributes, api)
@@ -121,7 +127,7 @@ class GitHubApi
     def keys
       keys = api.hash_get(api.get("repos/keys/#{@name}"), "public_keys")
       return false if keys.nil?
-      keys.map { |k| SSHPublicKey.new(k["title"], k["id"]) }
+      keys.map { |k| SSHPublicKey.new(k["title"], k["id"], k["key"]) }
     end
     
     def add_key(title, key)
@@ -131,7 +137,7 @@ class GitHubApi
       })
       keys = api.hash_get(result, "public_keys")
       return false if keys.nil?
-      keys.map { |k| SSHPublicKey.new(k["title"], k["id"]) }
+      keys.map { |k| SSHPublicKey.new(k["title"], k["id"], k["key"]) }
     end
     
     def remove_key(key_id)
@@ -145,7 +151,7 @@ class GitHubApi
     
   end
   
-  # Methods
+  # Helper Methods
   
   def get(path, opts = {})
     self.class.get(full_path_for(path), :query => with_auth(opts))

data/lib/gitauth/gh_mirror.rb

@@ -1,4 +1,5 @@
 require 'gitauth'
+require 'digest/sha2'
 
 dir = Pathname.new(__FILE__).dirname.dirname
 $:.unshift(dir) unless $:.include?(dir)
@@ -9,9 +10,9 @@ module GitAuth
   class GitHubMirror
     include GitAuth::Loggable
     
-    Project = Struct.new(:name, :github_clone_url, :repository)
+    Project = Struct.new(:name, :github_clone_url, :repository, :github)
     
-    VERSION = [0, 0, 2, 0]
+    VERSION = [0, 0, 3, 0]
     
     def initialize(username, token)
       @api = GitHubApi.new(username, token)
@@ -20,12 +21,13 @@ module GitAuth
     def projects
       @projects ||= @api.repositories.map do |repository|
         local_repo = GitAuth::Repo.get(repository.name)
-        Project.new(repository.name, github_url_for_repo(repository), local_repo)
+        Project.new(repository.name, github_url_for_repo(repository), local_repo, repository)
       end
     end
     
     def mirror!(p)
       mirrored?(p) ? update!(p) : clone!(p) 
+      mirror_deploy_keys(p)
     end
     
     def update!(p)
@@ -56,6 +58,39 @@ module GitAuth
       end
     end
     
+    def mirror_deploy_keys(p)
+      if p.github.private?
+        p.github.keys.each do |key|
+          u = user_for_key(key)
+          p.repository.readable_by(u) 
+        end
+      end
+      GitAuth::Repo.save!
+    end
+    
+    def mirror_user_keys
+      users = []
+      @api.keys.each { |key| users << user_for_key(key) }
+      users.compact!
+      projects.each do |project|
+        users.each do |user|
+          project.repository.readable_by(user)
+        end
+      end
+      GitAuth::Repo.save!
+    end
+    
+    def update_public_project_authentication
+      users = GitAuth::User.all.select { |u| u.name =~ /^github-/i }
+      projects.each do |project|
+        next unless project.repository.public?
+        users.each do |user|
+          project.repository.readable_by(user)
+        end
+      end
+      GitAuth::Repo.save
+    end
+    
     class << self
       
       def run(options = {})
@@ -64,7 +99,12 @@ module GitAuth
         options.token = `git config --global github.token`.strip unless options.token?
         logger.info "Preparing to run GitHub mirror for #{options.user}"
         mirror = self.new(options.user, options.token)
+        logger.info "Mirroring all repositories"
         mirror.mirror_all
+        logger.info "Mirroring user keys"
+        mirror.mirror_user_keys
+        logger.info "Updating key access to public repositories"
+        mirror.update_public_project_authentication
       rescue Exception => e
         logger.fatal "Got Exception: #{e.class.name} - #{e.message}"
         e.backtrace.each { |l| logger.fatal "--> #{l}" }
@@ -87,6 +127,19 @@ module GitAuth
       repo.repository.present?
     end
     
+    def user_for_key(k)
+      name = "github-#{Digest::SHA256.hexdigest(k.key)[0, 8]}"
+      if u = GitAuth::User.get(name)
+        u
+      else
+        if GitAuth::User.create(name, false, k.key)
+          GitAuth::User.get(name)
+        else
+          raise "Unable to create user for key '#{k.key}'"
+        end
+      end
+    end
+    
   end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: Sutto-gitauth-gh
 version: !ruby/object:Gem::Version 
-  version: 0.0.2.0
+  version: 0.0.3.0
 platform: ruby
 authors: 
 - Darcy Laycock
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-09-17 00:00:00 -07:00
+date: 2009-09-18 00:00:00 -07:00
 default_executable: gitauth-gh
 dependencies: 
 - !ruby/object:Gem::Dependency