uff_db_loader 1.0.1 → 1.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b3810d2c3cfd3142f0449b86a72121062cb0cbff2a6ec15f7aef872cc8cb912b
-  data.tar.gz: 5e370eef8b819a1501ade5f3e5da94e961c465782f858324fad3a5f17c8bf065
+  metadata.gz: a5f7c50d83a5de12e47520cccc509935f3944ae5d4aec252e27a085483487833
+  data.tar.gz: 961477955168e96b46559142a99d5ff139c4a865920ad32d006beec1a9f0482a
 SHA512:
-  metadata.gz: 702acb4e846ef4630064b1b08c127fd40cee5b48700d99e3553a829f7ab80a8a3a04c99825c0148fb1f1d75a814435ab390aff5cd785374d18f93a445dc0f25e
-  data.tar.gz: abe992067bc2a0f329047c47c96246cdfd8f594451bf6b57a5f76d7b17ad8ca02a896c861d76dd07dfd445e0c67dfdd849f585366158e849915718cc628058b6
+  metadata.gz: ecc86542755c63a15f5b617af89816632bda117dd24fc2e8eb145994ef6b5eef33a8cae83e907ab5279f0ac49135e7187555112df5d57f267b8b11380d4ee93f
+  data.tar.gz: dafc87709eae316b58a94b92894dff04357fcd750b3056114873b8fd6ed5ef42cae3655db24f44b3bf978a39933292b276f9a632c00c9eedb46c7fbccadfc927

data/lib/uff_db_loader/mysql.rb

@@ -13,5 +13,17 @@ module UffDbLoader
     def self.restore_command(database_name, result_file_path)
       "mysql -uroot #{database_name} < #{result_file_path}"
     end
+
+    def self.list_databases
+      ActiveRecord::Base.connection.execute("SHOW DATABASES;").to_a.flatten
+    end
+
+    def self.create_database(database_name)
+      ActiveRecord::Base.connection.execute("CREATE DATABASE #{database_name};")
+    end
+
+    def self.drop_database(database_name)
+      ActiveRecord::Base.connection.execute("DROP DATABASE IF EXISTS #{database_name};")
+    end
   end
 end

data/lib/uff_db_loader/postgresql.rb

@@ -13,5 +13,21 @@ module UffDbLoader
     def self.restore_command(database_name, result_file_path)
       "pg_restore --username postgres --clean --if-exists --no-owner --no-acl --dbname #{database_name} #{result_file_path}"
     end
+
+    def self.list_databases
+      ActiveRecord::Base
+        .connection
+        .execute("SELECT datname FROM pg_database;")
+        .values
+        .flatten
+    end
+
+    def self.create_database(database_name)
+      ActiveRecord::Base.connection.execute("CREATE DATABASE #{database_name};")
+    end
+
+    def self.drop_database(database_name)
+      ActiveRecord::Base.connection.execute("DROP DATABASE IF EXISTS #{database_name};")
+    end
   end
 end

data/lib/uff_db_loader/tasks/remote_database.rake

@@ -21,7 +21,7 @@ namespace :remote_database do
     puts "🤓 Reading from to #{result_file_path}"
 
     database_name = File.basename(result_file_path, ".*")
-    ActiveRecord::Base.connection.execute("CREATE DATABASE #{database_name};")
+    UffDbLoader.create_database(database_name)
 
     puts "🗂  Created database #{database_name}"
 
@@ -34,4 +34,15 @@ namespace :remote_database do
     puts "🆗 Go to #{UffDbLoader.config.database_config_file} and change the development database value to: #{database_name}"
     puts "🧑🏾‍🏫 Don't forgot to restart the Rails server after changing the database config (`rails restart`)"
   end
+
+  desc "Delete all downloaded db dumps and emove all databases created by UffDbLoader"
+  task prune: :environment do
+    UffDbLoader.databases.each do |database_name|
+      puts "Dropping #{database_name}"
+      UffDbLoader.drop_database(database_name)
+    end
+
+    puts "Removing dumps from #{UffDbLoader.config.dumps_directory}"
+    UffDbLoader.prune_dump_directory
+  end
 end

data/lib/uff_db_loader/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module UffDbLoader
-  VERSION = "1.0.1"
+  VERSION = "1.1.1"
 end

data/lib/uff_db_loader.rb

@@ -18,12 +18,20 @@ module UffDbLoader
     yield(config)
   end
 
+  def self.dump_filename(environment)
+    File.join(
+      config.dumps_directory,
+      Time.now.strftime("#{config.app_name}_#{environment}_%Y_%m_%d_%H_%M_%S.#{config.database_system.dump_extension}")
+    )
+  end
+
   def self.dump_from(environment)
     FileUtils.mkdir_p(config.dumps_directory)
 
     puts "⬇️  Creating dump ..."
 
-    target = File.join(config.dumps_directory, Time.now.strftime("#{config.app_name}_#{environment}_%Y_%m_%d_%H_%M_%S.#{config.database_system.dump_extension}"))
+    target = dump_filename(environment)
+
     command_successful = system(dump_command(environment, target))
     raise "Command did not run succesful: #{dump_command(environment, target)}" unless command_successful
 
@@ -32,6 +40,10 @@ module UffDbLoader
     target
   end
 
+  def self.create_database(database_name)
+    config.database_system.create_database(database_name)
+  end
+
   def self.dump_command(environment, target)
     config
       .database_system
@@ -54,6 +66,20 @@ module UffDbLoader
     config.database_system.restore_command(database_name, result_file_path)
   end
 
+  def self.prune_dump_directory
+    FileUtils.rm_rf("#{config.dumps_directory}/.", secure: true)
+  end
+
+  def self.drop_database(database_name)
+    config.database_system.drop_database(database_name)
+  end
+
+  def self.databases
+    config.database_system.list_databases.select do |line|
+      line =~ /#{config.app_name}_(#{config.environments.join("|")})_(\d|_)+/
+    end
+  end
+
   class ForbiddenEnvironmentError < StandardError; end
 
   class UnknownDatabaseSystem < StandardError; end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: uff_db_loader
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.1
 platform: ruby
 authors:
 - Andreas Hellwig
@@ -10,7 +10,7 @@ authors:
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-07-21 00:00:00.000000000 Z
+date: 2022-08-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tty-prompt
@@ -87,9 +87,9 @@ licenses:
 - MIT
 metadata:
   bug_tracker_uri: https://github.com/rmehner/uff_db_loader/issues
-  changelog_uri: https://github.com/rmehner/uff_db_loader/releases/tag/1.0.1
+  changelog_uri: https://github.com/rmehner/uff_db_loader/releases/tag/1.1.1
   homepage_uri: https://github.com/rmehner/uff_db_loader
-  source_code_uri: https://github.com/rmehner/uff_db_loader/tree/1.0.1
+  source_code_uri: https://github.com/rmehner/uff_db_loader/tree/1.1.1
 post_install_message:
 rdoc_options: []
 require_paths: