ralsina/tartrazine
1
0
Fork 0
mirror of https://github.com/ralsina/tartrazine.git synced 2024-09-20 07:21:22 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Use IO for output

Browse Source
This commit is contained in:
Roberto Alsina 2024-08-16 17:25:33 -03:00
parent ae03e4612e
commit bb952a44b8
4 changed files with 34 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/formatters/ansi.cr
View File

@ -11,10 +11,9 @@ module Tartrazine
"#{i + 1}".rjust(4).ljust(5) "#{i + 1}".rjust(4).ljust(5)
end end
def format(text : String, lexer : Lexer) : String def format(text : String, lexer : Lexer, outp : IO) : Nil
tokenizer = Tokenizer.new(lexer, text) tokenizer = Tokenizer.new(lexer, text)
i = 0 i = 0
output = String.build do |outp|
outp << line_label(i) if line_numbers? outp << line_label(i) if line_numbers?
tokenizer.each do |token| tokenizer.each do |token|
outp << colorize(token[:value], token[:type]) outp << colorize(token[:value], token[:type])
@ -24,21 +23,6 @@ module Tartrazine
end end
end end
end end
output
end
# def format(text : String, lexer : Lexer) : String
# output = String.build do |outp|
# lexer.group_tokens_in_lines(lexer.tokenize(text)).each_with_index do |line, i|
# label = line_numbers? ? "#{i + 1}".rjust(4).ljust(5) : ""
# outp << label
# line.each do |token|
# outp << colorize(token[:value], token[:type])
# end
# end
# end
# output
# end
def colorize(text : String, token : String) : String def colorize(text : String, token : String) : String
style = theme.styles.fetch(token, nil) style = theme.styles.fetch(token, nil)

23
src/formatters/html.cr
View File

@ -34,24 +34,21 @@ module Tartrazine
@weight_of_bold : Int32 = 600) @weight_of_bold : Int32 = 600)
end end
def format(text : String, lexer : Lexer) : String def format(text : String, lexer : Lexer, dst : IO) : Nil
text = format_text(text, lexer) pre, post = wrap_standalone
if standalone? dst << pre if standalone?
text = wrap_standalone(text) format_text(text, lexer, dst)
end dst << post if standalone?
text
end end
# Wrap text into a full HTML document, including the CSS for the theme # Wrap text into a full HTML document, including the CSS for the theme
def wrap_standalone(text) : String def wrap_standalone
output = String.build do |outp| output = String.build do |outp|
outp << "<!DOCTYPE html><html><head><style>" outp << "<!DOCTYPE html><html><head><style>"
outp << style_defs outp << style_defs
outp << "</style></head><body>" outp << "</style></head><body>"
outp << text
outp << "</body></html>"
end end
output {output.to_s, "</body></html>"}
end end
private def line_label(i : Int32) : String private def line_label(i : Int32) : String
@ -61,11 +58,9 @@ module Tartrazine
"<span #{line_id} #{line_class} style=\"user-select: none;\">#{line_label} </span>" "<span #{line_id} #{line_class} style=\"user-select: none;\">#{line_label} </span>"
end end
def format_text(text : String, lexer : Lexer) : String def format_text(text : String, lexer : Lexer, outp : IO) : Nil
# lines = lexer.group_tokens_in_lines(lexer.tokenize(text))
tokenizer = Tokenizer.new(lexer, text) tokenizer = Tokenizer.new(lexer, text)
i = 0 i = 0
output = String.build do |outp|
if surrounding_pre? if surrounding_pre?
pre_style = wrap_long_lines? ? "style=\"white-space: pre-wrap; word-break: break-word;\"" : "" pre_style = wrap_long_lines? ? "style=\"white-space: pre-wrap; word-break: break-word;\"" : ""
outp << "<pre class=\"#{get_css_class("Background")}\" #{pre_style}>" outp << "<pre class=\"#{get_css_class("Background")}\" #{pre_style}>"
@ -81,8 +76,6 @@ module Tartrazine
end end
outp << "</code></pre>" outp << "</code></pre>"
end end
output
end
# ameba:disable Metrics/CyclomaticComplexity # ameba:disable Metrics/CyclomaticComplexity
def style_defs : String def style_defs : String

5
src/formatters/json.cr
View File

@ -4,8 +4,9 @@ module Tartrazine
class Json < Formatter class Json < Formatter
property name = "json" property name = "json"
def format(text : String, lexer : Lexer, _theme : Theme) : String def format(text : String, lexer : Lexer, outp : IO) : Nil
lexer.tokenize(text).to_json tokenizer = Tokenizer.new(lexer, text)
outp << Tartrazine::Lexer.collapse_tokens(tokenizer.to_a).to_json
end end
end end
end end

8
src/main.cr
View File

@ -85,13 +85,11 @@ if options["-f"]
lexer = Tartrazine.lexer(name: options["-l"].as(String), filename: options["FILE"].as(String)) lexer = Tartrazine.lexer(name: options["-l"].as(String), filename: options["FILE"].as(String))
input = File.open(options["FILE"].as(String)).gets_to_end input = File.open(options["FILE"].as(String)).gets_to_end
output = formatter.format(input, lexer)
if options["-o"].nil? if options["-o"].nil?
puts output outf = STDOUT
else else
File.open(options["-o"].as(String), "w") do |outf| outf = File.open(options["-o"].as(String), "w")
outf << output
end
end end
formatter.format(input, lexer, outf)
end end