Reflection API

Rakefile

@@ -16,7 +16,16 @@ task default: :test
 
 configure = ->(task) do
   task.source_files =
-    FileList[%w[Gemfile Rakefile syntax_tree.gemspec lib/**/*.rb test/*.rb]]
+    FileList[
+      %w[
+        Gemfile
+        Rakefile
+        syntax_tree.gemspec
+        lib/**/*.rb
+        tasks/*.rake
+        test/*.rb
+      ]
+    ]
 
   # Since Syntax Tree supports back to Ruby 2.7.0, we need to make sure that we
   # format our code such that it's compatible with that version. This actually

bin/console

@@ -3,6 +3,7 @@
 
 require "bundler/setup"
 require "syntax_tree"
+require "syntax_tree/reflection"
 
 require "irb"
 IRB.start(__FILE__)

lib/syntax_tree/dsl.rb

@@ -210,12 +210,17 @@ def RAssign(value, operator, pattern)
     end
 
     # Create a new ClassDeclaration node.
-    def ClassDeclaration(constant, superclass, bodystmt)
+    def ClassDeclaration(
+      constant,
+      superclass,
+      bodystmt,
+      location = Location.default
+    )
       ClassDeclaration.new(
         constant: constant,
         superclass: superclass,
         bodystmt: bodystmt,
-        location: Location.default
+        location: location
       )
     end
 
@@ -225,12 +230,12 @@ def Comma(value)
     end
 
     # Create a new Command node.
-    def Command(message, arguments, block)
+    def Command(message, arguments, block, location = Location.default)
       Command.new(
         message: message,
         arguments: arguments,
         block: block,
-        location: Location.default
+        location: location
       )
     end
 
@@ -247,8 +252,8 @@ def CommandCall(receiver, operator, message, arguments, block)
     end
 
     # Create a new Comment node.
-    def Comment(value, inline)
-      Comment.new(value: value, inline: inline, location: Location.default)
+    def Comment(value, inline, location = Location.default)
+      Comment.new(value: value, inline: inline, location: location)
     end
 
     # Create a new Const node.
@@ -285,14 +290,21 @@ def CVar(value)
     end
 
     # Create a new DefNode node.
-    def DefNode(target, operator, name, params, bodystmt)
+    def DefNode(
+      target,
+      operator,
+      name,
+      params,
+      bodystmt,
+      location = Location.default
+    )
       DefNode.new(
         target: target,
         operator: operator,
         name: name,
         params: params,
         bodystmt: bodystmt,
-        location: Location.default
+        location: location
       )
     end
 
@@ -565,8 +577,8 @@ def MAssign(target, value)
     end
 
     # Create a new MethodAddBlock node.
-    def MethodAddBlock(call, block)
-      MethodAddBlock.new(call: call, block: block, location: Location.default)
+    def MethodAddBlock(call, block, location = Location.default)
+      MethodAddBlock.new(call: call, block: block, location: location)
     end
 
     # Create a new MLHS node.