From 39ff82deb7a82f1f9a1d55a06e2a7a2d20a92ebd Mon Sep 17 00:00:00 2001
From: Andrei Kaleshka <ka8725@gmail.com>
Date: Sat, 7 Sep 2024 00:57:34 +0200
Subject: [PATCH] fix cop

---
 lib/actual_db_schema/commands/rollback.rb | 40 +++++++++++++----------
 lib/actual_db_schema/failed_migration.rb  |  2 ++
 2 files changed, 24 insertions(+), 18 deletions(-)

diff --git a/lib/actual_db_schema/commands/rollback.rb b/lib/actual_db_schema/commands/rollback.rb
index 999cc58..08a8f36 100644
--- a/lib/actual_db_schema/commands/rollback.rb
+++ b/lib/actual_db_schema/commands/rollback.rb
@@ -4,6 +4,12 @@ module ActualDbSchema
   module Commands
     # Rolls back all phantom migrations
     class Rollback < Base
+      UNICODE_COLORS = {
+        red: 31,
+        green: 32,
+        yellow: 33
+      }.freeze
+
       def initialize(context, manual_mode: false)
         @manual_mode = manual_mode || manual_mode_default?
         super(context)
@@ -16,14 +22,11 @@ def call_impl
 
         return if ActualDbSchema.failed.empty?
 
-        puts ""
-        puts "\u2757\u2757\u2757 #{colorize('[ActualDbSchema]', :red)}"
-        puts ""
-        puts colorize("#{ActualDbSchema.failed.count} phantom migration(s) could not be rolled back automatically. Roll them back or fix manually:", :red)
+        puts_preamble
+        puts_into
         puts ""
         puts failed_migrations_list
-        puts "\u2757\u2757\u2757 #{colorize('[ActualDbSchema]', :red)}"
-        puts ""
+        puts_preamble
       end
 
       def failed_migrations_list
@@ -39,23 +42,24 @@ def failed_migrations_list
         end
       end
 
+      def puts_preamble
+        puts ""
+        puts %(\u2757\u2757\u2757 #{colorize("[ActualDbSchema]", :red)})
+        puts ""
+      end
+
+      def puts_into
+        msg = "#{ActualDbSchema.failed.count} phantom migration(s) could not be rolled back automatically."
+        msg += " Roll them back or fix manually:"
+        puts colorize(msg, :red)
+      end
+
       def manual_mode_default?
         ActualDbSchema.config[:auto_rollback_disabled]
       end
 
       def colorize(text, color)
-        code =
-          case color
-          when :red
-            31
-          when :green
-            32
-          when :yellow
-            33
-          else
-            37
-          end
-
+        code = UNICODE_COLORS.fetch(color, 37)
         "\e[#{code}m#{text}\e[0m"
       end
     end
diff --git a/lib/actual_db_schema/failed_migration.rb b/lib/actual_db_schema/failed_migration.rb
index e46f5b6..94390cd 100644
--- a/lib/actual_db_schema/failed_migration.rb
+++ b/lib/actual_db_schema/failed_migration.rb
@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module ActualDbSchema
   FailedMigration = Struct.new(:migration, :exception, keyword_init: true) do
     def filename