Pivotal tracker git status thing...

pivotalstatus.rb

#!/usr/bin/env ruby

require 'rubygems'
require 'pivotal-tracker'

class String
  ANSI_COLORS = [:black, :red, :green, :yellow, :blue, :magenta, :cyan, :white]
  def colorize(color)
    num = ANSI_COLORS.index(color) or raise("Bad color #{ color }")
    "\e[3#{ num }m" + self + "\e[0m"
  end
end

class PivotalStatus
  STATE_COLORS = Hash.new(:white).merge({
    "unstarted" => :magenta,
    "started"   => :blue,
    "finished"  => :green,
    "delivered" => :green,
    "accepted"  => :green,
    "rejected"  => :red,
  })

  def print
    if branches_and_story_ids = get_branches_and_story_ids()
      cleared_ids = []
      PivotalTracker::Project.all.each do |project|
        branches_and_story_ids.each do |branch, story_id|
          if story = project.stories.find(story_id)
            print_story(branch, story)
            cleared_ids << story_id
          end
        end
      end

      branches_and_story_ids_without_info = branches_and_story_ids.reject { |branch, story_id| cleared_ids.include?(story_id) }

      unless branches_and_story_ids_without_info.empty?
        puts
        puts "Could not find info for the following branches: #{ branches_and_story_ids_without_info.map(&:first).join(", ") }"
      end

    else
      $stderr.puts "No story ids found."
    end
  end

  private

  def get_branches_and_story_ids
    branches_and_story_ids = branches.select { |branch| branch.match(/^(t|s-)\d+/) }.map { |branch| [ branch, branch.gsub(/^(t|s)-?/, '').to_i ] }.uniq.sort_by(&:last)
    return nil if branches_and_story_ids.empty?
    branches_and_story_ids
  end

  def branches(include_remote = false)
    if include_remote
      `git branch -r 2>/dev/null`.scan(/^\s\sorigin\/(t\d+)$/).flatten
    else
      `git branch 2>/dev/null`.scan(/^[\s\*]\s(.*)$/).flatten
    end
  end

  def print_story(branch, story)
    puts format("%-15s %-10s - %s", branch, story.current_state, story.name).colorize(STATE_COLORS[story.current_state])

    story.tasks.all.each do |task|
      puts "  #{ task.description }".colorize(task.complete ? :green : :white)
    end
  end
end

begin
  PivotalTracker::Client.token = File.read(File.expand_path("~/.pivotal_api_token").strip)
  PivotalStatus.new.print
rescue Errno::ENOENT => e
  $stderr.puts e.message
  exit 1
rescue RestClient::Request::Unauthorized => e
  $stderr.puts "Unauthorized! You're probably using the wrong API key"
  exit 1
end