module LitmusPaper
  class App < Sinatra::Base
    get "/" do
      output = "Litmus Paper #{LitmusPaper::VERSION}\n\n"
      output += "Services monitored:\n"
      LitmusPaper.services.each do |service_name, service|
        output += "* #{service_name} (#{service.current_health.value})\n"
      end

      _text 200, output
    end

    delete "/down" do
      _delete_status_file(StatusFile.global_down_file)
    end

    post "/down" do
      _create_status_file(StatusFile.global_down_file)
    end

    delete "/up" do
      _delete_status_file(StatusFile.global_up_file)
    end

    post "/up" do
      _create_status_file(StatusFile.global_up_file)
    end

    get "/:service/status" do
      health = LitmusPaper.check_service(params[:service])
      if health.nil?
        _text 404, "NOT FOUND", { "X-Health" => "0" }
      else
        response_code = health.ok? ? 200 : 503
        body = "Health: #{health.value}\n"
        body << health.summary
        _text response_code, body, { "X-Health" => health.value.to_s }
      end
    end

    delete "/:service/down" do
      _delete_status_file(StatusFile.service_down_file(params[:service]))
    end

    post "/:service/down" do
      _create_status_file(StatusFile.service_down_file(params[:service]))
    end

    delete "/:service/up" do
      _delete_status_file(StatusFile.service_up_file(params[:service]))
    end

    post "/:service/up" do
      _create_status_file(StatusFile.service_up_file(params[:service]))
    end

    get "/test/error" do
      raise "an error"
    end

    error do
      _text 500, "Server Error"
    end

    def _create_status_file(status_file)
      status_file.create(params[:reason])
      _text 201, "File created"
    end

    def _delete_status_file(status_file)
      if status_file.exists?
        status_file.delete
        _text 200, "File deleted"
      else
        _text 404, "NOT FOUND"
      end
    end

    def _text(response_code, body, headers ={})
      [response_code, { "Content-Type" => "text/plain" }.merge(headers), body]
    end
  end
end