require "sinatra/base" require "sinatra/streaming" require "json" require "fog" require "exceptions/invalid_privileges" require "exceptions/invalid_record" require "exceptions/configuration_error" require "exceptions/record_not_found" require "exceptions/dependency_error" require "exceptions/conflict_exception" require "exceptions/parser_error" require "exceptions/validation_error" require "exceptions/knife_config_error" require "exceptions/access_error" require 'core/devops-logger' require_relative "../helpers/query" require 'swagger/blocks' module Devops class Api3 < Sinatra::Base include Swagger::Blocks include Sinatra::JSON helpers Sinatra::Streaming helpers Devops::API3::Helpers register Sinatra::DevopsAuth @@logger = nil @@access_logger = DevopsLogger.access_logger(File.join(DevopsConfig.config[:log_dir], "devops-api3.access.log")) set :root, File.expand_path("../../", __FILE__) set :public_folder, Proc.new { File.join(root, "public") } configure :production do config = DevopsConfig.config log_file = File.join(config[:log_dir], "devops-api3.log") @@logger = DevopsLogger.create(log_file, Logger::INFO) disable :dump_errors disable :show_exceptions #set :logging, Logger::INFO @@logger.info "Production mode" end configure :development do config = DevopsConfig.config log_file = File.join(config[:log_dir], "devops-api3.dev.log") @@logger = DevopsLogger.create(log_file, Logger::DEBUG) disable :raise_errors set :show_exceptions, false disable :show_exceptions enable :logging use Rack::CommonLogger, @@logger @@logger.info "Development mode" puts root puts public_folder end def logger @@logger end error Sinatra::NotFound do content_type 'text/plain' [404, "#{request.url} not found"] end # set current logger and call handlers def call env DevopsLogger.logger = @@logger res = super(env) @@access_logger.info(env["REQUEST_METHOD"] + " " + env["REQUEST_URI"] + " - from #{env["HTTP_USER_AGENT"]} (#{env["REMOTE_USER"]}) / #{res[0]}") if env["REMOTE_USER"] res end def handle_exception!(boom) if boom.is_a?(::Devops::Exception::DevopsError) boom.http_response elsif boom.is_a?(Mongoid::Errors::Validations) Devops::Exception::ValidationError.create_from_db_exception(boom).http_response elsif boom.is_a?(Mongoid::Errors::InvalidValue) Devops::Exception::ParserError.create_from_db_exception(boom).http_response elsif boom.is_a?(Sinatra::NotFound) super(boom) else @@logger.error boom.message + "\n\t" + boom.backtrace.join("\n\t") super(boom) end end error Devops::Exception::KnifeConfigError do e = env["sinatra.error"] logger.error e.message halt_response(e.message, 500) end error Excon::Errors::Error do e = env["sinatra.error"] logger.warn e.message halt_response(e.message, 400) end error ::Excon::Errors::Unauthorized do e = env["sinatra.error"] resp = e.response ct = resp.headers["Content-Type"] msg = unless ct.nil? if ct.include?("application/json") json = ::Chef::JSONCompat.from_json(resp.body) m = "ERROR: Unauthorized (#{json['error']['code']}): #{json['error']['message']}" logger.error(m) else end m else "Unauthorized: #{e.inspect}" end halt_response(msg, 500) end error Fog::Compute::AWS::Error do e = env["sinatra.error"] logger.error e.message halt_response(e.message, 500) end error do e = env["sinatra.error"] logger.error e.message halt_response(e.message, 500) end end end