activerecord-dbsnapshot 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6f6d0ba8702b72497080db038c570316e8fa86695435e2ad531db0e9ac21aae4
-  data.tar.gz: 95b154ff4dea18afd990dc301df252b391f6fac571213dfd8a4119f955bc172e
+  metadata.gz: 3e1622e0e7ec44461ef0dbb0e36def428afdf27e8b27a163b5bb494955ec168d
+  data.tar.gz: 1fefe1899a23195363a6c0b1af4600a224e515b71777a9403ed34f8b51f1caca
 SHA512:
-  metadata.gz: e2d8a26f0b16c39de9f17b4ef73a81881ae4c4deb609c36853094673a958a633baa0b7cdeabc18c9529c38b159aa3c235bf128970d8ff6e4f27d3e4b72cb428c
-  data.tar.gz: 97b58f4bfcb9feaaf99a6f5305aa0ddc22785d7b8413b9964008d218cf1e4713660e29255f82bfa5e7199b46b12d51bd642fa7dad9ecf1a37761705d6528c222
+  metadata.gz: 6f2ccb151614bf0fc8d9937b05b6187f25dc0e2f9fab8bcc872e5afd8f0613b9e30f43a34091399986b5fd13c6514fec17de5a9815bf43f02f726cbf8bfb4dc6
+  data.tar.gz: d03777d254b049ca37717b6e33f304f2177886e825ebb54142fa54078f31ca436e355bf8090dc8d564e5d4aedc08075e4ccab35011537b0119057c9d543f8132

data/lib/activerecord/dbsnapshot/version.rb

@@ -2,6 +2,6 @@
 
 module Activerecord
   module Dbsnapshot
-    VERSION = "0.1.1"
+    VERSION = "0.1.2"
   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

@@ -28,17 +28,19 @@ namespace :db do
       db_config = ActiveRecord::Base.connection_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_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.2
 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: