[ruby/prism] Add RBI and RBS templates for Prism

ruby/prism@b62305ff3e Co-authored-by: Stan Lo <st0012@users.noreply.github.com>
ruby · Oct 27, 2023 · 544f640 · 544f640
1 parent a8af5d3
commit 544f640
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 28 deletions.
diff --git a/lib/prism/prism.gemspec b/lib/prism/prism.gemspec
@@ -107,6 +107,8 @@ Gem::Specification.new do |spec|
     "src/util/pm_strpbrk.c",
     "src/prism.c",
     "prism.gemspec",
+    "sig/prism.rbs",
+    "rbi/prism.rbi"
   ]
 
   spec.extensions = ["ext/prism/extconf.rb"]

diff --git a/prism/templates/template.rb b/prism/templates/template.rb
@@ -290,33 +290,47 @@ def template(name, write_to: nil)
       template = File.expand_path("../#{filepath}", __dir__)
 
       erb = read_template(template)
-
-      non_ruby_heading = <<~HEADING
-      /******************************************************************************/
-      /* This file is generated by the templates/template.rb script and should not  */
-      /* be modified manually. See                                                  */
-      /* #{filepath + " " * (74 - filepath.size) } */
-      /* if you are looking to modify the                                           */
-      /* template                                                                   */
-      /******************************************************************************/
-      HEADING
-
-      ruby_heading = <<~HEADING
-      # frozen_string_literal: true
-      =begin
-      This file is generated by the templates/template.rb script and should not be
-      modified manually. See #{filepath}
-      if you are looking to modify the template
-      =end
-
-      HEADING
-
-      heading =
-        if File.extname(filepath.gsub(".erb", "")) == ".rb"
-          ruby_heading
-        else
-          non_ruby_heading
-        end
+      extension = File.extname(filepath.gsub(".erb", ""))
+
+      heading = case extension
+      when ".rb"
+        <<~HEADING
+        # frozen_string_literal: true
+        =begin
+        This file is generated by the templates/template.rb script and should not be
+        modified manually. See #{filepath}
+        if you are looking to modify the template
+        =end
+
+        HEADING
+      when ".rbs"
+        <<~HEADING
+        # This file is generated by the templates/template.rb script and should not be
+        # modified manually. See #{filepath}
+        # if you are looking to modify the template
+
+        HEADING
+      when ".rbi"
+          <<~HEADING
+          # typed: true
+
+          =begin
+          This file is generated by the templates/template.rb script and should not be
+          modified manually. See #{filepath}
+          if you are looking to modify the template
+          =end
+          HEADING
+      else
+        <<~HEADING
+        /******************************************************************************/
+        /* This file is generated by the templates/template.rb script and should not  */
+        /* be modified manually. See                                                  */
+        /* #{filepath + " " * (74 - filepath.size) } */
+        /* if you are looking to modify the                                           */
+        /* template                                                                   */
+        /******************************************************************************/
+        HEADING
+      end
 
       write_to ||= File.expand_path("../#{name}", __dir__)
       contents = heading + erb.result_with_hash(locals)
@@ -376,7 +390,9 @@ def locals
     "src/node.c",
     "src/prettyprint.c",
     "src/serialize.c",
-    "src/token_type.c"
+    "src/token_type.c",
+    "rbi/prism.rbi",
+    "sig/prism.rbs",
   ]
 end