add local_calls to MethodDefinition

Author: flyerhzm

lib/helpers/parse_ruby.rb

@@ -56,6 +56,12 @@
         end
       end
 
+      add_callback :call_node, at: 'start' do |node|
+        if node.receiver.nil? && definitions.current_node_type == "method"
+          definitions.node.local_calls.push(node.name.to_s)
+        end
+      end
+
       add_callback :def_node, at: 'start' do |node|
         # we can't handle `def self.inclueded` method
         if !node.receiver.nil? && node.name == :included
@@ -324,11 +330,12 @@ def to_h
 end
 
 class MethodDefinition
-  attr_reader :parent, :name
+  attr_reader :parent, :name, :local_calls
 
   def initialize(parent:, name:)
     @parent = parent
     @name = name
+    @local_calls = []
   end
 
   def call_method?(method_name)
@@ -341,6 +348,6 @@ def call_any_method?(method_names)
   end
 
   def to_h
-    { name: @name }
+    { name: @name, local_calls: local_calls }
   end
 end

spec/helpers/parse_ruby_spec.rb

@@ -27,6 +27,8 @@ def bot
           end
 
           def self.authenticate?(email, password)
+            user = find_by(email:)
+            user.valid_password?(password)
           end
 
           def user_type
@@ -59,7 +61,7 @@ def user_type
               superclass: "User",
               classes: [],
               modules: [],
-              methods: [{ name: "user_type" }],
+              methods: [{ name: "user_type", local_calls: [] }],
               static_methods: [],
               constants: [],
               included_modules: [],
@@ -72,15 +74,18 @@ def user_type
               singleton: {
                 constants: [],
                 methods: [
-                  { name: 'system' },
-                  { name: 'bot' }
+                  { name: 'system', local_calls: [] },
+                  { name: 'bot', local_calls: [] }
                 ],
                 ancestors: []
               },
               classes: [],
               modules: [],
-              methods: [{ name: "user_type" }],
-              static_methods: [{ name: 'authenticate?' }],
+              methods: [{ name: "user_type", local_calls: [] }],
+              static_methods: [{
+                name: 'authenticate?',
+                local_calls: ['find_by']
+              }],
               constants: [{ name: "ROLES" }],
               included_modules: ["Trackable"],
               ancestors: ["Trackable"]