1 files changed, 103 insertions, 0 deletions
diff --git a/tool/lrama/lib/lrama/grammar/symbol.rb b/tool/lrama/lib/lrama/grammar/symbol.rb
new file mode 100644
index 0000000000..deb67ad9a8
--- /dev/null
+++ b/tool/lrama/lib/lrama/grammar/symbol.rb
@@ -0,0 +1,103 @@
+# Symbol is both of nterm and term
+# `number` is both for nterm and term
+# `token_id` is tokentype for term, internal sequence number for nterm
+#
+# TODO: Add validation for ASCII code range for Token::Char
+
+module Lrama
+  class Grammar
+    class Symbol
+      attr_accessor :id, :alias_name, :tag, :number, :token_id, :nullable, :precedence,
+                    :printer, :destructor, :error_token, :first_set, :first_set_bitmap
+      attr_reader :term
+      attr_writer :eof_symbol, :error_symbol, :undef_symbol, :accept_symbol
+
+      def initialize(id:, term:, alias_name: nil, number: nil, tag: nil, token_id: nil, nullable: nil, precedence: nil, printer: nil, destructor: nil)
+        @id = id
+        @alias_name = alias_name
+        @number = number
+        @tag = tag
+        @term = term
+        @token_id = token_id
+        @nullable = nullable
+        @precedence = precedence
+        @printer = printer
+        @destructor = destructor
+      end
+
+      def term?
+        term
+      end
+
+      def nterm?
+        !term
+      end
+
+      def eof_symbol?
+        !!@eof_symbol
+      end
+
+      def error_symbol?
+        !!@error_symbol
+      end
+
+      def undef_symbol?
+        !!@undef_symbol
+      end
+
+      def accept_symbol?
+        !!@accept_symbol
+      end
+
+      def display_name
+        alias_name || id.s_value
+      end
+
+      # name for yysymbol_kind_t
+      #
+      # See: b4_symbol_kind_base
+      # @type var name: String
+      def enum_name
+        case
+        when accept_symbol?
+          name = "YYACCEPT"
+        when eof_symbol?
+          name = "YYEOF"
+        when term? && id.is_a?(Lrama::Lexer::Token::Char)
+          name = number.to_s + display_name
+        when term? && id.is_a?(Lrama::Lexer::Token::Ident)
+          name = id.s_value
+        when nterm? && (id.s_value.include?("$") || id.s_value.include?("@"))
+          name = number.to_s + id.s_value
+        when nterm?
+          name = id.s_value
+        else
+          raise "Unexpected #{self}"
+        end
+
+        "YYSYMBOL_" + name.gsub(/\W+/, "_")
+      end
+
+      # comment for yysymbol_kind_t
+      def comment
+        case
+        when accept_symbol?
+          # YYSYMBOL_YYACCEPT
+          id.s_value
+        when eof_symbol?
+          # YYEOF
+          alias_name
+        when (term? && 0 < token_id && token_id < 128)
+          # YYSYMBOL_3_backslash_, YYSYMBOL_14_
+          alias_name || id.s_value
+        when id.s_value.include?("$") || id.s_value.include?("@")
+          # YYSYMBOL_21_1
+          id.s_value
+        else
+          # YYSYMBOL_keyword_class, YYSYMBOL_strings_1
+          alias_name || id.s_value
+        end
+      end
+    end
+  end
+end