achoo 0.3

data/lib/achoo/timespan.rb

@@ -0,0 +1,119 @@
+require 'time'
+
+class Achoo; end
+
+class Achoo::Timespan
+
+  SECONDS_IN_A_DAY    = 86400
+  SECONDS_IN_AN_HOUR  = 3600
+  SECONDS_IN_A_MINUTE = 60
+
+  attr :start
+  attr :end
+
+  def initialize(start, end_)
+    raise ArgumentError.new('Nil in parameters not allowed') if start.nil? || end_.nil?
+
+    self.start = start
+    self.end   = end_
+  end
+
+  def start=(timeish)
+    @start = to_time(timeish)
+  end
+  
+  def end=(timeish)
+    @end = to_time(timeish)
+  end
+
+  def to_s
+    duration = duration_string
+    from_to  = from_to_string
+
+    sprintf("(%s) %s", duration, from_to)
+  end
+
+  def contains?(timeish_or_timespan)
+    if timeish_or_timespan.is_a? Achoo::Timespan
+      timespan = timeish_or_timespan
+      return start <= timespan.start && self.end >= timespan.end
+    else
+      time = to_time(timeish_or_timespan)
+      return start <= time && self.end >= time
+    end
+  end
+
+  def overlaps?(timespan)
+    start <= timespan.start && self.end >= timespan.start \
+      || start <= timespan.end && self.end >= timespan.end \
+      || contains?(timespan) || timespan.contains?(self)
+  end
+
+  private 
+  
+  def to_time(timeish)
+    case timeish
+    when Time
+      timeish.clone
+    when DateTime
+      Time.local(timeish.year, timeish.month, timeish.day, timeish.hour,timeish.minute, timeish.second)
+    when Date
+      Time.local(timeish.year, timeish.month, timeish.day)
+    else
+      if timeish.respond_to?(:to_s)
+        Time.parse(timeish.to_s)
+      else
+        raise ArgumentError.new("Don't know how to convert #{timeish.class} to Time")
+      end
+    end
+  end
+
+
+  def duration_string
+    delta = @end - @start
+    d     = delta.to_i / SECONDS_IN_A_DAY
+
+    delta = delta - d*SECONDS_IN_A_DAY
+    h     = delta.to_i / SECONDS_IN_AN_HOUR
+
+    delta = delta - h*SECONDS_IN_AN_HOUR
+    m     = delta.to_i / SECONDS_IN_A_MINUTE
+
+    sprintf "%d+%02d:%02d", d, h, m
+  end
+
+  def from_to_string
+    today      = Date.today
+    start_date = start.send(:to_date)
+    end_date   = self.end.send(:to_date)
+
+    format = if start_date == today
+               "Today"
+             elsif start_date.month == today.month && 
+                 start_date.year == today.year
+               "%a %e."
+             elsif start_date.year == today.year
+               "%a %e. %b"
+             else
+               "%a %e. %b %Y"
+             end
+    from = start.strftime(format << " %R")
+
+    format = if end_date == start_date
+               "%R"
+             elsif end_date == today
+               "Today %R"
+             elsif end_date.month == today.month && 
+               end_date.year == today.year
+               "%a %e. %R"
+             elsif end_date.year == today.year
+               "%a %e. %b %R"
+             else
+               "%a %e. %b %Y %R"
+             end
+    to = self.end.strftime(format)
+
+    sprintf "%s - %s", from, to
+  end
+
+end

data/lib/achoo/ui.rb

@@ -0,0 +1,10 @@
+class Achoo; class UI; end; end
+
+Achoo::UI.autoload :Commands, 'achoo/ui/commands'
+Achoo::UI.autoload :Common, 'achoo/ui/common'
+Achoo::UI.autoload :DateChooser, 'achoo/ui/date_chooser'
+Achoo::UI.autoload :DateChoosers, 'achoo/ui/date_choosers'
+Achoo::UI.autoload :ExceptionHandling, 'achoo/ui/exception_handling'
+Achoo::UI.autoload :MonthChooser, 'achoo/ui/month_chooser'
+Achoo::UI.autoload :OptionallyRangedDateChooser, 'achoo/ui/optionally_ranged_date_chooser'
+Achoo::UI.autoload :RegisterHours, 'achoo/ui/register_hours'

data/lib/achoo/ui/commands.rb

@@ -0,0 +1,51 @@
+require 'achoo/hour_administration_form'
+require 'achoo/lock_month_form'
+require 'achoo/ui'
+
+module Achoo::UI::Commands
+
+  include Achoo::UI::DateChoosers
+  include Achoo::UI::Common
+
+  def show_registered_hours_for_day(agent)
+    date = date_chooser
+    form = Achoo::HourAdministrationForm.new(agent)
+    form.show_registered_hours_for_day(date)
+  end
+
+  def show_registered_hours_for_week(agent)
+    date = date_chooser
+    form = Achoo::HourAdministrationForm.new(agent)
+    form.show_registered_hours_for_week(date)
+  end
+
+
+  def show_flexi_time(agent)
+    date = date_chooser
+    form = Achoo::HourAdministrationForm.new(agent)
+    balance = form.flexi_time(date)
+    puts "Flexi time balance: #{Achoo::Term::underline(balance)}"
+  end
+
+
+  def lock_month(agent)
+    month = month_chooser
+    form   = Achoo::LockMonthForm.new(agent)
+    form.lock_month(month)
+    form.print_values
+    if confirm
+      form.submit
+    else
+      puts "Cancelled"
+    end
+  end
+
+  
+  def show_holiday_report(agent)
+    page = agent.get(RC[:holiday_report_url])
+    page.body.match(/<b>(\d+,\d+)<\/b>/)
+    puts "Balance: #{Achoo::Term::underline($1)}"
+  end
+
+
+end

data/lib/achoo/ui/common.rb

@@ -0,0 +1,15 @@
+require 'achoo/term'
+
+class Achoo; class UI; end; end
+
+module Achoo::UI::Common
+
+  def confirm
+    answer = Achoo::Term::ask "Submit? [Y/n]"
+    answer.downcase!
+    return answer == 'y' || answer == ''
+  end
+
+end
+
+

data/lib/achoo/ui/date_chooser.rb

@@ -0,0 +1,75 @@
+require 'achoo/term'
+
+class Achoo; class UI; end; end
+
+class Achoo::UI::DateChooser
+
+  PROMPT = "Date ([today] | ?)"
+  FORMAT = "        today | (+|-)n | [[[YY]YY]-[M]M]-[D]D"
+
+  def choose
+    loop do
+      answer = Achoo::Term::ask PROMPT 
+      begin
+        date = handle_answer(answer)
+        return date if date
+      rescue ArgumentError => e
+        puts e
+      end
+    end
+  end
+
+  def parse_date(date_str, base=Date.today)
+    raise ArgumentError.new('Invalid date') if date_str.nil?
+
+    # Today (default)
+    if date_str == 'today' || date_str.empty?
+      return Date.today
+    end
+    
+    # Base offset
+    case date_str.chars.first
+    when '-'
+      return base - Integer(date_str[1..-1])
+    when '+'
+      return base + Integer(date_str[1..-1])
+    end
+    
+    # 
+    date = date_str.split('-').collect {|d| d.to_i}
+    case date.length
+    when 1
+      return Date.civil(base.year, base.month, *date)
+    when 2
+      return Date.civil(base.year, *date)
+    when 3
+      date[0] += 2000 if date[0] < 100
+      return Date.civil(*date)
+    end
+    
+    raise ArgumentError.new('Invalid date')
+  end
+
+  private
+
+  def handle_answer(answer)
+    if answer == '?'
+      print_help_message
+      return false
+    else
+      return parse_date(answer)
+    end
+  end
+  
+  def print_help_message   
+    puts "Accepted formats:"
+    puts date_format_help_string
+    puts
+    system 'cal -3m'
+  end
+
+  def date_format_help_string
+    FORMAT
+  end
+
+end

data/lib/achoo/ui/date_choosers.rb

@@ -0,0 +1,21 @@
+require 'achoo/ui/date_chooser'
+require 'achoo/ui/optionally_ranged_date_chooser'
+require 'achoo/ui/month_chooser'
+
+class Achoo; class UI; end; end
+
+module Achoo::UI::DateChoosers
+
+  def date_chooser
+    Achoo::UI::DateChooser.new.choose
+  end
+
+  def optionally_ranged_date_chooser
+    Achoo::UI::OptionallyRangedDateChooser.new.choose
+  end
+
+  def month_chooser
+    Achoo::UI::MonthChooser.new.choose
+  end
+
+end

data/lib/achoo/ui/exception_handling.rb

@@ -0,0 +1,20 @@
+require 'achoo/term'
+
+class Achoo; class UI; end; end;
+
+module Achoo::UI::ExceptionHandling
+
+  def handle_exception(user_message, e)
+    Achoo::Term::warn(user_message) + get_exception_reason(e)
+  end
+
+  def handle_fatal_exception(user_message, e)
+    puts Achoo::Term::fatal(user_message) + get_exception_reason(e)
+    exit 1
+  end
+
+  def get_exception_reason(e)
+    "\nReason: \n\t" + e.message.gsub("\n", "\n\t") + "\n---\n\t" + e.backtrace.join("\n\t")
+  end
+
+end

data/lib/achoo/ui/month_chooser.rb

@@ -0,0 +1,27 @@
+require 'achoo/term'
+
+class Achoo; class UI; end; end
+
+class Achoo::UI::MonthChooser
+
+  def choose
+    default = one_month_ago
+    period  = Achoo::Term::ask "Period ([#{default}] | YYYYMM)"
+    period  = default if !period || period.empty?
+    # FIX validate YYYYMM
+    period
+  end
+
+  def one_month_ago
+    now   = Time.now
+    year  = now.year
+
+    # Use -2 + 1 to shift range from 0-11 to 1-12 
+    month = (now.month - 2)%12 + 1
+    year -= 1 if month > now.month
+
+    sprintf "%d%02d", year, month
+  end
+
+end    
+

data/lib/achoo/ui/optionally_ranged_date_chooser.rb

@@ -0,0 +1,33 @@
+require 'achoo/ui/date_chooser'
+
+class Achoo::UI::OptionallyRangedDateChooser < Achoo::UI::DateChooser
+
+  def parse_date_range(date_range_str)
+    start_date_str, finish_date_str = *date_range_str.split('->')
+    start_date = parse_date(start_date_str.strip)
+    finish_date = parse_date(finish_date_str.strip, start_date)
+    
+    if start_date >= finish_date
+      raise ArgumentError.new('Invalid date range')
+    end
+    
+    [start_date, finish_date]
+  end
+
+  private 
+
+  def handle_answer(answer)
+    if answer.include? '->'
+      return parse_date_range(answer)
+    else 
+      return super
+    end
+  end
+
+  def date_format_help_string
+    return "    DATE [-> DATE]\n" \
+        << "    DATE:\n"          \
+        << FORMAT
+  end
+
+end

data/lib/achoo/ui/register_hours.rb

@@ -0,0 +1,137 @@
+require 'achoo/awake'
+require 'achoo/hour_registration_form'
+require 'achoo/hour_registration_form_ranged'
+require 'achoo/ical'
+require 'achoo/term'
+require 'achoo/ui'
+require 'achoo/vcs'
+require 'stringio'
+
+module Achoo::UI::RegisterHours
+
+  include Achoo::UI::DateChoosers
+  include Achoo::UI::ExceptionHandling
+
+  def register_hours(agent)
+    date       = optionally_ranged_date_chooser
+
+    puts "Fetching data ..."
+    form = if date.class == Date
+             Achoo::HourRegistrationForm
+           else
+             Achoo::HourRegistrationFormRanged
+           end.new(agent)
+
+    form.date    = date
+    form.project = project_chooser(form)
+    form.phase   = phase_chooser(form)
+    print_remark_help(date) if date.class == Date
+    form.remark  = remark_chooser
+    print_hours_help(date) if date.class == Date
+    form.hours   = hours_chooser
+
+    answer = Achoo::Term.ask("Do you want to change the defaults for worktime period and/or billing percentage? [N/y]").downcase
+    if answer == 'y'
+      form.workperiod = workperiod_chooser(form)
+      form.billing    = billing_chooser(form)
+    end
+
+    form.print_values
+    if confirm
+      puts "Submitting ..."
+      form.submit
+    else
+      puts "Cancelled"
+    end
+  end
+
+
+  def phase_chooser(form)
+    phases = form.phases_for_selected_project
+    puts "Phases"
+    answer = Achoo::Term.choose('Phase', phases.collect {|p| p[1] })
+    phases[answer.to_i-1][0]
+  end
+
+
+  def workperiod_chooser(form)
+    periods = form.worktime_periods
+    puts "Worktime periods"
+    answer = Achoo::Term.choose('Period [1]', periods.collect {|p| p[1] }, nil, [''])
+    answer = '1' if answer.empty?
+    periods[answer.to_i-1][0]
+  end
+
+  def billing_chooser(form)
+    options = form.billing_options
+    puts "Billing options"
+    answer = Achoo::Term.choose('Billing [1]', options.collect {|p| p[1] }, nil, [''])
+    answer = '1' if answer.empty?
+    options[answer.to_i-1][0]
+  end
+
+
+  def print_hours_help(date)
+    puts "Awake log:"
+    begin
+      awake = Achoo::Awake.new
+      awake.at(date)
+      puts
+    rescue Exception => e
+      print handle_exception("Failed to retrieve awake log.", e)
+    end
+  end
+
+  def hours_chooser
+    answer = Achoo::Term::ask 'Hours [7:30]'
+    return answer == '' ? '7.5' : answer
+  end
+
+
+  def print_remark_help(date)
+    puts "VCS logs for #{date}:"
+    begin
+      Achoo::VCS.print_logs_for(date, RC[:vcs_dirs])
+    rescue Exception => e
+      puts handle_exception("Failed to retrieve VCS logs.", e)
+    end
+    puts '-' * 80
+    puts "Calendar events for #{date}:"
+    puts '---'
+    begin
+      RC[:ical].each do |config|
+        Achoo::ICal.from_http_request(config).print_events(date)
+      end
+    rescue Exception => e
+      puts handle_exception("Failed to retrieve calendar events.", e)
+    end
+  end
+
+  def remark_chooser
+    Achoo::Term::ask 'Remark'
+  end
+
+
+  def project_chooser(form)
+    puts 'Recently used projects'
+    projects = form.recent_projects
+    answer = Achoo::Term.choose('Project [1]', projects.collect { |p| p[1] },
+                       'Other', [''])
+    case answer
+    when ''
+      projects[0][0]
+    when '0'
+      return all_projects_chooser(form)
+    else
+      return projects[answer.to_i-1][0]
+    end
+  end
+
+
+  def all_projects_chooser(form)
+    projects = form.all_projects
+    answer = Achoo::Term.choose('Project', projects.collect { |p| p[1] })
+    projects[answer.to_i-1][0]
+  end
+
+end