add extend_modules

Author: flyerhzm

lib/helpers/parse_ruby.rb

@@ -50,6 +50,9 @@
         if node.receiver.nil? && node.name == :prepend && definitions.current_node_type == "class" && !node.arguments.nil? && %i[constant_read_node constant_path_node].include?(node.arguments.arguments.first.type)
           definitions.add_prepend_module(node.arguments.arguments.first.to_source)
         end
+        if node.receiver.nil? && node.name == :extend && definitions.current_node_type == "class" && !node.arguments.nil? && %i[constant_read_node constant_path_node].include?(node.arguments.arguments.first.type)
+          definitions.add_extend_module(node.arguments.arguments.first.to_source)
+        end
       end
 
       add_callback :call_node, at: 'start' do |node|
@@ -142,6 +145,10 @@ def add_prepend_module(name)
     @node.prepend_modules.push(name)
   end
 
+  def add_extend_module(name)
+    @node.extend_modules.push(name)
+  end
+
   def add_method(name)
     method_definition = MethodDefinition.new(parent: @node, name: name)
     @node.methods.push(method_definition)
@@ -286,7 +293,7 @@ def to_h
 end
 
 class ClassDefinition < BaseDefinition
-  attr_reader :parent, :name, :superclass, :modules, :classes, :methods, :static_methods, :constants, :include_modules, :prepend_modules
+  attr_reader :parent, :name, :superclass, :modules, :classes, :methods, :static_methods, :constants, :include_modules, :prepend_modules, :extend_modules
   attr_accessor :singleton, :ancestors
 
   def initialize(parent:, name:, superclass:)
@@ -300,6 +307,7 @@ def initialize(parent:, name:, superclass:)
     @constants = []
     @include_modules = []
     @prepend_modules = []
+    @extend_modules = []
     @ansestors = []
   end
 
@@ -314,6 +322,7 @@ def to_h
       constants: @constants,
       include_modules: @include_modules,
       prepend_modules: @prepend_modules,
+      extend_modules: @extend_modules,
       singleton: @singleton&.to_h,
       ancestors: @ancestors
     }

spec/helpers/parse_ruby_spec.rb

@@ -16,6 +16,7 @@ module Synvert
         class User
           include Trackable
           prepend Authenticatable
+          extend ClassMethods
 
           ROLES = %w[user admin].freeze
 
@@ -66,6 +67,7 @@ def user_type
               constants: [],
               include_modules: [],
               prepend_modules: [],
+              extend_modules: [],
               singleton: nil,
               ancestors: ["Synvert::User", "Authenticatable", "Trackable"]
             },
@@ -90,6 +92,7 @@ def user_type
               constants: [{ name: "ROLES" }],
               include_modules: ["Trackable"],
               prepend_modules: ["Authenticatable"],
+              extend_modules: ["ClassMethods"],
               ancestors: ["Authenticatable", "Trackable"]
             }
           ],