milton 1.0.0

data/History.txt

@@ -0,0 +1,6 @@
+=== 1.0.0 / 2009-02-23
+
+* 1 major enhancement
+
+  * Birthday!
+

data/Manifest.txt

@@ -0,0 +1,8 @@
+History.txt
+Manifest.txt
+README.txt
+Rakefile
+bin/milton
+bin/recorder
+lib/milton.rb
+test/test_milton.rb

data/README.txt

@@ -0,0 +1,63 @@
+= milton
+
+* http://seattlerb.rubyforge.org/milton
+
+== DESCRIPTION:
+
+Milton fills out your ADP ezLaborManager timesheet
+
+== FEATURES/PROBLEMS:
+
+* Fills out timesheets for arbitrary weeks
+* Does not account for time off
+
+== SYNOPSIS:
+
+To fill out the current week:
+
+  milton
+
+To view your timesheet for the current week:
+
+  milton --view
+
+To fill out a timesheet for an arbitrary week:
+
+  milton --date=02/25/2009
+
+To view a timesheet for an arbitrary week:
+
+  milton --view --date=02/25/2009
+
+== REQUIREMENTS:
+
+* ADP ezLaborManager client name, username and password
+
+== INSTALL:
+
+* sudo gem install milton
+
+== LICENSE:
+
+(The MIT License)
+
+Copyright (c) 2009 Aaron Patterson, Eric Hodel
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+'Software'), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Rakefile

@@ -0,0 +1,12 @@
+# -*- ruby -*-
+
+require 'rubygems'
+require 'hoe'
+require './lib/milton.rb'
+
+Hoe.new('milton', Milton::VERSION) do |p|
+  p.developer('Aaron Patterson', 'aaronp@rubyforge.org')
+  p.extra_deps      = [['mechanize', '>= 0.9.2']]
+end
+
+# vim: syntax=Ruby

data/bin/milton

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby -w
+
+require 'milton'
+
+Milton.run
+

data/bin/recorder

@@ -0,0 +1,27 @@
+#!/usr/bin/env ruby -w
+
+require 'webrick'
+require 'webrick/httpproxy'
+require 'logger'
+
+last_request_time = Time.now
+human_requests = Logger.new('requests.log')
+
+s = WEBrick::HTTPProxyServer.new(
+  :Logger               => nil,
+  :Port                 => 8080,
+  :ProxyContentHandler  => lambda { |req,res|
+    # If the last request time is ten seconds ago, let's say that a human
+    # made it
+    if Time.now - last_request_time >= 10
+      human_requests.debug("#{req}")
+    end
+    last_request_time = Time.now
+  }
+)
+
+Signal.trap('INT') do
+  s.shutdown
+end
+
+s.start

data/lib/milton.rb

@@ -0,0 +1,293 @@
+require 'rubygems'
+require 'mechanize'
+require 'logger'
+require 'cgi'
+require 'date'
+require 'optparse'
+require 'optparse/date'
+require 'yaml'
+
+##
+# Milton fills out timesheets for the ADP ezLaborManager.
+
+class Milton
+
+  VERSION = '1.0.0'
+
+  def self.load_config config_file
+    unless File.exist? config_file then
+      open config_file, 'wb' do |f|
+        f.write YAML.dump({
+          'client_name' => 'Your client name',
+          'username'    => 'Your username',
+          'password'    => 'Your password',
+        })
+      end
+
+      raise "Please fill out #{config_file}. We've created a template there for you to edit."
+    end
+
+    YAML.load_file config_file
+  end
+
+  def self.parse_args argv
+    options = {
+      'date' => nil,
+      'view' => false
+    }
+
+    opts = OptionParser.new do |opt|
+      opt.program_name = File.basename $0
+      opt.version = Milton::VERSION
+      opt.release = nil
+      opt.banner = <<-EOF
+Usage: #{opt.program_name} [options]
+
+Milton fills out your ADP timesheet for you.  By default it fills it out for
+the current week with eight hours/day.
+      EOF
+
+      opt.separator nil
+
+      opt.on('--view',
+             'Only view your current timesheet') do |value|
+        options['view'] = value
+      end
+
+      opt.on('--date=DATE', Date,
+             'Select week by day') do |value|
+        options['date'] = value
+      end
+
+      opt.on('--month [DATE]', Date, 'Select month by day') do |value|
+        options['view'] = true  # For your safety, you can only view months
+        options['month'] = value || Date.today
+      end
+
+      opt.on('--fuck-the-man',
+             'Do not include lunch in your timesheet') do |value|
+        options['rows_per_day'] = 1
+      end
+    end
+
+    opts.parse! argv
+
+    options
+  end
+
+  def self.run argv = ARGV
+    config_file = File.join Gem.user_home, '.milton'
+
+    options = parse_args argv
+
+    config = load_config config_file
+
+    options.merge! config
+
+    new.run options
+  end
+
+  def initialize &block
+    @agent = WWW::Mechanize.new
+    @page = nil
+    @username = nil
+    yield self if block_given?
+  end
+
+  ##
+  # Sets the client name +name+
+
+  def client_name= name
+    page = @agent.get('http://workforceportal.elabor.com/ezLaborManagerNetRedirect/clientlogin.aspx')
+    @page = page.form('ClientLoginForm') { |form|
+      form.txtClientName = name
+      form.hdnTimeZone = 'Pacific Standard Time'
+      form['__EVENTTARGET'] = 'btnSubmit'
+    }.submit
+    @page = @page.form_with(:action => /ezlmportaldc2.adp.com/).submit
+    @page = @page.form_with(:action => /adp\.com/).submit
+  end
+
+  ##
+  # Logs in +username+ with +password+
+
+  def login username, password
+    @username = username
+
+    @page = @page.form('Login') { |form|
+      form['txtUserID']     = username
+      form['txtPassword']   = password
+      form['__EVENTTARGET'] = 'btnLogin'
+    }.submit
+    change_password if @page.body =~ /Old Password/
+    @page = @page.link_with(:text => 'Time Sheet').click
+  end
+
+  ##
+  # Selects the current week's timesheet
+
+  def select_current_week
+    select_week_of Date.today
+  end
+
+  def rows_per_day= rows = 2
+    @rows_per_day = rows
+    @page = @page.form('Form1') { |form|
+      form['__EVENTTARGET'] = 'SETNOOFROWS'
+      form['__EVENTARGUMENT'] = rows.to_s
+      form['__PageDirty']   = 'False'
+    }.submit
+  end
+
+  def run config
+    self.client_name = config['client_name']
+    login config['username'], config['password']
+
+    date  = config['date']
+    month = config['month']
+
+    if date then
+      select_week_of date
+    elsif month
+      select_month_of month
+    else
+      select_current_week
+    end
+
+    unless config['view'] then
+      self.rows_per_day = config['rows_per_day'] || 2
+      fill_timesheet
+    end
+
+    extract_timesheet
+  end
+
+  ##
+  # Fills in timesheet rows that don't already have data
+
+  def fill_timesheet
+    rows = []
+    last_date = nil
+
+    parse_timesheet.each do |data|
+      next if data[0].to_i > 0
+
+      department  = data[6]
+      employee_id = data[7]
+      date        = Date.parse(CGI.unescape(data[1])).strftime('%m/%d/%Y')
+
+      start, finish = starting_and_ending_timestamp(date, last_date)
+
+      rows << ['0','','False','True','False','False','False',
+      "#{date} 12:00:00 AM",
+      start,'',
+      finish,
+      '8','',
+      department,
+      employee_id,
+      '','','','','','','','','','','','','','','','','','','','','EDIT','','','','','2','','0','False']
+
+      # This reset is for the timestamp calculations.
+      last_date = date
+    end
+
+    @page = @page.form('Form1') { |form|
+      ## FIXME: Fill out this form
+      form['hdnRETURNDATA'] = rows.map { |row|
+        row.map { |value|
+          CGI.escape(value)
+        }.join('~~')
+      }.join('~||~')
+
+      form['__EVENTTARGET'] = 'TG:btnSubmitTop'
+      form['__PageDirty']   = 'True'
+    }.submit
+  end
+
+  ##
+  # Prints out your timesheet for the selected time frame
+
+  def extract_timesheet
+    timesheet = parse_timesheet
+
+    department  = timesheet.first[6]
+    employee_id = timesheet.first[7]
+
+    puts "Employee #{@username} id #{employee_id}, department #{department}"
+
+    puts "-" * 80
+
+    timesheet.each do |row|
+      if row[0] == '0' then
+        puts "#{row[2]} no time entered"
+      else
+        puts "#{row[2]} #{row[3]} to #{row[4]} for %2s hours" % row[5]
+      end
+    end
+  end
+
+  ##
+  # Selects the timesheet for the week containing +date+
+
+  def select_week_of date
+    monday = date - date.wday + 1
+    friday = monday + 4
+    select_range(monday, friday)
+  end
+
+  def select_month_of date
+    first = date - date.mday + 1
+    last  = Date.new(first.year, first.month, -1)
+    select_range(first, last)
+  end
+
+  private
+
+  def change_password
+    puts "Your password needs to be updated.  :-("
+    puts "\t#{@page.uri}"
+    exit
+  end
+
+  def select_range start, finish
+    @page = @page.form('Form1') { |form|
+      form['__EVENTTARGET'] = 'ctrlDtRangeSelector'
+      form['ctrlDtRangeSelector:SelectionItem'] = '3' # Set to this week
+      form['ctrlDtRangeSelector:BeginDate']     = start.strftime('%m/%d/%Y')
+      form['ctrlDtRangeSelector:EndDate']       = finish.strftime('%m/%d/%Y')
+      form['__PageDirty']   = 'False'
+    }.submit
+  end
+
+  ##
+  # Returns an array of arrays containing: row id, day start time, date, start
+  # time, end time, hours, department, employee id.  All values are strings.
+
+  def parse_timesheet
+    @page.body.scan(/TCMS.oTD.push\((\[.*\])\)/).map do |match|
+      match[0].gsub(/"/, '').split(',').map { |x|
+        CGI.unescape(x.strip).delete('[]')
+      }.values_at(0, 7, 8, 9, 11, 12, 14, 15)
+    end
+  end
+
+  ##
+  # Returns the starting and ending EZLabor-style timestamps for the
+  # current date row in the timesheet.
+  def starting_and_ending_timestamp(date, last_date)
+    if @rows_per_day == 2
+      if last_date == date
+        start_timestamp = "#{date} 08:30 AM"
+        end_timestamp = '12:00 PM'
+      else
+        start_timestamp = "#{date} 12:30 PM"
+        end_timestamp = '05:00 PM'
+      end
+    else
+      start_timestamp = "#{date} 08:30 AM"
+      end_timestamp = '04:30 PM'
+    end
+    return start_timestamp, end_timestamp
+  end
+end
+

metadata

@@ -0,0 +1,86 @@
+--- !ruby/object:Gem::Specification 
+name: milton
+version: !ruby/object:Gem::Version 
+  version: 1.0.0
+platform: ruby
+authors: 
+- Aaron Patterson
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-06-04 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mechanize
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.2
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.12.2
+    version: 
+description: Milton fills out your ADP ezLaborManager timesheet
+email: 
+- aaronp@rubyforge.org
+executables: 
+- milton
+- recorder
+extensions: []
+
+extra_rdoc_files: 
+- History.txt
+- Manifest.txt
+- README.txt
+files: 
+- History.txt
+- Manifest.txt
+- README.txt
+- Rakefile
+- bin/milton
+- bin/recorder
+- lib/milton.rb
+- test/test_milton.rb
+has_rdoc: true
+homepage: http://seattlerb.rubyforge.org/milton
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --main
+- README.txt
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: milton
+rubygems_version: 1.3.3
+signing_key: 
+specification_version: 3
+summary: Milton fills out your ADP ezLaborManager timesheet
+test_files: 
+- test/test_milton.rb