#!/usr/bin/env ruby

require 'pathname'
root = Pathname.new(__FILE__).realpath.parent.parent
$:.unshift root.join('lib') if $0 == __FILE__

require 'optparse'
require "date"
require 'brand2csv'

def help
  <<EOS
#$0 ver.#{Brand2csv::VERSION}
Usage:
  #{File.basename(__FILE__)} timespan
    Find all brands registered in switzerland during the given timespan.
    The following examples valid timespan periods:
      brand2csv 01.01.2013 "b*" #will search for all brand starting with "b"
      brand2csv 1.10.2005-31.10.2005 #this will work as well from version 0.1.9
    The results are stored in the file <date_selected>.csv.
    The trademark name is either a real brand name or a link to an image.
  --swiss_only Fetch only trademarks from swiss owner
EOS
end

def validates_timespan(arg)
  valid = true
  timespan = ""
  dates = arg.gsub(/[^\d\.-]/, '').split("-")
  catch (:error) do
    dates.each_with_index do |d, i|
      sep = (dates.length > 1 && i != 0) ? "-" : ""
      begin
        Date.parse(d)
        timespan << sep + d
      rescue ArgumentError
        valid = false
        elms = d.split(".")
        prms = [elms[2], elms[1], -1].map(&:to_i)
        begin
          cand = Date.new(*prms).strftime("%d.%m.%Y")
          if elms[0] == (elms - cand.to_s.split(".")).first
            timespan << sep + cand.to_s
          else
            raise
          end
        rescue ArgumentError
          timespan = "" # unknown
          throw :error
        end
      end
    end
  end
  message = nil
  unless valid
    if timespan.empty?
      message = "Timespan is invalid"
    else
      message = "Did you mean #{timespan} ?"
    end
  end
  [valid, message]
end

parser = OptionParser.new
opts = {}
parser.on('--swiss_only') {|v| opts[:swiss_only] = true }
parser.on_tail('-h', '--help') { puts help; exit }

args = ARGV.dup
begin
  parser.parse!(args)
rescue OptionParser::MissingArgument,
       OptionParser::InvalidArgument,
       OptionParser::InvalidOption
  puts help
  exit 1
end

unless args.size >= 1
  puts help
  exit 1
end

unless args.empty?
  valid,message = validates_timespan(args[0])
  unless valid
    puts message
    exit 1
  end
end

begin
  Brand2csv::run(args[0], args[1], opts[:swiss_only])
rescue Interrupt
  puts "Unterbrochen. Breche mit Fehler ab"
  exit 1
end

puts "#{__FILE__} completed successfully" if $VERBOSE