Format json with JQ_COLORS
#14
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -3,7 +3,6 @@ | |
| # MIT License | ||
| # | ||
| # TODO: | ||
| # ".end" lex, require whitespace/end around ident? | ||
| # how test associativity 1|2|3? | ||
| # add some term builder helper, _term("TermTypeArray"; {query: ...}) etc? | ||
|
|
@@ -1118,6 +1117,81 @@ def parse: | |
| // error("parse error: \(.)") | ||
| ); | ||
|
|
||
| def _tojson($opts): | ||
| # color order: null, false, true, number, string, array, object, field | ||
| def _color($id): | ||
| if $opts.colors != null then "\u001b[\($opts.colors[$id])m" | ||
| else empty end; | ||
| def _reset_color: | ||
| if $opts.colors != null then "\u001b[0m" | ||
| else empty end; | ||
| def _wrap_color($id): | ||
| if $opts.colors != null then _color($id) + . + _reset_color | ||
| else . end; | ||
| def _f($opts; $indent): | ||
| def _r($prefix): | ||
| ( type as $t | ||
| | if $t == "null" then tojson | _wrap_color(0) | ||
thaliaarchi marked this conversation as resolved.
Outdated
Show resolved
Hide resolved
|
||
| elif $t == "string" then tojson | _wrap_color(4) | ||
| elif $t == "number" then tojson | _wrap_color(3) | ||
| elif $t == "boolean" then | ||
| if . then "true" | _wrap_color(2) | ||
| else "false" | _wrap_color(1) | ||
| end | ||
| elif $t == "array" then | ||
| if length == 0 then "[]" | _wrap_color(5) | ||
| else | ||
| [ _color(5), "[", $opts.compound_newline | ||
| , ( [ .[] | ||
| | $prefix, $indent | ||
| , _r($prefix+$indent) | ||
| , _color(5), $opts.array_sep | ||
| ] | ||
| | .[0:-1] | ||
| ) | ||
| , $opts.compound_newline | ||
| , $prefix, ("]" | _wrap_color(5)) | ||
| ] | ||
| end | ||
| elif $t == "object" then | ||
| if length == 0 then "{}" | _wrap_color(6) | ||
| else | ||
| [ _color(6), "{", $opts.compound_newline | ||
| , ( [ to_entries[] | ||
| | $prefix, $indent, _reset_color | ||
| , (.key | tojson | _wrap_color(7)) | ||
| , ($opts.key_sep | _wrap_color(6)) | ||
| , (.value | _r($prefix+$indent)) | ||
| , _color(6), $opts.object_sep | ||
| ] | ||
| | .[0:-1] | ||
| ) | ||
| , $opts.compound_newline | ||
| , $prefix, ("}" | _wrap_color(6)) | ||
| ] | ||
| end | ||
| else _internal_error("unknown type \($t)") | ||
| end | ||
| ); | ||
| _r(""); | ||
| ( ( { indent: 0, | ||
| key_sep: ":", | ||
| object_sep: ",", | ||
| array_sep: ",", | ||
| compound_newline: "", | ||
| } + $opts | ||
| | if .indent > 0 then | ||
| ( .key_sep = ": " | ||
| | .object_sep = ",\n" | ||
| | .array_sep = ",\n" | ||
| | .compound_newline = "\n" | ||
| ) | ||
| end | ||
| ) as $o | ||
| | _f($o; $o.indent * " ") | ||
| | if type == "array" then flatten | join("") end | ||
| ); | ||
| def _tojson: _tojson({}); | ||
|
|
||
| def undefined_func_error: | ||
| error("undefined function \(.name)"); | ||
|
|
@@ -1308,66 +1382,6 @@ def eval_ast($query; $path; $env; undefined_func): | |
| catch | ||
| error("fromjson only supports constant literals"); | ||
|
|
||
| ( . as $input | ||
| | $query.term.func as {$name, $args} | ||
| | func_name($name; $args) as $name | ||
|
|
@@ -1447,7 +1461,6 @@ def eval_ast($query; $path; $env; undefined_func): | |
| # TODO: implement in jqjq? | ||
| elif $name == "tostring/0" then [[null], tostring] | ||
| elif $name == "tojson/0" then [[null], _tojson] | ||
| elif $name == "fromjson/0" then [[null], _fromjson] | ||
| # TODO: make args general | ||
| # note "null | error" is same as empty | ||
|
|
@@ -1521,6 +1534,7 @@ def eval_ast($query; $path; $env; undefined_func): | |
| elif $name == "match/2" then match(a0; a1) | [[null], .] | ||
| elif $name == "test/2" then test(a0; a1) | [[null], .] | ||
| elif $name == "gsub/2" then gsub(a0; a1) | [[null], .] | ||
| elif $name == "gsub/3" then gsub(a0; a1; a2) | [[null], .] | ||
| elif $name == "atan2/2" then [[null], atan2(a0; a1)] | ||
| elif $name == "copysign/2" then [[null], copysign(a0; a1)] | ||
| elif $name == "drem/2" then [[null], drem(a0; a1)] | ||
|
|
@@ -2456,35 +2470,60 @@ def fromjqtest: | |
| def jqjq($args; $env): | ||
| def _parse_args: | ||
| def _f: | ||
| .[0] as $a | | ||
| if length == 0 then empty | ||
| elif $a == "-h" or $a == "--help" then {help: true}, (.[1:] | _f) | ||
| elif $a == "--jq" then {jq: .[1]}, (.[2:] | _f) | ||
| elif $a == "--lex" then {lex: true}, (.[1:] | _f) | ||
| elif $a == "--no-builtins" then {no_builtins: true}, (.[1:] | _f) | ||
| elif $a == "-n" or $a == "--null-input" then {null_input: true}, (.[1:] | _f) | ||
| elif $a == "--parse" then {parse: true}, (.[1:] | _f) | ||
| elif $a == "--repl" then {repl: true}, (.[1:] | _f) | ||
| elif $a == "--run-tests" then {run_tests: true}, (.[1:] | _f) | ||
| elif $a == "-s" or $a == "--slurp" then {slurp: true}, (.[1:] | _f) | ||
| elif $a == "-C" or $a == "--color-output" then {color_output: true}, (.[1:] | _f) | ||
| elif $a == "-M" or $a == "--monochrome-output" then {monochrome_output: true}, (.[1:] | _f) | ||
| elif $a == "--" then {filter: .[1]}, (.[2:] | _f) | ||
| elif $a | startswith("-") then error("unknown argument: \($a)") | ||
| else {filter: $a}, (.[1:] | _f) | ||
| end; | ||
| ( [_f] | ||
| | add | ||
| ); | ||
|
|
||
| # get the ANSI color codes for printing values | ||
| # corresponds to jv_set_colors and its usage in main | ||
| def parse_colors($opts; $env): | ||
|
||
| # color order: null, false, true, number, string, array, object, field | ||
| ( ["0;90", "0;39", "0;39", "0;39", "0;32", "1;39", "1;39", "1;34"] as $default | ||
| | if $env | has("JQ_COLORS") then | ||
| ( ($env.JQ_COLORS | split(":")[:8]) as $custom | ||
| | if $custom | all(length <= 12 and test("^[0-9;]*$")) then | ||
wader marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| $custom + $default[$custom | length:] | ||
| else "Failed to set $JQ_COLORS\n" | stderr | $default | ||
| end | ||
| ) | ||
| else $default | ||
| end | ||
| | if $opts.monochrome_output or | ||
| (($opts.color_output | not) and ($env.NO_COLOR | . != null and . != "")) then | ||
thaliaarchi marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| null | ||
| end | ||
| ); | ||
|
|
||
| def _help: | ||
| ( "jqjq - jq implementation of jq" | ||
| , "Usage: jqjq [OPTIONS] [--] [EXPR]" | ||
| , " --jq PATH jq implementation to run with" | ||
| , " --lex Lex EXPR" | ||
| , " --no-builtins Don't include builtins" | ||
| , " --null-input / -n Null input" | ||
| , " --parse Lex then parse EXPR" | ||
| , " --repl REPL" | ||
| , " --run-tests Run jq tests from stdin" | ||
| , " --slurp / -s Slurp inputs into an array" | ||
| , " --color-output / -C Force colored output" | ||
| , " --monochrome-output / -M Disable colored output" | ||
| ); | ||
|
|
||
| def _repl: | ||
|
|
@@ -2494,7 +2533,8 @@ def jqjq($args; $env): | |
| if . == "break" then empty | ||
| else error | ||
| end; | ||
| ( parse_colors({}; $env) as $colors | ||
| | builtins_env as $builtins_env | ||
| | _repeat_break( | ||
| ( "> " | ||
| , ( try input | ||
|
|
@@ -2503,7 +2543,7 @@ def jqjq($args; $env): | |
| | null | ||
| | try | ||
| ( eval($expr; {"$ENV": $env}; $builtins_env) | ||
| | _tojson({$colors}) | ||
| , "\n" | ||
| ) | ||
| catch | ||
|
|
@@ -2584,7 +2624,7 @@ def jqjq($args; $env): | |
| }; | ||
| if ($l | type) == "object" then | ||
| ( .line = false | ||
| | if $l.error then .errors += 1 | ||
| elif $l.ok then .oks += 1 | ||
| elif $l.end then .end = true | ||
| else . | ||
|
|
@@ -2612,13 +2652,14 @@ def jqjq($args; $env): | |
| elif $opts.slurp then [inputs] | ||
| else inputs | ||
| end; | ||
| parse_colors($opts; $env) as $colors | ||
| | ( if $opts.no_builtins then {} | ||
| else builtins_env | ||
| end | ||
| ) as $builtins_env | ||
| | _inputs | ||
| | eval($opts.filter; {"$ENV": $env}; $builtins_env) | ||
wader marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| | _tojson({$colors}) | ||
| ); | ||
|
|
||
| ( ( { filter: "." | ||
thaliaarchi marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
|
||
Uh oh!
There was an error while loading. Please reload this page.