achoo 0.3 → 0.4.1

data/lib/achoo/binary.rb

@@ -1,7 +0,0 @@
-class Achoo
-  module Binary
-    autoload :CStruct,    'achoo/binary/cstruct'
-    autoload :UTMPRecord, 'achoo/binary/utmp_record'
-  end
-end
-

data/lib/achoo/binary/cstruct.rb

@@ -1,60 +0,0 @@
-require 'achoo/binary'
-
-class Achoo::Binary::CStruct
-
-  def initialize(bytes=nil)
-    @values = []
-    unpack(bytes) unless bytes.nil?
-  end
-
-  class << self
-
-    attr :template
-
-    def inherited(subclass)
-      subclass.instance_variable_set(:@template, '') 
-      subclass.instance_variable_set(:@count, 0)
-    end 
-
-    def char(name);  add_type(name, :char,  'c', 0); end
-    def short(name); add_type(name, :short, 's', 0); end
-    def long(name);  add_type(name, :long,  'l', 0); end
-    def quad(name);  add_type(name, :quad,  'q', 0); end
-
-    def string(name, length); add_type(name, :string, 'A', '', length); end
-
-    def bin_size
-      @bin_size ||= template.split('').select {|c| c =~ /[[:alpha:]]/}.map do |c|
-        c == 'A' ? '' : 0
-      end.pack(template).length
-    end
-
-    private 
-
-    def add_type(name, type, temp, zero, length=nil)
-      template << temp
-      template << length.to_s if type == :string
-      index = @count
-      @count += 1
-
-      send(:define_method, name) do
-        @values[index]
-      end
-
-      send(:define_method, "#{name}=") do |val|
-        @values[index] = val
-      end
-    end
-
-  end
-
-  def unpack(str)
-    @values = str.unpack(self.class.template)
-  end
-
-  def pack
-    t = self.class.template.tr('A', 'a')
-    @values.pack(t)
-  end
-
-end

data/lib/achoo/binary/utmp_record.rb

@@ -1,59 +0,0 @@
-require 'achoo/binary'
-
-class Achoo::Binary::UTMPRecord < Achoo::Binary::CStruct
-  long   :record_type
-  long   :process_id
-  string :device_name,        32
-  string :inittab_id,          4
-  string :username,           32
-  string :hostname,          256
-  short  :termination_status
-  short  :exit_status
-  long   :session_id
-  long   :seconds
-  long   :milliseconds
-  long   :ip_address1
-  long   :ip_address2
-  long   :ip_address3
-  long   :ip_address4
-  string :unused,             20
-
-
-  TYPE_MAP = [:empty,
-              :run_lvl,
-              :boot,
-              :new_time,
-              :old_time,
-              :init,
-              :login,
-              :normal,
-              :term,
-              :account,
-             ]
-  
-  def time
-    return nil if seconds.nil?
-    @time ||= Time.at(seconds, milliseconds)
-  end
-  
-  def record_type_symbol
-    TYPE_MAP[record_type]
-  end
-
-  def record_type_symbol=(sym)
-    @values[0] = (TYPE_MAP.find_index(sym))
-  end
-  
-  def to_s
-    sprintf "%s  %-7s  %-8s %s", time.strftime('%F_%T'), record_type_symbol, username, device_name
-  end
-
-  def boot_event?
-    record_type_symbol == :boot
-  end
-  
-  def halt_event?
-    record_type_symbol == :term && device_name == ':0'
-  end
-
-end

data/lib/achoo/form.rb

@@ -1,18 +0,0 @@
-
-class Achoo; end
-
-class Achoo::Form
-
-  def date=(date)
-    # Day and month must be prefixed with '0' if single
-    # digit. Date.day and Date.month doesn't do this. Use strftime
-    day_field.value   = date.strftime('%d')
-    month_field.value = date.strftime('%m')
-    year_field.value  = date.year
-  end
-
-  def date
-    Date.new(year_field.value.to_i, month_field.value.to_i, 
-             day_field.value.to_i)
-  end
-end

data/lib/achoo/hour_administration_form.rb

@@ -1,131 +0,0 @@
-require 'achoo/form'
-require 'achoo/term/table'
-
-class Achoo::HourAdministrationForm < Achoo::Form
-
-  def initialize(agent)
-    @agent = agent
-    @page  = nil
-  end
-
-  def show_registered_hours_for_day(date)
-    show_registered_hours(date, 'dayview', '#rl_1 tr')
-  end
-  
-  def show_registered_hours_for_week(date)
-    show_registered_hours(date, 'weekview', '//form[@name="weekview"]/following::table/tr')
-  end
-
-  def flexi_time(date)
-    set_page_to_view_for_date('dayview', date)
-
-    @page.body.match(/Flexi time balance: (-?\d+:\d+)/)[1]
-  end
-
-  private
-  
-  def show_registered_hours(date, view, query)
-    set_page_to_view_for_date(view, date)
-
-    columns     = choose_source_columns(view, query)
-    source_rows = @page.search(query)
-    headers     = extract_headers(source_rows, columns, view)
-    data_rows   = extract_data_rows(source_rows, columns)
-    summaries   = extract_summaries(source_rows, data_rows, columns)
-
-    Achoo::Term::Table.new(headers, data_rows, summaries).print
-  end
-  
-  def set_page_to_view_for_date(view, date)
-    @page ||= @agent.get(RC[:hour_admin_url])
-
-    link = @page.link_with(:text => view.capitalize)
-    @form = @page.form(view)
-    unless link.nil?
-      puts "Fetching #{view} ..."
-      @page = link.click
-      @form = @page.form(view)
-    end
-    unless date == self.date
-      @page = get_page_for(date)
-      @form = @page.form(view)
-    end
-
-    @page
-  end    
-
-  def choose_source_columns(view, query)
-    columns = nil
-    if view == 'dayview'
-      # Ignore 'Billing billed', 'Billing marked', and 'Billing total'
-      columns = [0,1,2,3,6,8,9]
-      # Achievo prepends an extra column dynamically if there are
-      # data rows.
-      unless @page.search(query + ' td').empty?
-        columns.collect! {|c| c + 1}
-      end
-    end
-    columns
-  end
-
-  def extract_headers(source_rows, columns, view)
-    headers = source_rows.first.css('th')
-    unless columns.nil?
-      headers = headers.to_a.values_at(*columns)
-    end
-    headers = headers.map {|th| th.content.strip}
-    if view == 'weekview'
-      headers = headers.map {|th| th.gsub(/\s+/, ' ') }
-    end
-    headers
-  end
-
-  def extract_data_rows(source_rows, columns)
-    data_rows = []
-    source_rows.each do |tr|
-      cells = tr.css('td')
-      next if cells.empty?
-      unless columns.nil?
-        cells = cells.to_a.values_at(*columns)
-      end
-      data_rows << fix_empty_cells(cells.map {|td| td.content.strip})
-    end
-    data_rows
-  end
-
-  def extract_summaries(source_rows, data_rows, columns)
-    summaries = nil
-    unless data_rows.empty?
-      summaries = source_rows.last.css('th')
-      unless columns.nil?
-        summaries = summaries.to_a.values_at(*columns)
-      end
-      summaries = summaries.map {|th| th.content.strip }
-      fix_empty_cells(summaries)
-    end
-    summaries
-  end
-
-  def get_page_for(date)
-    puts "Fetching data for #{date} ..."
-    self.date = date
-    @page = @form.submit
-  end
-
-  def day_field
-    @form.field_with(:name => 'viewdate[day]')
-  end
-  
-  def month_field
-    @form.field_with(:name => 'viewdate[month]')
-  end
-  
-  def year_field
-    @form.field_with(:name => 'viewdate[year]')
-  end
-
-  def fix_empty_cells(row)
-    row.collect! {|c| c == "\302\240" ? '  ' : c} # UTF-8 NO-BREAK-SPACE
-  end
-
-end  

data/lib/achoo/hour_registration_form.rb

@@ -1,227 +0,0 @@
-require 'achoo/form'
-require 'achoo/hour_administration_form'
-
-class Achoo::HourRegistrationForm < Achoo::Form
-
-  def initialize(agent)
-    @agent = agent
-    @page  = @agent.get(RC[:hour_registration_url])
-    @form  = @page.form('entryform')
-
-    if @form.nil?
-      # Happens if the user has viewed a day or week report for a
-      # locked month. Fetching todays day report should fix this in
-      # most cases.
-
-      # FIX Ugly call to a private method using send()
-      haf    = Achoo::HourAdministrationForm.new(@agent)
-      @page  = haf.send(:set_page_to_view_for_date, 'dayview', Date.today)
-      @form  = @page.form('entryform')
-    end
-
-    if @form.nil?
-      raise "Failed to retrieve the hour registration form.\nThe likely cause is that you have locked the current month, which is a silly thing to do."
-    end
-
-    @projects_seen = {}
-    @phases_seen   = {}
-
-    # Need to preselect this for some reason
-    @form.field_with(:name => 'billpercent').options.first.select
-    # Preselecting this one as well, just in case
-    @form.field_with(:name => 'workperiod').options.first.select
-  end
-
-  def project
-    extract_number_from_projectid(@form.projectid)
-  end
-
-  def project=(projectid)
-    @form.projectid = "project.id='#{projectid}'"
-  end
-
-  def remark=(remark)
-    @form.remark = remark
-  end
-
-  def hours=(hours)
-    @form.time = hours
-  end
-
-  def phase
-    @form.phaseid.match(/phase\.id='(\d+)'/)[1]
-  end
-
-  def phase=(phaseid)
-    @form.phaseid   = "phase.id='#{phaseid}'"
-  end
-
-  def workperiod=(workperiod)
-    @form.workperiod = "workperiod.id='#{workperiod}'"
-  end
-
-  def billing=(billing)
-    @form.billpercent = "billpercent.id='#{billing}'"
-  end
-
-
-  def worktime_periods
-    @form.field_with(:name => 'workperiod').options.collect do |opt|
-      [opt.value.match(/workperiod\.id='(\d+)'/)[1], opt.text]
-    end
-  end
-
-  def billing_options
-    @form.field_with(:name => 'billpercent').options.collect do |opt|
-      [opt.value.match(/billpercent\.id='(\d+)'/)[1], opt.text]
-    end
-  end
-
-  def phases_for_selected_project
-    partial_page = retrieve_project_phases_page
-    page         = create_page_from_partial(partial_page)
-    field        = page.forms.first.field_with(:name => 'phaseid')
-    
-    phases = []
-    if field.respond_to?(:options)
-      field.options.each do |opt|
-        phases << [extract_number_from_phaseid(opt.value), opt.text]
-      end
-    else
-      partial_page.body.match(/(^[^<]+)&nbsp;&nbsp;</)
-      phases << [extract_number_from_phaseid(field.value), $1]
-    end
-    
-    phases.each {|p| @phases_seen[p[0]] = p[1]}
-
-    return phases
-  end
-  
-  def recent_projects
-    projects = []
-    @form.field_with(:name => 'projectid').options.each do |opt|
-      val = opt.value["project.id='".length..-2]
-      projects << [val, opt.text]
-    end
-
-    projects.each {|p| @projects_seen[p[0]] = p[1]}
-
-    projects
-  end
-
-  def all_projects
-    puts "Getting project page #1..."
-    projects_page = @agent.get(projects_url)
-    projects = scrape_projects(projects_page)
-
-    i = 2
-    while (link = projects_page.link_with(:text => 'Next'))
-      puts "Getting project page ##{i}..."
-      projects_page = link.click
-      projects.merge!(scrape_projects(projects_page))
-      i += 1
-    end
-
-    projects.keys.sort.collect do |name|
-      id   = projects[name][0]
-      text = "#{projects[name][1]}: #{name}"
-      @projects_seen[id] = text
-      [id, text]
-    end
-  end
-
-  def print_values
-    format = "%10s: \"%s\"\n"
-    printf format, 'date',     date_to_s
-    printf format, 'project',  @projects_seen[project]
-    printf format, 'phase',    @phases_seen[phase]
-    printf format, 'remark',   @form.remark
-    printf format, 'hours',    @form.time
-    printf format, 'worktime', @form.field_with(:name => 'workperiod').options.first.text
-    printf format, 'billing',  @form.field_with(:name => 'billpercent').options.first.text
-
-    # @form.fields.each do |field|
-    #   printf format, field.name, field.value
-    # end
-
-  end
-
-  def submit
-    @form.submit()
-  end
-
-  private
-
-  def date_to_s
-    date.strftime("%Y-%m-%d")
-  end
-
-  def retrieve_project_phases_page
-    old = {
-      :atkaction => @form.atkaction,
-      :action    => @form.action,
-    }
-    
-    @form.action = RC[:url]+"/dispatch.php?atkpartial=attribute.phaseid.refresh"
-    @form.atkaction = 'add'
-    partial_page = @form.submit
-    @form.action    = old[:action]
-    @form.atkaction = old[:atkaction]
-    
-    return partial_page
-  end
-
-  def create_page_from_partial(partial_page)
-    body = "<html><head></head><body><form>#{partial_page.body}</form></body></html>"
-    page = Mechanize::Page.new(nil, {'content-type' => 'text/html; charset=iso-8859-1'},
-                               body, nil, @agent)
-  end
-
-  def day_field
-    @form.field_with(:name => 'activitydate[day]')
-  end
-
-  def month_field
-    @form.field_with(:name => 'activitydate[month]')
-  end
-
-  def year_field
-    @form.field_with(:name => 'activitydate[year]')
-  end
-
-  def extract_number_from_projectid(projectid)
-    projectid.match(/project\.id='(\d+)'/)[1]
-  end
-
-  def extract_number_from_phaseid(projectid)
-    projectid.match(/phase\.id='(\d+)'/)[1]
-  end
-
-  def projects_url
-    atk_submit_to_url(@page.link_with(:text => 'Select project').href)
-  end
-
-  def atk_submit_to_url(atk_submit)
-    href = atk_submit['javascript:atkSubmit("__'.length..-3]
-    href.gsub!('_13F', '?')
-    href.gsub!('_13D', '=')
-    href.gsub!('_126', '&')
-    href.gsub!('_125', '%')
-    return RC[:url] + '/' + href
-  end
-
-  def scrape_projects(projects_page)
-    projects = {}
-    projects_page.search('table#rl_1 tr').each do |tr|
-      cells = tr.search('td')
-      next if cells.empty?
-      projects[cells[1].text.strip] = [
-        cells[1].at_css('a').attribute('href').to_s.match('project.id%3D%27(\d+)%27')[1],
-        cells[0].text.strip,
-      ]
-    end
-
-    return projects
-  end
-
-end