lib/lrama/command.rb in lrama-0.5.2 vs lib/lrama/command.rb in lrama-0.5.3

@@ -1,97 +1,60 @@
 require 'optparse'
 
 module Lrama
   class Command
-    def run(argv)
-      opt = OptionParser.new
+    def initialize(argv)
+      @argv = argv
 
-      # opt.on('-h') {|v| p v }
-      opt.on('-V', '--version') {|v| puts Lrama::VERSION ; exit 0 }
+      @version = nil
+      @skeleton = "bison/yacc.c"
+      @header = false
+      @header_file = nil
+      @report = []
+      @report_file = nil
+      @outfile = "y.tab.c"
+      @trace = []
+      @error_recovery = false
+      @grammar_file = nil
+      @report_file = nil
+      @trace_opts = nil
+      @report_opts = nil
+    end
 
-      # Tuning the Parser
-      skeleton = "bison/yacc.c"
+    def run
+      parse_option
 
-      opt.on('-S', '--skeleton=FILE') {|v| skeleton = v }
-      opt.on('-t') {  } # Do nothing
-
-      # Output Files:
-      header = false
-      header_file = nil
-      report = []
-      report_file = nil
-      outfile = "y.tab.c"
-
-      opt.on('-h', '--header=[FILE]') {|v| header = true; header_file = v }
-      opt.on('-d') { header = true }
-      opt.on('-r', '--report=THINGS') {|v| report = v.split(',') }
-      opt.on('--report-file=FILE')    {|v| report_file = v }
-      opt.on('-v') {  } # Do nothing
-      opt.on('-o', '--output=FILE')   {|v| outfile = v }
-
-      # Hidden
-      trace = []
-      opt.on('--trace=THINGS') {|v| trace = v.split(',') }
-
-      # Error Recovery
-      error_recovery = false
-      opt.on('-e') {|v| error_recovery = true }
-
-      opt.parse!(argv)
-
-      trace_opts = validate_trace(trace)
-      report_opts = validate_report(report)
-
-      grammar_file = argv.shift
-
-      if !report.empty? && report_file.nil? && grammar_file
-        report_file = File.dirname(grammar_file) + "/" + File.basename(grammar_file, ".*") + ".output"
+      if @version
+        puts Lrama::VERSION
+        exit 0
       end
 
-      if !header_file && header
-        case
-        when outfile
-          header_file = File.dirname(outfile) + "/" + File.basename(outfile, ".*") + ".h"
-        when grammar_file
-          header_file = File.dirname(grammar_file) + "/" + File.basename(grammar_file, ".*") + ".h"
-        end
-      end
+      Report::Duration.enable if @trace_opts[:time]
 
-      if !grammar_file
-        abort "File should be specified\n"
-      end
-
-      Report::Duration.enable if trace_opts[:time]
-
       warning = Lrama::Warning.new
-      if grammar_file == '-'
-        grammar_file = argv.shift or abort "File name for STDIN should be specified\n"
-        y = STDIN.read
-      else
-        y = File.read(grammar_file)
-      end
-      grammar = Lrama::Parser.new(y).parse
-      states = Lrama::States.new(grammar, warning, trace_state: (trace_opts[:automaton] || trace_opts[:closure]))
+      grammar = Lrama::Parser.new(@y.read).parse
+      states = Lrama::States.new(grammar, warning, trace_state: (@trace_opts[:automaton] || @trace_opts[:closure]))
       states.compute
       context = Lrama::Context.new(states)
 
-      if report_file
+      if @report_file
         reporter = Lrama::StatesReporter.new(states)
-        File.open(report_file, "w+") do |f|
-          reporter.report(f, **report_opts)
+        File.open(@report_file, "w+") do |f|
+          reporter.report(f, **@report_opts)
         end
       end
 
-      File.open(outfile, "w+") do |f|
+      File.open(@outfile, "w+") do |f|
         Lrama::Output.new(
           out: f,
-          output_file_path: outfile,
-          template_name: skeleton,
-          grammar_file_path: grammar_file,
-          header_file_path: header_file,
+          output_file_path: @outfile,
+          template_name: @skeleton,
+          grammar_file_path: @grammar_file,
+          header_file_path: @header_file,
           context: context,
           grammar: grammar,
+          error_recovery: @error_recovery,
         ).render
       end
 
       if warning.has_error?
         exit 1
@@ -141,8 +104,64 @@
           raise "Invalid trace option \"#{t}\"."
         end
       end
 
       return h
+    end
+
+    def parse_option
+      opt = OptionParser.new
+
+      # opt.on('-h') {|v| p v }
+      opt.on('-V', '--version') {|v| @version = true }
+
+      # Tuning the Parser
+      opt.on('-S', '--skeleton=FILE') {|v| @skeleton = v }
+      opt.on('-t') {  } # Do nothing
+
+      # Output Files:
+      opt.on('-h', '--header=[FILE]') {|v| @header = true; @header_file = v }
+      opt.on('-d') { @header = true }
+      opt.on('-r', '--report=THINGS') {|v| @report = v.split(',') }
+      opt.on('--report-file=FILE')    {|v| @report_file = v }
+      opt.on('-v') {  } # Do nothing
+      opt.on('-o', '--output=FILE')   {|v| @outfile = v }
+
+      # Hidden
+      opt.on('--trace=THINGS') {|v| @trace = v.split(',') }
+
+      # Error Recovery
+      opt.on('-e') {|v| @error_recovery = true }
+
+      opt.parse!(@argv)
+
+      @trace_opts = validate_trace(@trace)
+      @report_opts = validate_report(@report)
+
+      @grammar_file = @argv.shift
+
+      if !@grammar_file
+        abort "File should be specified\n"
+      end
+
+      if @grammar_file == '-'
+        @grammar_file = @argv.shift or abort "File name for STDIN should be specified\n"
+        @y = STDIN
+      else
+        @y = File.open(@grammar_file, 'r')
+      end
+
+      if !@report.empty? && @report_file.nil? && @grammar_file
+        @report_file = File.dirname(@grammar_file) + "/" + File.basename(@grammar_file, ".*") + ".output"
+      end
+
+      if !@header_file && @header
+        case
+        when @outfile
+          @header_file = File.dirname(@outfile) + "/" + File.basename(@outfile, ".*") + ".h"
+        when @grammar_file
+          @header_file = File.dirname(@grammar_file) + "/" + File.basename(@grammar_file, ".*") + ".h"
+        end
+      end
     end
   end
 end