achoo 0.3

data/Rakefile

@@ -0,0 +1,64 @@
+# encoding: utf-8
+
+require 'rake/gempackagetask'
+require 'rake/rdoctask'
+require 'rake/testtask'
+require 'rubygems'
+
+load File.dirname(__FILE__) + '/achoo.gemspec'
+
+task :default => ['test:unit']
+
+namespace 'build' do
+
+  Rake::GemPackageTask.new(Spec) do |pkg|
+    pkg.need_tar = true
+  end
+
+end
+
+Rake::TestTask.new do |t|
+  t.libs << "test/lib"
+  t.test_files = FileList['test/**/test*.rb']
+  #t.verbose = true
+end
+
+namespace 'test' do
+
+  Rake::TestTask.new do |t|
+    t.name = :unit
+    t.libs << "test/lib"
+    t.test_files = FileList['test/unit/test*.rb']
+  end
+
+  Rake::TestTask.new do |t|
+    t.name = :acceptance
+    t.libs << "test/lib"
+    t.test_files = FileList['test/acceptance/test*.rb']
+  end
+
+  desc 'Measures test coverage'
+  task :coverage do
+    rm_f('coverage')
+    system("rcov -T -Ilib test/unit/test_*.rb")
+  end
+
+end
+
+namespace 'doc' do
+
+  Rake::RDocTask.new do |rd|
+    rd.title = 'Achoo --- The Achievo CLI'
+    rd.main = "README.rdoc"
+    rd.rdoc_files.include("README.rdoc", "lib/**/*.rb")
+    rd.template = `allison --path`.chop + '.rb'
+    rd.options << '--line-numbers' << '--inline-source'
+    rd.rdoc_dir = 'doc'
+  end
+
+end
+
+desc 'Install development dependencies'
+task :setup do
+  system 'gem install shoulda rack thin redgreen allison'
+end

data/bin/achoo

@@ -0,0 +1,48 @@
+#!/usr/bin/env ruby
+
+require 'achoo'
+require 'achoo/rc_loader'
+require 'optparse'
+
+include Achoo::RCLoader
+
+def main
+  trap("INT") { exit }
+
+  abort "UTF-8 terminal required" unless utf8?
+
+  options = {
+    :log     => false,
+    :rc_file => "#{ENV['HOME']}/.achoo",
+  }.merge(parse_args)
+
+  load_rc(options[:rc_file])
+
+  achoo = Achoo.new(options[:log])
+  achoo.start
+end
+
+def parse_args
+  options = {}
+
+  optparse = OptionParser.new do|opts|
+    opts.banner = "Usage: $0 [options]"
+
+    opts.on( '-l', '--log', 'Write http communication to achoo_http.log' ) do
+      options[:log] = true
+    end
+
+    opts.on('--rcfile FILE', 'FIX' ) do |file|
+      options[:rc_file] = file
+    end
+  end
+
+  optparse.parse!
+  options
+end
+
+def utf8?
+  `locale charmap`.chop == 'UTF-8'
+end
+
+main

data/bin/awake

@@ -0,0 +1,31 @@
+#! /usr/bin/env ruby
+
+require 'achoo/awake'
+require 'date'
+require 'optparse'
+
+def main
+  awake = Achoo::Awake.new
+  
+  options = {
+    :all => false,
+  }
+
+  optparse = OptionParser.new do|opts|
+    opts.banner = "Usage: $0 [options]"
+
+    opts.on( '-a', '--all', 'FIX' ) do
+      options[:all] = true
+    end
+  end
+
+  optparse.parse!
+
+  if options[:all]
+    awake.all
+  else
+    awake.at(Date.today)
+  end
+end
+
+main

data/bin/ical

@@ -0,0 +1,17 @@
+#! /usr/bin/env ruby
+
+require 'achoo/ical'
+require 'achoo/rc_loader'
+require 'optparse'
+
+include Achoo::RCLoader
+
+def main
+  load_rc
+
+  RC[:ical].each do |config|
+    Achoo::ICal.from_http_request(config).print_events(Date.parse(ARGV[0]))
+  end
+end
+
+main

data/bin/vcs_commits

@@ -0,0 +1,31 @@
+#! /usr/bin/env ruby
+
+require 'achoo/rc_loader'
+require 'achoo/vcs'
+require 'date'
+require 'optparse'
+
+include Achoo::RCLoader
+
+def main
+  load_rc
+
+  options = {
+    :date => Date.today,
+  }
+  
+  optparse = OptionParser.new do|opts|
+    opts.banner = "Usage: $0 [options]"
+  
+    opts.on( '-d', '--date [DATE]', 'FIX' ) do |date|
+      options[:date] = Date.parse(date)
+    end
+  end
+  
+  optparse.parse!
+  
+  Achoo::VCS.print_logs_for(options[:date], RC[:vcs_dirs])
+
+end
+
+main

data/lib/achoo.rb

@@ -0,0 +1,140 @@
+class Achoo; end
+
+require 'achoo/term'
+require 'achoo/ui'
+require 'logger'
+require 'mechanize'
+
+
+
+class Achoo
+
+  include Achoo::UI::Commands
+  include Achoo::UI::ExceptionHandling
+  include Achoo::UI::RegisterHours
+  
+  def initialize(log=false)
+    @agent = Mechanize.new
+    if log
+      @agent.log = Logger.new("achoo_http.log")
+    end
+  end
+
+  def start
+    begin
+      print_welcome
+      login
+      scrape_urls
+      command_loop
+    rescue SystemExit => e
+      raise
+    rescue Exception => e
+      handle_fatal_exception("Something bad happened. Shutting down.", e)
+    end
+  end
+
+
+  def print_welcome
+    puts Term.shadowbox("Welcome to Achoo!")
+  end
+
+
+  def command_loop
+    while true
+      begin
+        trap("INT", "DEFAULT");
+        answer = Term.choose('[1]',
+                             ["Register hours",
+                              "Show flexitime balance",
+                              "Day hour report",
+                              "Week hour report",
+                              "Holiday balance",
+                              "Lock month",
+                             ],
+                             "Exit",
+                             ['q', 'Q', ''])
+        dispatch(answer)
+      rescue Interrupt
+        puts # Add a new line in case we are prompting
+      end
+    end
+  end
+
+  
+  def dispatch(command)
+      case command
+      when '0', 'q', 'Q'
+        exit
+      when '1', ''
+        register_hours(@agent)
+      when '2'
+        show_flexi_time(@agent)
+      when '3'
+        show_registered_hours_for_day(@agent)
+      when '4'
+        show_registered_hours_for_week(@agent)
+      when '5'
+        show_holiday_report(@agent)
+      when '6'
+        lock_month(@agent)
+      end
+  end
+
+
+  def scrape_urls 
+    page = @agent.get(@agent.current_page.frames.find {|f| f.name == 'menu'}.href)
+    menu_links = page.search('a.menuItemLevel2')
+    
+    RC[:hour_registration_url] = menu_link_to_url(menu_links, 'Time Registration')
+    RC[:lock_months_url]       = menu_link_to_url(menu_links, 'Lock months')
+    RC[:holiday_report_url]    = menu_link_to_url(menu_links, 'Holiday report')
+    RC[:hour_admin_url]        = RC[:hour_registration_url]
+  end
+
+  def menu_link_to_url(menu_links, text)
+    a_tag = menu_links.find {|a| 
+      a.text.strip == text
+    }
+    
+    if a_tag.nil?
+      raise Exception.new("Could not find the '#{text}' link in Achievo.\nMake sure that language is 'English' and theme is 'no value' in Achievo's user preferences.\nYou must delete ~/.achoo_cookies.yml for these changes to take affect.")
+    end
+
+    url = a_tag.attribute('onclick').value.match(/window\.open\('([^']+)/)[1]
+    return "#{RC[:url]}#{url}"
+  end
+
+
+  def login
+    load_cookies
+
+    puts "Fetching data ..."
+    page = @agent.get(RC[:url])
+
+    return if page.forms.empty?
+
+    puts "Logging in ..."
+
+    form = page.forms.first
+    form.auth_user = RC[:user]
+    form.auth_pw   = RC[:password]
+    page = @agent.submit(form, form.buttons.first)
+
+    if page.body.match(/Username and\/or password are incorrect. Please try again./)
+      raise "Username and/or password are incorrect."
+    end
+
+    @agent.cookie_jar.save_as("#{ENV['HOME']}/.achoo_cookies.yml")
+  end
+
+
+  def load_cookies
+    cookies_file = "#{ENV['HOME']}/.achoo_cookies.yml"
+    if FileTest.exists? cookies_file
+      @agent.cookie_jar.load(cookies_file)
+    end
+  end
+
+
+end
+

data/lib/achoo/awake.rb

@@ -0,0 +1,130 @@
+require 'achoo/open_timespan'
+require 'achoo/timespan'
+require 'achoo/system'
+
+class Achoo; end
+
+class Achoo::Awake
+    
+  def initialize
+    log = array_merge(wtmp, suspend) {|a, b| a[0] >= b[0] }
+    log.unshift([Time.now, :now])
+    @sessions = sessions(log)
+  end
+
+  def at(date)
+    span = Achoo::Timespan.new(date, date+1)
+    @sessions.each do |s|
+      print_session(s, span) if s[0].overlaps?(span)
+    end
+  end
+  
+  def all
+    @sessions.each do |s|
+      print_session(s)
+    end
+  end
+
+  private
+
+  def print_session(s, span=nil)
+    puts "Powered on: " << s[0].to_s
+    s[1].each do |t|
+      puts "  Awake: " << t.to_s if span.nil? || t.overlaps?(span)
+    end
+  end
+
+  def sessions(log)
+    sessions = []
+    group(log).each do |g|
+      raise "Session doesn't begin with a boot event" unless g.last[1] == :boot
+      
+      session = []
+      case g.first[1]
+      when :now, :halt, :suspend
+        # We know the end of the session
+        session << Achoo::Timespan.new(g.last[0], g.first[0])
+      else # :awake, :boot
+        # We don't know the end of the session
+        session << Achoo::OpenTimespan.new(g.last[0], g.first[0])
+        g.unshift([-1, :crash])
+      end
+      
+      raise "Session must consist of even number of events. Found #{g.length}" unless g.length.even?
+
+
+      session << []
+      unless g.length == 2
+        i = 0
+        while i < g.length-1
+          klass = g[i][1] == :crash ? Achoo::OpenTimespan : Achoo::Timespan
+          session[1] << klass.new(g[i+1][0], g[i][0])
+          i += 2
+        end
+      end
+      
+      sessions << session
+    end
+    sessions
+  end
+
+  def array_merge(a1, a2)
+    array = []
+    until a1.empty? or a2.empty?
+      if yield(a1.first, a2.first)
+        array << a1.shift
+      else
+        array << a2.shift
+      end
+    end
+    array.concat(a1).concat(a2)
+    array
+  end
+
+  def suspend
+    log     = Achoo::System::PMSuspend.new.reverse
+    new_log = []
+    log.each do |entry|
+      new_log << [entry.time, entry.action == 'Awake.' ? :awake : :suspend]
+    end
+    new_log
+  end
+
+  def wtmp
+    log     = Achoo::System::Wtmp.new.reverse
+    new_log = []
+    log.each do |entry|
+      if entry.boot_event?
+        new_log << [entry.time, :boot]
+      elsif entry.halt_event?
+        new_log << [entry.time, :halt]
+      end
+    end
+    new_log
+  end
+
+  def group(log)
+    grouped_log = []
+    group = nil
+    log.each do |i|
+      if i[1] == :halt || i[1] == :now
+        group = [i]
+      else
+        if group.nil?
+          # Crashed
+          group = []
+        end
+        if i[1] == :boot
+          group << i
+          grouped_log << group
+          group = nil
+        else
+          group << i
+        end    
+      end
+    end
+    grouped_log
+  end
+
+
+end