#  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 BurndownChart

  attr_accessor :data
  
  def initialize(settings)
    @settings = settings
    @burndown_data = BurndownData.new settings

    @data = {
      "meta" => {
        "board_id" => nil,
        "sprint" => 1,
        "total_days" => 10,
        "weekend_lines" => [ 3.5, 8.5 ]
      },
      "days" => []
    }
  end
  
  def sprint
    @data["meta"]["sprint"]
  end
  
  def sprint= s
    @data["meta"]["sprint"] = s
  end
  
  def board_id
    @data["meta"]["board_id"]
  end
  
  def board_id= id
    @data["meta"]["board_id"] = id
  end
  
  def days
    @data["days"]
  end
  
  def add_data(burndown_data, date)
    new_entry = {
      "date" => date.to_s,
      "story_points" => {
        "total" => burndown_data.story_points.total,
        "open" => burndown_data.story_points.open
      },
      "tasks" => {
        "total" => burndown_data.tasks.total,
        "open" => burndown_data.tasks.open
      },
      "story_points_extra" => {
        "done" => burndown_data.extra_story_points.done
      },
      "tasks_extra" => {
        "done" => burndown_data.extra_tasks.done
      }
    }
    new_days = Array.new
    replaced_entry = false
    @data["days"].each do |entry|
      if entry["date"] == date.to_s
        new_days.push(new_entry)
        replaced_entry = true
      else
        new_days.push(entry)
      end
    end
    if !replaced_entry
      new_days.push(new_entry)
    end
    @data["days"] = new_days
  end
  
  def read_data filename
    @data = YAML.load_file filename
  end
  
  def write_data filename
    @data["days"].each do |day|
      [ "story_points_extra", "tasks_extra" ].each do |key|
        if day[key] && day[key]["done"] == 0
          day.delete key
        end
      end
    end

    File.open( filename, "w" ) do |file|
      file.write @data.to_yaml
    end
  end
  
  def burndown_data_filename
    "burndown-data-#{sprint.to_s.rjust(2,"0")}.yaml"
  end
  
  def setup(burndown_dir, board_id)
    self.board_id = board_id
    FileUtils.mkdir_p burndown_dir
    write_data File.join(burndown_dir, burndown_data_filename)
  end
  
  def update(burndown_dir)
    Dir.glob("#{burndown_dir}/burndown-data-*.yaml").each do |file|
      file =~ /burndown-data-(.*).yaml/
      current_sprint = $1.to_i
      if current_sprint > sprint
        self.sprint = current_sprint
      end
    end
    burndown_data_path = File.join(burndown_dir, burndown_data_filename)
    begin
      read_data burndown_data_path
      @burndown_data.board_id = board_id
      @burndown_data.fetch
      add_data(@burndown_data, Date.today)
      write_data burndown_data_path
      puts "Updated data for sprint #{self.sprint}"
    rescue Errno::ENOENT
      raise TrolloloError.new( "'#{burndown_data_path}' not found" )
    end
  end

  def create_next_sprint(burndown_dir)
    Dir.glob("#{burndown_dir}/burndown-data-*.yaml").each do |file|
      file =~ /burndown-data-(.*).yaml/
      current_sprint = $1.to_i
      if current_sprint > sprint
        self.sprint = current_sprint
      end
    end
    burndown_data_path = File.join(burndown_dir, burndown_data_filename)
    begin
      read_data burndown_data_path
      self.sprint = self.sprint + 1
      @data["days"] = []
      write_data File.join(burndown_dir, burndown_data_filename)
    rescue Errno::ENOENT
      raise TrolloloError.new( "'#{burndown_data_path}' not found" )
    end
  end
end