achoo 0.3

data/lib/achoo/hour_registration_form_ranged.rb

@@ -0,0 +1,45 @@
+require 'achoo/hour_registration_form'
+
+class Achoo::HourRegistrationFormRanged < Achoo::HourRegistrationForm
+
+  def initialize(agent)
+    super
+    
+    @page = @agent.get(atk_submit_to_url(@page.link_with(:text => 'Select range').href))
+    @form = @page.form('entryform')
+  end
+
+  def date=(date_range)
+    super(date_range[0])
+
+    to_day_field.value   = date_range[1].strftime('%d')
+    to_month_field.value = date_range[1].strftime('%m')
+    to_year_field.value  = date_range[1].year
+  end
+
+  def date
+    start = super
+    finish = Date.new(to_year_field.value.to_i, to_month_field.value.to_i, 
+                      to_day_field.value.to_i)
+    [start, finish]
+  end
+  
+  private
+
+  def to_day_field
+    @form.field_with(:name => 'todate[day]')
+  end
+
+  def to_month_field
+    @form.field_with(:name => 'todate[month]')
+  end
+
+  def to_year_field
+    @form.field_with(:name => 'todate[year]')
+  end
+
+  def date_to_s
+    date.map {|d| d.strftime("%Y-%m-%d")}.join(" -> ")
+  end
+    
+end

data/lib/achoo/ical.rb

@@ -0,0 +1,58 @@
+require 'achoo/timespan'
+require 'achoo/ui/exception_handling'
+require 'net/https'
+require 'ri_cal'
+
+class Achoo; end
+
+class Achoo::ICal
+
+  Achoo::UI::ExceptionHandling
+
+  def self.from_http_request(params)
+    http = Net::HTTP.new(params[:host], params[:port])
+    http.use_ssl = true
+    http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+    ics = http.start do |http|
+      request = Net::HTTP::Get.new(params[:path])
+      request.basic_auth(params[:user], params[:pass])
+      response = http.request(request)
+      response.body
+    end
+    self.new(ics)
+  end
+
+  def initialize(ics_str)
+    @calendar = RiCal.parse_string(ics_str).first
+  end
+
+  def print_events(date, io=$stdout)
+    arg_start = date
+    arg_end   = date + 1
+
+    @calendar.events.each do |e|
+      begin
+        if !e.x_properties['X-MICROSOFT-CDO-ALLDAYEVENT'].empty? && e.x_properties['X-MICROSOFT-CDO-ALLDAYEVENT'].first.value == 'TRUE'
+          # FIX handle this
+        elsif e.recurs?
+          e.occurrences({:overlapping => [arg_start, arg_end]}).each do |o|
+            print_event(o, io)
+          end
+        elsif e.dtstart >= arg_start && e.dtstart <= arg_end \
+          || e.dtend  >= arg_start && e.dtend <= arg_end
+          print_event(e, io)
+        end
+      rescue Exception => e
+        handle_exception("Failed to process calendar event", e)
+      end
+    end
+  end
+
+  private
+
+  def print_event(e, io)
+    dti = Achoo::Timespan.new(e.dtstart.to_s, e.dtend.to_s)
+    io.printf "%s: %s\n", dti, e.summary
+  end
+end
+

data/lib/achoo/lock_month_form.rb

@@ -0,0 +1,40 @@
+class Achoo; end
+
+class Achoo::LockMonthForm
+
+  def initialize(agent)
+    @agent = agent
+  end
+
+  def lock_month(period)
+    page  = @agent.get(RC[:lock_months_url])
+    @form  = page.form('entryform')
+
+    @form.period = period
+    unless user_select.nil?
+      user_select.options.each do |opt|
+        if opt.text.match(/\(#{RC[:user]}\)$/)
+          opt.select
+        end
+      end
+    end
+
+  end
+
+  def print_values
+    puts "Month: #{@form.period}"
+    puts " User: #{user_select.value}" unless user_select.nil?
+
+  end
+  
+  def submit
+    @form.submit
+  end
+
+  private
+
+  def user_select
+    @form.field_with(:name => 'userid')
+  end
+  
+end

data/lib/achoo/open_timespan.rb

@@ -0,0 +1,13 @@
+require 'achoo/timespan'
+
+class Achoo; end
+
+class Achoo::OpenTimespan < Achoo::Timespan
+
+  def to_s
+    s = super
+    s.sub!(/^\([^)]+\)/, '(?+??:??)')
+    s.sub!(/- .*$/, '- ?')
+    s
+  end
+end

data/lib/achoo/rc_loader.rb

@@ -0,0 +1,55 @@
+require 'achoo/term'
+
+class Achoo; end
+
+module Achoo::RCLoader
+
+  def load_rc(rc_file="#{ENV['HOME']}/.achoo")
+    #create_empty_rc_if_not_exists(rc_file)
+    file_permissions_secure?(rc_file)
+
+    load rc_file
+
+    verify_rc_contents(rc_file)
+  end
+
+  private
+
+  def file_permissions_secure?(rc_file)
+    # FIX test to is to strict
+    if File.stat(rc_file).mode != 0100600
+      puts Achoo::Term.fatal "Insecure permissions on #{rc_file}"
+      exit 1
+    end
+  end
+
+  def create_empty_rc_if_not_exists(rc_file)
+    return if FileTest.exist?(rc_file)
+
+    FileUtils.touch(rc_file)
+    FileUtils.chmod(0600, rc_file)
+  end
+
+  def verify_rc_contents(rc_file)
+    unless Object.const_defined?('RC')
+      puts Achoo::Term.fatal "Malformed run control file: No RC constant defined"
+      exit 1
+    end
+
+    %w(url user password).each do |key|
+      unless RC.has_key?(key.to_sym)
+        puts Achoo::Term.fatal "Missing mandatory run control configuration variable: #{key}"
+        exit 1
+      end
+    end
+
+    %w(vcs_dirs ical).each do |key|
+      unless RC.has_key?(key.to_sym)
+        puts Achoo::Term.warn "Missing run control configuration variable: #{key}. " \
+             + "Add it to #{rc_file} to get rid of this warning"
+        RC[key] = []
+      end
+    end
+  end
+
+end

data/lib/achoo/system.rb

@@ -0,0 +1,6 @@
+class Achoo
+  module System
+    autoload :Wtmp,      'achoo/system/wtmp.rb'
+    autoload :PMSuspend, 'achoo/system/pm_suspend.rb'
+  end
+end

data/lib/achoo/system/pm_suspend.rb

@@ -0,0 +1,30 @@
+require 'achoo/system'
+require 'time'
+
+class Achoo::System::PMSuspend < Array
+
+  class LogEntry
+    attr :time
+    attr :action
+    
+    def initialize(time_str, action)
+      @action = action
+      @time   = Time.parse(time_str)
+    end
+  end
+
+  def initialize(glob='/var/log/pm-suspend.log*')
+    super()
+    Dir.glob(glob).sort.reverse.each do |file|
+      next if file =~ /\.gz$/ # FIX uncompress?
+      File.open(file, 'r') do |fh|
+        fh.readlines.each do |l|
+          l.chop!
+          next unless l =~ /Awake|performing suspend/
+          self << LogEntry.new(*l.split(': '))
+        end
+      end
+    end
+  end
+
+end

data/lib/achoo/system/wtmp.rb

@@ -0,0 +1,18 @@
+require 'achoo/binary'
+require 'achoo/system'
+
+class Achoo::System::Wtmp < Array
+
+  def initialize(glob='/var/log/wtmp*')
+    super()
+    chunk_size = Achoo::Binary::UTMPRecord.bin_size
+    Dir.glob(glob).sort.reverse.each do |file|
+      File.open(file, 'r') do |io|
+        while (bytes = io.read(chunk_size)) 
+          self << Achoo::Binary::UTMPRecord.new(bytes)
+        end
+      end
+    end
+  end
+  
+end

data/lib/achoo/term.rb

@@ -0,0 +1,76 @@
+# encoding: utf-8
+
+class Achoo; class Term; end; end
+
+Achoo::Term.autoload :Menu,  'achoo/term/menu'
+Achoo::Term.autoload :Table, 'achoo/term/table'
+
+class Achoo::Term
+
+  def self.bold(text); "\e[1m#{text}\e[0m"; end
+
+  def self.underline(text); "\e[4m#{text}\e[0m"; end
+
+  def self.warn(text); "\e[1;33m#{text}\e[0m"; end
+
+  def self.fatal(text); "\e[1;31m#{text}\e[0m"; end
+
+  def self.password
+    `stty -echo`
+    pas = ask('Password')
+    `stty echo`
+    pas
+  end
+
+  def self.ask(question='')
+    answer = nil
+    loop do
+      print bold("#{question}> ")
+      $stdout.flush
+      answer = gets
+
+      # Answer is nil if user hits C-d on an empty input
+      if answer.nil?
+        puts
+        exit
+      end
+      
+      answer.strip! unless answer.nil?
+
+      # FIX move this to achoo.rb?
+      unless $stdin.tty?
+        puts answer
+      end
+      break unless a_little_something(answer)
+    end
+    answer
+  end
+
+  def self.choose(question, entries, special=nil, additional_valid_answers=[])
+    menu = Achoo::Term::Menu.new(question, entries, special, additional_valid_answers)
+    menu.print_ask_and_validate
+  end
+
+  private
+
+  def self.a_little_something(answer)
+    return false if answer.nil?
+
+    case answer.downcase
+    when 'bless you!', 'gesundheit!'
+      puts "Thank you!"
+      return true
+    else
+      return false
+    end
+  end
+
+  def self.shadowbox(text)
+    x =  "┌──────────────────────────────────────────┐ \n"
+    x << "│ #{text.center(40)} " <<                 "│▒\n"
+    x << "└──────────────────────────────────────────┘▒\n"
+    x << "  ▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒\n"
+    x
+  end
+
+end

data/lib/achoo/term/menu.rb

@@ -0,0 +1,55 @@
+require 'achoo/term'
+
+class Achoo
+  class Term
+    class Menu
+
+      def initialize(question, entries, special=nil, additional_valid_answers=[])
+        @question = question
+        @entries  = entries
+        @special  = special
+
+        @valid = {}
+        @valid['0'] = true unless @special.nil?
+        1.upto(@entries.length).each {|i| @valid[i.to_s] = true}
+        additional_valid_answers.each {|a| @valid[a] = true}
+      end
+
+      def print_ask_and_validate()
+        return nil if @entries.empty?
+
+        print_menu()
+        return '1' if only_one_option?
+        
+        loop do
+          answer = Achoo::Term.ask(@question)
+          if @valid[answer]
+            return answer
+          else
+            puts "Invalid value. Must be one of " << @valid.keys.sort.join(',')
+          end
+        end
+      end
+      
+      private
+
+      def print_menu
+        format = menu_item_format
+        @entries.each_with_index do |entry, i|
+          printf format, i+1, entry
+        end
+        printf format, 0, @special unless @special.nil?
+      end
+
+      def only_one_option?
+        @entries.length == 1 && @special.nil?
+      end
+      
+      def menu_item_format
+        max_digits = Math.log10(@entries.length).floor + 1
+        " %#{max_digits}d. %s\n"
+      end
+
+    end
+  end
+end

data/lib/achoo/term/table.rb

@@ -0,0 +1,101 @@
+# encoding: utf-8
+
+if RUBY_VERSION < "1.9"
+  $KCODE = 'u'
+  require 'jcode'
+end
+
+# Unicode box drawing characters
+#
+# http://en.wikipedia.org/wiki/Box-drawing_characters
+#
+#
+#      0 1 2 3 4 5 6 7 8 9 A B C D E F
+#
+# 2500 ─ ━ │ ┃ ┄ ┅ ┆ ┇ ┈ ┉ ┊ ┋ ┌ ┍ ┎ ┏
+#
+# 2510 ┐ ┑ ┒ ┓ └ ┕ ┖ ┗ ┘ ┙ ┚ ┛ ├ ┝ ┞ ┟
+#
+# 2520 ┠ ┡ ┢ ┣ ┤ ┥ ┦ ┧ ┨ ┩ ┪ ┫ ┬ ┭ ┮ ┯
+#
+# 2530 ┰ ┱ ┲ ┳ ┴ ┵ ┶ ┷ ┸ ┹ ┺ ┻ ┼ ┽ ┾ ┿
+#
+# 2540 ╀ ╁ ╂ ╃ ╄ ╅ ╆ ╇ ╈ ╉ ╊ ╋ ╌ ╍ ╎ ╏
+#
+# 2550 ═ ║ ╒ ╓ ╔ ╕ ╖ ╗ ╘ ╙ ╚ ╛ ╜ ╝ ╞ ╟
+#
+# 2560 ╠ ╡ ╢ ╣ ╤ ╥ ╦ ╧ ╨ ╩ ╪ ╫ ╬ ╭ ╮ ╯
+#
+# 2570 ╰ ╱ ╲ ╳ ╴ ╵ ╶ ╷ ╸ ╹ ╺ ╻ ╼ ╽ ╾ ╿
+
+class Achoo; class Term; end; end
+
+class Achoo::Term::Table
+
+  def initialize(headers, data_rows, summaries=nil)
+    @headers   = headers
+    @data_rows = data_rows
+    @summaries = summaries
+  end
+
+  def print(io=$stdout)
+    lengths   = calculate_table_cell_widths
+    format    = build_format(lengths)
+    headers   = center_table_headers(lengths)
+    separator = lengths.map {|length| '─'*(length+2)}
+    
+
+    io.print '┌' << separator.join('┬') << "┐\n"
+    io.print '│ ' << headers.join(' │ ') << " │\n"
+    io.print '├' << separator.join('┼') << "┤\n"
+    @data_rows.each {|r| io.printf format, *r }
+    unless @summaries.nil? || @data_rows.length == 1
+      io.print '├' << separator.join('┼') << "┤\n"
+      io.printf format, *@summaries
+    end
+    io.print '└' << separator.join('┴') << "┘\n"
+  end
+
+  private
+
+  def center_table_headers(lengths)
+    headers = @headers.dup
+    lengths.each_with_index do |len,i|
+      headers[i] = headers[i].center(len)
+    end
+    headers
+  end
+
+  def calculate_table_cell_widths
+    lengths = []
+    @headers.each_with_index do |h, i|
+      lengths[i] = RUBY_VERSION < '1.9' ? h.jlength : h.length
+    end
+    @data_rows.each do |r|
+      r.each_with_index do |d, i|
+        len = RUBY_VERSION < '1.9' ? d.jlength : d.length
+        lengths[i] = [len, lengths[i]].max
+      end
+    end
+    lengths
+  end
+
+  def build_format(lengths)
+    is_column_left_justified = Array.new(lengths.count)
+    is_column_left_justified.fill(false)
+    
+    @data_rows.each do |r|
+      r.each_index do |c|
+        if !r[c].strip.empty? && !r[c].match(/^\d+[:.,]?\d*$/)
+          is_column_left_justified[c] = true
+        end
+      end
+    end
+    
+    lengths.reduce('│') do |f, l|
+      justify = is_column_left_justified.shift ? '-' : ''
+      f + " %#{justify}#{l}s │"
+    end + "\n"
+  end
+
+end