rubocop-migration 0.3.2 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d5fe85da04542f564b53e8c0d779c4258ca907f99a7583919aa02007d719f33
-  data.tar.gz: 06bd03d9e0afdf5618ad2bdfffbfa2ac465821f65009e2c949051bd04e3defae
+  metadata.gz: 819776e7d688a0d112d1ad7ecc2994125de7bfa7e4b8fe27105f21d2a8061565
+  data.tar.gz: fa2c6bfce21e4dd4ad206e6fb049162ee25b016bc2be01c22782d3dc0871e255
 SHA512:
-  metadata.gz: 5943d60b5a214eecc77d326cf94b7f44fb270592822687f05b7165ca4b6165086ae6491f523a27dab620fb3e83ee6c7e03058d14b93cbecbc9dcf1334c47f333
-  data.tar.gz: 7b20c8d2766ff2803ae32a31f598e447fa5fd4282c9507fded7181d5c85d139fe8de603f6000e4dbe49c73b963a07f6cef12b8ec248eba925a621024086c2b86
+  metadata.gz: b9fecf517eab16cd9f6c16e302d7a392c0b263a75c164b53a99337b017d4416894fde4eac40874ca53e4fa61c77783f4ef5cbde64007b5388dc6ceefd79d644d
+  data.tar.gz: da06bac2b33c7018964708c0aec62808262e14478b09ff2af0303f6ce00034a57d2331a64ed9fd973ec34012525ec7051b362c71843cbf23f27e17d6f30d5248

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rubocop-migration (0.3.2)
+    rubocop-migration (0.4.0)
       activesupport
       rubocop (>= 1.34)
       rubocop-rails
@@ -57,7 +57,7 @@ GEM
     rubocop-performance (1.15.0)
       rubocop (>= 1.7.0, < 2.0)
       rubocop-ast (>= 0.4.0)
-    rubocop-rails (2.16.1)
+    rubocop-rails (2.17.0)
       activesupport (>= 4.2.0)
       rack (>= 1.1)
       rubocop (>= 1.33.0, < 2.0)

data/README.md

@@ -6,14 +6,11 @@ RuboCop extension focused on ActiveRecord migration.
 
 ## Usage
 
-~~Install `rubocop-migration` gem:~~
-
-This gem is not yet published to rubygems.org.
-See [#1](https://github.com/r7kamura/rubocop-migration/issues/1) for more details.
+Install `rubocop-migration` gem:
 
 ```ruby
 # Gemfile
-gem 'rubocop-migration', require: false, github: 'r7kamura/rubocop-migration', tag: 'v0.3.1'
+gem 'rubocop-migration', require: false
 ```
 
 then require `rubocop-migration` and enable the cops you want to use in your .rubocop.yml:
@@ -53,6 +50,8 @@ Please read the comments of the respective cop classes for more information.
 
 ## Acknowledgements
 
-This gem was heavily inspired by the following gem:
+This gem was heavily inspired by [ankane/strong_migrations](https://github.com/ankane/strong_migrations).
+
+The gem `rubocop-migration` was originally developed at [wealthsimple/rubocop-migration](https://github.com/wealthsimple/rubocop-migration), and later the gem name was transferred to this repository.
 
-- [ankane/strong_migrations](https://github.com/ankane/strong_migrations)
+Some cops were originally created at [r7kamura/sevencop](https://github.com/r7kamura/sevencop) then moved to this repository.

data/lib/rubocop/cop/migration/change_column_null.rb

@@ -40,12 +40,13 @@ module RuboCop
 
         RESTRICT_ON_SEND = %i[
           change_column_null
+          change_null
         ].freeze
 
         # @param node [RuboCop::AST::SendNode]
         # @return [void]
         def on_send(node)
-          return if in_second_migration?(node)
+          return if called_with_validate_constraint?(node)
 
           add_offense(node) do |corrector|
             autocorrect(corrector, node)
@@ -54,26 +55,6 @@ module RuboCop
 
         private
 
-        # @!method parse_table_name_and_column_name(node)
-        #   @param node [RuboCop::AST::SendNode]
-        #   @return [Array<Symbol>, nil]
-        def_node_matcher :parse_table_name_and_column_name, <<~PATTERN
-          (send
-            nil?
-            _
-            ({str sym} $_)
-            ({str sym} $_)
-            ...
-          )
-        PATTERN
-
-        # @!method remove_check_constraint?(node)
-        #   @param node [RuboCop::AST::SendNode]
-        #   @return [Boolean]
-        def_node_matcher :remove_check_constraint?, <<~PATTERN
-          (send nil? :remove_check_constraint ...)
-        PATTERN
-
         # @!method validate_constraint?(node)
         #   @param node [RuboCop::AST::SendNode]
         #   @return [Boolean]
@@ -88,39 +69,138 @@ module RuboCop
           corrector,
           node
         )
-          table_name, column_name = parse_table_name_and_column_name(node)
+          case node.method_name
+          when :change_column_null
+            autocorrect_change_column_null(corrector, node)
+          when :change_null
+            autocorrect_change_null(corrector, node)
+          end
+        end
+
+        # @param corrector [RuboCop::Cop::Corrector]
+        # @param node [RuboCop::AST::SendNode]
+        # @return [void]
+        def autocorrect_change_column_null(
+          corrector,
+          node
+        )
           corrector.replace(
             node,
-            format(
-              "add_check_constraint :%<table>s, '%<column>s IS NOT NULL', name: '%<constraint>s', validate: false",
-              column: column_name,
-              constraint: "#{table_name}_#{column_name}_is_not_null",
-              table: table_name
+            format_add_check_constraint(
+              column_name: find_column_name_from_change_column_null(node),
+              table_name: find_table_name_from_change_column_null(node)
+            )
+          )
+        end
+
+        # @param corrector [RuboCop::Cop::Corrector]
+        # @param node [RuboCop::AST::SendNode]
+        # @return [void]
+        def autocorrect_change_null(
+          corrector,
+          node
+        )
+          corrector.replace(
+            node.location.selector.with(
+              end_pos: node.location.expression.end_pos
+            ),
+            format_check_constraint(
+              column_name: find_column_name_from_change_null(node),
+              table_name: find_table_name_from_change_null(node)
             )
           )
         end
 
         # @param node [RuboCop::AST::SendNode]
         # @return [Boolean]
-        def called_after_validate_constraint?(node)
-          node.left_siblings.any? do |sibling|
+        def called_with_validate_constraint?(node)
+          case node.method_name
+          when :change_column_null
+            node
+          when :change_null
+            find_ancestor_change_table(node)
+          end.left_siblings.any? do |sibling|
             validate_constraint?(sibling)
           end
         end
 
         # @param node [RuboCop::AST::SendNode]
-        # @return [Boolean]
-        def called_before_remove_check_constraint?(node)
-          node.right_siblings.any? do |sibling|
-            remove_check_constraint?(sibling)
+        # @return [RuboCop::AST::BlockNode]
+        def find_ancestor_change_table(node)
+          node.each_ancestor(:block).find do |ancestor|
+            ancestor.method?(:change_table)
           end
         end
 
         # @param node [RuboCop::AST::SendNode]
-        # @return [Boolean]
-        def in_second_migration?(node)
-          called_after_validate_constraint?(node) ||
-            called_before_remove_check_constraint?(node)
+        # @return [String]
+        def find_column_name_from_change_column_null(node)
+          node.arguments[1].value.to_s
+        end
+
+        # @param node [RuboCop::AST::SendNode]
+        # @return [String]
+        def find_column_name_from_change_null(node)
+          node.arguments[0].value.to_s
+        end
+
+        # @parm node [RuboCop::AST::SendNode]
+        # @return [String]
+        def find_table_name_from_change_column_null(node)
+          node.arguments[0].value.to_s
+        end
+
+        # @param node [RuboCop::AST::SendNode]
+        # @return [String]
+        def find_table_name_from_change_null(node)
+          find_ancestor_change_table(node).send_node.arguments[0].value.to_s
+        end
+
+        # @param column_name [String]
+        # @param table_name [String]
+        # @return [String]
+        def format_add_check_constraint(
+          column_name:,
+          table_name:
+        )
+          format(
+            'add_check_constraint :%<table_name>s, %<arguments>s',
+            arguments: format_check_constraint_arguments(
+              column_name: column_name,
+              table_name: table_name
+            ),
+            table_name: table_name
+          )
+        end
+
+        # @param column_name [String]
+        # @param table_name [String]
+        # @return [String]
+        def format_check_constraint(
+          column_name:,
+          table_name:
+        )
+          format(
+            'check_constraint %<arguments>s',
+            arguments: format_check_constraint_arguments(
+              column_name: column_name,
+              table_name: table_name
+            )
+          )
+        end
+
+        # @param coumn_name [String]
+        # @param table_name [String]
+        # @return [String]
+        def format_check_constraint_arguments(
+          column_name:,
+          table_name:
+        )
+          format(
+            "'%<column_name>s IS NOT NULL', name: '%<constraint_name>s', validate: false",
+            column_name: column_name,
+            constraint_name: "#{table_name}_#{column_name}_is_not_null"
+          )
         end
       end
     end

data/lib/rubocop/migration/version.rb

@@ -2,6 +2,6 @@
 
 module Rubocop
   module Migration
-    VERSION = '0.3.2'
+    VERSION = '0.4.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rubocop-migration
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.4.0
 platform: ruby
 authors:
 - Ryo Nakamura
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-10-19 00:00:00.000000000 Z
+date: 2022-10-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -61,7 +61,6 @@ extra_rdoc_files: []
 files:
 - ".rspec"
 - ".rubocop.yml"
-- CHANGELOG.md
 - CODE_OF_CONDUCT.md
 - Gemfile
 - Gemfile.lock

data/CHANGELOG.md

@@ -1,5 +0,0 @@
-## [Unreleased]
-
-## [0.1.0] - 2022-10-17
-
-- Initial release