active_record_mysql_repl 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d60c9d4baca1fb49f423f128a5015c348e17301c593895afa9083c2ee47b495
-  data.tar.gz: 4060d8523d63a3db240d6cd96cefcc6f1e0a0e3e4c2e35691d736005f86f90e3
+  metadata.gz: 5c3f840dac882b20496edf1781709416bf8e81f4ed801683892b6a54431cbd4c
+  data.tar.gz: 5f947005433e288cc043e0375849d1ff51e335b863e81fbf890cb66fc3056658
 SHA512:
-  metadata.gz: 9fa3e154a372791095e0b6e4766e7a23d3a1c3f4f74fda84a0d4aeac33d2ef202080911a447c4a9ebe6b7a47eb09a9138630715e46b7bb482dbac5b1d2528fe3
-  data.tar.gz: e8599c8c29441696533ace9baa75163a24a271cd8b74a67e7038af2ec8e16c540ef869029438344fe75aa2f455b8012d61459efa26046ebcf86987314ef66c2d
+  metadata.gz: ae1b915b0668004ac07f597da9fae0a9b35b0103699e7e224fcb3b25ce2924700818850de6fb70fc5ecc3fe6d99ead005c1471bcb7f89744f942567b3092c66d
+  data.tar.gz: 37091f71892c3cf3c1fdf62aed7202ab44f72ad5cb87bf1f2e8f99bff52c0f007434e226d565229f03ed312f42e9ce23d6d7b27904e72b27d272ea2c3d29cba3

data/README.md

@@ -383,7 +383,6 @@ https://github.com/nogahighland/active_record_mysql_repl/blob/main/sample_config
 <details><summary>output:</summary>
 
 ```rb
-
 D, [2024-12-12T18:23:41.382606 #2816] DEBUG -- :   Order Load (8.6ms)  SELECT `orders`.* FROM `orders` ORDER BY `orders`.`id` DESC LIMIT 1
 D, [2024-12-12T18:23:41.389954 #2816] DEBUG -- :   User Load (2.4ms)  SELECT `users`.* FROM `users` WHERE `users`.`id` = '1' LIMIT 1
 D, [2024-12-12T18:23:41.392060 #2816] DEBUG -- :   UserProfile Load (1.8ms)  SELECT `user_profiles`.* FROM `user_profiles` WHERE `user_profiles`.`id` = '1' LIMIT 1
@@ -397,22 +396,148 @@ D, [2024-12-12T18:23:41.392060 #2816] DEBUG -- :   UserProfile Load (1.8ms)  SEL
 
 ---
 
+Even though `users.login_id` column exists, because the column is not an external key it should be ignored from the association.
+The column is ignored because the custom association is defined here and `User#login` causes a `NoMethodError`
 
+https://github.com/nogahighland/active_record_mysql_repl/blob/f52f04770425723f632220bb8031bf4b402020a6/sample_config/.army.sample/associations.sample.yml#L10-L13
 
 ```rb
-[10] test(main)> Order.last.item.brand
+[1] test(main)> User.last.login
 ```
 
 <details><summary>output:</summary>
 
 ```rb
-D, [2024-12-11T23:46:28.715219 #96076] DEBUG -- :   Order Load (4.1ms)  SELECT `orders`.* FROM `orders` ORDER BY `orders`.`id` DESC LIMIT 1
-D, [2024-12-11T23:46:28.717319 #96076] DEBUG -- :   Item Load (1.7ms)  SELECT `items`.* FROM `items` WHERE `items`.`id` = '1' LIMIT 1
-D, [2024-12-11T23:46:28.731254 #96076] DEBUG -- :   Brand Load (3.1ms)  SELECT `brands`.* FROM `brands` WHERE `brands`.`id` = '1' LIMIT 1
-#<Brand:0x000000012a411ab0> {
-  :id   => "1",
-  :name => "brand1"
-}
+D, [2024-12-12T22:17:49.291618 #37808] DEBUG -- :   User Load (2.5ms)  SELECT `users`.* FROM `users` ORDER BY `users`.`id` DESC LIMIT 1
+NoMethodError: undefined method `login' for an instance of User
+from /Users/hiroki.kishi/develop/private/active_record_mysql_repl/vendor/bundle/ruby/3.3.0/gems/activemodel-7.2.2/lib/active_model/attribute_methods.rb:512:in `method_missing'
+```
+
+</details>
+
+---
+
+You can use transaction by globally defined `transaction`.
+
+```rb
+[7] test(main)> u2 = User.new(id:2, profile_id: up2, login_id: 'login2')
+[9] test(main)> up2 = UserProfile.new(id:2, name: 'user2');
+[10] test(main)> transaction { [u2, up2].map(&:save) }
+```
+
+<details><summary>output:</summary>
+
+```sql
+[9] test(main)> up2 = UserProfile.new(id:2, name: 'user2');
+[10] test(main)> transaction { [u2, up2].map(&:save) }
+D, [2024-12-12T22:25:32.266790 #37808] DEBUG -- :   TRANSACTION (2.2ms)  BEGIN
+D, [2024-12-12T22:25:32.272857 #37808] DEBUG -- :   User Create (8.3ms)  INSERT INTO `users` (`id`, `login_id`) VALUES ('2', 'login2')
+D, [2024-12-12T22:25:32.275213 #37808] DEBUG -- :   UserProfile Create (2.1ms)  INSERT INTO `user_profiles` (`id`, `name`) VALUES ('2', 'user2')
+D, [2024-12-12T22:25:32.281765 #37808] DEBUG -- :   TRANSACTION (6.4ms)  COMMIT
+[
+  [0] true,
+  [1] true
+]
+```
+
+</details>
+
+---
+
+The following example shows the following steps:
+
+1. Find the users with id 1 and 2
+2. Change the login_id of the users
+3. Show the changes
+4. Save the changes in a transaction
+
+```rb
+[11] test(main)> u1, u2 = User.find([1, 2])
+```
+
+<details><summary>Output:</summary>
+
+```rb
+D, [2024-12-12T22:27:55.979311 #37808] DEBUG -- :   User Load (11.7ms)  SELECT `users`.* FROM `users` WHERE `users`.`id` IN ('1', '2')
+[
+  [0] #<User:0x0000000144b325a0> {
+    :id         => "1",
+    :login_id   => "user1",
+    :profile_id => 1
+  },
+  [1] #<User:0x0000000144b32460> {
+    :id         => "2",
+    :login_id   => "login2",
+    :profile_id => nil
+  }
+]
+```
+
+</details>
+
+```rb
+[12] test(main)> [u1, u2].each { |u| u.login_id = "login_id#{u.id}" }
+```
+
+<details><summary>Output:</summary>
+
+```rb
+[
+  [0] #<User:0x0000000144b325a0> {
+    :id         => "1",
+    :login_id   => "login_id1",
+    :profile_id => 1
+  },
+  [1] #<User:0x0000000144b32460> {
+    :id         => "2",
+    :login_id   => "login_id2",
+    :profile_id => nil
+  }
+]
+```
+
+</details>
+
+```rb
+[13] test(main)> [u1, u2].map(&:changes)
+```
+
+<details><summary>Output:</summary>
+
+```rb
+[
+  [0] {
+    "login_id" => [
+      [0] "user1",
+      [1] "login_id1"
+    ]
+  },
+  [1] {
+    "login_id" => [
+      [0] "login2",
+      [1] "login_id2"
+    ]
+  }
+]
+```
+
+</details>
+
+```rb
+[14] test(main)> transaction { [u1, u2].map(&:save)  }
+```
+
+<details><summary>Output:</summary>
+
+```rb
+D, [2024-12-12T22:29:14.179176 #37808] DEBUG -- :   TRANSACTION (1.2ms)  BEGIN
+D, [2024-12-12T22:29:14.188392 #37808] DEBUG -- :   User Update (9.1ms)  UPDATE `users` SET `users`.`login_id` = 'login_id1' WHERE `users`.`id` = '1'
+D, [2024-12-12T22:29:14.191429 #37808] DEBUG -- :   User Update (2.6ms)  UPDATE `users` SET `users`.`login_id` = 'login_id2' WHERE `users`.`id` = '2'
+D, [2024-12-12T22:29:14.196802 #37808] DEBUG -- :   TRANSACTION (5.2ms)  COMMIT
+[
+  [0] true,
+  [1] true
+]
 ```
 
 </details>
@@ -600,7 +725,7 @@ D, [2024-12-11T23:49:22.406631 #96381] DEBUG -- :    (1.3ms)  select 1 as num
 You can define your own extension script. For example `.upcase_name` is defined on `UserProfile` by the sample extension which is specified in the `.army.sample.yml` file.
 
 - https://github.com/nogahighland/active_record_mysql_repl/blob/9f7c91774b176e1204ed434dad2867721982c660/sample_config/.army.sample.yml#L5
-- https://github.com/nogahighland/active_record_mysql_repl/blob/main/sample_config/.army.sample/extensions/hello.rb
+- https://github.com/nogahighland/active_record_mysql_repl/blob/acaa44684ea4fd135225580431417d6ccd41a988/sample_config/.army.sample/extensions/hello.rb#L1-L9
 
 ```rb
 [3] test(main)> UserProfile.last.upcase_name

data/lib/active_record_mysql_repl/cli/erd.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'rails_erd/diagram/graphviz'
+require "rails_erd/diagram/graphviz"
 
 module ActiveRecordMysqlRepl
   module Erd

data/lib/active_record_mysql_repl/cli/main.rb

@@ -1,10 +1,10 @@
 #!/usr/bin/env ruby
 
-require 'pry'
-require 'colorize'
-require 'active_support'
-require 'active_support/core_ext'
-require 'fileutils'
+require "pry"
+require "colorize"
+require "active_support"
+require "active_support/core_ext"
+require "fileutils"
 
 module ActiveRecordMysqlRepl
   module CLI
@@ -15,34 +15,34 @@ module ActiveRecordMysqlRepl
           return
         end
 
-        if args[0] == '--zsh-completion'
+        if args[0] == "--zsh-completion"
           puts ZshCompletion.generate
           return
         end
 
         opts = CLI::Options.parse(args)
-        army_config = Config.load(opts[:c] || File.join(Dir.home, '.army.yml'))
+        army_config = Config.load(opts[:c] || File.join(Dir.home, ".army.yml"))
 
         db_configs = Database::Configs.load(army_config.database_config)
         db_config_key = opts[:d]
         db_config = db_configs[db_config_key]
         return puts db_configs.keys unless db_config
 
-        generate_erf = opts[:e] == 'erd'
+        generate_erf = opts[:e] == "erd"
 
         SSHTunnel.tunnel(db_config) do |port|
           Database::Connection.connect(db_config, port) do
             Database::Loader.load_tables(db_config_key, army_config, port)
 
             if generate_erf
-              require 'rails_erd/diagram/graphviz'
+              require "rails_erd/diagram/graphviz"
               puts "Generating ERD for #{db_config_key}_erd.pdf".gray
               RailsERD::Diagram::Graphviz.create
-              FileUtils.mv('erd.pdf', "#{db_config_key}_erd.pdf")
-              return
+              FileUtils.mv("erd.pdf", "#{db_config_key}_erd.pdf")
+              next
             end
 
-            require 'active_record_mysql_repl/extensions'
+            require "active_record_mysql_repl/extensions"
             if army_config.extensions_dir.present?
               puts "Loading custom extensions from #{army_config.extensions_dir}".gray
               Extensions.load_external(army_config.extensions_dir)
@@ -57,8 +57,8 @@ module ActiveRecordMysqlRepl
       end
 
       def self.clear_screen
-        system('clear')
-        system('cls')
+        system("clear")
+        system("cls")
       end
     end
   end

data/lib/active_record_mysql_repl/cli/zsh_completion.rb

@@ -1,29 +1,29 @@
 # frozen_string_literal: true
 
 COMPLETION = <<~COMPLETION
-#!/usr/bin/env bash
-
-function _army() {
-  _arguments \
-    '-c[path to .armyrc file]:.armyrc:_files' \
-    '-d[Database name]:database:->database' \
-    '-e[output erd]:output erd:->erd'
-
-  case "$state" in
-    database)
-      slice=("${words[@]:1}")
-      _values 'Database name' $(_database ${slice})
-      ;;
-    erd)
-      _values 'Generate ERD' erd ''
-  esac
-}
-
-function _database {
-  army ${@}
-}
-
-compdef _army army
+  #!/usr/bin/env bash
+  
+  function _army() {
+    _arguments \
+      '-c[path to .armyrc file]:.armyrc:_files' \
+      '-d[Database name]:database:->database' \
+      '-e[output erd]:output erd:->erd'
+  
+    case "$state" in
+      database)
+        slice=("${words[@]:1}")
+        _values 'Database name' $(_database ${slice})
+        ;;
+      erd)
+        _values 'Generate ERD' erd ''
+    esac
+  }
+  
+  function _database {
+    army ${@}
+  }
+  
+  compdef _army army
 COMPLETION
 
 module ActiveRecordMysqlRepl

data/lib/active_record_mysql_repl/cli.rb

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
-require_relative 'cli/options'
-require_relative 'cli/main'
-require_relative 'cli/zsh_completion'
-require_relative 'cli/erd'
+require_relative "cli/options"
+require_relative "cli/main"
+require_relative "cli/zsh_completion"
+require_relative "cli/erd"
 
 module ActiveRecordMysqlRepl
   module CLI; end

data/lib/active_record_mysql_repl/config.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'yaml'
+require "yaml"
 
 module ActiveRecordMysqlRepl
   class Config
@@ -9,7 +9,7 @@ module ActiveRecordMysqlRepl
     end
 
     def database_config
-      raise 'database_config is not defined' unless @database_config
+      raise "database_config is not defined" unless @database_config
       File.join(@army_config_dir, @database_config)
     end
 
@@ -29,10 +29,10 @@ module ActiveRecordMysqlRepl
 
     def initialize(path, config)
       @army_config_dir = File.dirname(File.absolute_path(path))
-      @database_config = config['database_config']
-      @associations = config['associations']
-      @extensions_dir = config['extensions_dir']
-      @pryrc = config['pryrc']
+      @database_config = config["database_config"]
+      @associations = config["associations"]
+      @extensions_dir = config["extensions_dir"]
+      @pryrc = config["pryrc"]
     end
   end
 end

data/lib/active_record_mysql_repl/database/association.rb

@@ -36,18 +36,18 @@ module ActiveRecordMysqlRepl
           columns.each do |column|
             next if association_setting.present? && association_settings.ignore_columns(table_name).include?(column.name)
 
-            associatable = column.name.gsub(/_id$/, '') if column.name.end_with?('_id')
-            next if associatable.blank? || associatable == 'class' # reserved word
+            associatable = column.name.gsub(/_id$/, "") if column.name.end_with?("_id")
+            next if associatable.blank? || associatable == "class" # reserved word
 
-            if analyzed_tables.keys.include?(associatable.pluralize)
+            if analyzed_tables.key?(associatable.pluralize)
               table.belongs_to << associatable.singularize if associatable
               analyzed_tables[associatable.pluralize].has_many << table_name.pluralize
             else
-              associatable_table_name = associatable.split('_').last
-              if analyzed_tables.keys.include?(associatable_table_name.pluralize)
-                table.belongs_to << { name: associatable, class_name: associatable_table_name.classify, foreign_key: :id }
+              associatable_table_name = associatable.split("_").last
+              table.belongs_to << if analyzed_tables.key?(associatable_table_name.pluralize)
+                {name: associatable, class_name: associatable_table_name.classify, foreign_key: :id}
               else
-                table.belongs_to << { name: associatable, class_name: table_name.singularize.classify, foreign_key: :id }
+                {name: associatable, class_name: table_name.singularize.classify, foreign_key: :id}
               end
             end
           end
@@ -71,12 +71,12 @@ module ActiveRecordMysqlRepl
       end
 
       def [](table)
-        table = (@association.keys - ['ignore_columns']) & [table]
+        table = (@association.keys - ["ignore_columns"]) & [table]
         @association[table.first] if table.present?
       end
 
       def ignore_columns(table)
-        @association.fetch('ignore_columns', {}).fetch(table, [])
+        @association.fetch("ignore_columns", {}).fetch(table, [])
       end
     end
   end

data/lib/active_record_mysql_repl/database/configs.rb

@@ -27,7 +27,7 @@ module ActiveRecordMysqlRepl
     private
 
     class Config
-      attr_reader *%i[
+      attr_reader(*%i[
         bastion
         ssh_user
         remote_host
@@ -36,7 +36,7 @@ module ActiveRecordMysqlRepl
         user
         password
         prompt_color
-      ]
+      ])
 
       def initialize(config)
         @bastion = config["bastion"]

data/lib/active_record_mysql_repl/database/connection.rb

@@ -7,28 +7,26 @@ module ActiveRecordMysqlRepl
 
       def self.connect(db_config, port)
         conn = {
-          adapter: 'mysql2',
-          host: '127.0.0.1',
+          adapter: "mysql2",
+          host: "127.0.0.1",
           port: port,
           username: db_config.user,
           password: db_config.password,
-          database: db_config.database,
+          database: db_config.database
         }
 
-        ActiveRecord::Base.logger = Logger.new(STDOUT)
-        conn = ActiveRecord::Base.establish_connection(conn)
+        ActiveRecord::Base.logger = Logger.new($stdout)
+        ActiveRecord::Base.establish_connection(conn)
 
         puts "Ensureing connection to #{db_config.database} on port 127.0.0.1:#{port}".gray
 
         tables = nil
-        (1..MAX_RETRY-1).to_a.each do |time|
-          begin
-            tables = ActiveRecord::Base.connection.tables
-          rescue => e
-            puts "#{e}, Retry #{time}/#{MAX_RETRY}".red
-              sleep time*time
-            next
-          end
+        (1..MAX_RETRY - 1).to_a.each do |time|
+          tables = ActiveRecord::Base.connection.tables
+        rescue => e
+          puts "#{e}, Retry #{time}/#{MAX_RETRY}".red
+          sleep time * time
+          next
         end
 
         if tables.blank?

data/lib/active_record_mysql_repl/database/loader.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require 'json'
-require 'active_record'
-require 'active_support/all'
+require "json"
+require "active_record"
+require "active_support/all"
 
 module ActiveRecordMysqlRepl
   module Database
@@ -13,13 +13,12 @@ module ActiveRecordMysqlRepl
         tables = ActiveRecord::Base.connection.tables
         association_settings = army_config.associations.present? ? Associations.load(army_config.associations) : {}
         association_setting = association_settings[db_config_key]
-        analyzed_tables = begin
-                            if association_setting.present?
-                              Associations.analyze(tables, association_setting)
-                            else
-                              Associations.analyze(tables)
-                            end
-                          end
+        analyzed_tables = if association_setting.present?
+          Associations.analyze(tables, association_setting)
+        else
+          Associations.analyze(tables)
+        end
+
         skipped = []
         analyzed_tables.each do |analyzed_table|
           # defer model definition for tables with `has_many: :xxx, through: xxx` associations

data/lib/active_record_mysql_repl/database.rb

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
-require_relative 'database/configs'
-require_relative 'database/association'
-require_relative 'database/connection'
-require_relative 'database/loader'
+require_relative "database/configs"
+require_relative "database/association"
+require_relative "database/connection"
+require_relative "database/loader"
 
 module ActiveRecordMysqlRepl
   module Database; end

data/lib/active_record_mysql_repl/extensions/active_record.rb

@@ -5,29 +5,28 @@ module ActiverecordMysqlRepl
     module ActiveRecord
       module Base
         def as(name = nil)
-          self.reflect_on_all_associations(name).map(&:name)
+          reflect_on_all_associations(name).map(&:name)
         end
 
         def describe
           [
-            "# #{self.table_name}",
-            exec_sql("DESCRIBE #{self.table_name}").tab(:h),
-            exec_sql("SHOW INDEX FROM #{self.table_name}").tab(:h)
+            "# #{table_name}",
+            exec_sql("DESCRIBE #{table_name}").tab(:h),
+            exec_sql("SHOW INDEX FROM #{table_name}").tab(:h)
           ].join("\n")
         end
 
-        alias desc describe
-        alias d describe
+        alias_method :desc, :describe
+        alias_method :d, :describe
 
         def show_ddl
-          exec_sql("SHOW CREATE TABLE #{self.table_name}")[0]['Create Table']
+          exec_sql("SHOW CREATE TABLE #{table_name}")[0]["Create Table"]
         end
 
-        alias ddl show_ddl
+        alias_method :ddl, :show_ddl
       end
 
       ::ActiveRecord::Base.extend(Base)
     end
   end
 end
-

data/lib/active_record_mysql_repl/extensions/global.rb

@@ -26,4 +26,3 @@ module ActiverecordMysqlRepl
     Kernel.include(Global)
   end
 end
-

data/lib/active_record_mysql_repl/extensions/hash.rb

@@ -4,17 +4,20 @@ module ActiverecordMysqlRepl
   module Extensions
     module Hash
       def method_missing(name, *args)
-        if (self.keys & [name, name.to_s]).present?
+        if (keys & [name, name.to_s]).present?
           self[name] || self[name.to_s]
-        elsif name.to_s.end_with?('=') && self.keys.include?(name.to_s[0..-2])
+        elsif name.to_s.end_with?("=") && keys.include?(name.to_s[0..-2])
           name = name[0..-2]
-          self[name] = args.first if self.key?(name)
-          self[name.to_sym] = args.first if self.key?(name.to_sym)
+          self[name] = args.first if key?(name)
+          self[name.to_sym] = args.first if key?(name.to_sym)
         end
       end
 
+      def respond_to_missing?
+        true
+      end
+
       ::Hash.include(Hash)
     end
   end
 end
-

data/lib/active_record_mysql_repl/extensions/object.rb

@@ -1,28 +1,28 @@
 # frozen_string_literal: true
 
-require 'terminal-table'
-require 'csv'
-require 'clipboard'
-require 'json'
+require "terminal-table"
+require "csv"
+require "clipboard"
+require "json"
 
 module ActiverecordMysqlRepl
   module Extensions
     module Object
       def tabulate(orientation = nil)
-        arr = if self.is_a?(Array)
-                self
-              elsif self.is_a?(::ActiveRecord::Relation)
-                self.to_a
-              else
-                [self]
-              end
+        arr = if is_a?(Array)
+          self
+        elsif is_a?(::ActiveRecord::Relation)
+          to_a
+        else
+          [self]
+        end
 
         arr = arr.map do |e|
           if e.is_a?(::ActiveRecord::Base)
-            values =  e.attributes.transform_values do |v|
+            values = e.attributes.transform_values do |v|
               next JSON.pretty_generate(v) if v.is_a?(Enumerable) && v.size > 0
-              next 'NULL' if v.nil?
-              next '""' if v == ''
+              next "NULL" if v.nil?
+              next '""' if v == ""
               v.to_s
             end
             next values
@@ -34,17 +34,17 @@ module ActiverecordMysqlRepl
         raise "#{arr} contains Hashes with different keys".red unless arr.map(&:keys).uniq.size == 1
 
         orientation = case orientation
-                      when :vertical, :v
-                        :vertical
-                      when :horizontal, :h
-                        :horizontal
-                      else
-                        if arr.first.keys.size > 5
-                          :vertical
-                        else
-                          :horizontal
-                        end
-                      end
+        when :vertical, :v
+          :vertical
+        when :horizontal, :h
+          :horizontal
+        else
+          if arr.first.keys.size > 5
+            :vertical
+          else
+            :horizontal
+          end
+        end
 
         t = Terminal::Table.new
 
@@ -54,7 +54,7 @@ module ActiverecordMysqlRepl
           t.rows = arr.map(&:values)
 
         when :vertical
-          t.headings = ['Name', 'Value']
+          t.headings = ["Name", "Value"]
           arr.each.with_index do |row, i|
             row.each { |col, val| t.add_row [col, val] }
             t.add_separator if i < arr.size - 1
@@ -64,25 +64,25 @@ module ActiverecordMysqlRepl
         t.to_s
       end
 
-      alias tab tabulate
-      alias table tabulate
-      alias to_table tabulate
+      alias_method :tab, :tabulate
+      alias_method :table, :tabulate
+      alias_method :to_table, :tabulate
 
       def csv(orientation = nil)
-        arr = if self.is_a?(Array)
-                self
-              elsif self.is_a?(::ActiveRecord::Relation)
-                self.to_a
-              else
-                [self]
-              end
+        arr = if is_a?(Array)
+          self
+        elsif is_a?(::ActiveRecord::Relation)
+          to_a
+        else
+          [self]
+        end
 
         arr = arr.map do |e|
           if e.is_a?(::ActiveRecord::Base)
-            values =  e.attributes.transform_values do |v|
+            values = e.attributes.transform_values do |v|
               next JSON.pretty_generate(v) if v.is_a?(Enumerable) && v.size > 0
-              next 'NULL' if v.nil?
-              next '""' if v == ''
+              next "NULL" if v.nil?
+              next '""' if v == ""
               v.to_s
             end
             next values
@@ -94,17 +94,17 @@ module ActiverecordMysqlRepl
         raise "#{arr} contains Hashes with different keys".red unless arr.map(&:keys).uniq.size == 1
 
         orientation = case orientation
-                      when :vertical, :v
-                        :vertical
-                      when :horizontal, :h
-                        :horizontal
-                      else
-                        if arr.first.keys.size > 5
-                          :vertical
-                        else
-                          :horizontal
-                        end
-                      end
+        when :vertical, :v
+          :vertical
+        when :horizontal, :h
+          :horizontal
+        else
+          if arr.first.keys.size > 5
+            :vertical
+          else
+            :horizontal
+          end
+        end
 
         CSV.generate do |csv|
           case orientation
@@ -121,11 +121,11 @@ module ActiverecordMysqlRepl
       end
 
       def copy
-        Clipboard.copy(self.to_s)
+        Clipboard.copy(to_s)
       end
-      alias cp copy
+      alias_method :cp, :copy
 
-      alias j to_json
+      alias_method :j, :to_json
 
       def jp
         JSON.pretty_generate(JSON.parse(to_json))
@@ -135,4 +135,3 @@ module ActiverecordMysqlRepl
 
   ::Object.include(Extensions::Object)
 end
-

data/lib/active_record_mysql_repl/extensions/tabler.rb

@@ -13,11 +13,11 @@ module ActiverecordMysqlRepl
         end
 
         model.column_names.each do |column|
-          define_method("#{model.table_name.singularize}_by_#{column}") do
+          define_method(:"#{model.table_name.singularize}_by_#{column}") do
             model.where(column => self)
           end
 
-          define_method("#{model.table_name.singularize}_#{column}_like") do
+          define_method(:"#{model.table_name.singularize}_#{column}_like") do
             model.where("#{column} like ?", "%#{self}%")
           end
         end
@@ -29,4 +29,3 @@ module ActiverecordMysqlRepl
     Enumerable.include Tabler
   end
 end
-

data/lib/active_record_mysql_repl/extensions.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require_relative 'extensions/object'
-require_relative 'extensions/global'
-require_relative 'extensions/active_record'
-require_relative 'extensions/hash'
-require_relative 'extensions/tabler'
+require_relative "extensions/object"
+require_relative "extensions/global"
+require_relative "extensions/active_record"
+require_relative "extensions/hash"
+require_relative "extensions/tabler"
 
 module ActiveRecordMysqlRepl
   module Extensions
@@ -15,4 +15,3 @@ module ActiveRecordMysqlRepl
     end
   end
 end
-

data/lib/active_record_mysql_repl/ssh_tunnel.rb

@@ -1,14 +1,16 @@
 # frozen_string_literal: true
 
-require 'net/ssh'
-require 'net/ssh/gateway'
+require "net/ssh"
+require "net/ssh/gateway"
 
 module ActiveRecordMysqlRepl
   module SSHTunnel
     EPHEMERAL_PORT = 0
 
     def self.tunnel(db_config)
-      return yield(db_config.port) if block_given? unless db_config.bastion
+      unless db_config.bastion
+        return yield(db_config.port) if block_given?
+      end
 
       puts "Establishing ssh tunnel to #{db_config.remote_host}:#{db_config.port} via #{db_config.ssh_user}#{db_config.bastion}".gray
 

data/lib/active_record_mysql_repl/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveRecordMysqlRepl
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/active_record_mysql_repl.rb

@@ -1,11 +1,10 @@
 # frozen_string_literal: true
 
 require_relative "active_record_mysql_repl/version"
-require_relative 'active_record_mysql_repl/config'
-require_relative 'active_record_mysql_repl/cli'
-require_relative 'active_record_mysql_repl/database'
-require_relative 'active_record_mysql_repl/ssh_tunnel'
-require_relative 'active_record_mysql_repl/version'
+require_relative "active_record_mysql_repl/config"
+require_relative "active_record_mysql_repl/cli"
+require_relative "active_record_mysql_repl/database"
+require_relative "active_record_mysql_repl/ssh_tunnel"
 # extensions should be loaded after database connection is established
 # require_relative 'active_record_mysql_repl/extensions'
 

data/sample_config/.army.sample/associations.sample.yml

@@ -8,6 +8,6 @@ test:
         class_name: UserProfile
 
   ignore_columns:
-    user:
+    users:
     - login_id
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_record_mysql_repl
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Hiroki Kishi
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-12-12 00:00:00.000000000 Z
+date: 2024-12-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: byebug
@@ -38,6 +38,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.50.0
+- !ruby/object:Gem::Dependency
+  name: standard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: activerecord
   requirement: !ruby/object:Gem::Requirement