activerecord-dbsnapshot 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6f6d0ba8702b72497080db038c570316e8fa86695435e2ad531db0e9ac21aae4
-  data.tar.gz: 95b154ff4dea18afd990dc301df252b391f6fac571213dfd8a4119f955bc172e
+  metadata.gz: 6176a0c48f0c782707247c24aeb2e4d93f9a497729042c9652aeeb3336ee640e
+  data.tar.gz: f29644ac30106b66e5b32b691c11990a75c85bd9a6cc461cab6e92c83d2dd133
 SHA512:
-  metadata.gz: e2d8a26f0b16c39de9f17b4ef73a81881ae4c4deb609c36853094673a958a633baa0b7cdeabc18c9529c38b159aa3c235bf128970d8ff6e4f27d3e4b72cb428c
-  data.tar.gz: 97b58f4bfcb9feaaf99a6f5305aa0ddc22785d7b8413b9964008d218cf1e4713660e29255f82bfa5e7199b46b12d51bd642fa7dad9ecf1a37761705d6528c222
+  metadata.gz: 341b735fe8abf9069eb08e3950d8869c98ce7a6a0b4159f9cdaebe392e5f537842fb4494dadc3871f460de2681cf5d96247bca9c4f2c3492ee1935686882dfcb
+  data.tar.gz: d164865468a7a414588c082d82e9783b08b62c4f6e125cc8fd000112e444339498c0344e34c64f98a63001fbb63af9204725ce48f541907b82f4096eef577b6a

data/lib/activerecord/dbsnapshot/version.rb

@@ -2,6 +2,6 @@
 
 module Activerecord
   module Dbsnapshot
-    VERSION = "0.1.1"
+    VERSION = "0.1.3"
   end
 end

data/lib/activerecord/dbsnapshot.rb

@@ -26,8 +26,7 @@ module ActiveRecord
         require latest_config_file
 
         klass_name = File.basename(latest_config_file, '.rb').gsub(/^[^_]+_/, '').camelize
-        klass = klass_name.constantize
-        klass
+        klass_name.constantize
       end
 
       # Finds snapshot configuration by name
@@ -42,6 +41,19 @@ module ActiveRecord
         ActiveRecord::DbSnapshot.new
       end
 
+      def build_restore_command(db_config, file)
+        command = "pg_restore"
+        command << " --host=#{db_config[:host]}" if db_config[:host]
+        command << " --port=#{db_config[:port]}" if db_config[:port]
+        command << " --username=#{db_config[:username]}" if db_config[:username]
+        command << " --dbname=#{db_config[:database]}" if db_config[:database]
+        command << " --no-owner" # Optional: Prevents restoring ownership information
+        command << " --disable-triggers" # Optional: Disable triggers during data restore
+        command << " --clean" # Optional: Clean (drop) existing database objects before restore
+        command << " #{file}"    # SQL file to restore
+        command
+      end
+
       # Builds the dump command based on configuration
       def build_dump_command(config, db_config, output_file)
         pg_dump_cmd = "pg_dump"
@@ -59,8 +71,8 @@ module ActiveRecord
       end
 
       # Executes the dump command
-      def execute_dump_command(dump_command)
-        system(dump_command)
+      def execute_command(command)
+        system(command)
       end
 
       private

data/lib/tasks/snapshot_tasks.rake

@@ -25,20 +25,22 @@ namespace :db do
         next
       end
 
-      db_config = ActiveRecord::Base.connection_config
+      db_config = ActiveRecord::Base.connection_db_config
       output_file = Rails.root.join('db', 'snapshots', "#{config.snapshot_name}.sql")
 
-      puts output_file
-
-
-      dump_command = ActiveRecord::DbSnapshot.build_dump_command(config, db_config, output_file)
-      ActiveRecord::DbSnapshot.execute_dump_command(dump_command)
+      command = ActiveRecord::DbSnapshot.build_dump_command(config, db_config, output_file)
+      ActiveRecord::DbSnapshot.execute_command(command)
 
       puts "Database dump completed: #{output_file}"
     end
 
     desc "Restore a snapshot SQL file into the database"
     task :restore, [:name] => :environment do |_, args|
+      unless `which pg_restore`.present?
+        puts "Error: pg_restore not found. Ensure PostgreSQL client tools are installed."
+        next
+      end
+
       name = args[:name]
       filename = "db/snapshots/#{name}.sql"
 
@@ -58,8 +60,9 @@ namespace :db do
       end
 
       # Restore the SQL file into the database
-      sql = File.read(filename)
-      ActiveRecord::Base.connection.execute(sql)
+      db_config = ActiveRecord::Base.connection_db_config
+      command = ActiveRecord::DbSnapshot.build_restore_command(db_config, filename)
+      ActiveRecord::DbSnapshot.execute_command(command)
 
       puts "Snapshot '#{name}' restored successfully."
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-dbsnapshot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Ryan Mammina
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-06 00:00:00.000000000 Z
+date: 2024-03-07 00:00:00.000000000 Z
 dependencies: []
 description:
 email:
@@ -56,7 +56,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
+rubygems_version: 3.4.18
 signing_key:
 specification_version: 4
 summary: Dump and Restore database snapshots.