pg_rls 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d2174ba1b02af4d50da23cd95648ca0098b376cef2f30796a159dbfc6b1ccf8f
-  data.tar.gz: e375b3153b94a1ed679f8ab69def4067f7b3c151a6d17c70584c17782498a9af
+  metadata.gz: 278c5a74c59b9b7b98709a4d36bd50d30facc864b140b231e5aa7d5a1bc26cfd
+  data.tar.gz: 30fab69ef11cf047c3c25213463160f66959c0d6f2da97a09e1bbfb6365247aa
 SHA512:
-  metadata.gz: f76546f92cbb5ab1b8f59741df0e504878ee68c755f2fe3791e576789a7f88edf17644929da8752882eb0c8f1dcce4f57fc090cd60763f96d9fb6e4f584f20d1
-  data.tar.gz: f2f377d2f78ddba9ca9776dc64880f00875993f4ed02364e85d3156ebbdeb7ae1a4bb75a925707ff7d39f2f61449a4a56c9f425ae55db40c1dd00bdf7eb2ded3
+  metadata.gz: 6df51d2edd3e608a4a9a0bdfd5c42c21cf9462b85e6a3b572f151991990d292b6e6fd625578254923145b836ea4129aacffcc08aeca0ca03eda9537be4dc8caa
+  data.tar.gz: 6f409cafedc4a4208e110cca2c8e13d76175272e4c4d022b1e87d3c1b3991e2dfe875f1fa80b6a1141654afa8fe31cbe6dd3c99faf29b5eccb50970e81207e76

data/lib/pg_rls/version.rb

@@ -2,6 +2,6 @@
 
 # :nocov:
 module PgRls
-  VERSION = "1.0.0"
+  VERSION = "1.0.1"
 end
 # :nocov:

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_rls
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Daniel Laloush
@@ -87,13 +87,11 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".rspec"
 - ".rubocop.yml"
 - ".ruby-version"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
 - Guardfile
-- LICENSE.txt
 - README.md
 - Rakefile
 - Steepfile
@@ -107,19 +105,8 @@ files:
 - app/models/pg_rls/tenant/switchable.rb
 - assets/logo.svg
 - docker-compose.yml
-- lib/generators/pg_rls.rb
 - lib/generators/pg_rls/active_record/active_record_generator.rb
-- lib/generators/pg_rls/active_record/templates/abstract_base_class.rb.tt
-- lib/generators/pg_rls/active_record/templates/convert_migration.rb.tt
-- lib/generators/pg_rls/active_record/templates/convert_migration_backport.rb.tt
-- lib/generators/pg_rls/active_record/templates/init_convert_migration.rb.tt
-- lib/generators/pg_rls/active_record/templates/init_migration.rb.tt
-- lib/generators/pg_rls/active_record/templates/init_model.rb.tt
-- lib/generators/pg_rls/active_record/templates/migration.rb.tt
-- lib/generators/pg_rls/active_record/templates/model.rb.tt
-- lib/generators/pg_rls/base.rb
 - lib/generators/pg_rls/install/install_generator.rb
-- lib/generators/pg_rls/install_generator.rb
 - lib/generators/pg_rls/pg_rls_generator.rb
 - lib/generators/pg_rls/templates/USAGE
 - lib/generators/pg_rls/templates/app/models/abstract_base_class.rb.tt
@@ -130,10 +117,7 @@ files:
 - lib/generators/pg_rls/templates/db/migrate/convert_to_pg_rls_tenant_table.rb.tt
 - lib/generators/pg_rls/templates/db/migrate/create_pg_rls_table.rb.tt
 - lib/generators/pg_rls/templates/db/migrate/create_pg_rls_tenant_table.rb.tt
-- lib/generators/templates/README
-- lib/generators/templates/pg_rls.rb.tt
 - lib/pg_rls.rb
-- lib/pg_rls/Rakefile
 - lib/pg_rls/active_record.rb
 - lib/pg_rls/active_record/connection_adapters.rb
 - lib/pg_rls/active_record/connection_adapters/connection_pool.rb
@@ -155,32 +139,12 @@ files:
 - lib/pg_rls/active_support.rb
 - lib/pg_rls/active_support/string_ext.rb
 - lib/pg_rls/connection_config.rb
-- lib/pg_rls/current/context.rb
-- lib/pg_rls/database/admin_statements.rb
-- lib/pg_rls/database/configurations.rb
-- lib/pg_rls/database/prepared.rb
-- lib/pg_rls/database/tasks/admin_database.rake
 - lib/pg_rls/deprecation.rb
 - lib/pg_rls/engine.rb
 - lib/pg_rls/error.rb
-- lib/pg_rls/errors/index.rb
-- lib/pg_rls/errors/rake_only_error.rb
-- lib/pg_rls/errors/tenant_not_found.rb
 - lib/pg_rls/generators/.keep
-- lib/pg_rls/logger.rb
-- lib/pg_rls/middleware.rb
-- lib/pg_rls/middleware/set_reset_connection.rb
-- lib/pg_rls/middleware/sidekiq.rb
-- lib/pg_rls/middleware/sidekiq/client.rb
-- lib/pg_rls/middleware/sidekiq/server.rb
-- lib/pg_rls/multi_tenancy.rb
 - lib/pg_rls/railtie.rb
-- lib/pg_rls/schema/down_statements.rb
-- lib/pg_rls/schema/dumper.rb
-- lib/pg_rls/schema/statements.rb
-- lib/pg_rls/schema/up_statements.rb
 - lib/pg_rls/tasks/.keep
-- lib/pg_rls/tenant.rb
 - lib/pg_rls/version.rb
 - rbs_collection.lock.yaml
 - rbs_collection.yaml

data/.rspec

@@ -1,3 +0,0 @@
---format documentation
---color
---require spec_helper

data/LICENSE.txt

@@ -1,21 +0,0 @@
-The MIT License (MIT)
-
-Copyright (c) 2021 Daniel Laloush
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in
-all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-THE SOFTWARE.

data/lib/generators/pg_rls/active_record/templates/abstract_base_class.rb.tt

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-<% module_namespacing do -%>
-class <%= abstract_class_name %> < ApplicationRecord
-  self.abstract_class = true
-
-  connects_to database: { <%= ActiveRecord.writing_role %>: :<%= database -%> }
-end
-<% end -%>

data/lib/generators/pg_rls/active_record/templates/convert_migration.rb.tt

@@ -1,11 +0,0 @@
-# frozen_string_literal: true
-
-class PgRlsConvert<%= table_name.camelize %> < ActiveRecord::Migration<%= migration_version %>
-  def up
-    convert_to_rls_table :<%= table_name %>
-  end
-
-  def down
-    revert_rls_table :<%= table_name %>
-  end
-end

data/lib/generators/pg_rls/active_record/templates/convert_migration_backport.rb.tt

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-class PgRlsBackport<%= table_name.camelize %> < ActiveRecord::Migration<%= migration_version %>
-  def up
-    # Suggested Code:
-    # PgRls.on_each_tenant do |tenant|
-    #   tenant.<%= table_name %>.in_batches(of: 100) do |<%= table_name %>|
-    #     <%= table_name %>.each { |<%= table_name.singularize %>| <%= table_name.singularize %>.update_attribute('tenant_id', tenant.tenant_id) }
-    #   end
-    # end
-  end
-end

data/lib/generators/pg_rls/active_record/templates/init_convert_migration.rb.tt

@@ -1,11 +0,0 @@
-# frozen_string_literal: true
-
-class PgRlsConvertTenant<%= PgRls.table_name.camelize %> < ActiveRecord::Migration<%= migration_version %>
-  def up
-    convert_to_rls_tenant_table :<%= table_name %>
-  end
-
-  def down
-    revert_rls_tenant_table :<%= table_name %>
-  end
-end

data/lib/generators/pg_rls/active_record/templates/init_migration.rb.tt

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-class PgRlsCreateTenant<%= PgRls.table_name.camelize %> < ActiveRecord::Migration<%= migration_version %>
-  def up
-    create_rls_tenant_table :<%= table_name %>, id: :uuid do |t|
-      t.string :name
-      t.string :logo
-
-      t.string :identification
-      t.string :subdomain
-      t.string :domain
-
-      t.timestamps
-    end
-
-    add_index :<%= table_name %>, :name,            unique: true
-    add_index :<%= table_name %>, :identification,  unique: true
-    add_index :<%= table_name %>, :domain,          unique: true
-    add_index :<%= table_name %>, :subdomain,       unique: true
-  end
-
-  def down
-    drop_rls_tenant_table :<%= table_name %>
-  end
-end

data/lib/generators/pg_rls/active_record/templates/init_model.rb.tt

@@ -1,24 +0,0 @@
-# frozen_string_literal: true
-
-<% module_namespacing do -%>
-class <%= PgRls.class_name.camelize %> < <%= parent_class_name.classify %>
-<% attributes.select(&:reference?).each do |attribute| -%>
-  belongs_to :<%= attribute.name %><%= ", polymorphic: true" if attribute.polymorphic? %>
-<% end -%>
-<% attributes.select(&:rich_text?).each do |attribute| -%>
-  has_rich_text :<%= attribute.name %>
-<% end -%>
-<% attributes.select(&:attachment?).each do |attribute| -%>
-  has_one_attached :<%= attribute.name %>
-<% end -%>
-<% attributes.select(&:attachments?).each do |attribute| -%>
-  has_many_attached :<%= attribute.name %>
-<% end -%>
-<% attributes.select(&:token?).each do |attribute| -%>
-  has_secure_token<% if attribute.name != "token" %> :<%= attribute.name %><% end %>
-<% end -%>
-<% if attributes.any?(&:password_digest?) -%>
-  has_secure_password
-<% end -%>
-end
-<% end -%>

data/lib/generators/pg_rls/active_record/templates/migration.rb.tt

@@ -1,17 +0,0 @@
-# frozen_string_literal: true
-
-class PgRlsCreate<%= table_name.camelize %> < ActiveRecord::Migration<%= migration_version %>
-  def up
-    create_rls_table :<%= table_name %><%= primary_key_type %> do |t|
-      <% attributes.each do |attribute| -%>
-      t.<%= attribute.type %> :<%= attribute.name %>
-      <% end -%>
-
-      t.timestamps null: false
-    end
-  end
-
-  def down
-    drop_rls_table :<%= table_name %>
-  end
-end

data/lib/generators/pg_rls/active_record/templates/model.rb.tt

@@ -1,24 +0,0 @@
-# frozen_string_literal: true
-
-<% module_namespacing do -%>
-class <%= class_name %> < <%= parent_class_name.classify %>
-<% attributes.select(&:reference?).each do |attribute| -%>
-  belongs_to :<%= attribute.name %><%= ", polymorphic: true" if attribute.polymorphic? %>
-<% end -%>
-<% attributes.select(&:rich_text?).each do |attribute| -%>
-  has_rich_text :<%= attribute.name %>
-<% end -%>
-<% attributes.select(&:attachment?).each do |attribute| -%>
-  has_one_attached :<%= attribute.name %>
-<% end -%>
-<% attributes.select(&:attachments?).each do |attribute| -%>
-  has_many_attached :<%= attribute.name %>
-<% end -%>
-<% attributes.select(&:token?).each do |attribute| -%>
-  has_secure_token<% if attribute.name != "token" %> :<%= attribute.name %><% end %>
-<% end -%>
-<% if attributes.any?(&:password_digest?) -%>
-  has_secure_password
-<% end -%>
-end
-<% end -%>

data/lib/generators/pg_rls/base.rb

@@ -1,36 +0,0 @@
-# frozen_string_literal: true
-
-module PgRls
-  # Main Definition for Generator
-  module Base
-    protected
-
-    def nested_parent_name
-      @class_path.join('/')
-    end
-
-    def nested_parent_id
-      "#{nested_parent_name}_id"
-    end
-
-    def nested_parent_class_name
-      nested_parent_name.classify
-    end
-
-    def plural_nested_parent_name
-      nested_parent_name.pluralize
-    end
-
-    def class_path
-      []
-    end
-
-    def regular_class_path
-      []
-    end
-
-    def controller_class_path
-      []
-    end
-  end
-end

data/lib/generators/pg_rls/install_generator.rb

@@ -1,90 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators/base'
-require 'securerandom'
-
-module PgRls
-  module Generators
-    MissingORMError = Class.new(Thor::Error)
-    # Installer Generator
-    class InstallGenerator < Rails::Generators::Base
-      def initialize(*args)
-        tenant_model_or_table = args.first
-        if tenant_model_or_table.present?
-          PgRls.table_name = tenant_model_or_table.first.pluralize
-          PgRls.class_name = tenant_model_or_table.first.singularize
-        end
-        super
-      end
-      ENVIRONMENT_LINE = 'Rails.application.initialize!'
-      ENVIRONMENT_PATH = 'config/environment.rb'
-
-      APPLICATION_LINE = 'class Application < Rails::Application'
-      APPLICATION_PATH = 'config/application.rb'
-
-      APPLICATION_RECORD_LINE = 'class ApplicationRecord < ActiveRecord::Base'
-      APPLICATION_RECORD_PATH = 'app/models/application_record.rb'
-
-      APPLICATION_CONTROLLER_LINE = 'class ApplicationController < ActionController::Base'
-      APPLICATION_CONTROLLER_PATH = 'app/controllers/application_controller.rb'
-
-      source_root File.expand_path('../templates', __dir__)
-
-      desc 'Creates a PgRls initializer and copy locale files to your application.'
-
-      hook_for :orm, required: true
-
-      def orm_error_message
-        <<~ERROR
-          An ORM must be set to install PgRls in your application.
-          Be sure to have an ORM like Active Record or loaded in your
-          app or configure your own at `config/application.rb`.
-            config.generators do |g|
-              g.orm :your_orm_gem
-            end
-        ERROR
-      end
-
-      def copy_initializer
-        raise MissingORMError, orm_error_message unless options[:orm]
-
-        inject_include_to_application_controller
-        template 'pg_rls.rb.tt', 'config/initializers/pg_rls.rb'
-      end
-
-      def inject_include_to_environment
-        return if environment_already_included?
-
-        gsub_file(ENVIRONMENT_PATH, /(#{Regexp.escape(ENVIRONMENT_LINE)})/mio) do |match|
-          "require_relative 'initializers/pg_rls'\n#{match}"
-        end
-      end
-
-      def inject_include_to_application_controller
-        return if aplication_controller_already_included?
-
-        gsub_file(APPLICATION_CONTROLLER_PATH, /(#{Regexp.escape(APPLICATION_CONTROLLER_LINE)})/mio) do |match|
-          "#{match}\n  include PgRls::MultiTenancy\n"
-        end
-      end
-
-      def aplication_controller_already_included?
-        File.readlines(APPLICATION_CONTROLLER_PATH).grep(/include PgRls::MultiTenancy/).any?
-      end
-
-      def environment_already_included?
-        File.readlines(ENVIRONMENT_PATH).grep(%r{require_relative 'initializers/pg_rls'}).any?
-      end
-
-      def initialize_error_text
-        <<~ERROR
-          TO DO
-        ERROR
-      end
-
-      def show_readme
-        readme 'README' if behavior == :invoke
-      end
-    end
-  end
-end

data/lib/generators/pg_rls.rb

@@ -1,19 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators/named_base'
-require 'rails/generators/active_model'
-require 'rails/generators/active_record/migration'
-require 'active_record'
-
-module PgRls
-  module Generators # :nodoc:
-    class PgRlsGenerator < Rails::Generators::NamedBase # :nodoc:
-      include PgRls::Generators::Migration
-
-      # Set the current directory as base for the inherited generators.
-      def self.base_root
-        __dir__
-      end
-    end
-  end
-end

data/lib/generators/templates/README

@@ -1,22 +0,0 @@
-README
-===============================================================================
-WARNING!!
-
-PgRls required that ActiveRecord format migration as SQL
-
-Once you remove a tenant all of his data would be removed as well
-
-If you're setting a custom user, make sure to regenerate the structure.sql since
-Postgresql policies are created on each user
-
-===============================================================================
-
-to generate secure model run
-
-rails g pg_rls model_name
-
-or
-
-rails generate pg_rls model_name
-
-===============================================================================

data/lib/generators/templates/pg_rls.rb.tt

@@ -1,48 +0,0 @@
-# frozen_string_literal: true
-
-require 'pg_rls'
-
-PgRls.setup do |config|
-  ActiveRecord::ConnectionAdapters::AbstractAdapter.include PgRls::Schema::Statements
-
-  # Do not remove this value after initialization
-  config.class_name = :<%= PgRls.class_name %>
-  config.table_name = :<%= PgRls.table_name %>
-  config.search_methods = <%= PgRls.search_methods %>
-  # If you are using `solid_queue`, `solid_cache`, or `solid_cable` with a sharding configuration,
-  # we recommend excluding these shards from Row-Level Security (RLS) to avoid the need to reset
-  # RLS on each shard.
-  #
-  # By default, RLS will be enabled for all shards.
-  # You can specify which shards to exclude from RLS using the `config.excluded_shards` option:
-  #
-  #   config.excluded_shards = []
-  #
-  # Note: While it's technically possible to leave `solid_cache` and `solid_cable` under RLS,
-  # it is generally unnecessary and may introduce complexity without added benefit.
-
-  ##
-  ## Uncomment this lines if you have a custome user per environment
-  ## don't forget to grant the required privilange in order for it to run
-  ##
-  ## Remember that PgRls is adding triggers that would set the RLS to the default user
-  ## make sure you recreate the structure.sql on each environment
-  ##
-  # config.username = Rails.application.credentials.dig(:database, :username)
-  # config.password = Rails.application.credentials.dig(:database, :password)
-
-  ## ------------------------------ Middleware SetResetConnection -----------------------------
-  ## Uncomment this lines if you're using SetResetConnection Middleware
-  #
-  # config.session_store_server = Rails.application.config_for(:redis).session
-  #
-  ## Uncomment this line if you're not using warden as your authentication system or if you
-  ## changed the default warden key. Devise, uses warden authentication.
-  # config.session_store_default_warden_key = '2'
-  #
-  ## Uncomment this line if you're setting a diferent session key than stablished under your
-  ## redis server configuration
-  # config.session_key_prefix = '_hub_session'
-  ##
-  ## ------------------------------ Middleware SetResetConnection -----------------------------
-end

data/lib/pg_rls/Rakefile

@@ -1,7 +0,0 @@
-# frozen_string_literal: true
-
-require_relative '../pg_rls'
-
-path = File.expand_path(__dir__)
-
-Dir.glob("#{path}/database/tasks/**/*.rake").each { |f| import f }

data/lib/pg_rls/current/context.rb

@@ -1,10 +0,0 @@
-# frozen_string_literal: true
-
-module PgRls
-  # Current Context
-  module Current
-    class Context < ActiveSupport::CurrentAttributes
-      attribute :tenant
-    end
-  end
-end

data/lib/pg_rls/database/admin_statements.rb

@@ -1,28 +0,0 @@
-# frozen_string_literal: true
-
-module PgRls
-  module Admin
-    module ActiveRecord
-      module Migrator
-        def initialize(*args)
-          PgRls.instance_variable_set(:@as_db_admin, true)
-          super
-        end
-      end
-
-      module Tasks
-        module DatabaseTasks
-          def resolve_configuration(configuration)
-            PgRls.instance_variable_set(:@as_db_admin, true) unless PgRls.as_db_admin?
-            super
-          end
-
-          def migration_class
-            PgRls.instance_variable_set(:@as_db_admin, true) unless PgRls.as_db_admin?
-            super
-          end
-        end
-      end
-    end
-  end
-end

data/lib/pg_rls/database/configurations.rb

@@ -1,46 +0,0 @@
-# frozen_string_literal: true
-
-require 'active_record/database_configurations'
-
-module ActiveRecord
-  # ActiveRecord::DatabaseConfigurations
-  class DatabaseConfigurations
-    class HashConfig
-      def initialize(env_name, name, configuration_hash)
-        @env_name = env_name
-        @name = name
-        @configuration_hash = configuration_hash
-      end
-
-      def configuration_hash
-        return @configuration_hash if PgRls.excluded_shards.include?(@name.to_s)
-
-        reset_pg_rls_configuration if db_changed?
-
-        return admin_configuration_hash if PgRls.as_db_admin?
-
-        rls_configuration_hash
-      end
-
-      def admin_configuration_hash
-        @admin_configuration_hash ||= @configuration_hash
-      end
-
-      def rls_configuration_hash
-        @rls_configuration_hash ||= @configuration_hash.deep_dup.tap do |config|
-          config[:username] = PgRls.username
-          config[:password] = PgRls.password
-        end.freeze
-      end
-
-      def db_changed?
-        admin_configuration_hash[:database] != @configuration_hash[:database]
-      end
-
-      def reset_pg_rls_configuration
-        @rls_configuration_hash = nil
-        @admin_configuration_hash = nil
-      end
-    end
-  end
-end

data/lib/pg_rls/database/prepared.rb

@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-module PgRls
-  module Database
-    # Prepare database for test unit
-    module Prepared
-      class << self
-        def grant_user_credentials(name: PgRls.username, schema: 'public')
-          PgRls.admin_execute <<-SQL.squish
-            DO
-            $do$
-            BEGIN
-              IF NOT EXISTS (
-                SELECT table_catalog, table_schema, table_name, privilege_type
-                  FROM   information_schema.table_privileges
-                  WHERE  grantee = '#{name}'
-              ) THEN
-                  GRANT ALL PRIVILEGES ON TABLE schema_migrations TO #{name};
-                  GRANT USAGE ON SCHEMA #{schema} TO #{name};
-                  ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema}
-                    GRANT USAGE, SELECT
-                    ON SEQUENCES TO #{name};
-                  ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema}
-                    GRANT SELECT, INSERT, UPDATE, DELETE
-                    ON TABLES TO #{name};
-                  GRANT SELECT, INSERT, UPDATE, DELETE
-                    ON ALL TABLES IN SCHEMA #{schema}
-                    TO #{name};
-                  GRANT USAGE, SELECT
-                    ON ALL SEQUENCES IN SCHEMA #{schema}
-                    TO #{name};
-                END IF;
-            END;
-            $do$;
-          SQL
-        end
-      end
-    end
-  end
-end

data/lib/pg_rls/database/tasks/admin_database.rake

@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-# OVERIDE RAILS TASK
-Rake::TaskManager.class_eval do
-  def alias_task(fq_name)
-    new_name = "#{fq_name}:original"
-    @tasks[new_name] = @tasks.delete(fq_name) unless @tasks[fq_name].nil?
-  end
-end
-
-def alias_task(fq_name)
-  Rake.application.alias_task(fq_name)
-end
-
-def override_task(*args, &)
-  name, _params, _deps = Rake.application.resolve_args(args.dup)
-  fq_name = Rake.application.instance_variable_get(:@scope).to_a.reverse.push(name).join(':')
-  alias_task(fq_name)
-  Rake::Task.define_task(*args, &)
-end
-
-namespace :db do
-  include PgRls::Schema::UpStatements
-
-  override_task :load_config do
-    PgRls.instance_variable_set(:@as_db_admin, true)
-    Rake::Task['db:load_config:original'].invoke
-  end
-
-  override_task grant_usage: :load_config do
-    create_rls_user
-  end
-
-  namespace :test do
-    override_task grant_usage: :load_config do
-      PgRls.instance_variable_set(:@as_db_admin, true)
-      create_rls_user
-    end
-  end
-end