47 lines
1.1 KiB
Ruby
47 lines
1.1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module WebHooks
|
||
|
|
module RecursionDetection
|
||
|
|
class UUID
|
||
|
|
HEADER = "#{::Gitlab::WebHooks::GITLAB_EVENT_HEADER}-UUID"
|
||
|
|
|
||
|
|
include Singleton
|
||
|
|
|
||
|
|
attr_accessor :request_uuid
|
||
|
|
|
||
|
|
def initialize
|
||
|
|
self.new_uuids_for_hooks = {}
|
||
|
|
end
|
||
|
|
|
||
|
|
class << self
|
||
|
|
# Back the Singleton with RequestStore so it is isolated to this request.
|
||
|
|
def instance
|
||
|
|
Gitlab::SafeRequestStore[:web_hook_recursion_detection_uuid] ||= new
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
# Returns a UUID, which will be either:
|
||
|
|
#
|
||
|
|
# - The UUID that was recycled from the request headers if the request was made by a webhook.
|
||
|
|
# - A new UUID initialized for the webhook.
|
||
|
|
def uuid_for_hook(hook)
|
||
|
|
request_uuid || new_uuid_for_hook(hook)
|
||
|
|
end
|
||
|
|
|
||
|
|
def header(hook)
|
||
|
|
{ HEADER => uuid_for_hook(hook) }
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
attr_accessor :new_uuids_for_hooks
|
||
|
|
|
||
|
|
def new_uuid_for_hook(hook)
|
||
|
|
new_uuids_for_hooks[hook.id] ||= SecureRandom.uuid
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|