modlib/luon.lua

187 lines
5.4 KiB
Lua
Raw Normal View History

local assert, next, pairs, pcall, error, type, table_insert, table_concat, string_format, string_match, setmetatable, select, setfenv, math_huge, loadfile, loadstring
= assert, next, pairs, pcall, error, type, table.insert, table.concat, string.format, string.match, setmetatable, select, setfenv, math.huge, loadfile, loadstring
local count_objects = modlib.table.count_objects
2021-07-06 21:56:20 +02:00
-- Build a table with the succeeding character from A-Z
local succ = {}
for char = ("A"):byte(), ("Z"):byte() - 1 do
succ[string.char(char)] = string.char(char + 1)
end
local function quote(string)
return string_format("%q", string)
end
local _ENV = {}
setfenv(1, _ENV)
local metatable = {__index = _ENV}
_ENV.metatable = metatable
2021-07-06 21:56:20 +02:00
function new(self)
return setmetatable(self, metatable)
end
function aux_write(_self, _object)
-- returns reader, arguments
return
end
aux_read = {}
function write(self, value, write)
2021-07-06 21:56:20 +02:00
local reference = {"A"}
local function increment_reference(place)
if not reference[place] then
reference[place] = "B"
elseif reference[place] == "Z" then
reference[place] = "A"
return increment_reference(place + 1)
else
reference[place] = succ[reference[place]]
end
end
local references = {}
local to_fill = {}
-- TODO sort objects by count, give frequently referenced objects shorter references
2021-07-07 18:49:14 +02:00
for object, count in pairs(count_objects(value)) do
local type_ = type(object)
2021-07-07 20:02:11 +02:00
if count >= 2 and (type_ ~= "string" or #reference + 2 < #object) then
2021-07-06 21:56:20 +02:00
local ref = table_concat(reference)
write(ref)
write"="
2021-07-07 18:49:14 +02:00
write(type_ == "table" and "{}" or quote(object))
2021-07-06 21:56:20 +02:00
write";"
2021-07-07 18:49:14 +02:00
references[object] = ref
2021-07-06 21:56:20 +02:00
if type_ == "table" then
to_fill[object] = ref
2021-07-06 21:56:20 +02:00
end
increment_reference(1)
end
end
local function is_short_key(key)
2021-07-06 23:43:03 +02:00
return not references[key] and type(key) == "string" and string_match(key, "^[%a_][%a%d_]*$")
2021-07-06 21:56:20 +02:00
end
local function dump(value)
-- Primitive types
if value == nil then
return write"nil"
end
if value == true then
return write"true"
end
if value == false then
return write"false"
end
local type_ = type(value)
if type_ == "number" then
2021-07-06 23:43:03 +02:00
return write(string_format("%.17g", value))
2021-07-06 21:56:20 +02:00
end
-- Reference types: table and string
local ref = references[value]
if ref then
-- Referenced
return write(ref)
2021-07-06 21:56:20 +02:00
elseif type_ == "string" then
return write(quote(value))
elseif type_ == "table" then
local first = true
write"{"
local len = #value
for i = 1, len do
if not first then write";" end
dump(value[i])
first = false
end
for k, v in next, value do
if type(k) ~= "number" or k % 1 ~= 0 or k < 1 or k > len then
if not first then write";" end
if is_short_key(k) then
write(k)
else
write"["
dump(k)
write"]"
end
write"="
dump(v)
first = false
end
end
write"}"
else
-- TODO move aux_write to start, to allow dealing with metatables etc.?
(function(func, ...)
-- functions are the only way to deal with varargs
if not func then
return error("unsupported type: " .. type_)
end
write(func)
write"("
local n = select("#", ...)
local args = {...}
for i = 1, n - 1 do
dump(args[i])
write","
end
if n > 0 then
write(args[n])
end
write")"
end)(self:aux_write(value))
2021-07-06 21:56:20 +02:00
end
end
for table, ref in pairs(to_fill) do
for k, v in pairs(table) do
write(ref)
if is_short_key(k) then
write"."
write(k)
else
write"["
dump(k)
write"]"
end
write"="
dump(v)
write";"
end
2021-07-06 21:56:20 +02:00
end
write"return "
dump(value)
2021-07-06 21:56:20 +02:00
end
function write_file(self, value, file)
return self:write(value, function(text)
2021-07-06 21:56:20 +02:00
file:write(text)
end)
end
function write_string(self, value)
2021-07-06 21:56:20 +02:00
local rope = {}
self:write(value, function(text)
2021-07-06 21:56:20 +02:00
table_insert(rope, text)
end)
return table_concat(rope)
end
function read(self, ...)
2021-07-06 21:56:20 +02:00
local read = assert(...)
-- math.huge is serialized to inf, 0/0 is serialized to -nan
setfenv(read, setmetatable({inf = math_huge, nan = 0/0}, {__index = self.aux_read}))
2021-07-06 21:56:20 +02:00
local success, value_or_err = pcall(read)
if success then
return value_or_err
end
return nil, value_or_err
end
function read_file(self, path)
return self:read(loadfile(path))
2021-07-06 21:56:20 +02:00
end
function read_string(self, string)
return self:read(loadstring(string))
2021-07-06 21:56:20 +02:00
end
return _ENV