383 lines
10 KiB
Lua
383 lines
10 KiB
Lua
local type_order = {["function"] = 5, boolean = 2, number = 1, string = 3, table = 4, thread = 7, userdata = 6}
|
|
local function sort_keys(_0_0, _1_0)
|
|
local _1_ = _0_0
|
|
local a = _1_[1]
|
|
local _2_ = _1_0
|
|
local b = _2_[1]
|
|
local ta = type(a)
|
|
local tb = type(b)
|
|
if ((ta == tb) and ((ta == "string") or (ta == "number"))) then
|
|
return (a < b)
|
|
else
|
|
local dta = type_order[ta]
|
|
local dtb = type_order[tb]
|
|
if (dta and dtb) then
|
|
return (dta < dtb)
|
|
elseif dta then
|
|
return true
|
|
elseif dtb then
|
|
return false
|
|
else
|
|
return (ta < tb)
|
|
end
|
|
end
|
|
end
|
|
local function table_kv_pairs(t)
|
|
local assoc_3f = false
|
|
local i = 1
|
|
local kv = {}
|
|
local insert = table.insert
|
|
for k, v in pairs(t) do
|
|
if ((type(k) ~= "number") or (k ~= i)) then
|
|
assoc_3f = true
|
|
end
|
|
i = (i + 1)
|
|
insert(kv, {k, v})
|
|
end
|
|
table.sort(kv, sort_keys)
|
|
if (#kv == 0) then
|
|
return kv, "empty"
|
|
else
|
|
local function _2_()
|
|
if assoc_3f then
|
|
return "table"
|
|
else
|
|
return "seq"
|
|
end
|
|
end
|
|
return kv, _2_()
|
|
end
|
|
end
|
|
local function count_table_appearances(t, appearances)
|
|
if (type(t) == "table") then
|
|
if not appearances[t] then
|
|
appearances[t] = 1
|
|
for k, v in pairs(t) do
|
|
count_table_appearances(k, appearances)
|
|
count_table_appearances(v, appearances)
|
|
end
|
|
else
|
|
appearances[t] = ((appearances[t] or 0) + 1)
|
|
end
|
|
end
|
|
return appearances
|
|
end
|
|
local function save_table(t, seen)
|
|
local seen0 = (seen or {len = 0})
|
|
local id = (seen0.len + 1)
|
|
if not seen0[t] then
|
|
seen0[t] = id
|
|
seen0.len = id
|
|
end
|
|
return seen0
|
|
end
|
|
local function detect_cycle(t, seen, _3fk)
|
|
if ("table" == type(t)) then
|
|
seen[t] = true
|
|
local _2_0, _3_0 = next(t, _3fk)
|
|
if ((nil ~= _2_0) and (nil ~= _3_0)) then
|
|
local k = _2_0
|
|
local v = _3_0
|
|
return (seen[k] or detect_cycle(k, seen) or seen[v] or detect_cycle(v, seen) or detect_cycle(t, seen, k))
|
|
end
|
|
end
|
|
end
|
|
local function visible_cycle_3f(t, options)
|
|
return (options["detect-cycles?"] and detect_cycle(t, {}) and save_table(t, options.seen) and (1 < (options.appearances[t] or 0)))
|
|
end
|
|
local function table_indent(t, indent, id)
|
|
local opener_length = nil
|
|
if id then
|
|
opener_length = (#tostring(id) + 2)
|
|
else
|
|
opener_length = 1
|
|
end
|
|
return (indent + opener_length)
|
|
end
|
|
local pp = nil
|
|
local function concat_table_lines(elements, options, multiline_3f, indent, table_type, prefix)
|
|
local indent_str = ("\n" .. string.rep(" ", indent))
|
|
local open = nil
|
|
local function _2_()
|
|
if ("seq" == table_type) then
|
|
return "["
|
|
else
|
|
return "{"
|
|
end
|
|
end
|
|
open = ((prefix or "") .. _2_())
|
|
local close = nil
|
|
if ("seq" == table_type) then
|
|
close = "]"
|
|
else
|
|
close = "}"
|
|
end
|
|
local oneline = (open .. table.concat(elements, " ") .. close)
|
|
if (not options["one-line?"] and (multiline_3f or ((indent + #oneline) > options["line-length"]))) then
|
|
return (open .. table.concat(elements, indent_str) .. close)
|
|
else
|
|
return oneline
|
|
end
|
|
end
|
|
local function pp_associative(t, kv, options, indent, key_3f)
|
|
local multiline_3f = false
|
|
local id = options.seen[t]
|
|
if (options.level >= options.depth) then
|
|
return "{...}"
|
|
elseif (id and options["detect-cycles?"]) then
|
|
return ("@" .. id .. "{...}")
|
|
else
|
|
local visible_cycle_3f0 = visible_cycle_3f(t, options)
|
|
local id0 = (visible_cycle_3f0 and options.seen[t])
|
|
local indent0 = table_indent(t, indent, id0)
|
|
local slength = nil
|
|
local function _3_()
|
|
local _2_0 = rawget(_G, "utf8")
|
|
if _2_0 then
|
|
return _2_0.len
|
|
else
|
|
return _2_0
|
|
end
|
|
end
|
|
local function _4_(_241)
|
|
return #_241
|
|
end
|
|
slength = ((options["utf8?"] and _3_()) or _4_)
|
|
local prefix = nil
|
|
if visible_cycle_3f0 then
|
|
prefix = ("@" .. id0)
|
|
else
|
|
prefix = ""
|
|
end
|
|
local items = nil
|
|
do
|
|
local tbl_0_ = {}
|
|
for _, _6_0 in pairs(kv) do
|
|
local _7_ = _6_0
|
|
local k = _7_[1]
|
|
local v = _7_[2]
|
|
local _8_
|
|
do
|
|
local k0 = pp(k, options, (indent0 + 1), true)
|
|
local v0 = pp(v, options, (indent0 + slength(k0) + 1))
|
|
multiline_3f = (multiline_3f or k0:find("\n") or v0:find("\n"))
|
|
_8_ = (k0 .. " " .. v0)
|
|
end
|
|
tbl_0_[(#tbl_0_ + 1)] = _8_
|
|
end
|
|
items = tbl_0_
|
|
end
|
|
return concat_table_lines(items, options, multiline_3f, indent0, "table", prefix)
|
|
end
|
|
end
|
|
local function pp_sequence(t, kv, options, indent)
|
|
local multiline_3f = false
|
|
local id = options.seen[t]
|
|
if (options.level >= options.depth) then
|
|
return "[...]"
|
|
elseif (id and options["detect-cycles?"]) then
|
|
return ("@" .. id .. "[...]")
|
|
else
|
|
local visible_cycle_3f0 = visible_cycle_3f(t, options)
|
|
local id0 = (visible_cycle_3f0 and options.seen[t])
|
|
local indent0 = table_indent(t, indent, id0)
|
|
local prefix = nil
|
|
if visible_cycle_3f0 then
|
|
prefix = ("@" .. id0)
|
|
else
|
|
prefix = ""
|
|
end
|
|
local items = nil
|
|
do
|
|
local tbl_0_ = {}
|
|
for _, _3_0 in pairs(kv) do
|
|
local _4_ = _3_0
|
|
local _0 = _4_[1]
|
|
local v = _4_[2]
|
|
local _5_
|
|
do
|
|
local v0 = pp(v, options, indent0)
|
|
multiline_3f = (multiline_3f or v0:find("\n"))
|
|
_5_ = v0
|
|
end
|
|
tbl_0_[(#tbl_0_ + 1)] = _5_
|
|
end
|
|
items = tbl_0_
|
|
end
|
|
return concat_table_lines(items, options, multiline_3f, indent0, "seq", prefix)
|
|
end
|
|
end
|
|
local function concat_lines(lines, options, indent, force_multi_line_3f)
|
|
if (#lines == 0) then
|
|
if options["empty-as-sequence?"] then
|
|
return "[]"
|
|
else
|
|
return "{}"
|
|
end
|
|
else
|
|
local oneline = nil
|
|
local _2_
|
|
do
|
|
local tbl_0_ = {}
|
|
for _, line in ipairs(lines) do
|
|
tbl_0_[(#tbl_0_ + 1)] = line:gsub("^%s+", "")
|
|
end
|
|
_2_ = tbl_0_
|
|
end
|
|
oneline = table.concat(_2_, " ")
|
|
if (not options["one-line?"] and (force_multi_line_3f or oneline:find("\n") or ((indent + #oneline) > options["line-length"]))) then
|
|
return table.concat(lines, ("\n" .. string.rep(" ", indent)))
|
|
else
|
|
return oneline
|
|
end
|
|
end
|
|
end
|
|
local function pp_metamethod(t, metamethod, options, indent)
|
|
if (options.level >= options.depth) then
|
|
if options["empty-as-sequence?"] then
|
|
return "[...]"
|
|
else
|
|
return "{...}"
|
|
end
|
|
else
|
|
local _ = nil
|
|
local function _2_(_241)
|
|
return visible_cycle_3f(_241, options)
|
|
end
|
|
options["visible-cycle?"] = _2_
|
|
_ = nil
|
|
local lines, force_multi_line_3f = metamethod(t, pp, options, indent)
|
|
options["visible-cycle?"] = nil
|
|
local _3_0 = type(lines)
|
|
if (_3_0 == "string") then
|
|
return lines
|
|
elseif (_3_0 == "table") then
|
|
return concat_lines(lines, options, indent, force_multi_line_3f)
|
|
else
|
|
local _0 = _3_0
|
|
return error("__fennelview metamethod must return a table of lines")
|
|
end
|
|
end
|
|
end
|
|
local function pp_table(x, options, indent)
|
|
options.level = (options.level + 1)
|
|
local x0 = nil
|
|
do
|
|
local _2_0 = nil
|
|
if options["metamethod?"] then
|
|
local _3_0 = x
|
|
if _3_0 then
|
|
local _4_0 = getmetatable(_3_0)
|
|
if _4_0 then
|
|
_2_0 = _4_0.__fennelview
|
|
else
|
|
_2_0 = _4_0
|
|
end
|
|
else
|
|
_2_0 = _3_0
|
|
end
|
|
else
|
|
_2_0 = nil
|
|
end
|
|
if (nil ~= _2_0) then
|
|
local metamethod = _2_0
|
|
x0 = pp_metamethod(x, metamethod, options, indent)
|
|
else
|
|
local _ = _2_0
|
|
local _4_0, _5_0 = table_kv_pairs(x)
|
|
if (true and (_5_0 == "empty")) then
|
|
local _0 = _4_0
|
|
if options["empty-as-sequence?"] then
|
|
x0 = "[]"
|
|
else
|
|
x0 = "{}"
|
|
end
|
|
elseif ((nil ~= _4_0) and (_5_0 == "table")) then
|
|
local kv = _4_0
|
|
x0 = pp_associative(x, kv, options, indent)
|
|
elseif ((nil ~= _4_0) and (_5_0 == "seq")) then
|
|
local kv = _4_0
|
|
x0 = pp_sequence(x, kv, options, indent)
|
|
else
|
|
x0 = nil
|
|
end
|
|
end
|
|
end
|
|
options.level = (options.level - 1)
|
|
return x0
|
|
end
|
|
local function number__3estring(n)
|
|
local _2_0 = string.gsub(tostring(n), ",", ".")
|
|
return _2_0
|
|
end
|
|
local function colon_string_3f(s)
|
|
return s:find("^[-%w?^_!$%&*+./@|<=>]+$")
|
|
end
|
|
local function pp_string(str, options, indent)
|
|
local escs = nil
|
|
local _2_
|
|
if (options["escape-newlines?"] and (#str < (options["line-length"] - indent))) then
|
|
_2_ = "\\n"
|
|
else
|
|
_2_ = "\n"
|
|
end
|
|
local function _4_(_241, _242)
|
|
return ("\\%03d"):format(_242:byte())
|
|
end
|
|
escs = setmetatable({["\""] = "\\\"", ["\11"] = "\\v", ["\12"] = "\\f", ["\13"] = "\\r", ["\7"] = "\\a", ["\8"] = "\\b", ["\9"] = "\\t", ["\\"] = "\\\\", ["\n"] = _2_}, {__index = _4_})
|
|
return ("\"" .. str:gsub("[%c\\\"]", escs) .. "\"")
|
|
end
|
|
local function make_options(t, options)
|
|
local defaults = {["detect-cycles?"] = true, ["empty-as-sequence?"] = false, ["escape-newlines?"] = false, ["line-length"] = 80, ["metamethod?"] = true, ["one-line?"] = false, ["prefer-colon?"] = false, ["utf8?"] = true, depth = 128}
|
|
local overrides = {appearances = count_table_appearances(t, {}), level = 0, seen = {len = 0}}
|
|
for k, v in pairs((options or {})) do
|
|
defaults[k] = v
|
|
end
|
|
for k, v in pairs(overrides) do
|
|
defaults[k] = v
|
|
end
|
|
return defaults
|
|
end
|
|
local function _2_(x, options, indent, colon_3f)
|
|
local indent0 = (indent or 0)
|
|
local options0 = (options or make_options(x))
|
|
local tv = type(x)
|
|
local function _4_()
|
|
local _3_0 = getmetatable(x)
|
|
if _3_0 then
|
|
return _3_0.__fennelview
|
|
else
|
|
return _3_0
|
|
end
|
|
end
|
|
if ((tv == "table") or ((tv == "userdata") and _4_())) then
|
|
return pp_table(x, options0, indent0)
|
|
elseif (tv == "number") then
|
|
return number__3estring(x)
|
|
else
|
|
local function _5_()
|
|
if (colon_3f ~= nil) then
|
|
return colon_3f
|
|
elseif ("function" == type(options0["prefer-colon?"])) then
|
|
return options0["prefer-colon?"](x)
|
|
else
|
|
return options0["prefer-colon?"]
|
|
end
|
|
end
|
|
if ((tv == "string") and colon_string_3f(x) and _5_()) then
|
|
return (":" .. x)
|
|
elseif (tv == "string") then
|
|
return pp_string(x, options0, indent0)
|
|
elseif ((tv == "boolean") or (tv == "nil")) then
|
|
return tostring(x)
|
|
else
|
|
return ("#<" .. tostring(x) .. ">")
|
|
end
|
|
end
|
|
end
|
|
pp = _2_
|
|
local function view(x, options)
|
|
return pp(x, make_options(x, options), 0)
|
|
end
|
|
return view
|