2018-09-14 05:42:05 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2017-06-21 14:59:13 +00:00
|
|
|
module WithPerformanceBar
|
|
|
|
extend ActiveSupport::Concern
|
|
|
|
|
2019-08-26 13:57:59 +00:00
|
|
|
included do
|
2019-08-27 11:40:44 +00:00
|
|
|
before_action :set_peek_enabled_for_current_request
|
2019-08-26 13:57:59 +00:00
|
|
|
end
|
|
|
|
|
2019-08-27 11:40:44 +00:00
|
|
|
private
|
2017-06-21 14:59:13 +00:00
|
|
|
|
2019-08-27 11:40:44 +00:00
|
|
|
def set_peek_enabled_for_current_request
|
2018-09-20 19:32:54 +00:00
|
|
|
Gitlab::SafeRequestStore.fetch(:peek_enabled) { cookie_or_default_value }
|
2018-01-16 18:13:31 +00:00
|
|
|
end
|
|
|
|
|
2019-08-27 11:40:44 +00:00
|
|
|
# Needed for Peek's routing to work;
|
|
|
|
# Peek::ResultsController#restrict_non_access calls this method.
|
|
|
|
def peek_enabled?
|
|
|
|
Gitlab::PerformanceBar.enabled_for_request?
|
|
|
|
end
|
2018-01-16 18:13:31 +00:00
|
|
|
|
|
|
|
def cookie_or_default_value
|
2021-06-01 06:09:35 +00:00
|
|
|
if cookies[:perf_bar_enabled].blank? && Rails.env.development?
|
|
|
|
cookies[:perf_bar_enabled] = 'true'
|
|
|
|
end
|
2019-08-27 11:40:44 +00:00
|
|
|
|
2021-06-01 06:09:35 +00:00
|
|
|
cookie_enabled = cookies[:perf_bar_enabled] == 'true'
|
2021-04-27 12:10:12 +00:00
|
|
|
cookie_enabled && Gitlab::PerformanceBar.allowed_for_user?(current_user)
|
2017-06-21 14:59:13 +00:00
|
|
|
end
|
|
|
|
end
|