tdiary/rpaproxy-sinatra

# -*- coding: utf-8 -*-

require 'sinatra'
require 'mongoid'
require 'omniauth'
require 'omniauth-twitter'
require 'haml'

require './models/user.rb'
require './models/proxy.rb'
require './models/log.rb'
require './models/stat.rb'
require './models/client.rb'

if production?
    require 'memcachier'
    require 'rack/session/dalli'
    use Rack::Session::Dalli, cache: Dalli::Client.new, expire_after: 2592000
else
    use Rack::Session::Pool, expire_after: 2592000
end

use OmniAuth::Builder do
    provider :twitter, ENV['TWITTER_KEY'], ENV['TWITTER_SECRET']
    provider :developer unless production?
end

set :haml, { format: :html5, escape_html: true }

configure do
    Mongoid.load!("config/mongoid.yml")
    Mongo::Logger.level = Logger::FATAL
end

configure :production do
    require 'newrelic_rpm'
    require 'newrelic_moped'
    raise StandardError.new("not found ENV['TWITTER_KEY']") unless ENV['TWITTER_KEY']
    raise StandardError.new("not found ENV['TWITTER_SECRET']") unless ENV['TWITTER_SECRET']
end

helpers do
    def current_user
        @current_user ||= User.where(uid: session[:user_id]).first if session[:user_id]
    end

    def locales
        ['jp', 'us', 'ca', 'de', 'fr', 'uk', 'es', 'it', 'cn']
    end

    def forbidden?
        client = Client.find_or_initialize_by(atag: params['AssociateTag'])
        client.update_status
        if client.status == Client::Status::ACTIVE
            client.save if client.changed? and client.created_at
            false
        else
            client.save if client.changed?
            true
        end
    end
end

before do
    content_type :html, 'charset' => 'utf-8'
end

before '/profile*' do
    redirect '/' unless current_user
end

before '/proxy*' do
    redirect '/' unless current_user
end

# ログアウト
get '/logout' do
    session[:user_id] = nil
    redirect '/'
end

# トップページ
get '/' do
    haml :index
end

# ログイン処理
get '/auth/twitter/callback' do
    auth = request.env['omniauth.auth']
    @current_user = User.find_or_create_with_omniauth(auth)
    session[:user_id] = @current_user.uid
    request.logger.info "[INFO] @#{current_user.screen_name} logged in"
    redirect '/profile'
end

# ログイン処理（開発用）
post '/auth/developer/callback' do
    auth = request.env['omniauth.auth']
    auth['uid'] = auth['info']['nickname'] = auth['info']['name']
    request.logger.info request.env['omniauth.auth'].inspect
    @current_user = User.find_or_create_with_omniauth(auth)
    session[:user_id] = @current_user.uid
    request.logger.info "[INFO] @#{current_user.screen_name} logged in"
    redirect '/profile'
end

# ログイン後の画面
get '/profile' do
    haml :profile
end

# プロフィール更新
put '/profile/:id' do
    user = User.find(params[:id])
    raise StandardError.new("error") unless current_user.id == user.id
    user.name = params[:name]
    user.url = params[:url]
    if user.save
        request.logger.info "[INFO] updated profile by @#{current_user.screen_name}"
    else
        request.logger.error "[ERROR] failed to update profile. #{user.errors.full_messages}"
    end
    redirect '/profile'
end

# プロキシ一覧
get '/proxies' do
    @proxies = Proxy.all
    haml :proxies
end

post '/proxy' do
    # create a new proxy
    # TODO: エラー処理
    proxy = Proxy.new_with_yaml(params[:endpoint])
    proxy.user = current_user
    begin
        if proxy.valid_endpoint? && proxy.save
            request.logger.info "[INFO] added proxy by @#{current_user.screen_name}"
        else
            request.logger.error "[ERROR] failed to add proxy. #{proxy.errors.full_messages}"
        end
    rescue StandardError => e
        request.logger.error "[ERROR] failed to add proxy. #{e.class}: #{e.message}"
    end
    redirect '/profile'
end

put '/proxy/:id' do
    # update an existing proxy
    proxy = Proxy.find(params[:id])
    raise StandardError.new("error") unless current_user.id == proxy.user.id
    proxy.name = params[:name]
    # proxy.endpoint = params[:endpoint]
    begin
        if proxy.save
            request.logger.info "[INFO] updated proxy by @#{current_user.screen_name}"
        else
            request.logger.error "[ERROR] failed to update proxy. #{proxy.errors.full_messages}"
        end
    rescue StandardError => e
        request.logger.error "[ERROR] failed to update proxy. #{e.class}: #{e.message}"
    end
    redirect '/profile'
end

delete '/proxy/:id' do
    proxy = Proxy.find(params[:id])
    raise StandardError.new("error") unless current_user.id == proxy.user.id
    proxy.destroy
    request.logger.info "[INFO] deleted proxy by @#{current_user.screen_name}"
    redirect '/profile'
end

# リバースプロキシ http://rpaproxy.heroku.com/rpaproxy/jp/
get %r{/rpaproxy/([\w]{2})/} do |locale|
    # deny bot
    if forbidden?
        env['QUERY_STRING'] = request.params.reject{|k,v| k == 'AssociateTag' }.to_query
    end
    # FIXME: 全件取得しているのを最適化したい
    proxies = Proxy.random(locale)
    res = nil
    proxies.each do |proxy|
        start_time = Time.now
        res = proxy.fetch(locale, request.query_string)
        if res
            Log.create(
                atag: params['AssociateTag'],
                locale: locale,
                created_at: Time.now,
                response: Time.now - start_time,
                proxy: proxy,
                success: true)
            break
        end
    end
    unless res
        # TODO: トータルの失敗回数を増分
        Log.create(
            atag: params['AssociateTag'],
            locale: locale,
            created_at: Time.now,
            success: false)
        request.logger.error "[ERROR] failed to return response"
        halt 503, "proxy unavailable"
    end
    redirect res['location'], 302
end

get '/stats' do
    Stat.destroy_all
    @stats = locales.map{|locale| Stat.create_by_logs(locale) }
    haml :stats
end

get '/logs' do
    redirect '/' unless current_user
    @logs = Log.desc('$natural').limit(100).reverse
    haml :logs
end

get '/users' do
    redirect '/' unless current_user
    @users = User.all.select{|user| user.proxies.size > 0 }
    haml :users
end

get '/debug' do
    raise StandardError.new('デバッグ')
end