112 lines
2.8 KiB
Ruby
112 lines
2.8 KiB
Ruby
require 'travis'
|
|
require 'backports'
|
|
require 'rack'
|
|
require 'rack/protection'
|
|
require 'rack/contrib'
|
|
require 'rack/cache'
|
|
require 'active_record'
|
|
require 'redis'
|
|
require 'gh'
|
|
require 'hubble'
|
|
require 'hubble/middleware'
|
|
require 'newrelic_rpm'
|
|
|
|
# Rack class implementing the HTTP API.
|
|
# Instances respond to #call.
|
|
#
|
|
# run Travis::Api::App.new
|
|
#
|
|
# Requires TLS in production.
|
|
module Travis::Api
|
|
class App
|
|
autoload :AccessToken, 'travis/api/app/access_token'
|
|
autoload :Base, 'travis/api/app/base'
|
|
autoload :Endpoint, 'travis/api/app/endpoint'
|
|
autoload :Extensions, 'travis/api/app/extensions'
|
|
autoload :Helpers, 'travis/api/app/helpers'
|
|
autoload :Middleware, 'travis/api/app/middleware'
|
|
autoload :Responders, 'travis/api/app/responders'
|
|
|
|
Rack.autoload :SSL, 'rack/ssl'
|
|
|
|
# Used to track if setup already ran.
|
|
def self.setup?
|
|
@setup ||= false
|
|
end
|
|
|
|
# Loads all endpoints and middleware and hooks them up properly.
|
|
# Calls #setup on any middleware and endpoint.
|
|
#
|
|
# This method is not threadsafe, but called when loading
|
|
# the environment, so no biggy.
|
|
def self.setup(options = {})
|
|
setup! unless setup?
|
|
Endpoint.set(options)
|
|
end
|
|
|
|
def self.new(options = {})
|
|
setup(options) if options
|
|
super()
|
|
end
|
|
|
|
attr_accessor :app
|
|
|
|
def initialize
|
|
@app = Rack::Builder.app do
|
|
use Hubble::Rescuer, env: Travis.env, codename: ENV['CODENAME'] if Endpoint.production? && ENV['HUBBLE_ENDPOINT']
|
|
use Rack::Protection::PathTraversal
|
|
use Rack::SSL if Endpoint.production?
|
|
use ActiveRecord::QueryCache
|
|
|
|
if memcache_servers = ENV['MEMCACHE_SERVERS']
|
|
use Rack::Cache,
|
|
verbose: true,
|
|
metastore: "memcached://#{memcache_servers}",
|
|
entitystore: "memcached://#{memcache_servers}"
|
|
end
|
|
|
|
use Rack::Deflater
|
|
use Rack::PostBodyContentTypeParser
|
|
use Rack::JSONP
|
|
use ActiveRecord::ConnectionAdapters::ConnectionManagement
|
|
|
|
use Rack::Config do |env|
|
|
env['travis.global_prefix'] = env['SCRIPT_NAME']
|
|
end
|
|
|
|
Middleware.subclasses.each { |m| use(m) }
|
|
Endpoint.subclasses.each { |e| map(e.prefix) { run(e.new) } }
|
|
end
|
|
end
|
|
|
|
# Rack protocol
|
|
def call(env)
|
|
app.call(env)
|
|
end
|
|
|
|
private
|
|
|
|
def self.setup!
|
|
setup_travis
|
|
load_endpoints
|
|
setup_endpoints
|
|
@setup = true
|
|
end
|
|
|
|
def self.setup_travis
|
|
Travis::Amqp.config = Travis.config.amqp
|
|
Travis::Database.connect
|
|
Travis.services = Travis::Services
|
|
end
|
|
|
|
def self.load_endpoints
|
|
Backports.require_relative_dir 'app/middleware'
|
|
Backports.require_relative_dir 'app/endpoint'
|
|
end
|
|
|
|
def self.setup_endpoints
|
|
Base.subclasses.each(&:setup)
|
|
end
|
|
end
|
|
end
|