require 'rubygems'
require 'active_record'
require 'optparse'

options = {}
OptionParser.new do |opts|
	opts.banner = "Usage: main.rb [options]"

	opts.on("-h", "--hostname HOST", "Specify host name") do |v|
	options[:host] = v
	end
	opts.on("-a", "--adapter ADAPTER", "Specify database adapter") do |v|
	options[:adapter] = v
	end
	opts.on("-db", "--database DBNAME", "Specify database name") do |v|
	options[:database] = v
	end
	opts.on("-u", "--user USER", "Specify database user") do |v|
	options[:user] = v
	end
	opts.on("-p", "--password PASS", "Specify database password") do |v|
	options[:password] = v
	end
	opts.on("-q", "--query QUERY", "Run your query") do |v|
	options[:query] = v
	end
	opts.on("-f", "--filename FILENAME", "Specify filename") do |v|
	options[:filename] = v
	end
end.parse!

p 'Execute '+options[:query]+' on database '+options[:database]
ActiveRecord::Base.establish_connection(
    :adapter => options[:adapter],
    :database => options[:database],
    :username => options[:user],
    :password => options[:password],
    :host => options[:host]
)

File.open(options[:filename], "w") { |f| f.write ActiveRecord::Base.connection.exec_query(options[:query]).to_json}
p 'Successfully saved to '+options[:filename]