rubocop-oneoff_codemod 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c69e11eb83dcf36feec593f77035bd1bdf3bfe9c3b59e56062654ee4dfb219f9
-  data.tar.gz: e3d18f33723b0063e80000e83451ef3e0de9eb6a22654d116238d2d64aad96f9
+  metadata.gz: 8f98a73003521d6a8f5aa5811db5db534a45dcfdb5e2ff6bfbca3f28efcb59cd
+  data.tar.gz: 96038200d22d10f9aba5d7ada387359b6912244a18410458d9a71791325962b3
 SHA512:
-  metadata.gz: b8cbe64d5d41996e8ac63e63b6b4e5fbb646481861da5fc954ba6e2c8167b6ac9975c92b97b1ead00428676d92d046dad2433238b20961f7db9cd6c5e9459377
-  data.tar.gz: a7139b363ba718b531c86d2fcd1df9ef7201c9f33cee57108d0c0b2e3e759c5b52771cfdf65207da89f081a13530e87ee6d41fe21bba93a9a55fd3614bd45252
+  metadata.gz: f33f033ec49e89d02c79e4b2b0a7216f487ced3d3471d050b3c26da3e657f5fae5bb6d819062d376c34b50b61de90c49d2c5cb3b711ea277ceb884a654162bbd
+  data.tar.gz: a985602a87cf54e8b50798b1179f311c9835a086b1f7fb58c86d78d2cd3d47f22491cfcdac78086131ddd2e61578577f9835718710edc336cd1f883c7217de20

data/lib/rubocop/oneoff_codemod/cops/keep_unique.rb

@@ -8,7 +8,8 @@ module RuboCop
         extend AutoCorrector
         include IgnoredNode
 
-        MSG = "keep-unique"
+        MSG = "Keep array items unique, removing duplicates."
+        COMMAND = "keep-unique"
 
         def initialize(*args)
           super
@@ -17,9 +18,9 @@ module RuboCop
 
         def on_new_investigation
           processed_source.comments.each do |comment|
-            next unless comment.text == "# @keep-unique"
+            next unless comment.text == "# @#{COMMAND}"
 
-            @commands[comment.location.line] = :keep_unique
+            @commands[comment.location.line] = COMMAND
 
             add_offense(comment.location.expression) do |corrector|
               corrector.replace(comment.location.expression, "#")
@@ -27,8 +28,12 @@ module RuboCop
           end
         end
 
+        def command_exists?(node)
+          @commands[node.location.line - 1] == COMMAND
+        end
+
         def on_array(node)
-          return unless @commands[node.location.line - 1] == :keep_unique
+          return unless command_exists? node
           return unless node.loc.begin.source == "[" || node.loc.begin.source.start_with?("%w")
 
           add_offense(node) do |corrector|

data/lib/rubocop/oneoff_codemod/cops/rbs_prototype.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "rbs"
+
+module RuboCop
+  module Cop
+    module Codemod
+      # KeepUnique
+      class RbsPrototype < Base
+        extend AutoCorrector
+        include IgnoredNode
+
+        MSG = "Run rbs prototype and generates boilerplate signature declaration."
+        COMMAND = "rbs-prototype"
+
+        def initialize(*args)
+          super
+          @commands = {}
+        end
+
+        def find_node_on_line(node, target_line)
+          if node.loc&.line == target_line && node.is_a?(RuboCop::AST::DefNode)
+            node
+          else
+            node.child_nodes.find do |child_node|
+              find_node_on_line child_node, target_line
+            end
+          end
+        end
+
+        def on_new_investigation
+          processed_source.comments.each do |comment|
+            next unless comment.text == "# @#{COMMAND}"
+
+            target_node = find_node_on_line(processed_source.ast, comment.location.line + 1)
+            target_method = target_node.method_name
+
+            add_offense(comment.location.expression) do |corrector|
+              parser = RBS::Prototype::RB.new
+              parser.parse processed_source.raw_source
+              string = []
+              parser.decls.each do |decl|
+                decl.members.each do |member|
+                  string << member.overloads.map(&:method_type).join(" | ") if target_method == member.name
+                end
+              end
+
+              corrector.replace(comment.location.expression, "# #{string.join}")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rubocop/oneoff_codemod/version.rb

@@ -2,6 +2,6 @@
 
 module RuboCop
   module OneoffCodemod
-    VERSION = "0.0.4"
+    VERSION = "0.0.5"
   end
 end

data/lib/rubocop/oneoff_codemod.rb

@@ -6,3 +6,4 @@ require_relative "oneoff_codemod/version"
 require_relative "oneoff_codemod/plugin"
 
 require_relative "oneoff_codemod/cops/keep_unique"
+require_relative "oneoff_codemod/cops/rbs_prototype"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rubocop-oneoff_codemod
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Koji NAKAMURA
@@ -23,6 +23,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: rbs
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rubocop
   requirement: !ruby/object:Gem::Requirement
@@ -61,6 +75,7 @@ files:
 - config/default.yml
 - lib/rubocop/oneoff_codemod.rb
 - lib/rubocop/oneoff_codemod/cops/keep_unique.rb
+- lib/rubocop/oneoff_codemod/cops/rbs_prototype.rb
 - lib/rubocop/oneoff_codemod/plugin.rb
 - lib/rubocop/oneoff_codemod/version.rb
 - sig/rubocop/oneoff_codemod.rbs