bus.rb

require 'sinatra'
require 'net/http'
require 'json'
require 'erb'
set :views, File.dirname(__FILE__) + '/templates'
set :public, File.dirname(__FILE__) + '/static'

STOPS = JSON.parse(File.read("./bus_stops.json"))["markers"]

get '*' do
  'The site is broken. Sorry. Hopefully will be up again in a few weeks. <a href="https://github.com/james/TFL-Live-Bus/issues/5">Longer explanation here</a>'
end

get '/' do
  if params[:search] && params[:search] != ""
    @search_results = STOPS.select {|x| x["name"] =~ /#{params[:search]}/i}
  elsif params[:lat] && params[:lon] && params[:lat] != "" && params[:lon] != ""
    @search_results = STOPS.select { |x| approximate_distance_between(x, params) < 0.005 }
    @search_results.sort!{ |a, b| approximate_distance_between(a, params) <=> approximate_distance_between(b, params) }
  end
  erb :index
end

get '/nearby' do
  erb :nearby
end

get '/stop/:stop_id' do |stop_id|
  @stop = STOPS.select {|x| x["id"] =~ /#{stop_id}/i}.first
  get_stop_json(stop_id)
  erb :stop
end

get '/stop/:stop_id/partial' do |stop_id|
  get_stop_json(stop_id)
  erb :indicator_table
end

get '/stop/:stop_id/curl' do |stop_id|
  get_stop_json(stop_id)
  @json["arrivals"].collect{|x| "#{x["routeId"]} | #{x["estimatedWait"]}\n"}
end

get '/stop/:stop_id/jsonp' do |stop_id|
  headers 'Content-Type' => "text/javascript"
  "bus_json(#{make_request(stop_id)})"
end


def get_stop_json(stop_id)
  response = make_request(stop_id)
  @json = JSON.parse(response)
end

def make_request(stop_id)
  Net::HTTP.get(URI.parse("http://countdown.tfl.gov.uk/stopBoard/#{stop_id}/"))
end

def approximate_distance_between(stop, coords)
  lat_diff = (stop["lat"].to_f - coords[:lat].to_f).abs
  lon_diff = (stop["lng"].to_f - coords[:lon].to_f).abs
  diff = Math.sqrt(lat_diff**2 + lon_diff**2)
end