190 lines
7.5 KiB
Ruby
190 lines
7.5 KiB
Ruby
|
namespace :cartodb do
|
||
|
|
namespace :services do
|
||
|
|
namespace :external do
|
||
|
|
namespace :google do
|
||
|
|
def add_feature_flag_if_not_exists(name, restricted = true)
|
||
|
|
ff = Carto::FeatureFlag.where(name: name).first
|
||
|
|
|
||
|
|
if ff.nil?
|
||
|
|
ff = Carto::FeatureFlag.new(name: name, restricted: restricted)
|
||
|
|
|
||
|
|
ff.id = Carto::FeatureFlag.order(:id).last.id + 1
|
||
|
|
ff.restricted = restricted
|
||
|
|
elsif ff.restricted != restricted
|
||
|
|
ff.restricted = restricted
|
||
|
|
end
|
||
|
|
|
||
|
|
ff.save
|
||
|
|
end
|
||
|
|
|
||
|
|
def add_feature_flag_to_user(feature_name, user)
|
||
|
|
ff = Carto::FeatureFlag.where(name: feature_name).first
|
||
|
|
raise "No such feature flag '#{feature_name}'" if ff.nil?
|
||
|
|
|
||
|
|
return if Carto::FeatureFlagsUser.where(feature_flag_id: ff.id, user_id: user.id).exists?
|
||
|
|
|
||
|
|
user.activate_feature_flag!(ff)
|
||
|
|
end
|
||
|
|
|
||
|
|
def remove_feature_flag_from_user(feature_name, user)
|
||
|
|
ff = Carto::FeatureFlag.where(name: feature_name).first
|
||
|
|
raise "No such feature flag '#{feature_name}'" if ff.nil?
|
||
|
|
|
||
|
|
ffu = Carto::FeatureFlagsUser.where(feature_flag_id: ff.id, user_id: user.id).first
|
||
|
|
|
||
|
|
if ffu.nil?
|
||
|
|
puts "[BUG] no feature flag user for feature '#{feature_name}' and '#{user.username}'"
|
||
|
|
else
|
||
|
|
ffu.destroy
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def remove_feature_flag_if_exists(name)
|
||
|
|
ff = Carto::FeatureFlag.where(name: name).first
|
||
|
|
|
||
|
|
ff.destroy unless ff.nil?
|
||
|
|
end
|
||
|
|
|
||
|
|
def prepend_client_equal_if_needed(key)
|
||
|
|
key.downcase.start_with?('client=') ? key : "client=#{key}"
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Enable Google Services for a given user'
|
||
|
|
task :enable_for_user, [:google_maps_key, :google_maps_private_key, :username] => [:environment] do |_, args|
|
||
|
|
add_feature_flag_if_not_exists('google_maps')
|
||
|
|
|
||
|
|
user = ::User.where(username: args[:username]).first
|
||
|
|
raise "No user with username '#{args[:username]}'" if user.nil?
|
||
|
|
|
||
|
|
puts "Enabling 'google_maps' feature flag for user '#{user.username}'"
|
||
|
|
add_feature_flag_to_user('google_maps', user)
|
||
|
|
|
||
|
|
puts "Enabling google services for user '#{user.username}'..."
|
||
|
|
user.google_maps_key = prepend_client_equal_if_needed(args[:google_maps_key])
|
||
|
|
user.google_maps_private_key = args[:google_maps_private_key] unless args[:google_maps_private_key].blank?
|
||
|
|
|
||
|
|
puts user.errors.full_messages unless user.save
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Enable Google Services for an organization'
|
||
|
|
task :enable_for_organization,
|
||
|
|
[:google_maps_key, :google_maps_private_key, :org_name] => [:environment] do |_, args|
|
||
|
|
add_feature_flag_if_not_exists('google_maps')
|
||
|
|
|
||
|
|
organization = ::Organization.where(name: args[:org_name]).first
|
||
|
|
|
||
|
|
raise "No organization with name '#{args[:org_name]}'" if organization.nil?
|
||
|
|
|
||
|
|
puts "Enabling 'google_maps' feature flag for users in '#{organization.name}'"
|
||
|
|
organization.users.each do |user|
|
||
|
|
puts "\tenabling 'google_maps' feature flag for user '#{user.username}'"
|
||
|
|
add_feature_flag_to_user('google_maps', user)
|
||
|
|
end
|
||
|
|
|
||
|
|
puts "Enabling google services for organization '#{organization.name}'..."
|
||
|
|
organization.google_maps_key = prepend_client_equal_if_needed(args[:google_maps_key])
|
||
|
|
|
||
|
|
unless args[:google_maps_private_key].blank?
|
||
|
|
organization.google_maps_private_key = args[:google_maps_private_key]
|
||
|
|
end
|
||
|
|
|
||
|
|
puts organization.errors.full_messages unless organization.save
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Enable Google Services for all users'
|
||
|
|
task :enable_for_all, [:google_maps_key, :google_maps_private_key] => [:environment] do |_, args|
|
||
|
|
add_feature_flag_if_not_exists('google_maps')
|
||
|
|
|
||
|
|
::Organization.all.each do |organization|
|
||
|
|
puts "Enabling 'google_maps' feature flag for users in '#{organization.name}'"
|
||
|
|
organization.users.each do |user|
|
||
|
|
puts "\tenabling 'google_maps' feature flag for user '#{user.username}'"
|
||
|
|
add_feature_flag_to_user('google_maps', user)
|
||
|
|
end
|
||
|
|
|
||
|
|
puts "Enabling google services for organization '#{organization.name}'..."
|
||
|
|
organization.google_maps_key = prepend_client_equal_if_needed(args[:google_maps_key])
|
||
|
|
|
||
|
|
unless args[:google_maps_private_key].blank?
|
||
|
|
organization.google_maps_private_key = args[:google_maps_private_key]
|
||
|
|
end
|
||
|
|
|
||
|
|
puts organization.errors.full_messages unless organization.save
|
||
|
|
end
|
||
|
|
|
||
|
|
# For users not in organizations
|
||
|
|
::User.where(organization_id: nil).each do |user|
|
||
|
|
puts "Enabling 'google_maps' feature flag for user '#{user.username}'"
|
||
|
|
add_feature_flag_to_user('google_maps', user)
|
||
|
|
|
||
|
|
puts "Enabling google services for non-organization user '#{user.username}'..."
|
||
|
|
user.google_maps_key = prepend_client_equal_if_needed(args[:google_maps_key])
|
||
|
|
user.google_maps_private_key = args[:google_maps_private_key] unless args[:google_maps_private_key].blank?
|
||
|
|
|
||
|
|
puts user.errors.full_messages unless user.save
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Disable Google Services for a given user'
|
||
|
|
task :disable_for_user, [:username] => [:environment] do |_, args|
|
||
|
|
user = ::User.where(username: args[:username]).first
|
||
|
|
|
||
|
|
raise "No user with username '#{args[:username]}'" if user.nil?
|
||
|
|
|
||
|
|
puts "Removing 'google_maps' feature flag from user '#{user.username}'"
|
||
|
|
remove_feature_flag_from_user('google_maps', user)
|
||
|
|
|
||
|
|
puts "Disabling google services for user '#{user.username}'..."
|
||
|
|
user.google_maps_key = nil
|
||
|
|
user.google_maps_private_key = nil
|
||
|
|
|
||
|
|
puts user.errors.full_messages unless user.save
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Disable Google Services for an organization'
|
||
|
|
task :disable_for_organization, [:org_name] => [:environment] do |_, args|
|
||
|
|
organization = ::Organization.where(name: args[:org_name]).first
|
||
|
|
|
||
|
|
raise "No organization with name '#{args[:org_name]}'" if organization.nil?
|
||
|
|
|
||
|
|
puts "Removing 'google_maps' feature flag from users in '#{organization.name}'"
|
||
|
|
organization.users.each do |user|
|
||
|
|
puts "\tremoving 'google_maps' feature flag from user '#{user.username}'"
|
||
|
|
remove_feature_flag_from_user('google_maps', user)
|
||
|
|
end
|
||
|
|
|
||
|
|
puts "Disabling google services for organization '#{organization.name}'..."
|
||
|
|
organization.google_maps_key = nil
|
||
|
|
organization.google_maps_private_key = nil
|
||
|
|
|
||
|
|
puts organization.errors.full_messages unless organization.save
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Disable Google Services for all users'
|
||
|
|
task disable_for_all: [:environment] do |_|
|
||
|
|
remove_feature_flag_if_exists('google_maps')
|
||
|
|
|
||
|
|
::Organization.all.each do |organization|
|
||
|
|
puts "Disabling google services for organization '#{organization.name}'..."
|
||
|
|
|
||
|
|
organization.google_maps_key = nil
|
||
|
|
organization.google_maps_private_key = nil
|
||
|
|
|
||
|
|
puts organization.errors.full_messages unless organization.save
|
||
|
|
end
|
||
|
|
|
||
|
|
# For users not in organizations
|
||
|
|
::User.where(organization_id: nil).each do |user|
|
||
|
|
puts "Disabling google services for non-organization user '#{user.username}'..."
|
||
|
|
|
||
|
|
user.google_maps_key = nil
|
||
|
|
user.google_maps_private_key = nil
|
||
|
|
|
||
|
|
puts user.errors.full_messages unless user.save
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|