28 lines
778 B
Ruby
28 lines
778 B
Ruby
|
module FeatureFlagHelper
|
||
|
|
def set_feature_flag(user, feature, state)
|
||
|
|
user.reload
|
||
|
|
previous_state = user.has_feature_flag?(feature)
|
||
|
|
if state != previous_state
|
||
|
|
ff = FeatureFlag[name: feature]
|
||
|
|
ffu = FeatureFlagsUser[feature_flag_id: ff.id, user_id: user.id]
|
||
|
|
if state
|
||
|
|
unless ffu
|
||
|
|
FeatureFlagsUser.new(feature_flag_id: ff.id, user_id: user.id).save
|
||
|
|
end
|
||
|
|
else
|
||
|
|
ff.update restricted: false unless ff.restricted
|
||
|
|
ffu.delete if ffu
|
||
|
|
end
|
||
|
|
user.reload
|
||
|
|
end
|
||
|
|
previous_state
|
||
|
|
end
|
||
|
|
|
||
|
|
def with_feature_flag(user, feature, state)
|
||
|
|
user.reload
|
||
|
|
previous_state = set_feature_flag user, feature, state
|
||
|
|
yield
|
||
|
|
set_feature_flag user, feature, previous_state if state != previous_state
|
||
|
|
end
|
||
|
|
end
|