-
Notifications
You must be signed in to change notification settings - Fork 19
/
parse.rb
55 lines (49 loc) · 1.31 KB
/
parse.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
$LOAD_PATH << File.expand_path("../lib", __FILE__)
require "twostroke"
require "bundler/setup"
require "pp"
require "coderay"
require "pry"
output = nil
def time(name)
start = Time.now.to_f
yield
finish = Time.now.to_f
puts "#{name}: #{((finish-start)*1000).round 1} ms"
end
def pretty(obj)
red=`tput setaf 1`
green=`tput setaf 2`
yellow=`tput setaf 3`
pink=`tput setaf 5`
blue=`tput setaf 6`
reset=`tput sgr0`
obj.pretty_inspect
.gsub(/<([A-Z][a-zA-Z]*(::[A-Za-z][A-Za-z]*)*)/) { |m| "<#{red}#{$1}#{reset}" }
.gsub(/([^:])(:[a-z]+)/i) { |m| "#{$1}#{pink}#{$2}#{reset}" }
.gsub(/"([^"]+)"/) { |m| "#{green}\"#{$1}\"#{reset}" }
.gsub(/=([\d\.\d]+)/) { |m| "=#{yellow}#{$1}#{reset}" }
.gsub(/(@[a-z_][a-z_0-9]*)/i) { |m| "#{blue}#{$1}#{reset}" }
end
lexer = Twostroke::Lexer.new(File.read ARGV.first)
if ARGV.include? "--tokens"
output = []
while lexer.str.size > 0
puts inspect(lexer.read_token)
end
elsif ARGV.include? "--bench"
time "lexing" do
lexer.read_token while lexer.str.size > 0
end
lexer = Twostroke::Lexer.new(File.read ARGV.first)
time "parsing" do
parser = Twostroke::Parser.new lexer
parser.parse
end
exit
else
parser = Twostroke::Parser.new lexer
parser.parse
output = parser.statements
end
puts pretty(output)