discourse/config/initializers/008-rack-cors.rb

53 lines
1.5 KiB
Ruby
Raw Normal View History

2017-12-07 07:30:50 +08:00
# frozen_string_literal: true
class Discourse::Cors
ORIGINS_ENV = "Discourse_Cors_Origins"
def initialize(app, options = nil)
@app = app
if GlobalSetting.enable_cors && GlobalSetting.cors_origin.present?
@global_origins = GlobalSetting.cors_origin.split(',').map(&:strip)
end
2017-12-07 07:30:50 +08:00
end
2017-12-07 07:30:50 +08:00
def call(env)
cors_origins = @global_origins || []
cors_origins += SiteSetting.cors_origins.split('|') if SiteSetting.cors_origins.present?
cors_origins = cors_origins.presence
2017-12-07 07:30:50 +08:00
if env['REQUEST_METHOD'] == ('OPTIONS') && env['HTTP_ACCESS_CONTROL_REQUEST_METHOD']
return [200, Discourse::Cors.apply_headers(cors_origins, env, {}), []]
end
2017-12-07 07:30:50 +08:00
env[Discourse::Cors::ORIGINS_ENV] = cors_origins if cors_origins
2017-12-07 07:30:50 +08:00
status, headers, body = @app.call(env)
headers ||= {}
2017-12-07 07:30:50 +08:00
Discourse::Cors.apply_headers(cors_origins, env, headers) if cors_origins
2017-12-07 07:30:50 +08:00
[status, headers, body]
end
def self.apply_headers(cors_origins, env, headers)
origin = nil
if cors_origins
if origin = env['HTTP_ORIGIN']
origin = nil unless cors_origins.include?(origin)
end
2017-12-07 07:30:50 +08:00
headers['Access-Control-Allow-Origin'] = origin || cors_origins[0]
headers['Access-Control-Allow-Headers'] = 'X-Requested-With, X-CSRF-Token, Discourse-Visible'
headers['Access-Control-Allow-Credentials'] = 'true'
end
2017-12-07 07:30:50 +08:00
headers
end
2017-12-07 07:30:50 +08:00
end
2017-12-07 07:30:50 +08:00
if GlobalSetting.enable_cors
Rails.configuration.middleware.insert_before ActionDispatch::Flash, Discourse::Cors
end