$LOAD_PATH << File.join(File.expand_path(File.dirname(__FILE__)), 'lib')

require "json"
require "sinatra/base"
require "sinatra/reloader"
require "remote_storage/riak"

class LiquorCabinet < Sinatra::Base

  include RemoteStorage::Riak

  def self.config=(config)
    @config = config
    configure_airbrake
  end

  def self.config
    return @config if @config
    config = File.read(File.expand_path('config.yml', File.dirname(__FILE__)))
    self.config = YAML.load(config)[ENV['RACK_ENV']]
  end

  configure :development do
    register Sinatra::Reloader
    enable :logging
    disable :protection
  end

  configure :production do
    disable :logging
    disable :protection
  end

  configure :staging do
    disable :logging
    disable :protection
  end

  ["/:user/*/:key", "/:user/:key", "/:user/*/", "/:user/"].each do |path|
    before path do
      headers 'Access-Control-Allow-Origin' => '*',
              'Access-Control-Allow-Methods' => 'GET, PUT, DELETE',
              'Access-Control-Allow-Headers' => 'Authorization, Content-Type, Origin'
      headers['Access-Control-Allow-Origin'] = env["HTTP_ORIGIN"] if env["HTTP_ORIGIN"]
      headers['Cache-Control'] = 'no-cache'

      @user, @key = params[:user], params[:key]
      @directory = params[:splat] && params[:splat].first || ""

      token = env["HTTP_AUTHORIZATION"] ? env["HTTP_AUTHORIZATION"].split(" ")[1] : ""

      authorize_request(@user, @directory, token, @key.blank?) unless request.options?
    end
  end

  ["/:user/*/:key", "/:user/:key"].each do |path|
    get path do
      get_data(@user, @directory, @key)
    end
  end

  ["/:user/*/", "/:user/"].each do |path|
    get path do
      get_directory_listing(@user, @directory)
    end
  end

  ["/:user/*/:key", "/:user/:key"].each do |path|
    put path do
      data = request.body.read

      if env['CONTENT_TYPE'] == "application/x-www-form-urlencoded"
        content_type = "text/plain; charset=utf-8"
      else
        content_type = env['CONTENT_TYPE']
      end

      put_data(@user, @directory, @key, data, content_type)
    end
  end

  ["/:user/*/:key", "/:user/:key"].each do |path|
    delete path do
      delete_data(@user, @directory, @key)
    end
  end

  ["/:user/*/:key", "/:user/:key", "/:user/*/", "/:user/"].each do |path|
    options path do
      halt 200
    end
  end

  private

  def self.configure_airbrake
    if @config['airbrake'] && @config['airbrake']['api_key']
      require "airbrake"

      Airbrake.configure do |airbrake|
        airbrake.api_key = @config['airbrake']['api_key']
      end

      use Airbrake::Rack
      enable :raise_errors
    end
  end

end