uff_db_loader 1.3.0 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5f807430f4dff47a2a9a98b6f360720a51a7f96c9af2554e0784383adaf0fe63
-  data.tar.gz: dfaf6dad64eea0e267872426f1426534703e52257c8e29a6440c174a94a94f72
+  metadata.gz: 28a567bb65dcdcc545605a5b8b921abe6fb6b4b666d97928e659bf535a1675ed
+  data.tar.gz: 4deffc2f8bad009cd16d04f08280765be75042da0f6e6e97d488addd22a51065
 SHA512:
-  metadata.gz: c1bed481bd67cc2d5dc3000b6d8b67d17ae0e3f7631302b717c436827862538ab679b52edea76777e8a6f2e1d103caca6c5b75796f64dcf457c7bec7e5a6b226
-  data.tar.gz: 05defe18ffd2c6719180c26fcab48c85de42cba697834162c9ca81a15e8394760afe8e5daf60e287bc7611b2d257d3a58cbe98778ea2c119275a0a692673bbb3
+  metadata.gz: 22b2e2a27364e1afccda77edf18218c9c0862ee0ef389126e745848fceb08fb9fb73b042e07a71a9bddacc2818262d36c9ba109cadc5840c1c3b6175190ccfb5
+  data.tar.gz: d8ddddf06be94aa6d2799e7beaeff371c2897cbb7cf6e83855ed112325cc2796732c23082214aa9aa938be023f881e354d2607197dd01a9923777962068c555f

data/README.md

@@ -6,7 +6,7 @@
 
 ## Installation
 
-Add this line to your application's Gemfile:
+Add this line to your application's `Gemfile`:
 
 ```ruby
 gem 'uff_db_loader'
@@ -20,14 +20,16 @@ Or install it yourself as:
 
     $ gem install uff_db_loader
 
+Run the installation script: 
+
+    $ bin/rails uff_db_loader:install
+
 ## Configuration
 
 You can configure the gem by running the following during the initialization of the Rails app:
 ```ruby
-# frozen_string_literal: true
-
 UffDbLoader.configure do |config|
-  config.environments = ['sandbox', 'production'] # default is "['staging', 'production']"
+  config.environments = ['staging', 'production']
   config.ssh_user = 'Francina'
   config.ssh_host = 'host.of.yoursite'
   config.db_name = 'twotter'
@@ -43,10 +45,13 @@ Make sure the app's database user has the superuser role. Otherwise the app will
 
 ## Usage
 
-`uff_db_loader` provides `rails uff_db_loader:dump` and `rails uff_db_loader:load` which will prompt for a configured environment.
-`dump` will only create and download a current database dump, while `load`, will do the same and restore the database content into a new database and gives instructions on how to use it in development.
-
+`uff_db_loader` can be called like `bin/rails uff_db_loader:<task>` where `<task>` is one of the following:
 
+- `dump`: Dumps a remote database from a selected environment and downloads it
+- `restore`: Restores a downloaded dump into a local database
+- `switch`: Selects a restored local database to use
+- `load`: Dumps a remote database from a selected environment and downloads it then restores and selects the database
+- `prune`: Delete all downloaded db dumps and removes all databases created by UffDbLoader
 
 ## Development
 

data/lib/configuration.rb

@@ -8,7 +8,7 @@ module UffDbLoader
     attr_accessor :environments, :ssh_host, :ssh_user, :db_name, :db_system, :app_name, :dumps_directory, :database_config_file
 
     def initialize
-      @environments = %w[staging production]
+      @environments = nil
       @ssh_host = nil
       @ssh_user = nil
       @db_name = nil

data/lib/uff_db_loader/tasks/uff_db_loader.rake

@@ -3,7 +3,21 @@
 require "tty-prompt"
 
 namespace :uff_db_loader do
-  desc "Dumps a remote database to #{UffDbLoader.config.dumps_directory}"
+  desc "Set up UffDbLoader"
+  task install: :environment do
+    UffDbLoader.create_initializer
+
+    puts "👶 Created a Rails initializer file at #{UffDbLoader.initializer_path}."
+
+    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 from a selected environment to #{UffDbLoader.config.dumps_directory}"
   task dump: :environment do
     prompt = TTY::Prompt.new
     environment = prompt.select("Which environment should we get the dump from?", UffDbLoader.config.environments)
@@ -11,8 +25,35 @@ namespace :uff_db_loader do
     UffDbLoader.dump_from(environment)
   end
 
-  desc "Gets a dump from remote and loads it into the local database"
+  desc "Restores a downloaded dump into a local database"
+  task restore: :environment do
+    UffDbLoader.ensure_installation!
+
+    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)
+
+    UffDbLoader.load_dump_into_database(database_name)
+  end
+
+  desc "Selects a restored local database to use"
+  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)
+    UffDbLoader.restart_rails_server
+
+    puts "♻️  Restarted rails server with new database."
+  end
+
+  desc "Dumps a remote database from a selected environment to #{UffDbLoader.config.dumps_directory}, then restores and selects the 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,27 +62,15 @@ namespace :uff_db_loader do
     puts "🤓 Reading from to #{result_file_path}"
 
     database_name = File.basename(result_file_path, ".*")
-    UffDbLoader.create_database(database_name)
-
-    puts "🗂  Created database #{database_name}"
-
-    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
-
-    puts "✅ Succesfully loaded #{result_file_path} into #{database_name}"
-
-    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
+    UffDbLoader.load_dump_into_database(database_name)
   end
 
-  desc "Delete all downloaded db dumps and emove all databases created by UffDbLoader"
+  desc "Delete all downloaded db dumps and removes all databases created by UffDbLoader"
   task prune: :environment do
+    # switch to default db so we can also drop the currently connected database
+    UffDbLoader.remember_database_name("")
+    ActiveRecord::Base.connection.reconnect!
+
     UffDbLoader.databases.each do |database_name|
       puts "Dropping #{database_name}"
       UffDbLoader.drop_database(database_name)

data/lib/uff_db_loader/templates/uff_db_loader_initializer.erb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+# Check out our README: https://github.com/rmehner/uff_db_loader/blob/main/README.md
+
+if defined?(UffDbLoader)
+  UffDbLoader.configure do |config|
+    config.ssh_user = 'SSH_USER'
+    config.ssh_host = 'HOST_OF_YOUR_SITE'
+    config.db_name = 'YOUR_DATABASE_NAME'
+    config.db_system = <%= used_database_system %> # Possible values are :postgresql and :mysql
+    config.environments = <%= environments %>
+
+    # Optional settings:
+    # config.app_name = 'my_app' # Defaults to the Rails app name
+    # config.dumps_directory = '/path/to/dumps' # Defaults to Rails.root.join('dumps')
+    # config.database_config_file = 'path/to/database.yml' # Defaults to Rails.root.join('config', 'database.yml')
+  end
+end

data/lib/uff_db_loader/version.rb

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

data/lib/uff_db_loader.rb

@@ -6,91 +6,183 @@ require "configuration"
 module UffDbLoader
   require "railtie"
 
-  def self.config
-    @configuration ||= Configuration.new
-  end
+  class << self
+    def config
+      @configuration ||= Configuration.new
+    end
 
-  def self.reset
-    @configuration = Configuration.new
-  end
+    def reset
+      @configuration = Configuration.new
+    end
 
-  def self.configure
-    yield(config)
-  end
+    def configure
+      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 restart_rails_server
+      system("bin/rails restart")
+    end
 
-  def self.dump_from(environment)
-    FileUtils.mkdir_p(config.dumps_directory)
+    def dump_from(environment)
+      FileUtils.mkdir_p(config.dumps_directory)
 
-    puts "⬇️  Creating dump ..."
+      puts "⬇️  Creating dump ..."
 
-    target = dump_filename(environment)
+      target = dump_file_path(Time.now.strftime("#{config.app_name}_#{environment}_%Y_%m_%d_%H_%M_%S)"))
 
-    command_successful = system(dump_command(environment, target))
-    raise "Command did not run succesful: #{dump_command(environment, target)}" unless command_successful
+      command_successful = system(dump_command(environment, target))
+      raise "Command did not run succesful: #{dump_command(environment, target)}" unless command_successful
 
-    puts "✅ Succesfully dumped to #{target}"
+      puts "✅ Succesfully dumped to #{target}"
 
-    target
-  end
+      target
+    end
 
-  def self.create_database(database_name)
-    config.database_system.create_database(database_name)
-  end
+    def ensure_valid_environment!(environment)
+      unless config.environments.include?(environment)
+        raise ForbiddenEnvironmentError, "Invalid environment: #{environment}."
+      end
+    end
 
-  def self.dump_command(environment, target)
-    config
-      .database_system
-      .dump_command_template
-      .gsub("%environment%", environment)
-      .gsub("%host%", config.ssh_host)
-      .gsub("%user%", config.ssh_user)
-      .gsub("%database%", config.database)
-      .gsub("%target%", target)
-      .gsub("%app_name%", config.app_name)
-  end
+    def prune_dump_directory
+      FileUtils.rm_rf("#{config.dumps_directory}/.", secure: true)
+    end
 
-  def self.ensure_valid_environment!(environment)
-    unless config.environments.include?(environment)
-      raise ForbiddenEnvironmentError, "Invalid environment: #{environment}."
+    def create_database(database_name)
+      config.database_system.create_database(database_name)
     end
-  end
 
-  def self.restore_command(database_name, result_file_path)
-    config.database_system.restore_command(database_name, result_file_path)
-  end
+    def drop_database(database_name)
+      config.database_system.drop_database(database_name)
+    end
 
-  def self.prune_dump_directory
-    FileUtils.rm_rf("#{config.dumps_directory}/.", secure: true)
-  end
+    def databases
+      config.database_system.list_databases.select do |line|
+        line =~ /#{config.app_name}_(#{config.environments.join("|")})_(\d|_)+/
+      end
+    end
 
-  def self.drop_database(database_name)
-    config.database_system.drop_database(database_name)
-  end
+    def setup_dynamic_database_name_in_config
+      old_database_name = Rails.configuration.database_configuration["development"]["database"]
+
+      return false if old_database_name.nil?
 
-  def self.databases
-    config.database_system.list_databases.select do |line|
-      line =~ /#{config.app_name}_(#{config.environments.join("|")})_(\d|_)+/
+      old_config = File.read(UffDbLoader.config.database_config_file)
+      new_config = old_config.sub(old_database_name, database_name_template(old_database_name))
+      File.write(UffDbLoader.config.database_config_file, new_config)
     end
-  end
 
-  def self.replace_database_name_in_config(new_database_name)
-    old_database_name = Rails.configuration.database_configuration["development"]["database"]
+    def current_database_name
+      File.read(database_name_file).strip.presence
+    rescue IOError, Errno::ENOENT => e
+      puts "Could not read #{database_name_file}. #{e.message} – Falling back to default database. 🥱"
+    end
 
-    return false if old_database_name.nil?
+    def remember_database_name(database_name)
+      File.write(database_name_file, database_name)
+    end
 
-    old_config = File.read(UffDbLoader.config.database_config_file)
-    new_config = old_config.sub(old_database_name, new_database_name)
-    File.write(UffDbLoader.config.database_config_file, new_config)
-  end
+    def 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 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)
 
-  class ForbiddenEnvironmentError < StandardError; end
+      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
 
-  class UnknownDatabaseSystem < StandardError; end
+      puts "✅ Succesfully loaded #{dump_file_path} into #{database_name}"
+
+      remember_database_name(database_name)
+      restart_rails_server
+
+      puts "♻️  Restarted rails server with new database."
+    end
+
+    def initializer_path
+      File.join(__dir__, "uff_db_loader", "templates", "uff_db_loader_initializer.erb")
+    end
+
+    def create_initializer
+      template = ERB.new(File.read(initializer_path))
+
+      File.write(
+        Rails.root.join("config", "initializers", "uff_db_loader.rb"),
+        template.result_with_hash(
+          used_database_system: used_database_system,
+          environments: environments
+        )
+      )
+    end
+
+    private
+
+    def database_name_file
+      Rails.root.join("tmp", "uff_db_loader_database_name")
+    end
+
+    def 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 dump_command(environment, target)
+      config
+        .database_system
+        .dump_command_template
+        .gsub("%environment%", environment)
+        .gsub("%host%", config.ssh_host)
+        .gsub("%user%", config.ssh_user)
+        .gsub("%database%", config.database)
+        .gsub("%target%", target)
+        .gsub("%app_name%", config.app_name)
+    end
+
+    def restore_command(database_name, result_file_path)
+      config.database_system.restore_command(database_name, result_file_path)
+    end
+
+    def database_name_template(old_database_name)
+      "<%= defined?(UffDbLoader) && UffDbLoader.current_database_name || '#{old_database_name}' %>"
+    end
+
+    def dump_file_path(database_name)
+      File.join(
+        config.dumps_directory,
+        "#{database_name}.#{config.database_system.dump_extension}"
+      )
+    end
+
+    def used_database_system
+      case Rails.configuration.database_configuration["development"]["adapter"]
+      when "mysql", "mysql2", "trilogy"
+        ":mysql"
+      when "postgresql"
+        ":postgresql"
+      else
+        puts "🙃 Could not automatically determine your used database system. Please adapt in the initializer."
+        ":unknown"
+      end
+    end
+
+    def environments
+      ActiveRecord::Base.configurations.configurations.to_a.map(&:env_name) - ["test", "development"]
+    end
+
+    class ForbiddenEnvironmentError < StandardError; end
+
+    class UnknownDatabaseSystem < StandardError; end
+
+    class InstallationDidNotRunError < StandardError; end
+  end
 end

data/uff_db_loader.gemspec

@@ -8,6 +8,7 @@ Gem::Specification.new do |spec|
 
   spec.summary = "Allows to dump, download and restore databases from docker servers."
   spec.description = "Allows to dump, download and restore databases from servers with a specific docker setup, that UFF uses commonly."
+  spec.post_install_message = "Please run `bin/rails uff_db_loader:install` to complete the installation."
   spec.homepage = "https://github.com/rmehner/uff_db_loader"
   spec.license = "MIT"
   spec.required_ruby_version = Gem::Requirement.new(">= 2.6.0")

metadata

@@ -1,16 +1,16 @@
 --- !ruby/object:Gem::Specification
 name: uff_db_loader
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 2.0.0
 platform: ruby
 authors:
 - Andreas Hellwig
 - Fynn Heintz
 - Robin Mehner
-autorequire: 
+autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-08-25 00:00:00.000000000 Z
+date: 2022-09-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tty-prompt
@@ -79,8 +79,8 @@ files:
 - lib/uff_db_loader.rb
 - lib/uff_db_loader/mysql.rb
 - lib/uff_db_loader/postgresql.rb
-- lib/uff_db_loader/tasks/remote_database.rake
 - lib/uff_db_loader/tasks/uff_db_loader.rake
+- lib/uff_db_loader/templates/uff_db_loader_initializer.erb
 - lib/uff_db_loader/version.rb
 - uff_db_loader.gemspec
 homepage: https://github.com/rmehner/uff_db_loader
@@ -88,10 +88,11 @@ 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/2.0.0
   homepage_uri: https://github.com/rmehner/uff_db_loader
-  source_code_uri: https://github.com/rmehner/uff_db_loader/tree/1.3.0
-post_install_message: 
+  source_code_uri: https://github.com/rmehner/uff_db_loader/tree/2.0.0
+post_install_message: Please run `bin/rails uff_db_loader:install` to complete the
+  installation.
 rdoc_options: []
 require_paths:
 - lib
@@ -106,8 +107,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.3.8
+signing_key:
 specification_version: 4
 summary: Allows to dump, download and restore databases from docker servers.
 test_files: []

data/lib/uff_db_loader/tasks/remote_database.rake

@@ -1,23 +0,0 @@
-# frozen_string_literal: true
-
-require "tty-prompt"
-
-namespace :remote_database do
-  desc "Dumps a remote database to #{UffDbLoader.config.dumps_directory}"
-  task dump: :environment do
-    puts "🧐 Please note this task is called 'uff_db_loader:dump' now."
-    Rake::Task["uff_db_loader:dump"].invoke
-  end
-
-  desc "Gets a dump from remote and loads it into the local database"
-  task load: :environment do
-    puts "🧐 Please note this task is called 'uff_db_loader:load' now."
-    Rake::Task["uff_db_loader:load"].invoke
-  end
-
-  desc "Delete all downloaded db dumps and emove all databases created by UffDbLoader"
-  task prune: :environment do
-    puts "🧐 Please note this task is called 'uff_db_loader:prune' now."
-    Rake::Task["uff_db_loader:prune"].invoke
-  end
-end