Add NewPerser support

ruby · Oct 14, 2023 · 124135b · 124135b
1 parent ace817b
commit 124135b
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 6 deletions.
diff --git a/lib/lrama/command.rb b/lib/lrama/command.rb
@@ -8,7 +8,7 @@ def run(argv)
       warning = Lrama::Warning.new
       text = options.y.read
       options.y.close if options.y != STDIN
-      grammar = Lrama::NewParser.new(text).parse
+      grammar = Lrama::NewParser.new(text, options.header_file).parse
       states = Lrama::States.new(grammar, warning, trace_state: (options.trace_opts[:automaton] || options.trace_opts[:closure]))
       states.compute
       context = Lrama::Context.new(states)

diff --git a/lib/lrama/new_parser.rb b/lib/lrama/new_parser.rb
diff --git a/parser.y b/parser.y
@@ -18,7 +18,7 @@ rule
                           }
                         "%}"
                           {
-                            @grammar.prologue = val[2].s_value
+                            @grammar.prologue = prologue(val[2])
                           }
                       | "%require" STRING
 
@@ -363,8 +363,9 @@ end
 
 ---- inner
 
-def initialize(text)
+def initialize(text, header_file)
   @text = text
+  @header_path = header_file ? header_file.sub("./", "") : nil
 end
 
 def parse
@@ -382,3 +383,11 @@ end
 def next_token
   @lexer.next_token
 end
+
+def prologue(s_value)
+  if @header_path
+    "\n#include \"#{@header_path}\"\n" + s_value
+  else
+    s_value
+  end
+end