github-key-upload 0.0.1 → 0.0.2

data/bin/github-key-upload

@@ -3,16 +3,56 @@
 require 'github-key-upload'
 require 'slop'
 
-opts = Slop.parse(:longest_flag => 8, :help => true) do
+def handle_exit_status(key)
+  if key
+    puts "OK"
+  else
+    puts "NOK"
+    exit(1)
+  end
+end
+
+opts = Slop.parse(:longest_flag => 1, :help => true) do
   banner "github-key-upload [options]\n"
   on :k, :key=, "Path to the file containing the public key to upload"
   on :t, :title=, "Key Title"
   on :u, :user=, "Github username"
   on :P, :password=, "Github password"
+  on :l, "--check-title", "Checks if the supplied key exists (by title)"
+  on :c, "--check-key", "Checks if the supplied key exists (by key)"
+  on :C, "--check-both", "Checks if the supplied key exists (exact combination of key and title)"
+end
+
+exit if opts[:help]
+
+if opts[:k].nil? && (opts[:c] || opts[:C] || opts[:l].nil?)
+  puts "You need to specify a path to a key (-k or --key)"
+  need_to_break = true
+end
+
+if opts[:t].nil? && (opts[:l] || opts[:C] || opts[:c].nil?)
+  puts "You need to specify a key title (-t or --title)"
+  need_to_break = true
+end
+
+if opts[:u].nil? || opts[:P].nil?
+  puts "You need to specify both username and password (-u or --user and -P or --password)"
+  need_to_break = true
 end
 
-if (opts[:k].nil? || opts[:u].nil? || opts[:P].nil? || opts[:t].nil?)
+if need_to_break
   puts opts.help
+  exit
+end
+
+if opts[:l].nil? && opts[:c].nil? && opts[:C].nil?
+  Github::Key::Upload.upload opts.to_hash
+elsif opts[:C] || (opts[:l] && opts[:c])
+  handle_exit_status(Github::Key::Upload.check_for_both opts.to_hash)
+elsif opts[:l]
+  handle_exit_status(Github::Key::Upload.check_for_title opts.to_hash)
+elsif opts[:c]
+  handle_exit_status(Github::Key::Upload.check_for_key opts.to_hash)
 else
-  Github::Key::Upload.upload(opts.to_hash)
+  puts opts.help
 end

data/lib/github-key-upload.rb

@@ -4,11 +4,63 @@ require "github_api"
 module Github
   module Key
     module Upload
-      def self.upload(options)
+      def self.create_auth(options)
         github = Github.new basic_auth: "#{options[:user]}:#{options[:password]}"
         github.oauth.create 'scopes' => ['user']
+        github
+      end
+
+      def self.upload(options)
+        github = create_auth(options)
         github.users.keys.create({title: options[:title], key: File.read(options[:key])})
       end
+
+      def self.check_for_both(options)
+        github = create_auth(options)
+        matching_key = nil
+
+        github.users.keys.all.each_page do |page|
+          if page.map {|k| [k["title"], k["key"]]}.include?(
+              [options[:title], File.read(options[:key]).split(" ")[0,2].join(" ")])
+            matching_key = true
+            break
+          end
+        end
+
+        matching_key
+      end
+
+      def self.check(options)
+        github = create_auth(options)
+        matching_key = nil
+        search_key = options[:search_key].to_s
+        search_proc = options[:search_proc]
+
+        github.users.keys.all.each_page do |page|
+          if page.map {|k| k[search_key]}.include?(search_proc.call(options))
+            matching_key = true
+            break
+          end
+        end
+
+        matching_key
+      end
+
+      def self.check_for_key(options)
+        check(options.merge({
+                              :search_key => :key,
+                              :search_proc => Proc.new {|opts|
+                                File.read(opts[:key]).split(" ")[0,2].join(" ")
+                              }
+                            }))
+      end
+
+      def self.check_for_title(options)
+        check(options.merge({
+                              :search_key => :title,
+                              :search_proc => Proc.new {|opts| opts[:title]}
+                            }))
+      end
     end
   end
 end

data/lib/github-key-upload/version.rb

@@ -1,7 +1,7 @@
 module Github
   module Key
     module Upload
-      VERSION = "0.0.1"
+      VERSION = "0.0.2"
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: github-key-upload
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-10-26 00:00:00.000000000 Z
+date: 2012-10-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: github_api