Initial commit

lib/tasks/api_key.rake

@@ -0,0 +1,122 @@
+namespace :carto do
+  namespace :api_key do
+    module ApiKeyRake
+      def for_each_api_key(query, description)
+        all_count = query.count
+        current_count = 0
+        reflection_seconds = ENV['ENABLE_FF_REFLECTION_SECONDS'] || 10
+
+        puts "############################"
+        puts "#"
+        puts "# #{description}"
+        puts "#"
+        puts "# #{all_count} api keys will be processed"
+        puts "#"
+        puts "# You have #{reflection_seconds} seconds to cancel "
+        puts "# the task before it starts"
+        puts "#"
+        puts "############################"
+
+        sleep reflection_seconds.to_i
+
+        query.find_each do |api_key|
+          begin
+            current_count += 1
+            puts "#{current_count} of #{all_count}"
+
+            yield api_key
+          rescue StandardError => e
+            puts "ERROR - API Key #{api_key.id}: #{e}"
+          end
+        end
+      end
+
+      def create_api_keys_for_user(carto_user)
+        if carto_user.api_keys.master.empty?
+          puts "INFO: Creating master API Key for user #{carto_user.username}"
+          carto_user.api_keys.create_master_key!
+        end
+
+        if carto_user.api_keys.default_public.empty?
+          puts "INFO: Creating default public API Key for user #{carto_user.username}"
+          carto_user.api_keys.create_default_public_key!
+        end
+      end
+
+      def rename_api_keys_for_user_id(user_id)
+        Carto::ApiKey.where(user_id: user_id).each do |api_key|
+          api_key.reload
+          n = 1
+          Carto::ApiKey.where(name: api_key.name).where('id != ?', api_key.id).each do |renamed_api_key|
+            renamed_api_key.update_column(:name, "#{renamed_api_key.name} #{n}")
+            n += 1
+          end
+        end
+      end
+    end
+
+    # This should only be used for development, before CartoDB/cartodb/pull/13396 migration
+    desc 'Rename Api Keys to avoid name collisions within the same user'
+    task rename: :environment do
+      include ApiKeyRake
+      Carto::ApiKey.uniq.pluck(:user_id).each { |user_id| rename_api_keys_for_user_id(user_id) }
+    end
+
+    desc 'Creates default API Keys for user'
+    task :create_default_for_emails, [:emails_file] => [:environment] do |_, args|
+      line = 0
+      File.foreach(args[:emails_file]) do |email|
+        email.strip!
+        line += 1
+
+        unless email =~ /.*@.*\..*/
+          puts "Row #{line} is not an email: #{email}"
+          next
+        end
+
+        user = User.first(email: email)
+
+        if user.nil?
+          puts "WARN: User #{email} not found"
+        else
+          begin
+            user.create_api_keys
+            puts "INFO: Api Keys created for #{user.email}"
+          rescue => e
+            puts "WARN: Unable to create api keys for user with email #{email}: #{e}"
+          end
+        end
+      end
+    end
+
+    desc 'Creates default API Keys for users who don\'t have them yet'
+    task create_default: :environment do
+      include ApiKeyRake
+      Carto::User.joins('LEFT JOIN api_keys ON api_keys.user_id = users.id').where(api_keys: { id: nil }).find_each { |u| create_api_keys_for_user(u) }
+    end
+
+    desc 'Creates default API Keys for users who don\'t have them yet under given organization'
+    task :create_default_for_organization, [:org_name] => [:environment] do |_, args|
+      include ApiKeyRake
+      puts "INFO: Creating default API Keys for all users in organization #{args[:org_name]}"
+      org = Carto::Organization.find_by_name(args[:org_name])
+      raise "ERROR: Organization not found" unless org
+      org.users.each { |u| create_api_keys_for_user(u) }
+    end
+
+    desc 'Creates grants for every API Key'
+    task create_api_key_grants: :environment do
+      include ApiKeyRake
+      MESSAGE = "Creating permissions for every API Key".freeze
+
+      for_each_api_key(Carto::ApiKey, MESSAGE) do |api_key|
+        if api_key.master?
+          api_key.grants = Carto::ApiKey::MASTER_API_KEY_GRANTS
+          api_key.save!
+        else
+          api_key.save_cdb_conf_info
+        end
+      end
+    end
+  end
+end