Merge pull request #1292 from eregon/use-kind-for-ruby-signature

ruby · Aug 20, 2023 · 613c6ff · 613c6ff
2 parents e39f861 + ffed352
commit 613c6ff
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/templates/template.rb b/templates/template.rb
@@ -23,6 +23,10 @@ def c_type
     end
   end
 
+  def ruby_type
+    options[:kind] || "Node"
+  end
+
   def java_type
     options[:kind] || "Node"
   end
@@ -42,7 +46,7 @@ class NodeParam < Param
   include KindTypes
 
   def rbs_class
-    "Node"
+    ruby_type
   end
 end
 
@@ -52,7 +56,7 @@ class OptionalNodeParam < Param
   include KindTypes
 
   def rbs_class
-    "Node?"
+    "#{ruby_type}?"
   end
 end