46 lines
985 B
Ruby
46 lines
985 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
# rubocop:disable Style/ClassVars
|
||
|
|
module Gitlab
|
||
|
|
module Testing
|
||
|
|
class RobotsBlockerMiddleware
|
||
|
|
@@block_requests = Concurrent::AtomicBoolean.new(false)
|
||
|
|
|
||
|
|
# Block requests according to robots.txt.
|
||
|
|
# Any new requests disallowed by robots.txt will return an HTTP 503 status.
|
||
|
|
def self.block_requests!
|
||
|
|
@@block_requests.value = true
|
||
|
|
end
|
||
|
|
|
||
|
|
# Allows the server to accept requests again.
|
||
|
|
def self.allow_requests!
|
||
|
|
@@block_requests.value = false
|
||
|
|
end
|
||
|
|
|
||
|
|
def initialize(app)
|
||
|
|
@app = app
|
||
|
|
end
|
||
|
|
|
||
|
|
def call(env)
|
||
|
|
request = Rack::Request.new(env)
|
||
|
|
|
||
|
|
if block_requests? && Gitlab::RobotsTxt.disallowed?(request.path_info)
|
||
|
|
block_request(env)
|
||
|
|
else
|
||
|
|
@app.call(env)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def block_requests?
|
||
|
|
@@block_requests.true?
|
||
|
|
end
|
||
|
|
|
||
|
|
def block_request(env)
|
||
|
|
[503, {}, []]
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|