Module: RackHelpers

Defined in:

lib/stack-service-base/rack_helpers.rb

Class Method Summary

• .rack_setup(app) ⇒ Object

Class Method Details

.rack_setup(app) ⇒ Object

# File 'lib/stack-service-base/rack_helpers.rb', line 152

def rack_setup(app)
  # PATCH: for the Grape Swagger
  # https://github.com/ruby-grape/grape-swagger/pull/905
  # https://github.com/ruby-grape/grape-swagger/issues/904
  GrapeSwagger::DocMethods::ParseParams.instance_eval do
    def parse_enum_or_range_values(values)
      case values
      when Proc
        parse_enum_or_range_values(values.call) if values.parameters.empty?
      when Range
        parse_range_values(values) # if values.first.is_a?(Integer) TODO
      when Array
        { enum: values }
      else
        { enum: [values] } if values
      end
    end
  end if defined? GrapeSwagger::DocMethods::ParseParams

  app.use Rack.middleware_klass do |env, app|
    code, headers, body = env['REQUEST_METHOD'] == 'OPTIONS' ? [200, {}, []] : app.call(env)
    if code == 404 && env['PATH_INFO'] == '/healthcheck'
      code, headers, body = [200, {'Content-Type' =>'application/json'}, [{ Status: 'Healthy' }.to_json ]]
    end
    [code, headers, body]
  end

  if defined? OpenTelemetry::Instrumentation::Rack::Instrumentation
    # use OpenTelemetry::Instrumentation::Rack::Middlewares::TracerMiddleware
    app.use *OpenTelemetry::Instrumentation::Rack::Instrumentation.instance.middleware_args
  end

  app.use Rack::Deflater
  app.use OTELTraceInfo if defined? OpenTelemetry::Trace

  unless defined?(PERFORMANCE) && PERFORMANCE
    app.use RequestProfile
    app.use HeadersLogger

    if defined? Rack::ODataCommonLogger
      app.use Rack::ODataCommonLogger, LOGGER # or use Rack::CommonLogger, LOGGER
    end
  end

  app.use Rack.middleware_klass do |env, app|
    code, headers, body = env['REQUEST_METHOD'] == 'OPTIONS' ? [200, {}, []] : app.call(env)

    # scheme = env['rack.url_scheme']
    # referer = URI.parse env['HTTP_REFERER']
    headers.merge!(
      # 'Access-Control-Allow-Origin' => "#{referer.scheme}://#{referer.host}",
      'Access-Control-Allow-Origin' => '*',
      'Access-Control-Allow-Methods' => 'GET, PUT, POST, PATCH, DELETE, HEAD, OPTIONS',
      'Access-Control-Allow-Headers' => '*',
      'Access-Control-Allow-Credentials' => 'true')
    [code, headers, body]
  end

  # unless @run
  #   run ->(_env) { [200, {'Content-Type' => 'text/plain'}, ['OK!?']] }
  # end
end