require 'OSM/StreamParser'
require 'socket'
require 'digest/sha1'
require 'cucumber/rake/task'
require 'sys/proctable'

SANDBOX = 'sandbox'
DATA_FOLDER = 'osm_data'

Cucumber::Rake::Task.new do |t|
  t.cucumber_opts = %w{--format pretty}
end

areas = {
  :kbh => { :country => 'denmark', :bbox => 'top=55.6972 left=12.5222 right=12.624 bottom=55.6376' },
  :frd => { :country => 'denmark', :bbox => 'top=55.7007 left=12.4765 bottom=55.6576 right=12.5698' },
  :regh => { :country => 'denmark', :bbox => 'top=56.164 left=11.792 bottom=55.403 right=12.731' },
  :dk => { :country => 'denmark', :bbox => nil },
  :skaane => { :counry => 'sweden', :bbox => 'top=56.55 left=12.4 bottom=55.3 right=14.6' }
}



osm_data_area_name = ARGV[1] ? ARGV[1].to_s.to_sym : :kbh
raise "Unknown data area." unless areas[osm_data_area_name]
osm_data_country = areas[osm_data_area_name][:country]
osm_data_area_bbox = areas[osm_data_area_name][:bbox]


task osm_data_area_name.to_sym {}   #define empty task to prevent rake from whining. will break if area has same name as a task


def each_process name, &block
  Sys::ProcTable.ps do |process|
    if process.comm.strip == name.strip
      yield process.pid.to_i, process.state.strip
    end
  end
end

def up?
  find_pid('osrm-routed') != nil
end

def find_pid name
  each_process(name) { |pid,state| return pid.to_i }
  return nil
end

def wait_for_shutdown name
  timeout = 10
  (timeout*10).times do
    return if find_pid(name) == nil
    sleep 0.1
  end
  raise "*** Could not terminate #{name}."
end

def write_server_ini osm_file
  s=<<-EOF
  Threads = 1
  IP = 0.0.0.0
  Port = 5000

  hsgrData=#{DATA_FOLDER}/#{osm_file}.osrm.hsgr
  nodesData=#{DATA_FOLDER}/#{osm_file}.osrm.nodes
  ramIndex=#{DATA_FOLDER}/#{osm_file}.osrm.ramIndex
  fileIndex=#{DATA_FOLDER}/#{osm_file}.osrm.fileIndex
  namesData=#{DATA_FOLDER}/#{osm_file}.osrm.names
  EOF
  File.open( 'server.ini', 'w') {|f| f.write( s ) }
end


desc "Rebuild and run tests."
task :default => [:build, :cucumber]

desc "Build using SConsstruct."
task :build do
  system "scons"
end

desc "Setup config files."
task :setup do
  Dir.mkdir "#{SANDBOX}/#{DATA_FOLDER}" unless File.exist? "#{SANDBOX}/#{DATA_FOLDER}"
  ['server.ini','speedprofile.ini','extractor.ini','contractor.ini'].each do |file|
    unless File.exist? "#{SANDBOX}/#{file}"
      puts "Copying #{file} template to sandbox/#{file}" 
      FileUtils.cp file, "#{SANDBOX}/#{file}"
    end
  end
end

desc "Download OSM data."
task :download => :setup do
  puts "Downloading..."
  raise "Error while downloading data." unless system "curl http://download.geofabrik.de/osm/europe/#{osm_data_country}.osm.pbf -o #{SANDBOX}/#{DATA_FOLDER}/#{osm_data_country}.osm.pbf"
  if osm_data_area_bbox
    puts "Cropping and converting to protobuffer..."
    raise "Error while cropping data." unless system "osmosis --read-pbf file=#{SANDBOX}/#{DATA_FOLDER}/#{osm_data_country}.osm.pbf --bounding-box #{osm_data_area_bbox} --write-pbf file=#{SANDBOX}/#{DATA_FOLDER}/#{osm_data_area_name}.osm.pbf omitmetadata=true"
  end
end

desc "Crop OSM data"
task :crop do
  if osm_data_area_bbox
    raise "Error while cropping data." unless system "osmosis --read-pbf file=#{SANDBOX}/#{DATA_FOLDER}/#{osm_data_country}.osm.pbf --bounding-box #{osm_data_area_bbox} --write-pbf file=#{SANDBOX}/#{DATA_FOLDER}/#{osm_data_area_name}.osm.pbf omitmetadata=true"
  end
end

desc "Reprocess OSM data."
task :process => :setup do
  Dir.chdir SANDBOX do
    raise "Error while extracting data." unless system "../osrm-extract #{DATA_FOLDER}/#{osm_data_area_name}.osm.pbf"
    puts
    raise "Error while preparing data." unless system "../osrm-prepare #{DATA_FOLDER}/#{osm_data_area_name}.osrm #{DATA_FOLDER}/#{osm_data_area_name}.osrm.restrictions"
    puts
  end
end

desc "Delete preprocessing files."
task :clean do
  File.delete *Dir.glob("#{SANDBOX}/#{DATA_FOLDER}/*.osrm")
  File.delete *Dir.glob("#{SANDBOX}/#{DATA_FOLDER}/*.osrm.*")
end

desc "Run all cucumber test"
task :test do
  system "cucumber"
  puts
end

desc "Run the routing server in the terminal. Press Ctrl-C to stop."
task :run => :setup do
  Dir.chdir SANDBOX do
    write_server_ini osm_data_area_name
    system "../osrm-routed"
  end
end

desc "Launch the routing server in the background. Use rake:down to stop it."
task :up => :setup do
  Dir.chdir SANDBOX do
    abort("Already up.") if up?
    write_server_ini osm_data_area_name
    pipe = IO.popen('../osrm-routed 1>>osrm-routed.log 2>>osrm-routed.log')
    timeout = 5
    (timeout*10).times do
      begin
        socket = TCPSocket.new('localhost', 5000)
        socket.puts 'ping'
      rescue Errno::ECONNREFUSED
        sleep 0.1
      end
    end
  end
end

desc "Stop the routing server."
task :down do
  pid = find_pid 'osrm-routed'
  abort("Already down.") unless pid
  Process.kill 'TERM', pid
end

desc "Kill all osrm-extract, osrm-prepare and osrm-routed processes."
task :kill do
  each_process('osrm-routed') { |pid,state| Process.kill 'KILL', pid }
  each_process('osrm-prepare') { |pid,state| Process.kill 'KILL', pid }
  each_process('osrm-extract') { |pid,state| Process.kill 'KILL', pid }
  wait_for_shutdown 'osrm-routed'
  wait_for_shutdown 'osrm-prepare'
  wait_for_shutdown 'osrm-extract'  
end

desc "Get PIDs of all osrm-extract, osrm-prepare and osrm-routed processes."
task :pid do
  each_process 'osrm-routed' do |pid,state|
    puts "#{pid}\t#{state}"
  end
end