#!/usr/bin/env ruby
require 'yaml'
require 'tmpdir'
require_relative '../services/geocoder/lib/hires_geocoder_factory'

TIMEOUT = 300 # seconds before raising a timeout error

def usage
  abort "usage: script/geocoder_test environment <non-batch>"
end

environment = ARGV[0]
non_batch = (ARGV[1] == 'non-batch')
usage unless environment

def load_config(environment)
  config_file_hash = YAML.load_file(File.expand_path('../../config/app_config.yml',  __FILE__))
  config_file_hash[environment]["geocoder"]
rescue StandardError => e
  raise "Missing or inaccessible config for environment #{environment} in config/app_config.yml: #{e.message}"
end

config = load_config(environment)
CartoDB::GeocoderConfig.instance.set config.merge('force_batch' => !non_batch)
input_file = File.expand_path('../../services/geocoder/spec/fixtures/nokia_input.csv',  __FILE__)
working_dir = Dir.mktmpdir
geocoder = CartoDB::HiresGeocoderFactory.get(input_file, working_dir)

puts "Runing #{(non_batch ? 'non batch' : 'batch')} geocoder..."
start = Time.now
geocoder.run
finish = Time.now
until geocoder.status == 'completed' do
  finish = Time.now
  raise "Geocoder FAILURE: Timeout" if (finish - start) > TIMEOUT
  geocoder.update_status
  sleep(2)
end
raise "Geocoder FAILURE" unless geocoder.status == 'completed'

Dir.chdir geocoder.dir
`unp #{geocoder.result} 2>&1`
file_path = (non_batch ? geocoder.result : Dir[File.join(geocoder.dir, '*_out.txt')][0])
response = File.read(file_path).split(',')

expected_position = [45.96, -66.60]
position = [response[4].to_f, response[5].to_f]
distance = Math.sqrt(expected_position.zip(position).map { |x| (x[1] - x[0])**2 }.reduce(:+))
raise "Geocoder FAILURE: wrong geocoded position #{position}, expected #{expected_position}, distance: #{distance}" if distance > 0.04

FileUtils.rm_f working_dir
puts "\e[32mGeocoder OK #{(finish - start)} secs\e[0m"