lazy_migrate 0.1.0 → 0.2.0

data/lib/lazy_migrate/migrator.rb

@@ -1,186 +0,0 @@
-# frozen_string_literal: true
-
-require 'tty-prompt'
-require 'active_record'
-require 'rails'
-
-module LazyMigrate
-  class Migrator
-    class << self
-      MIGRATE = 'migrate'
-      ROLLBACK = 'rollback'
-      UP = 'up'
-      DOWN = 'down'
-      REDO = 'redo'
-      BRING_TO_TOP = 'bring to top'
-
-      def run
-        loop do
-          catch(:done) do
-            on_done = -> { throw :done }
-
-            load_migration_paths
-            migrations = find_migrations
-
-            prompt = TTY::Prompt.new(active_color: :bright_green)
-            prompt.ok("\nDatabase: #{ActiveRecord::Base.connection_config[:database]}\n")
-
-            select_migration_prompt(prompt: prompt, migrations: migrations, on_done: on_done)
-          end
-        end
-      rescue TTY::Reader::InputInterrupt
-        puts
-      end
-
-      private
-
-      def select_migration_prompt(prompt:, migrations:, on_done:)
-        prompt.select('Pick a migration') do |menu|
-          migrations.each { |migration|
-            name = render_migration_option(migration)
-
-            menu.choice(
-              name,
-              -> {
-                select_action_prompt(
-                  on_done: on_done,
-                  prompt: prompt,
-                  context: context,
-                  migration: migration,
-                )
-              }
-            )
-          }
-        end
-      end
-
-      def select_action_prompt(on_done:, context:, prompt:, migration:)
-        if !migration[:has_file]
-          prompt.error("\nMigration file not found for migration #{migration[:version]}")
-          prompt_any_key(prompt)
-          on_done.()
-        end
-
-        option_map = obtain_option_map(context: context)
-
-        prompt.select("\nWhat would you like to do for #{migration[:version]} #{name}?") do |inner_menu|
-          options_for_migration(status: migration[:status]).each do |option|
-            inner_menu.choice(option, -> {
-              with_unsafe_error_capture do
-                option_map[option].(migration)
-              end
-              prompt_any_key(prompt)
-              on_done.()
-            })
-          end
-
-          inner_menu.choice('cancel', on_done)
-        end
-      end
-
-      def options_for_migration(status:)
-        common_options = [MIGRATE, ROLLBACK, BRING_TO_TOP]
-        specific_options = if status == 'up'
-          [DOWN, REDO]
-        else
-          [UP]
-        end
-        specific_options + common_options
-      end
-
-      def obtain_option_map(context:)
-        {
-          UP => ->(migration) { context.run(:up, migration[:version]) },
-          DOWN => ->(migration) { context.run(:down, migration[:version]) },
-          REDO => ->(migration) {
-            context.run(:down, migration[:version])
-            context.run(:up, migration[:version])
-          },
-          MIGRATE => ->(migration) { context.up(migration[:version]) },
-          ROLLBACK => ->(migration) {
-            # for some reason in https://github.com/rails/rails/blob/5-2-stable/activerecord/lib/active_record/migration.rb#L1221
-            # we stop before the selected version. I have no idea why.
-            # I could override the logic here but it wouldn't
-            # work when trying to rollback the final migration.
-            context.down(migration[:version])
-          },
-          BRING_TO_TOP => ->(migration) { bring_to_top(migration) },
-        }
-      end
-
-      def load_migration_paths
-        ActiveRecord::Migrator.migrations_paths.each do |path|
-          Dir[Rails.application.root.join("#{path}/**/*.rb")].each { |file| load file }
-        end
-      end
-
-      def find_migrations
-        context.migrations_status
-          .reverse
-          .map { |status, version, name|
-            # This depends on how rails reports a file is missing.
-            # This is no doubt subject to change so be wary.
-            has_file = name != '********** NO FILE **********'
-
-            { status: status, version: version.to_i, name: name, has_file: has_file }
-          }
-      end
-
-      def render_migration_option(migration)
-        "#{migration[:status].ljust(6)}#{migration[:version].to_s.ljust(16)}#{migration[:name].ljust(50)}"
-      end
-
-      # bring_to_top updates the version of a migration to bring it to the top of the
-      # migration list. If the migration had already been up'd it will mark the
-      # new migration file as upped as well. The former version number will be
-      # removed from the schema_migrations table.
-      def bring_to_top(migration)
-        filename = context.migrations.find { |m| m.version == migration[:version] }&.filename
-
-        if filename.nil?
-          raise("No file found for migration #{migration[:version]}")
-        end
-
-        last = last_version
-        new_version = ActiveRecord::Migration.next_migration_number(last ? last + 1 : 0)
-
-        # replace the version
-        basename = File.basename(filename)
-        dir = File.dirname(filename)
-        new_basename = "#{new_version}_#{basename.split('_')[1..].join('_')}"
-        new_filename = File.join(dir, new_basename)
-
-        File.rename(filename, new_filename)
-
-        if migration[:status] == 'up'
-          ActiveRecord::SchemaMigration.create(version: new_version)
-        end
-
-        ActiveRecord::SchemaMigration.find_by(version: migration[:version])&.destroy!
-      end
-
-      def last_version
-        context.migrations.last&.version
-      end
-
-      def prompt_any_key(prompt)
-        prompt.keypress("\nPress any key to continue")
-      end
-
-      def with_unsafe_error_capture
-        yield
-      rescue Exception => e # rubocop:disable Lint/RescueException
-        # I am aware you should not rescue 'Exception' exceptions but I think this is is an 'exceptional' use case
-        puts "\n#{e.class}: #{e.message}\n#{e.backtrace.take(5).join("\n")}"
-      end
-
-      def context
-        ActiveRecord::MigrationContext.new(Rails.root.join('db', 'migrate'))
-      end
-    end
-  end
-
-  def self.run
-    Migrator.show
-  end
-end