Formatting

.gitignore

@@ -6,3 +6,5 @@
 /pkg/
 /spec/reports/
 /tmp/
+
+test.rb

Gemfile

@@ -1,9 +1,10 @@
 # frozen_string_literal: true
 
-source 'https://rubygems.org'
+source "https://rubygems.org"
 
 gemspec
 
-gem 'benchmark-ips'
-gem 'parser'
-gem 'ruby_parser'
+gem "benchmark-ips"
+gem "parser"
+gem "ruby_parser"
+gem "stackprof"

Gemfile.lock

@@ -22,6 +22,7 @@ GEM
       simplecov_json_formatter (~> 0.1)
     simplecov-html (0.12.3)
     simplecov_json_formatter (0.1.3)
+    stackprof (0.2.17)
 
 PLATFORMS
   x86_64-darwin-19
@@ -35,6 +36,7 @@ DEPENDENCIES
   rake
   ruby_parser
   simplecov
+  stackprof
   syntax_tree!
 
 BUNDLED WITH

bin/profile

@@ -0,0 +1,19 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require 'stackprof'
+
+filepath = File.expand_path('../lib/syntax_tree', __dir__)
+require_relative filepath
+
+GC.disable
+
+StackProf.run(mode: :cpu, out: "tmp/profile.dump", raw: true) do
+  SyntaxTree.format(File.read("#{filepath}.rb"))
+end
+
+GC.enable
+
+`bundle exec stackprof --d3-flamegraph tmp/profile.dump > tmp/flamegraph.html`
+puts "open tmp/flamegraph.html"

exe/stree

@@ -1,5 +1,84 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-require_relative File.expand_path('../lib/syntax_tree', __dir__)
-pp SyntaxTree.new(ARGF.read).parse
+require_relative File.expand_path("../lib/syntax_tree", __dir__)
+
+help = <<~EOF
+  stree MDOE FILE
+
+  MODE: one of "a", "ast", "d", "doc", "f", "format", "w", or "write"
+  FILE: one or more paths to files to parse
+EOF
+
+if ARGV.length < 2
+  warn(help)
+  exit(1)
+end
+
+module SyntaxTree::CLI
+  class AST
+    def run(filepath)
+      pp SyntaxTree.parse(File.read(filepath))
+    end
+  end
+
+  class Doc
+    def run(filepath)
+      formatter = SyntaxTree::Formatter.new([])
+      SyntaxTree.parse(File.read(filepath)).format(formatter)
+      pp formatter.groups.first
+    end
+  end
+
+  class Format
+    def run(filepath)
+      puts SyntaxTree.format(File.read(filepath))
+    end
+  end
+
+  class Write
+    def run(filepath)
+      File.write(filepath, SyntaxTree.format(File.read(filepath)))
+    end
+  end
+end
+
+mode =
+  case ARGV.shift
+  when "a", "ast"
+    SyntaxTree::CLI::AST.new
+  when "d", "doc"
+    SyntaxTree::CLI::Doc.new
+  when "f", "format"
+    SyntaxTree::CLI::Format.new
+  when "w", "write"
+    SyntaxTree::CLI::Write.new
+  else
+    warn(help)
+    exit(1)
+  end
+
+queue = Queue.new
+ARGV.each { |pattern| Dir[pattern].each { |filepath| queue << filepath } }
+
+if queue.size <= 1
+  filepath = queue.shift
+  mode.run(filepath) if File.file?(filepath)
+  return
+end
+
+count = [8, queue.size].min
+threads =
+  count.times.map do
+    Thread.new do
+      loop do
+        filepath = queue.shift
+        break if filepath == :exit
+
+        mode.run(filepath) if File.file?(filepath)
+      end
+    end
+  end
+
+count.times { queue << :exit }
+threads.each(&:join)