fix parse logic

euglena1215 · Jan 5, 2024 · 6c4a5fc · 6c4a5fc
1 parent 1faa9e3
commit 6c4a5fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/packwerk_yard/parser.rb b/lib/packwerk_yard/parser.rb
@@ -14,11 +14,12 @@ def initialize(ruby_parser: Packwerk::Parsers::Ruby.new)
 
     def call(io:, file_path: "<unknown>")
       source_code = io.read
-      return to_ruby_ast(nil.inspect, file_path + ".yard") if source_code.nil?
+      file_path = file_path.gsub('.rb', '.yard.rb')
+      return to_ruby_ast(nil.inspect, file_path) if source_code.nil?
 
       types = extract_from_yard_to_types(source_code)
 
-      to_ruby_ast(types.map { |type| to_constant(type) }.compact.inspect, file_path + ".yard")
+      to_ruby_ast(types.map { |type| to_constant(type) }.compact.inspect, file_path)
     end
 
     def match?(path:)