#!/usr/bin/env ruby
 
$:.unshift "#{File.dirname(__FILE__)}/lib"
 
require 'people'
require 'hansard_parser'
require 'configuration'
require 'optparse'
require 'progressbar'
 
def parse_date(text)
  today = Date.today
  
  if text == "today"
    today
  elsif text == "yesterday"
    today - 1
  elsif text == "previous-working-day"
    # For Sunday (wday 0) and Monday (wday 1) the previous working day is last Friday otherwise it's
    # just the previous day
    if today.wday == 0
      today - 2
    elsif today.wday == 1
      today - 3
    else
      today - 1
    end
  else
    Date.parse(text)
  end
end
 
# Defaults
options = {:load_database => true, :proof => false, :force => false}
 
OptionParser.new do |opts|
  opts.banner = <<EOF
Usage: parse-speeches.rb [options] <from-date> [<to-date>]
    formatting of date:
      year.month.day or today or yesterday
EOF
  opts.on("--no-load", "Just generate XML and don't load up database") do |l|
    options[:load_database] = l
  end
  opts.on("--proof", "Only parse dates that are at proof stage. Will redownload and populate html cache for those dates.") do |l|
    options[:proof] = l
  end
  opts.on("--force", "On loading data into database delete records that are not in the XML") do |l|
    options[:force] = l
  end
end.parse!
 
if ARGV.size != 1 && ARGV.size != 2
  puts "Need to supply one or two dates"
  exit
end
    
from_date = parse_date(ARGV[0])
 
if ARGV.size == 1
  to_date = from_date
else
  to_date = parse_date(ARGV[1])
end
 
conf = Configuration.new
 
FileUtils.mkdir_p "#{conf.xml_path}/scrapedxml/representatives_debates"
FileUtils.mkdir_p "#{conf.xml_path}/scrapedxml/senate_debates"
 
# First load people back in so that we can look up member id's
people = PeopleCSVReader.read_members
 
parser = HansardParser.new(people)
 
progress = ProgressBar.new("parse-speeches", ((to_date - from_date + 1) * 2).to_i)
 
# Kind of helpful to start at the end date and go backwards when using the "--proof" option. So, always going to do this now.
date = to_date
while date >= from_date
  if conf.write_xml_representatives
    if options[:proof]
      parser.parse_date_house_only_in_proof(date, "#{conf.xml_path}/scrapedxml/representatives_debates/#{date}.xml", House.representatives)
    else
      parser.parse_date_house(date, "#{conf.xml_path}/scrapedxml/representatives_debates/#{date}.xml", House.representatives)
    end
  end
  progress.inc
  if conf.write_xml_senators
    if options[:proof]
      parser.parse_date_house_only_in_proof(date, "#{conf.xml_path}/scrapedxml/senate_debates/#{date}.xml", House.senate)
    else
      parser.parse_date_house(date, "#{conf.xml_path}/scrapedxml/senate_debates/#{date}.xml", House.senate)
    end
  end
  progress.inc
  date = date - 1
end
 
progress.finish
 
# And load up the database
if options[:load_database]
  command_options = " --from=#{from_date} --to=#{to_date}"
  command_options << " --debates" if conf.write_xml_representatives
  command_options << " --lordsdebates" if conf.write_xml_senators
  command_options << " --force" if options[:force]
  
  # Starts with 'perl' to be friendly with Windows
  system("perl #{conf.web_root}/twfy/scripts/xml2db.pl #{command_options}")
end