Space Station 14 Вики

Модуль:GetField: различия между версиями

Интерактивная навигация по истории
← Предыдущая правка
мНет описания правки
мНет описания правки
 
(не показано 13 промежуточных версий этого же участника)
Строка 1: Строка 1:
local p = {}
local p = {}


local function split(s, sep)
local cache = {}
if s == nil then return {} end
local entryCache = {}
local parts = {}
sep = sep or "%."
for part in string.gmatch(s, "([^" .. sep .. "]+)") do
table.insert(parts, part)
end
return parts
end


local function parse_indexed_part(part)
local function parse_indexed_part(part)
local key, idx = string.match(part, "^(.-)%[(%d+)%]$")
    local key, idx = string.match(part, "^(.-)%[(%d+)%]$")
if key then
    if key then
return key, tonumber(idx)
        return key, tonumber(idx)
end
    end
local num = tonumber(part)
    local num = tonumber(part)
if num then
    if num then
return nil, num
        return nil, num
end
    end
return part, nil
    return part, nil
end
end


local function get_by_path(tbl, path)
local function get_by_path(tbl, path)
if not tbl then return nil end
    if not tbl or path == "" then return nil end
local parts = split(path, "%.")
    local cur = tbl
local cur = tbl
    for part in string.gmatch(path, "([^%.]+)") do
for _, part in ipairs(parts) do
        local key, idx = parse_indexed_part(part)
local key, idx = parse_indexed_part(part)
        if key and key ~= "" then
if key and key ~= "" then
            if type(cur) ~= "table" then return nil end
local nextCur = nil
            local nextCur = cur[key]
if type(cur) == "table" then
            if nextCur == nil then
nextCur = cur[key]
                nextCur = cur["!type:" .. key]
if nextCur == nil then
            end
nextCur = cur["!type:" .. key]
            cur = nextCur
end
        end
end
        if idx then
cur = nextCur
            if type(cur) ~= "table" then return nil end
end
            cur = cur[idx]
if idx then
        end
if type(cur) ~= "table" then
        if cur == nil then return nil end
return nil
    end
end
    return cur
cur = cur[idx]
end
end
 
if cur == nil then
local function format_value(v)
return nil
    if v == nil then return "" end
end
    local t = type(v)
end
    if t == "string" or t == "number" or t == "boolean" then
return cur
        return tostring(v)
    elseif t == "table" then
        local max = 0
        local hasNonNumber = false
        for k in pairs(v) do
            if type(k) ~= "number" then
                hasNonNumber = true
                break
            else
                if k > max then max = k end
            end
        end
        if not hasNonNumber and max > 0 then
            local out = {}
            for i = 1, max do
                out[#out + 1] = format_value(v[i])
            end
            return table.concat(out, ", ")
        else
            local out = {}
            for k, val in pairs(v) do
                out[#out + 1] = tostring(k) .. ": " .. format_value(val)
            end
            return table.concat(out, ", ")
        end
    else
        return tostring(v)
    end
end
 
local function to_nowiki(v)
    local s = tostring(v)
    return "<nowiki>" .. s .. "</nowiki>"
end
end


local function is_array(t)
local function is_array(tbl)
if type(t) ~= "table" then return false end
    local max = 0
local i = 0
    local count = 0
for _ in pairs(t) do
    for k in pairs(tbl) do
i = i + 1
        if type(k) ~= "number" then
end
            return false
local n = 0
        end
for k in pairs(t) do
        if k > max then max = k end
if type(k) ~= "number" then
        count = count + 1
return false
    end
end
    return count > 0 and max == count
if k > n then n = k end
end
end
 
return n > 0
local function deep_copy(src)
    local dst = {}
    for k, v in pairs(src) do
        if type(v) == "table" then
            dst[k] = deep_copy(v)
        else
            dst[k] = v
        end
    end
    return dst
end
 
local function deep_merge(dst, src)
    for k, v in pairs(src) do
        if type(v) == "table" and type(dst[k]) == "table" then
            deep_merge(dst[k], v)
        elseif type(v) == "table" then
            dst[k] = deep_copy(v)
        else
            dst[k] = v
        end
    end
end
 
function p.findInProto(frame)
    local args = frame.args or {}
    local protoKey = args[1] or ""
    local fieldId = args[2] or ""
 
    if protoKey == "" or fieldId == "" then
        return "[]"
    end
 
    local moduleName = "Module:IanComradeBot/prototype.json/data"
 
    local data = cache[moduleName]
    if not data then
        local ok, loaded = pcall(mw.loadData, moduleName)
        if not ok or not loaded then
            return "[]"
        end
        data = loaded
        cache[moduleName] = data
    end
 
    local proto = data[protoKey]
    if type(proto) ~= "table" then
        return "[]"
    end
 
    local value = proto[fieldId]
    if value == nil then
        return "[]"
    end
 
    local out = {}
    local t = type(value)
    if t == "table" then
        for _, v in ipairs(value) do
            out[#out + 1] = v
        end
    else
        out[1] = value
    end
 
    return mw.text.jsonEncode(out)
end
end


local function format_value(v)
function p.flattenField(frame)
if v == nil then return "" end
    local args = frame.args or {}
local t = type(v)
    local id = args[1] or ""
if t == "string" or t == "number" or t == "boolean" then
    local pagePath = args[2] or ""
return tostring(v)
    if id == "" or pagePath == "" then return "" end
elseif t == "table" then
 
if is_array(v) then
    local baseUser = "IanComradeBot/"
local out = {}
    local moduleName = "Module:" .. baseUser .. pagePath .. "/data"
local max = 0
 
for k in pairs(v) do if type(k) == "number" and k > max then max = k end end
    local data = cache[moduleName]
for i = 1, max do
    if not data then
table.insert(out, format_value(v[i]))
        local ok, loaded = pcall(mw.loadData, moduleName)
end
        if not ok or not loaded then return "" end
return table.concat(out, ", ")
        data = loaded
else
        cache[moduleName] = data
local out = {}
    end
for k, val in pairs(v) do
 
table.insert(out, tostring(k) .. ": " .. format_value(val))
    local entry = data[id]
end
 
return table.concat(out, ", ")
    if entry == nil then
end
        local idsTable = data.id
else
        if type(idsTable) == "table" then
return tostring(v)
            local specific = idsTable[id]
end
            if type(specific) == "table" then
                local base = data["default"]
                if type(base) == "table" then
                    local merged = deep_copy(base)
                    deep_merge(merged, specific)
                    entry = merged
                else
                    entry = deep_copy(specific)
                end
            end
        end
    end
 
    if entry == nil then
        local base = data["default"]
        if type(base) == "table" then
            entry = deep_copy(base)
        else
            entry = {}
        end
    end
 
    if type(entry) ~= "table" then return "" end
 
    local parts = {}
    local function walk(tbl, prefix)
        local keys = {}
        for k in pairs(tbl) do keys[#keys + 1] = k end
        table.sort(keys, function(a, b) return tostring(a) < tostring(b) end)
        for _, k in ipairs(keys) do
            local v = tbl[k]
            local key = (prefix == "" and tostring(k) or prefix .. "." .. tostring(k))
            if type(v) == "table" then
                if is_array(v) then
                    local ok, json = pcall(mw.text.jsonEncode, v)
                    if ok and json then
                        parts[#parts + 1] = key .. "=" .. to_nowiki(json)
                    end
                else
                    local ok, json = pcall(mw.text.jsonEncode, v)
                    if ok and json then
                        parts[#parts + 1] = key .. "=" .. to_nowiki(json)
                    end
                    walk(v, key)
                end
            else
                parts[#parts + 1] = key .. "=" .. to_nowiki(v)
            end
        end
    end
 
    walk(entry, "")
 
    return table.concat(parts, "|")
end
end


function p.get(frame)
function p.get(frame)
local args = frame.args or {}
    local args = frame.args or {}
local id = args[1] or ""
    local id = args[1] or ""
local pagePath = args[2] or ""
    local pagePath = args[2] or ""
local keyPath = args[3] or ""
    local keyPath = args[3] or ""


local baseUser = "IanComradeBot/"
    if pagePath == "" then return "" end
local moduleName = "Module:" .. baseUser .. pagePath .. "/data"


local ok, data = pcall(mw.loadData, moduleName)
    local baseUser = "IanComradeBot/"
if not ok or not data then
    local moduleName = "Module:" .. baseUser .. pagePath .. "/data"
return ""
end


local entry = nil
    local data = cache[moduleName]
if id ~= "" then
    if not data then
entry = data[id]
        local ok, loaded = pcall(mw.loadData, moduleName)
end
        if not ok or not loaded then return "" end
if entry == nil then
        data = loaded
entry = data["default"]
        cache[moduleName] = data
end
    end
if entry == nil then
return ""
end


if keyPath == "" then
    local entryKey = moduleName .. "|" .. (id ~= "" and id or "default")
return format_value(entry)
    local entry = entryCache[entryKey]
end
    if not entry then
        if id ~= "" then entry = data[id] end
        if entry == nil then entry = data["default"] end
        entryCache[entryKey] = entry
    end
    if entry == nil then return "" end


local value = get_by_path(entry, keyPath)
    if keyPath == "" then
return format_value(value)
        return format_value(entry)
    end
 
    local value = get_by_path(entry, keyPath)
    return format_value(value)
end
end


function p.flattenField(frame)
function p.getTpl(frame)
local args = frame.args or {}
    local args = frame.args or {}
local id = args[1] or ""
    local id = args[1] or ""
local pagePath = args[2] or ""
    local pagePath = args[2] or ""
if id == "" or pagePath == "" then return "" end
    local tplPath = args[3] or ""
 
    if id == "" or pagePath == "" or tplPath == "" then
        return ""
    end
 
    local extra = p.flattenField({ args = { id, pagePath } }) or ""
    local tplStr = "{{" .. tostring(tplPath) .. "|id=" .. tostring(id)
    if extra ~= "" then
        tplStr = tplStr .. "|" .. extra
    end
    tplStr = tplStr .. "}}"
 
    if type(frame.preprocess) == "function" then
        return frame:preprocess(tplStr)
    end
 
    return tplStr
end
 
function p.getTplProto(frame)
    local args = frame.args or {}
    local searchId = args[1] or ""
    local protoId = args[2] or ""
    local tplPath = args[3] or ""
    local pagePath = "prototype/" .. protoId .. ".json"


local baseUser = "IanComradeBot/"
    if searchId == "" or protoId == "" or tplPath == "" then
local moduleName = "Module:" .. baseUser .. pagePath .. "/data"
        return ""
local ok, data = pcall(mw.loadData, moduleName)
    end
if not ok or not data then
return ""
end


local entry = data[id] or data["default"] or {}
    local idsJson = p.findInProto({ args = { searchId, protoId } })
    local ok, ids = pcall(mw.text.jsonDecode, idsJson or "")
    if not ok or type(ids) ~= "table" or #ids == 0 then
        return ""
    end


local flat = {}
    local out = {}
local function walk(tbl, prefix)
    for _, id in ipairs(ids) do
if type(tbl) ~= "table" then
        local tpl = p.getTpl({ args = { id, pagePath, tplPath } })
flat[prefix] = format_value(tbl)
        if tpl ~= "" then
return
            out[#out + 1] = tpl
end
        end
if prefix ~= "" then
    end
flat[prefix] = format_value(tbl)
end
if is_array(tbl) then
local max = 0
for k in pairs(tbl) do if type(k) == "number" and k > max then max = k end end
for i = 1, max do
local v = tbl[i]
local key = (prefix == "" and tostring(i) or prefix .. "." .. tostring(i))
if type(v) == "table" then
walk(v, key)
else
flat[key] = format_value(v)
end
end
else
for k, v in pairs(tbl) do
local key = (prefix == "" and tostring(k) or prefix .. "." .. tostring(k))
if type(v) == "table" then
walk(v, key)
else
flat[key] = format_value(v)
end
end
end
end


walk(entry, "")
    local result = table.concat(out, "\n")
    if type(frame.preprocess) == "function" then
        return frame:preprocess(result)
    end


local parts = {}
    return result
for k, v in pairs(flat) do
table.insert(parts, tostring(k) .. "=" .. tostring(v))
end
return table.concat(parts, "|")
end
end


return p
return p
Источник — https://station14.ru/wiki/Модуль:GetField