Lazy initialization of PrinterCode

PrinterCode needs tag for code translation.
It's possible to pass tag as argument by lazy initialization.

lib/lrama/grammar.rb

@@ -46,12 +46,12 @@ def add_percent_code(id:, code:)
       @percent_codes << PercentCode.new(id, code)
     end
 
-    def add_printer(ident_or_tags:, code:, lineno:)
-      @printers << Printer.new(ident_or_tags: ident_or_tags, code: code, lineno: lineno)
+    def add_printer(ident_or_tags:, token_code:, lineno:)
+      @printers << Printer.new(ident_or_tags: ident_or_tags, token_code: token_code, lineno: lineno)
     end
 
-    def add_error_token(ident_or_tags:, code:, lineno:)
-      @error_tokens << ErrorToken.new(ident_or_tags: ident_or_tags, code: code, lineno: lineno)
+    def add_error_token(ident_or_tags:, token_code:, lineno:)
+      @error_tokens << ErrorToken.new(ident_or_tags: ident_or_tags, token_code: token_code, lineno: lineno)
     end
 
     def add_term(id:, alias_name: nil, tag: nil, token_id: nil, replace: false)

lib/lrama/grammar/code/printer_code.rb

@@ -2,7 +2,10 @@ module Lrama
   class Grammar
     class Code
       class PrinterCode < Code
-        attr_accessor :tag
+        def initialize(type: nil, token_code: nil, tag: nil)
+          super(type: type, token_code: token_code)
+          @tag = tag
+        end
 
         private
 

lib/lrama/grammar/error_token.rb

@@ -1,9 +1,8 @@
 module Lrama
   class Grammar
-    class ErrorToken < Struct.new(:ident_or_tags, :code, :lineno, keyword_init: true)
+    class ErrorToken < Struct.new(:ident_or_tags, :token_code, :lineno, keyword_init: true)
       def translated_code(tag)
-        code.tag = tag
-        code.translated_code
+        Code::PrinterCode.new(type: :error_token, token_code: token_code, tag: tag).translated_code
       end
     end
   end

lib/lrama/grammar/printer.rb

@@ -1,9 +1,8 @@
 module Lrama
   class Grammar
-    class Printer < Struct.new(:ident_or_tags, :code, :lineno, keyword_init: true)
+    class Printer < Struct.new(:ident_or_tags, :token_code, :lineno, keyword_init: true)
       def translated_code(tag)
-        code.tag = tag
-        code.translated_code
+        Code::PrinterCode.new(type: :printer, token_code: token_code, tag: tag).translated_code
       end
     end
   end

parser.y

@@ -108,7 +108,7 @@ rule
                          {
                            @grammar.add_printer(
                              ident_or_tags: val[6],
-                             code: Grammar::Code::PrinterCode.new(type: :printer, token_code: val[3]),
+                             token_code: val[3],
                              lineno: val[3].line
                            )
                          }
@@ -124,7 +124,7 @@ rule
                          {
                            @grammar.add_error_token(
                              ident_or_tags: val[6],
-                             code: Grammar::Code::PrinterCode.new(type: :error_token, token_code: val[3]),
+                             token_code: val[3],
                              lineno: val[3].line
                            )
                          }

spec/lrama/grammar/code_spec.rb

@@ -60,26 +60,22 @@
       let(:tag) { token_class::Tag.new(s_value: '<val>') }
 
       it "translats '$$' to '((*yyvaluep).val)'" do
-        code = described_class.new(type: :printer, token_code: user_code_dollar_dollar)
-        code.tag = tag
+        code = described_class.new(type: :printer, token_code: user_code_dollar_dollar, tag: tag)
         expect(code.translated_code).to eq("print(((*yyvaluep).val));")
       end
 
       it "translats '@$' to '(*yylocationp)'" do
-        code = described_class.new(type: :printer, token_code: user_code_at_dollar)
-        code.tag = tag
+        code = described_class.new(type: :printer, token_code: user_code_at_dollar, tag: tag)
         expect(code.translated_code).to eq("print((*yylocationp));")
       end
 
       it "raises error for '$n'" do
-        code = described_class.new(type: :printer, token_code: user_code_dollar_n)
-        code.tag = tag
+        code = described_class.new(type: :printer, token_code: user_code_dollar_n, tag: tag)
         expect { code.translated_code }.to raise_error("$n can not be used in printer.")
       end
 
       it "raises error for '@n'" do
-        code = described_class.new(type: :printer, token_code: user_code_at_n)
-        code.tag = tag
+        code = described_class.new(type: :printer, token_code: user_code_at_n, tag: tag)
         expect { code.translated_code }.to raise_error("@n can not be used in printer.")
       end
     end

spec/lrama/parser_spec.rb

@@ -56,12 +56,12 @@
       expect(grammar.printers).to eq([
         Printer.new(
           ident_or_tags: [T::Tag.new(s_value: "<int>")],
-          code: Code::PrinterCode.new(type: :printer, token_code: T::UserCode.new(s_value: "\n    print_int();\n")),
+          token_code: T::UserCode.new(s_value: "\n    print_int();\n"),
           lineno: 15
         ),
         Printer.new(
           ident_or_tags: [T::Ident.new(s_value: "tNUMBER"), T::Ident.new(s_value: "tSTRING")],
-          code: Code::PrinterCode.new(type: :printer, token_code: T::UserCode.new(s_value: "\n    print_token();\n")),
+          token_code: T::UserCode.new(s_value: "\n    print_token();\n"),
           lineno: 18
         ),
       ])