initial commit

This commit is contained in:
John Bintz 2011-09-23 16:49:48 -04:00
commit c2ce35c701
7 changed files with 513 additions and 0 deletions

4
.gitignore vendored Normal file
View File

@ -0,0 +1,4 @@
*.gem
.bundle
Gemfile.lock
pkg/*

4
Gemfile Normal file
View File

@ -0,0 +1,4 @@
source "http://rubygems.org"
# Specify your gem's dependencies in pretty-tracker.gemspec
gemspec

1
Rakefile Normal file
View File

@ -0,0 +1 @@
require "bundler/gem_tasks"

468
bin/tracker Executable file
View File

@ -0,0 +1,468 @@
#!/usr/bin/env ruby
require 'rubygems'
gem 'pivotal-tracker', '~> 0.4.0'
require 'pivotal-tracker'
require 'thor'
require 'rainbow'
require 'tempfile'
require 'yaml'
CACHE_TIME = 60
GIT_PIVOTAL_PROJECT_ID_KEY = "pivotal.project-id"
CACHE_PATH = '~/.tracker-cache'
LAST_STORY_PATH = File.expand_path("#{CACHE_PATH}/last-story")
module PivotalTracker
class Story
def move(story)
Client.connection["/projects/#{project_id}/stories/#{id}/moves"].post(:move => { :move => :before, :target => story.id})
end
end
end
class Tracker < Thor
method_options :first => :boolean
method_options :project => :string
method_options :description => :string
method_options :api_token => :string
method_options %w{labels -l} => :string
method_options %w{skip_description --nd} => :boolean
method_options :external => :boolean
def initialize(*args)
super
@options = options.dup
if File.file?(path = File.expand_path('~/.trackerrc'))
@options.merge!(YAML.load_file(path))
end
api_token = %x{git config --get pivotal.api-token}.strip
api_token = @options[:api_token] if api_token.empty?
PivotalTracker::Client.token = api_token
@stdin_content = nil
if !$stdin.tty?
@stdin_content = $stdin.read
$stdin.close
end
end
started_color = 'f3f3d1'
COLORS = {
:feature => 'ffff44',
:chore => '999999',
:bug => 'ff2233',
:release => '0044cc',
:id => '22cc33',
:label => '33ee55',
:project => 'dd2211',
:story => 'ffffff',
:comment => 'ddcc11',
:description => '2255ee',
:started => started_color,
:finished => started_color,
:delivered => started_color,
:unscheduled => 'e4eff7',
:score => '2255cc'
}
desc "feature NAME <score>", "Create a new feature in the current project"
def feature(name, estimate = nil)
params = { :name => name, :story_type => :feature, :current_state => :unstarted }
params[:estimate] = estimate if estimate
create_story(params)
clear_cache!
end
desc "chore NAME", "Create a new chore in the current project"
def chore(name)
create_story(:name => name, :story_type => :chore, :current_state => :unstarted)
clear_cache!
end
desc "release NAME", "Create a new release in the current project"
def release(name)
create_story(:name => name, :story_type => :release, :current_state => :unstarted)
clear_cache!
end
desc "bug NAME", "Create a new bug in the current project"
def bug(name)
create_story(:name => name, :story_type => :bug, :current_state => :unstarted)
clear_cache!
end
desc "start NAME-OR-ID", "Start a story"
def start(name_or_id = nil)
with_story(name_or_id) do |story|
story.update(:current_state => :started)
puts "Story #{story.name} started"
clear_cache!
end
end
desc "deliver NAME-OR-ID", "Deliver a feature or bug"
def deliver(name_or_id = nil)
with_story(name_or_id) do |story|
story.update(:current_state => :delivered)
puts "Story #{story.name} delivered"
clear_cache!
end
end
desc "accept NAME-OR-ID", "Accept a feature or bug"
def accept(name_or_id)
with_story(name_or_id) do |story|
story.update(:current_state => :accepted)
puts "Story #{story.name} accepted"
clear_cache!
end
end
desc "finish NAME-OR-ID <COMMENT>", "Finish a story and add an optional comment"
def finish(name_or_id, comment = nil)
with_story(name_or_id) do |story|
comment = external(:comment) if !comment && @options[:external]
story.update(:current_state => :finished)
PivotalTracker::Note.new(:owner => story, :text => comment).create if comment && !comment.empty?
puts "Story #{story.name} finished"
clear_cache!
end
end
desc "labels NAME-OR-ID <labels>", "Update the labels on a story"
def labels(name_or_id, label_list)
with_story(name_or_id) do |story|
story.update(:labels => label_list)
puts "Labels for #{story.name.foreground(COLORS[:story])} changed to #{label_list.foreground(COLORS[:label])}"
clear_cache!
end
end
desc "score NAME-OR-ID <score>", "Update the score on a story"
def score(name_or_id, score)
with_story(name_or_id) do |story|
story.update(:estimate => score)
puts "Score for #{story.name.foreground(COLORS[:feature])} changed to #{score.foreground(COLORS[:score])}"
clear_cache!
end
end
desc "comment NAME-OR-ID <COMMENT>", "Add a comment to a story"
def comment(name_or_id, comment = nil)
with_story(name_or_id) do |story|
if !comment
if @stdin_content
comment = @stdin_content
else
comment = external(:comment, story_text_history(story)) if @options[:external]
end
end
if comment && !comment.empty?
PivotalTracker::Note.new(:owner => story, :text => comment).create
puts "Comment #{comment.foreground(COLORS[:comment])} added to #{story.name}"
clear_cache!
else
puts "No comment provided!"
end
end
end
desc "show <NAME-OR-ID>", "Show a single story with description and comments"
def show(name_or_id = nil)
with_story(name_or_id) do |story|
comments = []
wrap_text(story_chrono_history(story)) { |line| comments << line.foreground(COLORS[:comment]) }
output = <<-ANSI
#{story.name.foreground(COLORS[:story])} #{"(#{story.id})".foreground(COLORS[:id])}
#{comments.join("\n")}
ANSI
file = Tempfile.new('tracker')
file.puts output
file.close
system %{bash -c 'less -fr #{file.path}'}
end
end
desc "list <FILTER>", "List all stories"
def list(filter = nil)
search_filter = filter ? search_value(filter) : nil
non_accepted_stories.each do |story|
if !search_filter || story.name[search_filter]
case story.current_state
when 'unstarted'
change_section(:backlog)
when 'unscheduled'
change_section(:icebox)
else
change_section(:current)
end
print_story(story)
end
end
end
desc "delete NAME-OR-ID", "Delete a story"
def delete(name_or_id)
with_story(name_or_id) do |story|
story.delete
puts "Story #{story.name} deleted"
clear_cache!
end
end
desc "export-for PROJECT", "Get the bash export for the specified project"
def export_for(project_search)
if project(project_search)
puts "export TRACKER_PROJECT=#{project.id}"
end
end
desc "set-git-project PROJECT", "Set pivotal.project-id to the requested project"
def set_git_project(project_search)
if project(project_search)
system %{git config --local #{GIT_PIVOTAL_PROJECT_ID_KEY} #{project.id}}
puts "#{project.name} set for this git repo"
end
end
desc "project", "Get the current project as determined by current config"
def project
if project
puts "#{project.id} #{project.name}"
end
end
default_task :list
no_tasks do
def story_text_history(story)
description_and_notes(story).reverse.join("\n")
end
def story_chrono_history(story)
description_and_notes(story).join("\n")
end
def description_and_notes(story)
([ story.description ] + PivotalTracker::Note.all(story).collect(&:text))
end
def create_story(params)
if @options[:external] && !@options[:skip_description]
params[:description] ||= external(:story_description)
else
params[:description] ||= @options[:description]
end
params[:labels] ||= @options[:labels]
story = project.stories.create(params)
story.move(first_in_icebox) if @options[:first] && first_in_icebox
puts "#{story.story_type.capitalize} #{story.name.foreground(COLORS[story.story_type.to_sym])} created in #{project.name.foreground(COLORS[:project])}"
story
end
def external(type, initial_comments = nil)
file = Tempfile.new("tracker")
file.puts
file.puts "# Write your #{type} here, lines starting with # are ignored"
if initial_comments
wrap_text(initial_comments) { |line| file.puts "# #{line}" }
file.puts "#"
end
file.close
system %{bash -c "$EDITOR #{file.path}"}
lines = File.readlines(file.path).find_all { |line| line[0..0] != '#' }
lines.empty? ? nil : lines.join.strip
end
def wrap_text(text, columns = 78)
text.each_line do |line|
line = line.gsub(/ +/, ' ')
line_done = false
while line.length > 78 && !line_done
index = 78
done = false
while !done && !line_done
if line[index..index] == " "
yield line[0..index - 1]
line = line[(index + 1)..-1]
done = true
else
index -= 1
line_done = (index == 0)
end
end
end
yield line
end
end
def stories_cache
FileUtils.mkdir_p File.expand_path(CACHE_PATH)
target = File.expand_path("#{CACHE_PATH}/#{project.id}")
if !File.file?(target) || (File.mtime(target) + CACHE_TIME < Time.now)
stories = project.stories.all
File.open(target, 'wb') { |fh| Marshal.dump(stories, fh) }
else
stories = Marshal.load(File.read(target))
end
stories
end
def clear_cache!
FileUtils.rm_f File.expand_path("#{CACHE_PATH}/#{project.id}")
end
def non_accepted_stories
stories_cache.reject { |story| story.current_state == 'accepted' }
end
def with_story(name_or_id = nil)
if !name_or_id && File.file?(LAST_STORY_PATH)
name_or_id = File.read(LAST_STORY_PATH).to_i
end
if story = find_story(name_or_id)
yield story
FileUtils.mkdir_p File.expand_path(CACHE_PATH)
File.open(LAST_STORY_PATH, 'wb') { |fh| fh.print story.id }
else
puts "Story #{name_or_id} not found."
end
end
def project(search = nil)
return @project if @project
search ||= @options[:project] || ENV['TRACKER_PROJECT'] || %x{git config --local --get #{GIT_PIVOTAL_PROJECT_ID_KEY}}.strip
search = nil if search.empty?
if search
if search[%r{^\d+$}]
project = PivotalTracker::Project.find(search)
else
project = all_projects.find { |proj| proj.name[search_value(search)] }
end
if !project || !project.kind_of?(PivotalTracker::Project)
puts "Could not find project #{search.to_s.foreground(COLORS[:project])}. Valid projects:"
all_projects.each do |project|
print "(#{project.id})".foreground(COLORS[:id])
print " #{project.name.foreground(COLORS[:project])}"
puts
end
exit 1
end
else
puts "No project found! Set one via export-for or git-set-project."
exit 1
end
@project = project
end
def search_value(value)
search = Regexp.new(value, Regexp::IGNORECASE)
search = Regexp.new(value[1..-2]) if value[0..0] == '/'
search
end
def all_projects
@all_projects ||= PivotalTracker::Project.all
end
def find_story(name_or_id)
if !(story = project.stories.find(name_or_id.to_i))
search = search_value(name_or_id)
story = stories_cache.find { |story| story.name[search] }
end
story
end
def print_story(story)
@alternate = false if @alternate == nil
prefix = case story.story_type
when 'feature'
"*FEATURE*".foreground(COLORS[story.story_type.to_sym])
when 'chore'
"[CHORE] ".foreground(COLORS[story.story_type.to_sym])
when 'bug'
"%BUG% ".foreground(COLORS[story.story_type.to_sym]).bright
when 'release'
"#RELEASE#".foreground(COLORS[story.story_type.to_sym]).bright
end
suffix = case story.story_type
when 'feature'
if story.estimate.to_i >= 0
" #{"|" * story.estimate.to_i} #{story.estimate}".foreground(COLORS[:score])
else
''
end
else
''
end
max_story_name = width - 9 - 2 - 2 - 9 - 1 - suffix.gsub(/\033[^m]+m/, '').length
story_name = story.name[0..max_story_name]
story_id = "(#{story.id.to_s.rjust(9)})"
left = "#{prefix} #{story_id.foreground(COLORS[:id])} #{story_name.foreground(COLORS[story.current_state.to_sym] || COLORS[:story])} "
if !(%w{unscheduled unstarted}).include?(story.current_state)
left += "<#{story.current_state.foreground(COLORS[:project])}> "
end
left += "#{story.labels.foreground(COLORS[:label])} " if story.labels
left += "@".foreground(COLORS[:description]) if story.description && !story.description.empty?
right = "#{suffix}"
existing_length = (width - left.gsub(/\033[^m]+m/, '').length - right.gsub(/\033[^m]+m/, '').length)
center = " " * [ existing_length, 0 ].max
puts [ left, center, right ].join
@alternate = !@alternate
end
def width
@width ||= (%x{stty size}.strip.split.last.to_i - 2)
end
def change_section(which)
if @current_section != which
string = ("-- #{which.to_s.upcase} " + ("-" * width))[0..width]
puts string.foreground('7777cc')
@current_section = which
end
end
def first_in_icebox
project.stories.all(:current_state => 'unstarted').first
end
end
end
Tracker.start

7
lib/pretty-tracker.rb Normal file
View File

@ -0,0 +1,7 @@
require "pretty-tracker/version"
module Pretty
module Tracker
# Your code goes here...
end
end

View File

@ -0,0 +1,5 @@
module Pretty
module Tracker
VERSION = "0.0.1"
end
end

24
pretty-tracker.gemspec Normal file
View File

@ -0,0 +1,24 @@
# -*- encoding: utf-8 -*-
$:.push File.expand_path("../lib", __FILE__)
require "pretty-tracker/version"
Gem::Specification.new do |s|
s.name = "pretty-tracker"
s.version = Pretty::Tracker::VERSION
s.authors = ["John Bintz"]
s.email = ["john@coswellproductions.com"]
s.homepage = ""
s.summary = %q{TODO: Write a gem summary}
s.description = %q{TODO: Write a gem description}
s.rubyforge_project = "pretty-tracker"
s.files = `git ls-files`.split("\n")
s.test_files = `git ls-files -- {test,spec,features}/*`.split("\n")
s.executables = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
s.require_paths = ["lib"]
# specify any dependencies here; for example:
# s.add_development_dependency "rspec"
# s.add_runtime_dependency "rest-client"
end