#!/usr/bin/env ruby
require 'rubygems'
require 'open-uri'
require 'optparse'
require File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib', 'twitter_to_csv'))

options = { :csv => STDOUT, :fields => %w[text] }
parser = OptionParser.new do |opts|
  opts.banner = "Usage: #{File.basename($0)} [options]"
  opts.separator ""
  opts.separator "Specific options:"

  opts.on("-u", "--username USERNAME", "Twitter username") do |username|
    options[:username] = username
  end

  opts.on("-p", "--password PASSWORD", "Twitter password") do |password|
    options[:password] = password
  end

  opts.on("-c", "--csv FILE", "The CSV file to append to") do |csv|
    options[:csv_appending] = File.exists?(csv)
    options[:csv] = File.open(csv, 'a')
  end

  opts.on("-j", "--json FILE", "The JSON file to append to") do |json|
    options[:json] = File.open(json, 'a')
  end

  opts.on("-f", "--filter KEYWORDS", "Keywords to ask Twitter to filter on") do |filter|
    options[:filter] = filter.split(/\s*,\s*/)
  end

  opts.on("-x", "--fields FIELDS", "Fields to include in the CSV") do |fields|
    options[:fields] = fields.split(/\s*,\s*/)
  end

  opts.on("-e", "--require-english", "Attempt to filter out non-English tweets.", "This will have both false positives and false negatives.") do |e|
    options[:require_english] = e
  end

  opts.on("-v", "--[no-]verbose", "Run verbosely") do |v|
    options[:verbose] = v
  end

  opts.on_tail(nil, "--sample-fields NUMBER_OF_SAMPLES", "Record NUMBER_OF_SAMPLES tweets and then print out all","of the field names seen.  Use to find out what can be passed to.") do |samples|
    options[:sample_fields] = samples && samples.to_i
  end

  opts.on_tail("", "--replay-from-file FILENAME", "Replay tweets from a JSON dump file") do |replay_file|
    options[:replay_from_file] = replay_file
  end

  opts.on_tail("", "--url-columns NUMBER_OF_COLUMNS", "Extract up to NUMBER_OF_COLUMNS urls from the status and include them in the CSV") do |url_columns|
    options[:url_columns] = url_columns.to_i
  end

  opts.on_tail("-h", "--help", "Show this message") do
    STDERR.puts opts
    exit
  end

  opts.on_tail("--version", "Show version") do
    STDERR.puts "twitter_to_csv version #{TwitterToCsv::VERSION}"
    exit
  end
end

parser.parse!

unless (options[:username] && options[:password]) || options[:replay_from_file]
  STDERR.puts "Error: Twitter username and password are required fields.\n\n"
  STDERR.puts parser
  exit 1
end

TwitterToCsv::CsvBuilder.new(options).run