uff_db_loader 1.3.0 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5f807430f4dff47a2a9a98b6f360720a51a7f96c9af2554e0784383adaf0fe63
-  data.tar.gz: dfaf6dad64eea0e267872426f1426534703e52257c8e29a6440c174a94a94f72
+  metadata.gz: b64fcfeeeda9c2895262cabcf104828511ae1a17abe6c906c040f5b132c48bc3
+  data.tar.gz: 166953900412522babe8958ddbadef253b8e3558c487d67c53962679252dbd29
 SHA512:
-  metadata.gz: c1bed481bd67cc2d5dc3000b6d8b67d17ae0e3f7631302b717c436827862538ab679b52edea76777e8a6f2e1d103caca6c5b75796f64dcf457c7bec7e5a6b226
-  data.tar.gz: 05defe18ffd2c6719180c26fcab48c85de42cba697834162c9ca81a15e8394760afe8e5daf60e287bc7611b2d257d3a58cbe98778ea2c119275a0a692673bbb3
+  metadata.gz: bf80b9b22d70bf31fe2b8f2d4c9d2f84e9b23a5448e2a94fcdbfe43fdd5a4155a06da935846f3df41cf34528b1e71a17dcc697bce68a5263e09becad424ae31c
+  data.tar.gz: 74b533e7f45a5b5292d74c7660709f14b60f2a3c8dfa61f390314c4145f7cdcc00f079cd3c253f24f44599518b16eb00ea6edcb37640139f2e1ae9b214b64c19

data/lib/uff_db_loader/tasks/uff_db_loader.rake

@@ -3,6 +3,18 @@
 require "tty-prompt"
 
 namespace :uff_db_loader do
+  desc "Install uff_db_loader"
+  task install: :environment do
+    UffDbLoader.remember_database_name("") # ensure database file exists
+
+    if UffDbLoader.setup_dynamic_database_name_in_config
+      puts "🤖 Updated #{UffDbLoader.config.database_config_file}. Happy hacking, beep boop!"
+    else
+      puts "💩 Because YAML is a wonderful format, you need to adapt your config file by hand."
+      puts "🆗 Go to #{UffDbLoader.config.database_config_file} and change the development database value to: #{UffDbLoader.database_name_template("default_database_name")}"
+    end
+  end
+
   desc "Dumps a remote database to #{UffDbLoader.config.dumps_directory}"
   task dump: :environment do
     prompt = TTY::Prompt.new
@@ -13,6 +25,8 @@ namespace :uff_db_loader do
 
   desc "Gets a dump from remote and loads it into the local database"
   task load: :environment do
+    UffDbLoader.ensure_installation!
+
     prompt = TTY::Prompt.new
     environment = prompt.select("Which environment should we get the dump from?", UffDbLoader.config.environments)
     UffDbLoader.ensure_valid_environment!(environment)
@@ -21,26 +35,35 @@ namespace :uff_db_loader do
     puts "🤓 Reading from to #{result_file_path}"
 
     database_name = File.basename(result_file_path, ".*")
-    UffDbLoader.create_database(database_name)
+    UffDbLoader.load_dump_into_database(database_name)
+  end
 
-    puts "🗂  Created database #{database_name}"
+  desc "Loads an existing dump into the local database"
+  task restore: :environment do
+    UffDbLoader.ensure_installation!
 
-    command_successful = system(UffDbLoader.restore_command(database_name, result_file_path))
-    raise "Command did not run succesful: #{UffDbLoader.restore_command(database_name, result_file_path)}" unless command_successful
+    prompt = TTY::Prompt.new
+    existing_dumps = Dir.glob("#{UffDbLoader.config.dumps_directory}/#{UffDbLoader.config.app_name}*").map { |f| File.basename(f, ".*") }
+    database_name = prompt.select("Which dump should be restored?", existing_dumps)
 
-    puts "✅ Succesfully loaded #{result_file_path} into #{database_name}"
+    UffDbLoader.load_dump_into_database(database_name)
+  end
 
-    if UffDbLoader.replace_database_name_in_config(database_name)
-      system("bin/rails restart")
-      puts "🤖 Updated #{UffDbLoader.config.database_config_file} and restarted the rails server. Happy hacking, beep boop!"
-    else
-      puts "💩 Because YAML is a wonderful format, you need to adapt your config file by hand."
-      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 "Switches to an existing database"
+  task switch: :environment do
+    UffDbLoader.ensure_installation!
+
+    prompt = TTY::Prompt.new
+    databases = UffDbLoader.databases
+    new_database = prompt.select("Which database do you want to switch to?", databases)
+
+    UffDbLoader.remember_database_name(new_database)
+    system("bin/rails restart")
+
+    puts "♻️  Restarted rails server with new database."
   end
 
-  desc "Delete all downloaded db dumps and emove all databases created by UffDbLoader"
+  desc "Delete all downloaded db dumps and remove all databases created by UffDbLoader"
   task prune: :environment do
     UffDbLoader.databases.each do |database_name|
       puts "Dropping #{database_name}"

data/lib/uff_db_loader/version.rb

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

data/lib/uff_db_loader.rb

@@ -10,6 +10,10 @@ module UffDbLoader
     @configuration ||= Configuration.new
   end
 
+  def self.database_name_file
+    Rails.root.join("tmp", "uff_db_loader_database_name")
+  end
+
   def self.reset
     @configuration = Configuration.new
   end
@@ -80,17 +84,65 @@ module UffDbLoader
     end
   end
 
-  def self.replace_database_name_in_config(new_database_name)
+  def self.database_name_template(old_database_name)
+    "<%= UffDbLoader.current_database_name || '#{old_database_name}' %>"
+  end
+
+  def self.setup_dynamic_database_name_in_config
     old_database_name = Rails.configuration.database_configuration["development"]["database"]
 
     return false if old_database_name.nil?
 
     old_config = File.read(UffDbLoader.config.database_config_file)
-    new_config = old_config.sub(old_database_name, new_database_name)
+    new_config = old_config.sub(old_database_name, database_name_template(old_database_name))
     File.write(UffDbLoader.config.database_config_file, new_config)
   end
 
+  def self.current_database_name
+    File.read(database_name_file).strip.presence
+  rescue IOError, Errno::ENOENT => e
+    puts "Could not read #{database_name_file}. #{e.message}"
+  end
+
+  def self.remember_database_name(database_name)
+    File.write(database_name_file, database_name)
+  end
+
+  def self.ensure_installation!
+    unless File.read(UffDbLoader.config.database_config_file).include?("UffDbLoader.current_database_name")
+      raise InstallationDidNotRunError, "Please run bin/rails uff_db_loader:install"
+    end
+  end
+
+  def self.dump_file_path(database_name)
+    File.join(
+      config.dumps_directory,
+      "#{database_name}.#{config.database_system.dump_extension}"
+    )
+  end
+
+  def self.load_dump_into_database(database_name)
+    UffDbLoader.drop_database(database_name)
+    UffDbLoader.create_database(database_name)
+
+    puts "🗂  Created database #{database_name}"
+
+    dump_file_path = dump_file_path(database_name)
+
+    command_successful = system(restore_command(database_name, dump_file_path))
+    raise "Command did not run succesful: #{restore_command(database_name, dump_file_path)}" unless command_successful
+
+    puts "✅ Succesfully loaded #{dump_file_path} into #{database_name}"
+
+    remember_database_name(database_name)
+    system("bin/rails restart")
+
+    puts "♻️  Restarted rails server with new database."
+  end
+
   class ForbiddenEnvironmentError < StandardError; end
 
   class UnknownDatabaseSystem < StandardError; end
+
+  class InstallationDidNotRunError < StandardError; end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: uff_db_loader
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.4.0
 platform: ruby
 authors:
 - Andreas Hellwig
@@ -88,9 +88,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.3.0
+  changelog_uri: https://github.com/rmehner/uff_db_loader/releases/tag/1.4.0
   homepage_uri: https://github.com/rmehner/uff_db_loader
-  source_code_uri: https://github.com/rmehner/uff_db_loader/tree/1.3.0
+  source_code_uri: https://github.com/rmehner/uff_db_loader/tree/1.4.0
 post_install_message: 
 rdoc_options: []
 require_paths: