thymer.rb

require 'rubygems'
require 'icalendar'
require 'haml'
require 'sha1'
require 'open-uri'
require 'openssl'
require 'yaml'
require 'extensions.rb'

# LAME, dirty, etc, gives warning.
# For some reason Google calendar has a bum certificate, and so
# you need the following for your downloader not to give you shit.
# The internets tell me that the version of open-uri that ships with 1.9
# does not have this problem, but I have yet to make the jump from 1.8
OpenSSL::SSL::VERIFY_PEER = OpenSSL::SSL::VERIFY_NONE

# http://xkcd.com/282/
class Thymer
  def initialize
    @date_list = Hash.new()
    @ticket_list = Hash.new(0)
    @variables = YAML::load(open("config.yaml"))
    @template = Haml::Engine.new(File.read(@variables[:template]){ |io| io.read })
    parse(@variables[:calendar_uri])
  end

  def invoice(date, end_date = nil)
    start_date = Date.parse(date)
    end_date = Date.parse(end_date) unless end_date == nil
    end_date ||= start_date + 13

    @variables[:date] ||= Time.now.to_english
    @variables[:start_date] = start_date.to_english
    @variables[:end_date] = end_date.to_english
    @variables[:invoice_number] = eval(@variables[:invoice_number]).call
    @variables[:total_seconds] = 0
    @variables[:days] = []

    (start_date..end_date).each { |d|

      this_day = d.to_8601
      if (days_tickets = @date_list[this_day])
	day = Day.new
	day.date = d
	@variables[:days] << day

	days_tickets.keys.each { |ticket|


	  t = Ticket.new
	  t.description = ticket

	  t.time = days_tickets[ticket]
	  t.hours = t.time / 3600.0
	  @variables[:total_seconds] = @variables[:total_seconds] + t.time
	  @ticket_list[ticket] = @ticket_list[ticket] + t.time
	  day.tickets << t
	}
      end
    }

    @variables[:total_tickets] = []
    @ticket_list.each_pair do |desc, qty|
      t = Ticket.new
      t.description = desc
      t.time = qty
      t.hours = t.time / 3600.0
      @variables[:total_tickets] << t
    end

    @variables[:total_hours] = @variables[:total_seconds] / 3600

    puts "The following dates have been processed:"
    @variables[:days].each { |d| 
      puts d.date.to_english 
    }

    
    if @variables[:discount_unit_rate] != nil then
      @variables[:discount_exists] = true
      
      @variables[:discount_seconds] = @variables[:discount_hours] * 3600
      @variables[:discount_cost] = @variables[:discount_unit_rate] * @variables[:discount_seconds]
      
      @variables[:regular_priced_seconds] = @variables[:total_seconds] - @variables[:discount_seconds]
      puts "discount cost: #{@variables[:discount_cost]/3600}"

      @variables[:total_cost] = @variables[:discount_cost] + (@variables[:regular_priced_seconds] * @variables[:unit_rate])
      puts "total cost: #{ @variables[:total_cost]/3600}"
    else
      
      @variables[:discount_exists] = false
      @variables[:pre_vat_cost] = (@variables[:total_seconds] * @variables[:unit_rate]) / 3600.0
      @variables[:vat_cost] = @variables[:pre_vat_cost] * @variables[:vat_rate]
      @variables[:total_cost] = @variables[:pre_vat_cost] * (@variables[:vat_rate] + 1)

      puts "total cost: #{ @variables[:total_cost]}"
    end


    File.open("output/#{start_date.to_8601}_#{end_date.to_8601}.#{@variables[:template_suffix]}", "w"){ |io| io.puts @template.render(Object.new, @variables) }
    puts "Next invoice start date: #{(end_date+1).to_8601}"

  end


  # This is a static method because it gets called in the template; it's a 
  # holdover from before I wrapped this code into a class. Lame, I know. 
  # TODO:
  # Why did I write this in the first place? Should be deleted.
  def self.render_days(days)
    day_template = Haml::Engine.new(File.read("template/day.haml"){ |io| io.read })
    str = ""
    days.each { |day|
      str = str + day_template.render(Object.new, :day => day)
    }
    return str
  end


  private

  def parse(cal_uri)
    calendar = Icalendar.parse(open(cal_uri)).first

    calendar.events.each { |e| 
      prop = e.properties
      date = prop["dtstart"].to_8601
      task_name = prop["summary"]
      task_duration = prop["dtend"].to_unix - prop["dtstart"].to_unix

      if @date_list[date].nil?
	@date_list[date] = Hash.new(0)
      end

      @date_list[date][task_name] = @date_list[date][task_name] + task_duration
    }
  end

  
end

Thymer.new.invoice("2011-02-21")