Make NodeKindField subclass NodeKindField

* To avoid duplication.
ruby · Feb 28, 2024 · 12e5aae · 12e5aae
1 parent 33e987b
commit 12e5aae
Showing 1 changed file with 2 additions and 15 deletions.
diff --git a/templates/template.rb b/templates/template.rb
@@ -145,7 +145,7 @@ def rbi_class
 
   # This represents a field on a node that is a list of nodes. We pass them as
   # references and store them directly on the struct.
-  class NodeListField < Field
+  class NodeListField < NodeKindField
     def rbs_class
       if specific_kind
         "Array[#{specific_kind}]"
@@ -161,20 +161,7 @@ def rbi_class
     end
 
     def java_type
-      if specific_kind
-        "#{specific_kind}[]"
-      else
-        "Node[]"
-      end
-    end
-
-    # TODO: unduplicate with NodeKindField
-    def specific_kind
-      options[:kind] unless options[:kind].is_a?(Array)
-    end
-
-    def union_kind
-      options[:kind] if options[:kind].is_a?(Array)
+      "#{super}[]"
     end
   end