Passing actual code to PercentCode

yui-knk · Dec 25, 2023 · 5a43cf2 · 5a43cf2
1 parent 76a80b6
commit 5a43cf2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/lib/lrama/grammar.rb b/lib/lrama/grammar.rb
@@ -52,7 +52,7 @@ def initialize(rule_counter)
     end
 
     def add_percent_code(id:, code:)
-      @percent_codes << PercentCode.new(id.s_value, code)
+      @percent_codes << PercentCode.new(id.s_value, code.s_value)
     end
 
     def add_printer(ident_or_tags:, token_code:, lineno:)

diff --git a/lib/lrama/output.rb b/lib/lrama/output.rb
@@ -354,7 +354,7 @@ def percent_code(name)
       @grammar.percent_codes.select do |percent_code|
         percent_code.name == name
       end.map do |percent_code|
-        percent_code.code.s_value
+        percent_code.code
       end.join
     end
 

diff --git a/sig/lrama/grammar/percent_code.rbs b/sig/lrama/grammar/percent_code.rbs
@@ -2,9 +2,9 @@ module Lrama
   class Grammar
     class PercentCode
       attr_reader name: String
-      attr_reader code: Lexer::Token::UserCode
+      attr_reader code: String
 
-      def initialize: (String name, Lexer::Token::UserCode code) -> void
+      def initialize: (String name, String code) -> void
     end
   end
 end