rls_multi_tenant 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e90e7f280c785aa09f9f643da2420a38423f888592a750d4587dcc961ede1de1
-  data.tar.gz: cb1ec28221f70d26bcf6444be404feccf6098417325921692f8c95ba4d945a3c
+  metadata.gz: fcb89757e26727014390f0ce50eed58d22e760d4249508bf58c550533e0ef14c
+  data.tar.gz: 21d57b4c19ff0f5117bd62df654b2815eb75a54749adf8a689448816a29f3141
 SHA512:
-  metadata.gz: e28c40e201630a87955c890b4f417d43c0e7418674a6bf776d44cd4a39b28cb6c49b4873c6807e0e21247b8e120682a986e1bccf95cdc47d1be29f2b63c213c6
-  data.tar.gz: 986900e03f1234f1f19813f1680af7675861c35a437b64582512854bdc9f7d05828a78a6bf6755f656a2b05fa7e47447312c83d53796bea4929962f1ed18b70c
+  metadata.gz: 0c00dde13e466a6d466f8170b3732775654bdb1655d1985717d551667313d3a250cbaf74067a770f14600da7d64582d1d2b241e668f8f3b6f831752d3767f9f5
+  data.tar.gz: 4a3df9f10af7671d2ed572b8d3f1009155bff00014f6823e130a39b3f4d50a1b5de9e2a86ce271e38e7378e99185694435e6332012e7091a6b287c803ac81c5b

data/README.md

@@ -16,7 +16,7 @@ A Rails gem that provides PostgreSQL Row Level Security (RLS) based multi-tenanc
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'rls_multi_tenant', path: 'gems/rls_multi_tenant'
+gem 'rls_multi_tenant'
 ```
 
 And then execute:

data/lib/rls_multi_tenant/generators/install/install_generator.rb

@@ -38,11 +38,7 @@ module RlsMultiTenant
       end
 
       def create_app_user_migration
-        unless Dir.glob(File.join(destination_root, "db/migrate/*_create_app_user.rb")).any?
-          create_migration_with_timestamp("create_app_user", 2)
-        else
-          say "App user migration already exists, skipping creation", :yellow
-        end
+        create_app_user_migrations_for_all_databases
       end
 
       def create_tenant_migration
@@ -71,6 +67,50 @@ module RlsMultiTenant
         say "="*60, :green
       end
 
+      def create_app_user_migrations_for_all_databases
+        # Get database configuration for current environment
+        db_config = Rails.application.config.database_configuration[Rails.env]
+        
+        # Handle both single database and multiple databases configuration
+        databases_to_process = if db_config.is_a?(Hash) && db_config.key?('primary')
+          # Multiple databases configuration
+          db_config
+        else
+          # Single database configuration - treat as primary
+          { 'primary' => db_config }
+        end
+
+        databases_to_process.each do |db_name, config|
+          next if db_name == 'primary' # Skip primary database, handle it separately
+          
+          # Check if migrations_paths is defined for this database
+          if config['migrations_paths']
+            migration_paths = Array(config['migrations_paths'])
+            migration_paths.each do |migration_path|
+              migration_dir = File.join(destination_root, migration_path)
+              
+              # Check if migration already exists in this path
+              unless Dir.glob(File.join(migration_dir, "*_create_app_user.rb")).any?
+                FileUtils.mkdir_p(migration_dir) unless File.directory?(migration_dir)
+                create_migration_with_timestamp_for_path("create_app_user", 2, migration_path)
+                say "Created app user migration for #{db_name} in #{migration_path}", :green
+              else
+                say "App user migration already exists for #{db_name} in #{migration_path}, skipping creation", :yellow
+              end
+            end
+          else
+            say "No migrations_paths defined for database '#{db_name}', skipping app user migration", :yellow
+          end
+        end
+
+        # Handle primary database (default behavior)
+        unless Dir.glob(File.join(destination_root, "db/migrate/*_create_app_user.rb")).any?
+          create_migration_with_timestamp("create_app_user", 2)
+        else
+          say "App user migration already exists for primary database, skipping creation", :yellow
+        end
+      end
+
       private
 
       def create_migration_with_timestamp(migration_type, order)
@@ -86,6 +126,16 @@ module RlsMultiTenant
           template "create_tenant.rb", "db/migrate/#{timestamp}_create_tenants.rb"
         end
       end
+
+      def create_migration_with_timestamp_for_path(migration_type, order, migration_path)
+        base_timestamp = Time.current.strftime("%Y%m%d%H%M")
+        timestamp = "#{base_timestamp}#{sprintf('%02d', order)}"
+        
+        case migration_type
+        when "create_app_user"
+          template "create_app_user.rb", "#{migration_path}/#{timestamp}_create_app_user.rb"
+        end
+      end
     end
   end
 end

data/lib/rls_multi_tenant/generators/migration/migration_generator.rb

@@ -30,8 +30,7 @@ module RlsMultiTenant
       private
 
       def create_app_user_migration
-        timestamp = Time.current.strftime("%Y%m%d%H%M%S")
-        template "create_app_user.rb", "db/migrate/#{timestamp}_create_app_user.rb"
+        create_app_user_migrations_for_all_databases
       end
 
       def create_enable_rls_migration
@@ -49,6 +48,44 @@ module RlsMultiTenant
         template "enable_uuid_extension.rb", "db/migrate/#{timestamp}_enable_uuid_extension.rb"
       end
 
+      def create_app_user_migrations_for_all_databases
+        # Get database configuration for current environment
+        db_config = Rails.application.config.database_configuration[Rails.env]
+        
+        # Handle both single database and multiple databases configuration
+        databases_to_process = if db_config.is_a?(Hash) && db_config.key?('primary')
+          # Multiple databases configuration
+          db_config
+        else
+          # Single database configuration - treat as primary
+          { 'primary' => db_config }
+        end
+
+        databases_to_process.each do |db_name, config|
+          next if db_name == 'primary' # Skip primary database, handle it separately
+          
+          # Check if migrations_paths is defined for this database
+          if config['migrations_paths']
+            migration_paths = Array(config['migrations_paths'])
+            migration_paths.each do |migration_path|
+              migration_dir = File.join(destination_root, migration_path)
+              FileUtils.mkdir_p(migration_dir) unless File.directory?(migration_dir)
+              
+              timestamp = Time.current.strftime("%Y%m%d%H%M%S")
+              template "create_app_user.rb", "#{migration_path}/#{timestamp}_create_app_user.rb"
+              say "Created app user migration for #{db_name} in #{migration_path}", :green
+            end
+          else
+            say "No migrations_paths defined for database '#{db_name}', skipping app user migration", :yellow
+          end
+        end
+
+        # Handle primary database (default behavior)
+        timestamp = Time.current.strftime("%Y%m%d%H%M%S")
+        template "create_app_user.rb", "db/migrate/#{timestamp}_create_app_user.rb"
+        say "Created app user migration for primary database", :green
+      end
+
       def table_name
         @table_name ||= ask("Enter table name for RLS:")
       end

data/lib/rls_multi_tenant/generators/migration/templates/create_app_user.rb

@@ -18,6 +18,10 @@ class CreateAppUser < ActiveRecord::Migration[<%= Rails.version.to_f %>]
     execute "GRANT CONNECT ON DATABASE #{ActiveRecord::Base.connection.current_database} TO #{app_user};"
     execute "GRANT USAGE ON SCHEMA public TO #{app_user};"
     execute "GRANT CREATE ON SCHEMA public TO #{app_user};"
+
+    # Grant permissions on system tables
+    execute "GRANT SELECT ON TABLE schema_migrations TO #{app_user};"
+    execute "GRANT SELECT ON TABLE ar_internal_metadata TO #{app_user};"
   end
 
   def down

data/lib/rls_multi_tenant/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RlsMultiTenant
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rls_multi_tenant
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Coding Ways