# This script creates a new Historics query from a file containing CSDL.
#
# NB: Most of the error handling (exception catching) has been removed for
# the sake of simplicity. Nearly everything in this library may throw
# exceptions, and production code should catch them. See the documentation
# for full details.
#

# Function to display usage instructions with an optional error message.
def usage(message = '', end_of_story = true)
	puts message + '\n' unless message.length() == 0
	puts
	puts 'Usage: create-from-csdl <username> <api_key> \\'
	puts '            <csdl_filename> <start> <end> <sources> <sample> <name>'
	puts
	puts 'Where: csdl_filename = the stream hash the query should run'
	puts '       start         = the start date for the query (YYYYMMDDHHMMSS)'
	puts '       end           = the end date for the query (YYYYMMDDHHMMSS)'
	puts '       sources       = comma separated list of data sources (e.g. twitter)'
	puts '       sample        = the sample rate'
	puts '       name          = a friendly name for the query'
	puts
	puts 'Example'
	puts '       create-from-csdl csdl.txt 20120101000000 20120101235959 \\'
	puts '                      twitter 100 HistoricsQuery123'
	puts
	exit 1 unless not end_of_story
end

# Include the shared Env class
require File.dirname(__FILE__) + '/env'

# Create the env object. This reads the command line arguments, creates the
# user object, and provides access to both along with helper functions.
env = Env.new()

# Check that we have enough arguments
usage() unless env.args.size() == 6

# Read the arguments
csdl_filename = env.args[0]
start_date    = env.args[1]
end_date      = env.args[2]
sources       = env.args[3].split(',')
sample        = env.args[4]
name          = env.args[5]

# Parse the dates
start_date = DateTime.strptime(start_date, '%Y%m%d%H%M%S')
end_date   = DateTime.strptime(end_date,   '%Y%m%d%H%M%S')

# Read the CSDL
csdl = File.open(csdl_filename, 'r').read

begin
	# Create the definition
	definition = env.user.createDefinition(csdl)

	# Create the Historics query
	historic = definition.createHistoric(start_date, end_date, sources, sample, name)

	# Prepare the query
	historic.prepare()

	# Display the details
	env.displayHistoricDetails(historic)

	puts 'Rate limit remainining: ' + String(env.user.rate_limit_remaining)
 rescue DataSift::DataSiftError => err
	puts 'ERR: [' + err.class.name + '] ' + err.message
end