Data reader/saver

-- Загрузка всех нужных функций
local string = require "string"
local global = require "_G"
local table = require "table"
local table = require "table"
local coroutine = require "coroutine"
local open = require "io".open
local floor = require "math".floor
local error = global.error
local type = global.type
local load = global.load
local tonumber = global.tonumber
local setmetatable = global.setmetatable
local next = global.next
local ipairs = global.ipairs
local pcall = global.pcall
local format, match, sub, gsub, find, gmatch = string.format, string.match, string.sub, string.gsub, string.find, string.gmatch
local wrap, yield = coroutine.wrap, coroutine.yield
local insert, concat, remove = table.insert, table.concat, table.remove
string = nil
coroutine = nil
table = nil

local unserialize
function unserialize(value, n) -- Принимает значения ключей в файле в виде строки и возвращает значения Lua
    local t = sub(value, #value)
    if t == 'b' then -- Определение типа boolean
        return true, value ~= "0b"
    elseif match(value, "^[\"'](.+)[\"']$") then -- Определение типа строки
        local f = load("return "..value)
        if not (f and pcall(f)) then
            return f, "#"..n..": not a string format"
        end
        return true, f()
    elseif t == 'n' then -- Определение типа число
        local a = tonumber(sub(value, 1, #value-1))
        if not a then
            return a, "#"..n..": not a number format"
        end
        return true, a
    elseif t == 'x' then -- Определение типа шестнадцатиричное число
        local a = tonumber("0x"..sub(value, 1, #value-1))
        if not a then
            return a, "#"..n..": not a hex number format"
        end
        return true, a/1000
    elseif match(value, "^{(.-)}$") then -- Определение типа массив
        value = sub(value, 2, #value-1)
        local k = 1
        local a = { }
        local c, b = 0, ""
        for value in gmatch(value, "([^;,]+)") do
            if find(value, "{") then
                c = c + 1
            end
            if find(value, "}") then
                c = c - 1
            end
            if c == 0 then
                b = b..value
                b = match(b, "^%s*(.-)%s*$")
                local ok, err = unserialize(b, k)
                if not ok then
                    return ok, "#"..n..": in array field "..err
                end
                a[k] = err
                k = k+1
                b = ""
            else
                b = b..value..","
            end
        end
        return true, a
    elseif value == "" then -- Определение типа ничего
        return true, nil
    end
    return nil, "#"..n..": couldn't read unregistered type"
end

local serialize
function serialize(value, n, hex) -- Делает обратную работу функции unserialize. Аргумент hex определяет в какую СС будет кодироваться число.
    local t = type(value)
    if t == "boolean" then -- Определение типа boolean
        return true, value == true and "1b" or "0b"
    elseif t == "string" then -- Определение типа строки
        return true, '"'..value:gsub("\\", "\\\\"):gsub("\t", "\\t"):gsub("\b", "\\b"):gsub("\n", "\\n"):gsub("\"", "\\\"")..'"'
    elseif t == "number" then -- Определение типа число
        if hex then
            return true, format("%xx", floor(value*1000))
        else
            return true, value == floor(value) and format("%dn", value) or format("%.3fn", value)
        end
    elseif t == "table" then -- Определение типа массив
        local s = { }
        for key, value in ipairs(value) do
            local ok, err = serialize(value, key, hex)
            if not ok then
                return ok, err
            end
            s[#s+1] = err
        end
        s = "{"..concat(s, ", ").."}"
        return true, s
    elseif t == "nil" then -- Определение типа ничего
        return true, ""
    end
    return nil, "#"..n..": couldn't code unregistered type"
end

local function decode(String, n) -- Расшифровывает одну строку из файла
    local key, value, comment
    do -- Ниже можно увидеть адские шаблоны для строк
        local a, b = match(String, ".+=.-[\"'].+[\"'].-#()(.+)$")
        if not a then
            a, b = match(String, "#()(.+)")
        end
        if a then
            String = sub(String, 1, a-2)
            comment = b
        end
    end
    key, value = match(String, "^%s*(.-)%s*=%s*(.-)%s*$")
    if not key then
        key, value = match(String, "^%s*(.-)%s*="), ""
    end
    local ok = true
    if key then
        ok, value = unserialize(value, n)
    else value = nil
    end
    if not ok then
        return ok, value, nil, comment
    end
    return true, key, value, comment
end

local function code(key, value, comment, hex, n) -- Конвертирует значения Lua в строку с настройкой
    if key then
        local ok, a = serialize(value, n, hex)
        if not ok then
            return ok, a
        end
        return true, key.." = "..a..(comment and "  #"..comment or "")
    elseif comment then
        return true, "#"..comment
    end
end

flush = function(self, hex) -- Сохраняет данные в файл
    local file, err = open(self.path, "w")
    if not file then
        return file, err
    end
    local keys, values, comments, n = self.keys, self.values, self.comments, self.n
    for i = 1, n do
        local ok, value = code(keys[i], values[i], comments[i], hex, i)
        if ok then
            file:write(value)
        end
        if i ~= n then
            file:write("\n")
        end
        file:flush()
    end
    file:close()
end

-- Метатаблица для объекта с настройками
local meta = {
    __index = {
        object = "settings"; -- Для моих будующих программ может быть полезно
        save = flush;
        totable = function(self) -- Трансформирует объект в обычную таблицу Lua
            local keys, values, t = self.keys, self.values, { }
            for i = 1, self.n do
                if keys[i] then
                    t[keys[i]] = values[i]
                end
            end
            return t
        end;
        setLine = function(self, key, value, i) -- Устанавливает строку со значением
            local keys, values = self.keys, self.values
            i = i or self.n
            if type(i) ~= "number" then
                error("bad argument #3 to method 'setLine' (number expected, got "..type(i)..")", 2)
            end
            self.n = self.n < i and i or self.n
            if key == nil then
                keys[i] = nil
                values[i] = nil
                return
            end
            if type(key) == "string" then
                keys[i] = key
                values[i] = value
                return
            end
            error("bad argument #1 to method 'setLine' (string expected, got "..type(key)..")", 2)
        end;
        getLine = function(self, i) -- Возвращает всё, что храница в строке
            if type(i) ~= "number" then
                error("bad argument to method 'setLine' (number expected, got "..type(i)..")", 2)
            end
            return self.keys[i], self.values[i], self.comments[i]
        end;
        set = function(self, key, value) -- Устанавливает ключу значение
            local keys, values, n = self.keys, self.values, self.n
            if type(key) ~= "string" then
                error("bad argument #1 to method 'setLine' (string expected, got "..type(key)..")", 2)
            end
            for i = 1, n do
                if key == keys[i] then
                    values[i] = value
                    return
                end
            end
            n = n+1
            keys[n] = key
            values[n] = value
            self.n = n
        end;
        get = function(self, key) -- Возвращает значение ключа
            local keys, values, n = self.keys, self.values, self.n
            if type(key) ~= "string" then
                error("bad argument to method 'setLine' (string expected, got "..type(key)..")", 2)
            end
            for i = 1, n do
                if key == keys[i] then
                    return values[i]
                end
            end
            return nil
        end;
        setComment = function(self, i, comment) -- Устанавливает комментарий строки
            if type(i) ~= "number" then
                error("bad argument #1 to method 'setLine' (number expected, got "..type(i)..")", 2)
            end
            if type(comment) ~= "string" and comment ~= nil then
                error("bad argument #2 to method 'setLine' (string expected, got "..type(comment)..")", 2)
            end
            self.comments[i] = comment
        end;
        getComment = function(self, i) -- Возвращает комментарий строки
            if type(i) ~= "number" then
                error("bad argument to method 'setLine' (number expected, got "..type(i)..")", 2)
            end
            return self.comments[i]
        end;
    };
    __pairs = function(self) -- Возвращает итератор, который проходит по всем элементам
        local keys, values, comments = self.keys, self.values, self.comments
        return wrap(function()
            for i = 1, self.n do
                if keys[i] then
                    yield(keys[i], values[i], comments[i])
                end
            end
        end)
    end;
    __ipairs = function(self) -- Возвращает итератор, который проходит по всем строкам
        local keys, values, comments = self.keys, self.values, self.comments
        return wrap(function()
            for i = 1, self.n do
                yield(i, keys[i], values[i], comments[i])
            end
        end)
    end;
    __len = function(self) -- Возвращает количество строк, и обнавляет поле 'n' в объекте
        local n = 0
        for key, value in next, self.comments do
            n = n < key and key or n
        end
        for key, value in next, self.keys do
            n = n < key and key or n
        end
        for key, value in next, self.values do
            n = n < key and key or n
        end
        self.n = n
        return n
    end;
}

return {
    open = function(path) -- Возвращает объект, полученный из просканированного файла
        do
            local path = type(path)
            if path ~= "string" then
                error("bad argument (string expected, got "..path..")", 2)
            end
        end
        local file, err = open(path)
        if not file then
            return nil, err
        end
        local object = {
            path = path;
            keys = { };
            values = { };
            comments = { };
        }
        local keys, values, comments = object.keys, object.values, object.comments
        local n = 0
        for line in file:lines() do
            n = n + 1
            local ok, key, value, comment = decode(line, n)
            comments[n] = comment
            if key then
                keys[n] = key
                values[n] = value
            end
        end
        object.n = n
        return setmetatable(object, meta)
    end;
    convert = function(path, t) -- Конвертирует обычную таблицу Lua в объект настроек
        t = t or { }
        do
            local _t = type(path)
            if _t ~= "string" then
                error("bad argument #1 (string expected, got ".._t..")", 2)
            end
            _t = type(t)
            if _t ~= "table" then
                error("bad argument #2 (table/nil expected, got ".._t..")", 2)
            end
        end
        local i, keys, values = 0, { }, { }
        for key, value in next, t do
            i = i + 1
            if type(key) == "string" then
                keys[i] = key
                values[i] = value
            end
        end
        return setmetatable({
            n = i;
            path = path;
            keys = keys;
            values = values;
            comments = { };
        }, meta)
    end;
    decode = function(string, i) -- Декодер для пользователя
        if type(string) ~= "string" then
            error("bad argument (string expected, got "..type(string)..")", 2)
        end
        return decode(string, i or 1)
    end;
    code = function(key, value, comment, hex, i)
        if type(key) ~= "string" then
            error("bad argument #1 (string expected, got "..type(key)..")", 2)
        end
        if type(comment) ~= "string" then
            error("bad argument #3 (string expected, got "..type(comment)..")", 2)
        end
        return code(key, value, comment, hex, i or 1)
    end;
}