activerecord-dbsnapshot 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 54a0b9a73c7af8b2c2f02e7b6a15b476ce11544309d3e293f4078fdbf468bdc6
-  data.tar.gz: 4854a6d9eff890a5f9bd01f0d67422eba4620fc027bb8bd9bb9a0373dddc9115
+  metadata.gz: 3e1622e0e7ec44461ef0dbb0e36def428afdf27e8b27a163b5bb494955ec168d
+  data.tar.gz: 1fefe1899a23195363a6c0b1af4600a224e515b71777a9403ed34f8b51f1caca
 SHA512:
-  metadata.gz: 74857122ff508b761901ea8020826c1bd5831f8e4247a61d4d4bdc091d52ab45838116dc6b4666043a0869f3c0a7fca027f0823943d802b6542c6a1749719a90
-  data.tar.gz: 79d48c80d7880338520575b9dda121b9a21c1a6974c689fb3c9f73cacfcb739d43e20711533897582e39f4e7463713351cac3bef982c362256e1f284ac7919a5
+  metadata.gz: 6f2ccb151614bf0fc8d9937b05b6187f25dc0e2f9fab8bcc872e5afd8f0613b9e30f43a34091399986b5fd13c6514fec17de5a9815bf43f02f726cbf8bfb4dc6
+  data.tar.gz: d03777d254b049ca37717b6e33f304f2177886e825ebb54142fa54078f31ca436e355bf8090dc8d564e5d4aedc08075e4ccab35011537b0119057c9d543f8132

data/lib/activerecord/dbsnapshot/version.rb

@@ -2,6 +2,6 @@
 
 module Activerecord
   module Dbsnapshot
-    VERSION = "0.1.0"
+    VERSION = "0.1.2"
   end
 end

data/lib/activerecord/dbsnapshot.rb

@@ -26,10 +26,7 @@ module ActiveRecord
         require latest_config_file
 
         klass_name = File.basename(latest_config_file, '.rb').gsub(/^[^_]+_/, '').camelize
-        klass = klass_name.constantize
-
-        puts klass.snapshot_name, klass.excluded_tables.inspect
-        klass
+        klass_name.constantize
       end
 
       # Finds snapshot configuration by name
@@ -44,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"
@@ -61,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.0
+  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: