trollolo 0.0.3

data/lib/card.rb

@@ -0,0 +1,64 @@
+#  Copyright (c) 2013-2014 SUSE LLC
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of version 3 of the GNU General Public License as
+#  published by the Free Software Foundation.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, contact SUSE LLC.
+#
+#  To contact SUSE about this file by physical or electronic mail,
+#  you may find current contact information at www.suse.com
+
+class Card
+
+  attr_accessor :sp, :tasks, :tasks_done
+
+  def self.name_to_points(card_name)
+    card_name =~ /^\(([\d.]+)\)/
+    return nil if $1.nil?
+    $1.to_f
+  end
+
+  def initialize
+    @sp = nil
+    @extra = false
+  end
+  
+  def has_sp?
+    @sp != nil
+  end
+  
+  def extra?
+    @extra
+  end
+  
+  def set_extra
+    @extra = true
+  end
+  
+  def self.parse json
+    card = Card.new
+
+    title = json["name"]
+    card.sp = name_to_points(title)
+
+    labels = json["labels"]
+    labels.each do |label|
+      if label["name"] == "Under waterline"
+        card.set_extra
+      end
+    end
+
+    card.tasks = json["badges"]["checkItems"]
+    card.tasks_done = json["badges"]["checkItemsChecked"]
+
+    card
+  end
+
+end

data/lib/cli.rb

@@ -0,0 +1,251 @@
+#  Copyright (c) 2013-2014 SUSE LLC
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of version 3 of the GNU General Public License as
+#  published by the Free Software Foundation.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, contact SUSE LLC.
+#
+#  To contact SUSE about this file by physical or electronic mail,
+#  you may find current contact information at www.suse.com
+
+class Cli < Thor
+
+  default_task :global
+
+  class_option :version, :type => :boolean, :desc => "Show version"
+  class_option :verbose, :type => :boolean, :desc => "Verbose mode"
+  class_option :raw, :type => :boolean, :desc => "Raw mode"
+  class_option "board-id", :type => :string, :desc => "id of Trello board"
+
+  def self.settings= s
+    @@settings = s
+  end
+
+  desc "global", "Global options", :hide => true
+  def global
+    if options[:version]
+      puts "Trollolo: #{@@settings.version}"
+    else
+      Cli.help shell
+    end
+  end
+
+  desc "get-lists", "Get lists"
+  option "board-id", :desc => "Id of Trello board", :required => true
+  def get_lists
+    process_global_options options
+    require_trello_credentials
+
+    trello = Trello.new(board_id: options["board-id"], developer_public_key: @@settings.developer_public_key, member_token: @@settings.member_token)
+    lists = trello.lists
+
+    if @@settings.raw
+      puts JSON.pretty_generate lists
+    else
+      lists.each do |list|
+        puts "#{list[ "name" ]}"
+      end
+    end
+  end
+
+  desc "get-cards", "Get cards"
+  option "board-id", :desc => "Id of Trello board", :required => true
+  def get_cards
+    process_global_options options
+    require_trello_credentials
+
+    trello = Trello.new(board_id: options["board-id"], developer_public_key: @@settings.developer_public_key, member_token: @@settings.member_token)
+    
+    cards = trello.cards
+
+    if @@settings.raw
+      puts JSON.pretty_generate cards
+    else
+      burndown = BurndownData.new @@settings
+      burndown.board_id = options["board-id"]
+
+      todo_list_id = burndown.fetch_todo_list_id
+      doing_list_id = burndown.fetch_doing_list_id
+      done_list_id = burndown.fetch_done_list_id
+      
+      cards_todo = Array.new
+      cards_doing = Array.new
+      cards_done = Array.new
+
+      above_waterline = true
+      
+      cards.each do |card|
+        name = card["name"]
+        list = card["idList"]
+        puts "CARD #{name} (#{list})"
+        
+        if name == "Waterline"
+          above_waterline = false
+          next
+        end
+        
+        if Card.name_to_points(name).nil?
+          next
+        end
+        
+        if list == todo_list_id && above_waterline
+          cards_todo.push card
+        elsif list == doing_list_id
+          cards_doing.push card
+        elsif list == done_list_id
+          cards_done.push card
+        end
+      end
+
+      story_points_todo = 0
+      story_points_doing = 0
+      story_points_done = 0
+      
+      puts
+      
+      puts "Todo"
+      cards_todo.each do |card|
+        puts "  #{card["name"]}"
+        story_points_todo += Card.name_to_points(card["name"])
+      end
+
+      puts "Doing"
+      cards_doing.each do |card|
+        puts "  #{card["name"]}"
+        story_points_doing += Card.name_to_points(card["name"])
+      end
+
+      puts "Done"
+      cards_done.each do |card|
+        puts "  #{card["name"]}"
+        story_points_done += Card.name_to_points(card["name"])
+      end
+      
+      puts
+      
+      story_points_total = story_points_todo + story_points_doing + story_points_done
+      
+      puts "Done: #{story_points_done}/#{story_points_total} (#{story_points_doing} in progress)"
+    end
+  end
+
+  desc "get-checklists", "Get checklists"
+  option "board-id", :desc => "Id of Trello board", :required => true
+  def get_checklists
+    process_global_options options
+    require_trello_credentials
+
+    trello = Trello.new(board_id: options["board-id"], developer_public_key: @@settings.developer_public_key, member_token: @@settings.member_token)
+    
+    data = trello.checklists
+
+    puts JSON.pretty_generate data
+  end
+
+  desc "fetch-burndown-data", "Fetch data for burndown chart"
+  option "board-id", :desc => "Id of Trello board", :required => true
+  def fetch_burndown_data
+    process_global_options options
+    require_trello_credentials
+
+    burndown = BurndownData.new @@settings
+    burndown.board_id = options["board-id"]
+    burndown.fetch
+        
+    puts "Story points:"
+    puts "   Open: #{burndown.story_points.open}"
+    puts "   Done: #{burndown.story_points.done}"
+    puts "  Total: #{burndown.story_points.total}"    
+    puts "Tasks:"
+    puts "   Open: #{burndown.tasks.open}"
+    puts "   Done: #{burndown.tasks.done}"
+    puts "  Total: #{burndown.tasks.total}"    
+    puts
+    puts "Extra story points:"
+    puts "   Open: #{burndown.extra_story_points.open}"
+    puts "   Done: #{burndown.extra_story_points.done}"
+    puts "  Total: #{burndown.extra_story_points.total}"    
+    puts "Extra tasks:"
+    puts "   Open: #{burndown.extra_tasks.open}"
+    puts "   Done: #{burndown.extra_tasks.done}"
+    puts "  Total: #{burndown.extra_tasks.total}"
+  end
+  
+  desc "burndown-init", "Initialize burndown chart"
+  option :output, :aliases => :o, :desc => "Output directory", :required => true
+  option "board-id", :desc => "Id of Trello board", :required => true
+  def burndown_init command=nil
+    process_global_options options
+    require_trello_credentials
+
+    chart = BurndownChart.new @@settings
+    puts "Preparing directory..."
+    chart.setup(options[:output],options["board-id"])
+  end
+  
+  desc "burndown", "Update burndown chart"
+  option :output, :aliases => :o, :desc => "Output directory", :required => false
+  option :new_sprint, :aliases => :n, :desc => "Create new sprint"
+  def burndown
+    process_global_options options
+    require_trello_credentials
+
+    chart = BurndownChart.new @@settings
+    begin
+      if options[:new_sprint]
+        chart.create_next_sprint(options[:output] || Dir.pwd)
+      end
+      chart.update(options[:output] || Dir.pwd)
+    rescue TrolloloError => e
+      STDERR.puts e
+      exit 1
+    end
+  end
+  
+  desc "plot", "Plot burndown chart"
+  def plot(sprint_number)
+    process_global_options options
+
+    plot_helper = File.expand_path("../../scripts/create_burndown.py", __FILE__ )
+    system "python #{plot_helper} #{sprint_number}"
+  end
+
+  private
+  
+  def process_global_options options
+    @@settings.verbose = options[:verbose]
+    @@settings.raw = options[:raw]
+  end
+
+  def require_trello_credentials
+    write_back = false
+
+    if !@@settings.developer_public_key
+      puts "Put in Trello developer public key:"
+      @@settings.developer_public_key = STDIN.gets.chomp
+      write_back = true
+    end
+    
+    if !@@settings.member_token
+      puts "Put in Trello member token:"
+      @@settings.member_token = STDIN.gets.chomp
+      write_back = true
+    end
+
+    if write_back
+      @@settings.save_config
+    end
+    
+    if !@@settings.developer_public_key || !@@settings.member_token
+      STDERR.puts "Require trello credentials in config file"
+      exit 1
+    end
+  end
+end

data/lib/settings.rb

@@ -0,0 +1,54 @@
+#  Copyright (c) 2013-2014 SUSE LLC
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of version 3 of the GNU General Public License as
+#  published by the Free Software Foundation.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, contact SUSE LLC.
+#
+#  To contact SUSE about this file by physical or electronic mail,
+#  you may find current contact information at www.suse.com
+
+class Settings
+
+  attr_accessor :verbose, :raw
+  attr_accessor :developer_public_key, :member_token
+
+  def initialize config_file_path
+    @config_file_path = config_file_path
+    if File.exists? config_file_path
+      @config = YAML.load_file(config_file_path)
+
+      if @config
+        @developer_public_key = @config["developer_public_key"]
+        @member_token = @config["member_token"]
+      else
+        raise "Couldn't read config data from '#{config_file_path}'"
+      end
+    end
+    
+    @verbose = false
+    @raw = false
+  end
+
+  def save_config
+    @config = {}
+    @config["developer_public_key"] = @developer_public_key
+    @config["member_token"] = @member_token
+    
+    File.open(@config_file_path,"w") do |f|
+      f.write(@config.to_yaml)
+    end
+  end
+  
+  def version
+    Trollolo::VERSION
+  end
+
+end

data/lib/trello.rb

@@ -0,0 +1,66 @@
+#  Copyright (c) 2013-2014 SUSE LLC
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of version 3 of the GNU General Public License as
+#  published by the Free Software Foundation.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, contact SUSE LLC.
+#
+#  To contact SUSE about this file by physical or electronic mail,
+#  you may find current contact information at www.suse.com
+
+class Trello
+  class ApiError < StandardError; end
+
+  attr_accessor :board_id
+
+  attr_reader   :developer_public_key, :member_token
+
+  def initialize settings
+    @developer_public_key = settings.fetch(:developer_public_key)
+    @member_token         = settings.fetch(:member_token)
+    @board_id             = settings.fetch(:board_id)
+  end
+
+  def lists
+    get "lists"
+  end
+
+  def cards
+    get "cards"
+  end
+
+  def checklists
+    get "checklists"
+  end
+
+  private
+
+  def resource_url resource
+    "/1/boards/#{board_id}/#{resource}?key=#{developer_public_key}&token=#{member_token}"
+  end
+
+  # FIXME: we should handle time-outs gracefully.
+
+  def get resource
+    resp = http_client.get resource_url(resource)
+    unless resp.code == "200"
+      raise ApiError.new("Error occured while connecting to the trello API.")
+    end
+    JSON.parse(resp.body)
+  rescue JSON::ParserError => e
+    raise ApiError.new(e)
+  end
+
+  def http_client
+    http = Net::HTTP.new "trello.com", 443
+    http.use_ssl = true
+    http
+  end
+end

data/lib/trollolo.rb

@@ -0,0 +1,34 @@
+#  Copyright (c) 2013-2014 SUSE LLC
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of version 3 of the GNU General Public License as
+#  published by the Free Software Foundation.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, contact SUSE LLC.
+#
+#  To contact SUSE about this file by physical or electronic mail,
+#  you may find current contact information at www.suse.com
+
+require "thor"
+require "net/http"
+require "net/https"
+require "json"
+require "yaml"
+require "erb"
+
+require_relative 'version'
+require_relative 'cli'
+require_relative 'settings'
+require_relative 'trello'
+require_relative 'card'
+require_relative 'burndown_chart'
+require_relative 'burndown_data'
+
+class TrolloloError < StandardError
+end