playing with Ripper

Author: drewhoffer

Gemfile

@@ -12,3 +12,5 @@ gem "minitest", "~> 5.16"
 gem "rubocop", "~> 1.21"
 
 gem "thor", "~> 1.3"
+
+gem "prism", "~> 1.4"

Gemfile.lock

@@ -46,6 +46,7 @@ PLATFORMS
 
 DEPENDENCIES
   minitest (~> 5.16)
+  prism (~> 1.4)
   rake (~> 13.0)
   rbi_generator!
   rubocop (~> 1.21)

exe/rbi_generator

@@ -1,20 +1,15 @@
 require "thor"
-
+require_relative "generator"
 module RbiGenerator
   # Generate will take a filepath and then we generate the rbi file beside it
   class CLI < Thor
     desc "generate FILEPATH", "Generate an RBI file for the given Ruby file"
     def generate(file_path)
-      if File.exist?(file_path)
-        # Here you would implement the logic to generate the RBI file
-        # For now, we will just print a message
-        puts "Generating RBI for #{file_path}..."
-        # Example: File.write("#{file_path}.rbi", "Generated RBI content")
-      else
+      unless File.exist?(file_path)
         puts "File not found: #{file_path}"
+        exit(1)
       end
-    rescue StandardError => e
-      puts "An error occurred: #{e.message}"
+      Generator.new(file_path).generate
     end
   end
 end

lib/rbi_generator/cli.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require "ripper"
+require "prism"
+require "pp"
+
+module RbiGenerator
+  class Generator
+    def initialize(file_path)
+      @file_path = file_path
+      @rbi_path = @file_path.sub(/\.rb$/, ".rbi")
+    end
+
+    def generate
+      content = File.read(@file_path)
+      ast = Ripper.sexp(content)
+      pp ast
+      class_name = extract_class_name(ast)
+      method_names = extract_method_names(ast)
+      puts "Class Name: #{class_name}"
+      puts "Method Names: #{method_names.join(", ")}"
+      # puts "Method Names: #{method_names.join(", ")}"
+      # puts Ripper.sexp(content)
+    end
+
+    # Takes the syntax tree and extracts the class name
+    def extract_class_name(ast)
+      class_node = ast[1].find { |node| node[0] == :class }
+      class_node[1][1] if class_node
+    end
+
+    def extract_method_names(ast)
+      class_node = ast[1].find { |node| node[0] == :class }
+      body = class_node[3] # :bodystmt
+      return [] unless body && body[0].is_a?(Array)
+
+      method_defs = body[0].select { |node| node[0] == :def }
+      method_defs.map { |def_node| def_node[1][1] } # => ["hello", "goodbye"]
+    end
+  end
+end

lib/rbi_generator/generator.rb

@@ -0,0 +1,5 @@
+class MyTest
+  def hello; end
+
+  def test(x); end
+end