Merge pull request #444 from ydah/refactor-trace-reporter

Introduce the `Lrama::TraceReporter` class to organize the command.rb

lib/lrama.rb

@@ -15,5 +15,6 @@
 require_relative "lrama/state"
 require_relative "lrama/states"
 require_relative "lrama/states_reporter"
+require_relative "lrama/trace_reporter"
 require_relative "lrama/version"
 require_relative "lrama/warning"

lib/lrama/command.rb

@@ -44,15 +44,8 @@ def run(argv)
         end
       end
 
-      if options.trace_opts && options.trace_opts[:rules]
-        puts "Grammar rules:"
-        grammar.rules.each { |rule| puts rule.display_name }
-      end
-
-      if options.trace_opts && options.trace_opts[:actions]
-        puts "Grammar rules with actions:"
-        grammar.rules.each { |rule| puts rule.with_actions }
-      end
+      reporter = Lrama::TraceReporter.new(grammar)
+      reporter.report(**options.trace_opts)
 
       File.open(options.outfile, "w+") do |f|
         Lrama::Output.new(

lib/lrama/trace_reporter.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Lrama
+  class TraceReporter
+    def initialize(grammar)
+      @grammar = grammar
+    end
+
+    def report(**options)
+      _report(**options)
+    end
+
+    private
+
+    def _report(rules: false, actions: false, **_)
+      report_rules if rules
+      report_actions if actions
+    end
+
+    def report_rules
+      puts "Grammar rules:"
+      @grammar.rules.each { |rule| puts rule.display_name }
+    end
+
+    def report_actions
+      puts "Grammar rules with actions:"
+      @grammar.rules.each { |rule| puts rule.with_actions }
+    end
+  end
+end