flaviogranero/assisted_workflow

require "assisted_workflow/exceptions"
require "assisted_workflow/addons/base"
require 'pivotal_tracker'

# wrapper class to pivotal api client
module AssistedWorkflow::Addons
  class Pivotal < Base
    required_options :fullname, :token, :project_id
  
    def initialize(output, options = {})
      super

      PivotalTracker::Client.token = options["token"]
      begin
        @project = PivotalTracker::Project.find(options["project_id"])
      rescue
        raise AssistedWorkflow::Error, "pivotal project #{options["project_id"]} not found."
      end
      @fullname = options["fullname"]
      @username = options["username"]
    end
  
    def find_story(story_id)
      if story_id.to_i > 0
        log "loading story ##{story_id}"
        story = @project.stories.find(story_id)
        story.other_id = @username || @fullname
        story.other_id = story.other_id.to_s.downcase.split.join
        story
      end
    end
  
    def start_story(story, options = {})
      log "starting story ##{story.id}"
      update_story! story, options.merge(:current_state => "started")
    end
  
    def finish_story(story, options = {})
      log "finishing story ##{story.id}"
      if update_story! story, :current_state => finished_state(story)
        story.notes.create(:text => options[:note]) if options[:note]
      end
    end
  
    def pending_stories(options = {})
      log "loading pending stories"
      states = ["unstarted"]
      states << "started" if options[:include_started]
      @project.stories.all(:state => states, :owned_by => @fullname, :limit => 5)
    end
    
    def valid?
      !@project.nil?
    end
  
    private
  
    def finished_state(story)
      if story.story_type == "chore"
        "accepted"
      else
        "finished"
      end
    end
    
    def update_story!(story, attributes)
      if story
        story.update(attributes)
        raise AssistedWorkflow::Error, story.errors.first.to_s if story.errors.any?
        true
      end
    end
  end
end