wal 0.0.6 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 45034aa6d8fa0824843be7e7940e8e85e4363ca80c58cb0838c92774d46f37c7
-  data.tar.gz: 721a2121845c5eee3cb3f5751a439bfff28a6d54eebc8c12117090aee068c453
+  metadata.gz: 267c5bf23d2617d97504613afc0e4fd4ca533a2ac8fd0852fbe38a4e6671abe4
+  data.tar.gz: ebccdc39d41b731395678882e45c63afff1d72b5e607f91e142d871890776d7d
 SHA512:
-  metadata.gz: fd3f30a1ff2e8bb18bdc39edd89659d939c944ff7acaefbd9545227b06e728b93432e84f37877321d5df2d8a4a8163613f3287b13ed4dae700dbfe145a18e4fc
-  data.tar.gz: 3a86e518afdd3256769fd65c428387be1148e9494867c087d4dd8e32b03f2b50fb7e6c32ecaa00de9caae805beaac2edf9c194485fe2ccc1f943a0a9c175bfc6
+  metadata.gz: f050e89a405581fd20ee704ba89c44a99920baf3d3e2703e678a319c8845c06803b695b2a69a457b05ef8e60106de7e57c253ff43546ad7f01dde2c20e70d9e3
+  data.tar.gz: dce3f46c0bb2688532a46bdc364be1306ef91eb2b1267bd0e37b74a68f9f85d8dd8fe436ffb5b1f44b119da3b7e9605c96faf4fd7109bd30ac09102a8de3415c

data/exe/wal

@@ -47,7 +47,7 @@ elsif cli["start"]
     watcher = config["watcher"].constantize.new
     temporary = config["temporary"] || false
     publications = config["publications"] || []
-    replicator = config["--replicator"].presence&.constantize || Wal::Replicator
+    replicator = config["replicator"].presence&.constantize || Wal::Replicator
 
     Thread.new(slot, watcher, temporary, publications) do |replication_slot, watcher, use_temporary_slot, publications|
       replication_slot = "#{replication_slot}_#{SecureRandom.alphanumeric(4)}" if use_temporary_slot

data/lib/wal/generators/migration.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+require "rails/generators/active_record"
+
+module Wal
+  module Generators
+    class Migration < Rails::Generators::Base
+      include Rails::Generators::Migration
+
+      source_root File.expand_path("templates", __dir__)
+
+      argument :watcher, type: :string
+
+      def self.next_migration_number(dir)
+        ::ActiveRecord::Generators::Base.next_migration_number(dir)
+      end
+
+      def create_migration_file
+        migration_template("migration.rb.erb", "db/migrate/#{migration_class_name.underscore}.rb")
+      end
+
+      private
+
+      def activerecord_migration_class
+        if ActiveRecord::Migration.respond_to? :current_version
+          "ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]"
+        else
+          "ActiveRecord::Migration"
+        end
+      end
+
+      def publication_name
+        "#{watcher.underscore}_publication"
+      end
+
+      def class_name
+        watcher.camelize
+      end
+
+      def migration_class_name
+        "Set#{class_name}Publication"
+      end
+
+      def publication_tables
+        if (watcher = class_name.constantize) && watcher < Wal::RecordWatcher
+          tables = watcher
+            .change_callbacks
+            .map { |table, configs| [table.to_s, configs.flat_map { |config| config[:changed] || [] }.uniq] }
+            .to_h
+
+          watcher
+            .delete_callbacks
+            .keys
+            .reduce(tables) { |tables, table| tables.reverse_merge(table => []) }
+        else
+          []
+        end
+      end
+    end
+  end
+end

data/lib/wal/generators/templates/migration.rb.erb

@@ -0,0 +1,9 @@
+class <%= migration_class_name %> < <%= activerecord_migration_class %>
+  def change
+    define_publication :<%= publication_name %> do |p|
+<% publication_tables.each do |table, columns| -%>
+      p.table :<%= table %><%= columns.present? ? ", columns: #{columns}" : "" %>
+<% end -%>
+    end
+  end
+end

data/lib/wal/migration.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+module Wal
+  module Migration
+    class PublicationDefinition
+      def initialize(name)
+        @name = name
+        @tables = {}
+      end
+
+      def table(name, columns: [])
+        @tables[name] ||= [].to_set
+        @tables[name].merge(columns.map(&:to_s))
+      end
+
+      def tables
+        @tables.keys
+      end
+
+      def to_sql
+        tables_sql = @tables.reduce([]) do |sql, (table, columns)|
+          table_sql = "TABLE #{table}"
+          table_sql << " (#{columns.join(", ")})" unless columns.empty?
+          sql << table_sql
+        end
+        "ALTER PUBLICATION #{@name} SET #{tables_sql.join(", ")}"
+      end
+    end
+
+    def define_publication(name)
+      execute <<~SQL
+        DO $$
+        BEGIN
+        CASE WHEN NOT EXISTS (SELECT 1 FROM pg_publication WHERE pubname = '#{name}') THEN
+          CREATE PUBLICATION #{name};
+        END CASE;
+        END;
+        $$;
+      SQL
+
+      definition = PublicationDefinition.new(name)
+      yield definition
+
+      execute definition.to_sql
+    end
+  end
+end

data/lib/wal/railtie.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "rails/railtie"
+
+module Waldit
+  class Railtie < Rails::Railtie
+    generators do
+      require_relative "generators/migration"
+    end
+
+    config.before_configuration do
+      require_relative "migration"
+      ActiveRecord::Migration.include Wal::Migration
+      ActiveRecord::Schema.include Wal::Migration
+    end
+  end
+end

data/lib/wal/record_watcher.rb

@@ -65,6 +65,14 @@ module Wal
       @@delete_callbacks[table].push(block: block)
     end
 
+    def self.change_callbacks
+      @@change_callbacks
+    end
+
+    def self.delete_callbacks
+      @@delete_callbacks
+    end
+
     def on_record_changed(event)
       case event
       when InsertEvent

data/lib/wal/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Wal
-  VERSION = "0.0.6"
+  VERSION = "0.0.8"
 end

data/lib/wal.rb

@@ -8,6 +8,7 @@ require_relative "wal/record_watcher"
 require_relative "wal/streaming_watcher"
 require_relative "wal/replicator"
 require_relative "wal/active_record_context_extension"
+require_relative "wal/railtie"
 require_relative "wal/version"
 
 module Wal

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: wal
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.8
 platform: ruby
 authors:
 - Rodrigo Navarro
 bindir: exe
 cert_chain: []
-date: 2025-08-10 00:00:00.000000000 Z
+date: 2025-08-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -93,7 +93,11 @@ files:
 - exe/wal
 - lib/wal.rb
 - lib/wal/active_record_context_extension.rb
+- lib/wal/generators/migration.rb
+- lib/wal/generators/templates/migration.rb.erb
+- lib/wal/migration.rb
 - lib/wal/noop_watcher.rb
+- lib/wal/railtie.rb
 - lib/wal/record_watcher.rb
 - lib/wal/replicator.rb
 - lib/wal/streaming_watcher.rb