nex3-github 0.1.2 → 0.1.3

data/Manifest

@@ -1,17 +1,17 @@
 bin/github
 commands/commands.rb
 commands/helpers.rb
+lib/extensions.rb
 lib/github/command.rb
 lib/github/helper.rb
 lib/github.rb
 LICENSE
+Manifest
 README
 spec/command_spec.rb
-spec/helpers/owner_spec.rb
-spec/helpers/project_spec.rb
-spec/helpers/public_url_for_spec.rb
-spec/helpers/repo_for_spec.rb
-spec/helpers/user_and_repo_from_spec.rb
-spec/helpers/user_for_spec.rb
+spec/extensions_spec.rb
+spec/github_spec.rb
+spec/helper_spec.rb
 spec/spec_helper.rb
-Manifest
+spec/ui_spec.rb
+spec/windoze_spec.rb

data/README

@@ -51,3 +51,4 @@ Contributors
 - defunkt
 - maddox
 - halorgium
+- kballard

data/commands/commands.rb

@@ -1,66 +1,95 @@
-GitHub.register :helper do |name, comma_args|
-  comma_args ||= ''
-  puts helper.send(name, comma_args.split(/,/))
+desc "Open this repo's master branch in a web browser."
+command :home do |user|
+  if helper.project
+    helper.open helper.homepage_for(user || helper.owner, 'master')
+  end
 end
 
-GitHub.describe :home => "Open this repo's master branch in a web browser."
-GitHub.register :home do
+desc "Open this repo in a web browser."
+command :browse do |user, branch|
   if helper.project
-    exec "#{helper.open} #{helper.homepage_for(helper.owner, 'master')}"
+    # if one arg given, treat it as a branch name
+    # unless it maches user/branch, then split it
+    # if two args given, treat as user branch
+    # if no args given, use defaults
+    user, branch = user.split("/", 2) if branch.nil? unless user.nil?
+    branch = user and user = nil if branch.nil?
+    user ||= helper.branch_user
+    branch ||= helper.branch_name
+    helper.open helper.homepage_for(user, branch)
   end
 end
 
-GitHub.describe :browse => "Open this repo in a web browser."
-GitHub.register :browse do
+desc "Open the network page for this repo in a web browser."
+command :network do |user|
   if helper.project
-    exec "#{helper.open} #{helper.homepage_for(helper.branch_user, helper.branch_name)}"
+    user ||= helper.owner
+    helper.open helper.network_page_for(user)
   end
 end
 
-GitHub.describe :info => "Info about this project."
-GitHub.register :info do
+desc "Info about this project."
+command :info do
   puts "== Info for #{helper.project}"
   puts "You are #{helper.owner}"
-  puts "Currently tracking: "
+  puts "Currently tracking:"
   helper.tracking.each do |(name,user_or_url)|
     puts " - #{user_or_url} (as #{name})"
   end
 end
 
-GitHub.describe :track => "Track another user's repository."
-GitHub.register :track do |user|
+desc "Track another user's repository."
+flags :private => "Use git@github.com: instead of git://github.com/."
+command :track do |user|
   die "Specify a user to track" if user.nil?
   die "Already tracking #{user}" if helper.tracking?(user)
 
-  git "remote add #{user} #{helper.public_url_for(user)}"
+  if options[:private]
+    git "remote add #{user} #{helper.private_url_for(user)}"
+  else
+    git "remote add #{user} #{helper.public_url_for(user)}"
+  end
 end
 
-GitHub.describe :pull => "Pull from a remote.  Pass --merge to automatically merge remote's changes into your master."
-GitHub.register :pull do |user, branch|
+desc "Pull from a remote."
+flags :merge => "Automatically merge remote's changes into your master."
+command :pull do |user, branch|
   die "Specify a user to pull from" if user.nil?
-  GitHub.invoke(:track, user) unless helper.tracking?(user)
+  user, branch = user.split("/", 2) if branch.nil?
   branch ||= 'master'
-
-  puts "Switching to #{user}/#{branch}"
-  git "checkout #{user}/#{branch}" if git("checkout -b #{user}/#{branch}").error? 
+  GitHub.invoke(:track, user) unless helper.tracking?(user)
   
   if options[:merge]
-    git "pull #{user} #{branch}"
-    git "checkout master"
-    git_exec "merge #{user}/#{branch}"
+    git_exec "pull #{user} #{branch}"
   else
+    puts "Switching to #{user}/#{branch}"
+    git "checkout #{user}/#{branch}" if git("checkout -b #{user}/#{branch}").error?
     git_exec "pull #{user} #{branch}"
   end
 end
 
-GitHub.describe :clone => "Clone a repo.  Pass --ssh to clone from your own git@github.com schema."
-GitHub.register :clone do |user, repo|
+desc "Clone a repo."
+flags :ssh => "Clone using the git@github.com style url."
+command :clone do |user, repo, dir|
+  die "Specify a user to pull from" if user.nil?
   user, repo = user.split('/') unless repo
   die "Specify a repo to pull from" if repo.nil?
 
   if options[:ssh]
-    git_exec "clone git@github.com:#{user}/#{repo}.git"
+    git_exec "clone git@github.com:#{user}/#{repo}.git" + (dir ? " #{dir}" : "")
   else
-    git_exec "clone git://github.com/#{user}/#{repo}.git"
+    git_exec "clone git://github.com/#{user}/#{repo}.git" + (dir ? " #{dir}" : "")
+  end
+end
+
+desc "Generate the text for a pull request."
+command :'pull-request' do |user, branch|
+  if helper.project
+    die "Specify a user for the pull request" if user.nil?
+    user, branch = user.split('/', 2) if branch.nil?
+    branch ||= 'master'
+    GitHub.invoke(:track, user) unless helper.tracking?(user)
+
+    git_exec "request-pull #{user}/#{branch} origin"
   end
 end

data/commands/helpers.rb

@@ -1,26 +1,25 @@
-GitHub.helper :user_and_repo_from do |url|
+helper :user_and_repo_from do |url|
   case url
   when %r|^git://github\.com/([^/]+/[^/]+)$|: $1.split('/')
   when %r|^(?:ssh://)?(?:git@)?github\.com:([^/]+/[^/]+)$|: $1.split('/')
-  else ['', '']
   end
 end
 
-GitHub.helper :user_and_repo_for do |remote|
+helper :user_and_repo_for do |remote|
   user_and_repo_from(url_for(remote))
 end
 
-GitHub.helper :user_for do |remote|
-  user_and_repo_for(remote).first
+helper :user_for do |remote|
+  user_and_repo_for(remote).try.first
 end
 
-GitHub.helper :repo_for do |remote|
-  user_and_repo_for(remote).last
+helper :repo_for do |remote|
+  user_and_repo_for(remote).try.last
 end
 
-GitHub.helper :project do
+helper :project do
   repo = repo_for(:origin)
-  if repo == ""
+  if repo.nil?
     if url_for(:origin) == ""
       STDERR.puts "Error: missing remote 'origin'"
     else
@@ -31,38 +30,40 @@ GitHub.helper :project do
   repo.chomp('.git')
 end
 
-GitHub.helper :url_for do |remote|
+helper :url_for do |remote|
   `git config --get remote.#{remote}.url`.chomp
 end
 
-GitHub.helper :remotes do
+helper :remotes do
   regexp = '^remote\.(.+)\.url$'
-  `git config --get-regexp '#{regexp}'`.split(/\n/).map do |line|
+  `git config --get-regexp '#{regexp}'`.split(/\n/).inject({}) do |memo, line|
     name_string, url = line.split(/ /, 2)
     m, name = *name_string.match(/#{regexp}/)
-    [name, url]
+    memo[name.to_sym] = url
+    memo
   end
 end
 
-GitHub.helper :tracking do
-  remotes.map do |(name, url)|
+helper :tracking do
+  remotes.inject({}) do |memo, (name, url)|
     if ur = user_and_repo_from(url)
-      [name, ur.first]
+      memo[name] = ur.first
     else
-      [name, url]
+      memo[name] = url
     end
+    memo
   end
 end
 
-GitHub.helper :tracking? do |user|
-  tracking.include?(user)
+helper :tracking? do |user|
+  tracking.values.include?(user)
 end
 
-GitHub.helper :owner do
+helper :owner do
   user_for(:origin)
 end
 
-GitHub.helper :user_and_branch do
+helper :user_and_branch do
   raw_branch = `git rev-parse --symbolic-full-name HEAD`.chomp.sub(/^refs\/heads\//, '')
   user, branch = raw_branch.split(/\//, 2)
   if branch
@@ -72,23 +73,30 @@ GitHub.helper :user_and_branch do
   end
 end
 
-GitHub.helper :branch_user do
+helper :branch_user do
   user_and_branch.first
 end
 
-GitHub.helper :branch_name do
+helper :branch_name do
   user_and_branch.last
 end
 
-GitHub.helper :public_url_for do |user|
+helper :public_url_for do |user|
   "git://github.com/#{user}/#{project}.git"
 end
 
-GitHub.helper :homepage_for do |user, branch|
+helper :private_url_for do |user|
+  "git@github.com:#{user}/#{project}.git"
+end
+
+helper :homepage_for do |user, branch|
   "https://github.com/#{user}/#{project}/tree/#{branch}"
 end
 
-GitHub.helper :open do
-  Windoze ? 'start' : 'open'
+helper :network_page_for do |user|
+  "https://github.com/#{user}/#{project}/network"
+end
+
+helper :open do |url|
+  Launchy::Browser.new.visit url
 end
-    

data/lib/extensions.rb

@@ -0,0 +1,28 @@
+# define #try
+class Object
+  def try
+    self
+  end
+end
+
+class NilClass
+  klass = Class.new
+  klass.class_eval do
+    instance_methods.each { |meth| undef_method meth.to_sym unless meth =~ /^__(id|send)__$/ }
+    def method_missing(*args)
+      nil
+    end
+  end
+  NilProxy = klass.new
+  def try
+    NilProxy
+  end
+end
+
+# define #tap
+class Object
+  def tap(&block)
+    block.call(self)
+    self
+  end
+end

data/lib/github.rb

@@ -1,6 +1,9 @@
 $:.unshift File.dirname(__FILE__)
+require 'extensions'
 require 'github/command'
 require 'github/helper'
+require 'rubygems'
+require 'launchy'
 
 ##
 # Starting simple.
@@ -20,13 +23,22 @@ module GitHub
 
   BasePath = File.expand_path(File.dirname(__FILE__) + '/..')
 
-  def register(command, &block)
+  def command(command, &block)
     debug "Registered `#{command}`"
+    descriptions[command] = @next_description if @next_description
+    @next_description = nil
+    flag_descriptions[command].update @next_flags if @next_flags
+    @next_flags = nil
     commands[command.to_s] = Command.new(block)
   end
 
-  def describe(hash)
-    descriptions.update hash
+  def desc(str)
+    @next_description = str
+  end
+
+  def flags(hash)
+    @next_flags ||= {}
+    @next_flags.update hash
   end
 
   def helper(command, &block)
@@ -55,27 +67,38 @@ module GitHub
     @descriptions ||= {}
   end
 
+  def flag_descriptions
+    @flagdescs ||= Hash.new { |h, k| h[k] = {} }
+  end
+
   def options
     @options
   end
 
   def parse_options(args)
-    @debug = args.delete('--debug')
-    args.inject({}) do |memo, arg|
-      if arg =~ /^--([^=]+)=(.+)/
-        args.delete(arg)
+    idx = 0
+    args.clone.inject({}) do |memo, arg|
+      case arg
+      when /^--(.+?)=(.*)/
+        args.delete_at(idx)
         memo.merge($1.to_sym => $2)
-      elsif arg =~ /^--(.+)/
-        args.delete(arg)
+      when /^--(.+)/
+        args.delete_at(idx)
         memo.merge($1.to_sym => true)
+      when "--"
+        args.delete_at(idx)
+        return memo
       else
+        idx += 1
         memo
       end
     end
   end
 
   def load(file)
-    file[0] == ?/ ? super : super(BasePath + "/commands/#{file}")
+    file[0] == ?/ ? path = file : path = BasePath + "/commands/#{file}"
+    data = File.read(path)
+    GitHub.module_eval data, path
   end
 
   def debug(*messages)
@@ -87,13 +110,18 @@ module GitHub
   end
 end
 
-GitHub.register :default do
+GitHub.command :default do
   puts "Usage: github command <space separated arguments>", ''
   puts "Available commands:", ''
   longest = GitHub.descriptions.map { |d,| d.to_s.size }.max
   GitHub.descriptions.each do |command, desc|
-    command = "%-#{longest}s" % command
-    puts "  #{command} => #{desc}"
+    cmdstr = "%-#{longest}s" % command
+    puts "  #{cmdstr} => #{desc}"
+    flongest = GitHub.flag_descriptions[command].map { |d,| "--#{d}".size }.max
+    GitHub.flag_descriptions[command].each do |flag, fdesc|
+      flagstr = "#{" " * longest}  %-#{flongest}s" % "--#{flag}"
+      puts "  #{flagstr}: #{fdesc}"
+    end
   end
   puts
 end

data/lib/github/command.rb

@@ -1,6 +1,4 @@
-Windoze = RUBY_PLATFORM =~ /mswin|mingw/
-
-if Windoze
+if RUBY_PLATFORM =~ /mswin|mingw/
   begin
     require 'win32/open3'
   rescue LoadError

data/spec/command_spec.rb

@@ -0,0 +1,66 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe GitHub::Command do
+  before(:each) do
+    @command = GitHub::Command.new(proc { |x| puts x })
+  end
+
+  it "should return a GitHub::Helper" do
+    @command.helper.should be_instance_of(GitHub::Helper)
+  end
+
+  it "should call successfully" do
+    @command.should_receive(:puts).with("test").once
+    @command.call("test")
+  end
+
+  it "should return options" do
+    GitHub.should_receive(:options).with().once.and_return({:ssh => true})
+    @command.options.should == {:ssh => true}
+  end
+
+  it "should successfully call out to the shell" do
+    unguard(Kernel, :fork)
+    unguard(Kernel, :exec)
+    hi = @command.sh("echo hi")
+    hi.should == "hi"
+    hi.out?.should be(true)
+    hi.error?.should be(false)
+    hi.command.should == "echo hi"
+    bye = @command.sh("echo bye >&2")
+    bye.should == "bye"
+    bye.out?.should be(false)
+    bye.error?.should be(true)
+    bye.command.should == "echo bye >&2"
+  end
+
+  it "should return the results of a git operation" do
+    GitHub::Command::Shell.should_receive(:new).with("git rev-parse master").once.and_return do |*cmds|
+      s = mock("GitHub::Commands::Shell")
+      s.should_receive(:run).once.and_return("sha1")
+      s
+    end
+    @command.git("rev-parse master").should == "sha1"
+  end
+
+  it "should print the results of a git operation" do
+    @command.should_receive(:puts).with("sha1").once
+    GitHub::Command::Shell.should_receive(:new).with("git rev-parse master").once.and_return do |*cmds|
+      s = mock("GitHub::Commands::Shell")
+      s.should_receive(:run).once.and_return("sha1")
+      s
+    end
+    @command.pgit("rev-parse master")
+  end
+
+  it "should exec a git command" do
+    @command.should_receive(:exec).with("git rev-parse master").once
+    @command.git_exec "rev-parse master"
+  end
+
+  it "should die" do
+    @command.should_receive(:puts).once.with("=> message")
+    @command.should_receive(:exit!).once
+    @command.die "message"
+  end
+end