git-pivotal-tracker 0.9.0

data/lib/commands/block.rb

@@ -0,0 +1,59 @@
+require 'commands/base'
+
+module Commands
+  class Block < Base
+    Label = "blocked"
+    MessagePrefix = "Blocked:"
+    
+    def initialize(*args)
+      @story_id = args.shift if args.first =~ /^(\d+)$/
+      super(*args)
+    end
+
+    def run!
+      super
+
+      unless story_id
+        put "No story id was supplied and you aren't on a topic branch!"
+        return 1
+      end
+      
+      if story.labels.to_s.include?(Label)
+        put "Story #{story_id} is already blocked."
+        return 0
+      end
+      
+      message = options[:message].to_s
+
+      if message.empty?
+        loop do
+          put "What's the reason for blocking this story?"
+          message = input.gets.chomp
+          break unless message.empty?
+          put ""
+        end
+      end
+      
+      labels = story.labels.to_s.split(",").concat([Label]).join(",")
+      story.update :labels => labels
+      story.notes.create :author => full_name, :text => "#{MessagePrefix} #{message}"
+      put "Story #{story_id} has been blocked."
+
+      return 0
+    end
+
+  protected
+  
+    def on_parse(opts)
+      opts.on("-m [String]", "--message [String]", "The message to provide when blocking a story."){ |m| options[:message] = m }
+    end
+
+    def story_id
+      @story_id || current_branch[/\d+/]
+    end
+
+    def story
+      @story ||= project.stories.find(story_id)
+    end
+  end
+end

data/lib/commands/bug.rb

@@ -0,0 +1,19 @@
+require 'commands/pick'
+
+module Commands
+  class Bug < Pick
+
+    def type
+      "bug"
+    end
+    
+    def plural_type
+      "bugs"
+    end
+    
+    def branch_suffix
+      "bugfix"
+    end
+
+  end
+end

data/lib/commands/card.rb

@@ -0,0 +1,32 @@
+require 'commands/pick'
+
+module Commands
+  class Card < Pick
+    attr_accessor :story_id
+
+    def type
+      "story"
+    end
+    
+    def plural_type
+      "cards"
+    end
+    
+    def branch_suffix
+      if story.story_type == "bug"
+        "bugfix"
+      else
+        story.story_type
+      end
+    end
+
+    protected
+    
+    def story
+      return @story if @story
+      raise ArgumentError, "No story id was given!" unless story_id
+      project.stories.find(story_id)
+    end
+
+  end
+end

data/lib/commands/chore.rb

@@ -0,0 +1,19 @@
+require 'commands/pick'
+
+module Commands
+  class Chore < Pick
+
+    def type
+      "chore"
+    end
+    
+    def plural_type
+      "chores"
+    end
+    
+    def branch_suffix
+      "chore"
+    end
+
+  end
+end

data/lib/commands/feature.rb

@@ -0,0 +1,19 @@
+require 'commands/pick'
+
+module Commands
+  class Feature < Pick
+
+    def type
+      "feature"
+    end
+    
+    def plural_type
+      "features"
+    end
+    
+    def branch_suffix
+      "feature"
+    end
+
+  end
+end

data/lib/commands/finish.rb

@@ -0,0 +1,59 @@
+require 'commands/base'
+
+module Commands
+  class Finish < Base
+
+    def run!
+      super
+
+      unless story_id
+        put "Branch name must contain a Pivotal Tracker story id"
+        return 1
+      end
+
+      put "Marking Story #{story_id} as finished..."
+      if story.update(:current_state => finished_state)
+        topic_branch = current_branch
+        
+        put "Pushing #{topic_branch} to #{remote}"
+        sys "git push --set-upstream #{remote} #{topic_branch}"
+        
+        put "Pulling #{acceptance_branch}..."
+        sys "git checkout #{acceptance_branch}"
+        sys "git pull"
+
+        put "Merging #{topic_branch} into #{acceptance_branch}"
+        sys "git merge --no-ff #{topic_branch}"
+  
+        put "Pushing #{acceptance_branch} to #{remote}"
+        sys "git push"
+      
+        put "Now on #{acceptance_branch}."
+
+        return 0
+      else
+        put "Unable to mark Story #{story_id} as finished"
+
+        return 1
+      end
+    end
+
+  protected
+
+    def finished_state
+      if story.story_type == "chore"
+        "accepted"
+      else
+        "finished"
+      end
+    end
+
+    def story_id
+      match = current_branch[/\d+/] and match.to_i
+    end
+
+    def story
+      @story ||= project.stories.find(story_id)
+    end
+  end
+end

data/lib/commands/info.rb

@@ -0,0 +1,58 @@
+require 'commands/base'
+
+module Commands
+  class Info < Base
+    def initialize(*args)
+      @story_id = args.shift if args.first =~ /^(\d+)$/
+      super(*args)
+    end
+
+    def run!
+      super
+
+      unless story_id
+        put "No story id was supplied and you aren't on a topic branch!"
+        return 1
+      end
+
+      put "Story:         #{story.name}"
+      put "URL:           #{story.url}"
+      put "Labels:        #{story.labels.split(',').join(', ')}" if story.labels
+      put "State:         #{story.accepted_at ? 'accepted' : 'not accepted'}"
+      
+      colwidth = 74
+      
+      put "\nDescription:\n"
+      put wrap_text("#{story.description}\n", colwidth).gsub(/^/, '  ').chomp
+
+      if options[:comments]
+        put "\nComments:\n"
+        story.notes.all.each do |note|
+          @output.printf "  %-37s%37s\n\n", note.author, note.noted_at.strftime("%b %e, %Y %k:%M%p")
+          put wrap_text(note.text, colwidth - 2).gsub(/^/, '    ')
+        end
+      end
+      
+      return 0
+    end
+
+  protected
+  
+    def wrap_text(txt, col = 80)
+      txt.gsub(/(.{1,#{col}})( +|$\n?)|(.{1,#{col}})/,
+        "\\1\\3\n") 
+    end  
+
+    def on_parse(opts)
+      opts.on("-c", "--comments", "Display comments"){ |v| options[:comments] = v }
+    end
+
+    def story_id
+      @story_id || current_branch[/\d+/]
+    end
+
+    def story
+      @story ||= project.stories.find(story_id)
+    end
+  end
+end

data/lib/commands/map.rb

@@ -0,0 +1,10 @@
+module Commands
+  class Map < Hash
+    def [](arg)
+      super || (
+        key = keys.detect{ |e| e.respond_to?(:match) && e.match(arg) }
+        super key
+      )
+    end
+  end
+end

data/lib/commands/pick.rb

@@ -0,0 +1,76 @@
+require 'commands/base'
+
+module Commands
+  class Pick < Base
+
+    def type
+      raise Error("must define in subclass")
+    end
+
+    def plural_type
+      raise Error("must define in subclass")
+    end
+
+    def branch_suffix
+      raise Error("must define in subclass")
+    end
+
+    def run!
+      response = super
+      return response if response > 0
+
+      msg = "Retrieving latest #{plural_type} from Pivotal Tracker"
+      if options[:only_mine]
+        msg += " for #{options[:full_name]}"
+      end
+      put "#{msg}..."
+
+      unless story
+        put "No #{plural_type} available!"
+        return 0
+      end
+
+      put "Story: #{story.name}"
+      put "URL:   #{story.url}"
+
+      put "Updating #{type} status in Pivotal Tracker..."
+      story.update(:owned_by => options[:full_name], :current_state => :started)
+
+      if story.errors.empty?
+        suffix_or_prefix = ""
+        unless options[:quiet] || options[:defaults]
+          put "Enter branch name (will be #{options[:append_name] ? 'appended' : 'prepended'} by #{story.id}) [#{suffix_or_prefix}]: ", false
+          suffix_or_prefix = input.gets.chomp
+        end
+        suffix_or_prefix = branch_suffix if suffix_or_prefix == ""
+
+        if options[:append_name]
+          branch = "#{suffix_or_prefix}-#{story.id}"
+        else
+          branch = "#{story.id}-#{suffix_or_prefix}"
+        end
+        if get("git branch").match(branch).nil?
+          put "Switched to a new branch '#{branch}'"
+          sys "git checkout -b #{branch}"
+        end
+
+        return 0
+      else
+        put "Unable to mark #{type} as started"
+        put "\t" + story.errors.to_a.join("\n\t")
+
+        return 1
+      end
+    end
+
+  protected
+
+    def story
+      return @story if @story
+
+      conditions = { :story_type => type, :current_state => "unstarted", :limit => 1, :offset => 0 }
+      conditions[:owned_by] = options[:full_name] if options[:only_mine]
+      @story = project.stories.all(conditions).first
+    end
+  end
+end

data/lib/commands/start.rb

@@ -0,0 +1,35 @@
+require 'commands/map'
+require 'commands/bug'
+require 'commands/card'
+require 'commands/chore'
+require 'commands/feature'
+
+module Commands
+  class Start
+    COMMAND_MAP = Map.new.merge({
+      "bug"     => Commands::Bug,
+      "chore"   => Commands::Chore,
+      "feature" => Commands::Feature,
+      /^\d+$/   => Commands::Card
+    })
+    
+    class << self
+      def for(*args)
+        identifier = args.shift
+        construct_instance_for(identifier, args) || 
+          raise(ArgumentError, "Unknown card identifier given: #{identifier}")
+      end
+      
+      private
+      
+      def construct_instance_for(identifier, args)
+        if klass=COMMAND_MAP[identifier]
+          instance = klass.new(*args)
+          instance.story_id = identifier if instance.respond_to?(:story_id=)
+          instance
+        end
+      end
+    end
+  end
+
+end

data/lib/commands/unblock.rb

@@ -0,0 +1,55 @@
+require 'commands/base'
+require 'commands/block'
+
+module Commands
+  class Unblock < Base
+    PlaceholderLabel = "."
+    
+    def initialize(*args)
+      @story_id = args.shift if args.first =~ /^(\d+)$/
+      super(*args)
+    end
+
+    def run!
+      super
+
+      unless story_id
+        put "No story id was supplied and you aren't on a topic branch!"
+        return 1
+      end
+      
+      unless story.labels.to_s.include?(Block::Label)
+        put "Story #{story_id} is already unblocked."
+        return 0
+      end
+      
+      labels = story.labels.to_s.split(",") - [Block::Label]
+      
+      # this line is to work aroudn Pivotal Tracker's broken API for removing the last
+      # label on a card. http://community.pivotaltracker.com/pivotal/topics/api_v3_cannot_remove_labels_from_a_story_anymore
+      if labels.empty?
+        labels << PlaceholderLabel
+        put "Note: a '.' label will be placed on this card due to a bug in the v3 API of Pivotal Tracker."
+      end
+      
+      story.update :labels => labels.join(",")
+      put "Story #{story_id} has been unblocked."
+      
+      return 0
+    end
+
+  protected
+  
+    def on_parse(opts)
+      opts.on("-m [String]", "--message [String]", "The message to provide when blocking a story."){ |m| options[:message] = m }
+    end
+
+    def story_id
+      @story_id || current_branch[/\d+/]
+    end
+
+    def story
+      @story ||= project.stories.find(story_id)
+    end
+  end
+end