Space Station 14 Вики

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

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


local function split(s, sep)
local cache = {}
    if s == nil then return {} end
local entryCache = {}
    local parts = {}
local BASE_USER = "IanComradeBot/"
    sep = sep or "%."
 
     for part in string.gmatch(s, "([^" .. sep .. "]+)") do
local function get_module_name(pagePath)
         table.insert(parts, part)
     return "Module:" .. BASE_USER .. pagePath .. "/data"
end
 
local function load_cached_data(moduleName)
    local data = cache[moduleName]
    if data then
         return data
    end
 
    local ok, loaded = pcall(mw.loadData, moduleName)
    if not ok or not loaded then
        return nil
     end
     end
     return parts
 
    cache[moduleName] = loaded
     return loaded
end
end


Строка 24: Строка 37:


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 ipairs(parts) do
     for part in string.gmatch(path, "([^%.]+)") 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
            local nextCur = nil
             if type(cur) ~= "table" then return nil end
             if type(cur) == "table" then
            local nextCur = cur[key]
                nextCur = cur[key]
            if nextCur == nil then
                if nextCur == nil then
                nextCur = cur["!type:" .. key]
                    nextCur = cur["!type:" .. key]
                end
             end
             end
             cur = nextCur
             cur = nextCur
         end
         end
         if idx then
         if idx then
             if type(cur) ~= "table" then
             if type(cur) ~= "table" then return nil end
                return nil
            end
             cur = cur[idx]
             cur = cur[idx]
         end
         end
         if cur == nil then
         if cur == nil then return nil end
            return nil
        end
     end
     end
     return cur
     return cur
end
end


local function is_array(t)
local function format_value(v)
     if type(t) ~= "table" then return false end
     local okJson, json = pcall(mw.text.jsonEncode, v)
     local i = 0
    if okJson and json == "null" then
     for _ in pairs(t) do
        return "null"
         i = i + 1
    end
 
    if v == nil then return "" end
 
     local t = type(v)
     if t == "string" or t == "number" or t == "boolean" then
        return tostring(v)
    elseif t == "table" then
        local ok, json2 = pcall(mw.text.jsonEncode, v)
         if ok and json2 then
            return json2
        end
        return ""
    else
        return tostring(v)
     end
     end
     local n = 0
end
     for k in pairs(t) do
 
local function to_nowiki(v)
    return "<nowiki>" .. v .. "</nowiki>"
end
 
local function is_array(tbl)
    local max = 0
     local count = 0
     for k in pairs(tbl) do
         if type(k) ~= "number" then
         if type(k) ~= "number" then
             return false
             return false
         end
         end
         if k > n then n = k end
         if k > max then max = k end
        count = count + 1
    end
    return count > 0 and max == count
end
 
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
 
local function resolve_entry(data, id)
    if type(data) ~= "table" then
        return nil
    end
 
    if id and id ~= "" then
        local direct = data[id]
        if direct ~= nil then
            return direct
        end
 
        local idsTable = data.id
        if type(idsTable) == "table" then
            local specific = idsTable[id]
            if type(specific) == "table" then
                local base = data["default"]
                if type(base) == "table" then
                    local merged = deep_copy(base)
                    deep_merge(merged, specific)
                    return merged
                end
                return deep_copy(specific)
            end
        end
    end
 
    local base = data["default"]
    if type(base) == "table" then
        return deep_copy(base)
     end
     end
     return n > 0
     return nil
end
end


local function format_value(v)
local function contains_target(v, target)
     if v == nil then return "" end
     if type(v) == "table" then
    local t = type(v)
    if t == "string" or t == "number" or t == "boolean" then
        return tostring(v)
    elseif t == "table" then
         if is_array(v) then
         if is_array(v) then
            local out = {}
             for _, item in ipairs(v) do
            local max = 0
                if tostring(item) == target then
             for k in pairs(v) do if type(k) == "number" and k > max then max = k end end
                    return true
             for i = 1, max do
                end
                table.insert(out, format_value(v[i]))
            end
             return false
        end
 
        for _, item in pairs(v) do
            if tostring(item) == target then
                return true
             end
             end
            return table.concat(out, ", ")
        end
        else
        return false
            local out = {}
    end
            for k, val in pairs(v) do
 
                table.insert(out, tostring(k) .. ": " .. format_value(val))
    return tostring(v) == target
end
 
local function find_matching_ids(idsTable, keyPath, searchValue)
    local target = tostring(searchValue)
    local matches = {}
 
    for idKey, entry in pairs(idsTable) do
        if type(entry) == "table" then
            local v = get_by_path(entry, keyPath)
            if v ~= nil and contains_target(v, target) then
                matches[#matches + 1] = idKey
             end
             end
            return table.concat(out, ", ")
        end
    end
 
    return matches
end
 
local function preprocess_or_return(frame, text)
    if type(frame.preprocess) == "function" then
        return frame:preprocess(text)
    end
    return text
end
 
local function get_field_loose(entry, fieldId)
    local value = entry[fieldId]
    if value ~= nil then return value end
    if fieldId == "" then return nil end
 
    local first = string.sub(fieldId, 1, 1)
    local tail = string.sub(fieldId, 2)
    value = entry[string.lower(first) .. tail]
    if value ~= nil then return value end
 
    return entry[string.upper(first) .. tail]
end
 
function p.findInGenerator(frame)
    local args = frame.args or {}
    local searchId = args[1] or ""
    local kind = (args[2] or ""):lower()
    local fieldId = args[3] or ""
 
    if searchId == "" or fieldId == "" then
        return ""
    end
    if kind ~= "prototype" and kind ~= "component" then
        return ""
    end
 
    local storeName = (kind == "prototype") and "prototype_store.json" or "component_store.json"
    local moduleName = get_module_name(storeName)
    local data = load_cached_data(moduleName)
    if not data then
        return ""
    end
 
    local entry = data[searchId]
    if type(entry) ~= "table" then
        return ""
    end
 
    local value = get_field_loose(entry, 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
         end
     else
     else
         return tostring(v)
         out[1] = value
    end
 
    return mw.text.jsonEncode(out)
end
 
function p.flattenField(frame)
    local args = frame.args or {}
    local id = args[1] or ""
    local pagePath = args[2] or ""
    if id == "" or pagePath == "" then return "" end
 
    local moduleName = get_module_name(pagePath)
    local data = load_cached_data(moduleName)
    if not data then return "" end
 
    local entry = resolve_entry(data, id) or {}
 
    if type(entry) ~= "table" then return "" end
 
    local parts = {}
    local function append_table_json(key, value)
        local ok, json = pcall(mw.text.jsonEncode, value)
        if ok and json then
            parts[#parts + 1] = key .. "=" .. to_nowiki(json)
        end
    end
 
    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 kStr = tostring(k)
            local key = (prefix == "" and kStr or prefix .. "." .. kStr)
            if type(v) == "table" then
                if next(v) == nil then
                else
                    append_table_json(key, v)
                    if not is_array(v) then
                        walk(v, key)
                    end
                end
            else
                parts[#parts + 1] = key .. "=" .. tostring(v)
            end
        end
     end
     end
    walk(entry, "")
    return table.concat(parts, "|")
end
end


Строка 100: Строка 310:
     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 moduleName = get_module_name(pagePath)
    local data = load_cached_data(moduleName)
    if not data then return "" end
 
    local entryKey = moduleName .. "|" .. (id ~= "" and id or "default")
    local entry = entryCache[entryKey]
    if not entry then
        entry = resolve_entry(data, id)
        entryCache[entryKey] = entry
    end
    if entry == nil then return "" end
 
    if keyPath == "" then
        return format_value(entry)
    end
 
    local value = get_by_path(entry, keyPath)
    return format_value(value)
end
 
function p.getId(frame)
    local args = frame.args or {}
    local searchValue = args[1] or ""
    local pagePath = args[2] or ""
    local keyPath = args[3] or ""
 
    if searchValue == "" or pagePath == "" or keyPath == "" then
        return ""
    end
 
    local moduleName = get_module_name(pagePath)
    local data = load_cached_data(moduleName)
    if not data then return "[]" end


     local ok, data = pcall(mw.loadData, moduleName)
     local idsTable = data.id
     if not ok or not data then
     if type(idsTable) ~= "table" then
         return ""
         return ""
     end
     end


     local entry = nil
     local matches = find_matching_ids(idsTable, keyPath, searchValue)
     if id ~= "" then
 
         entry = data[id]
     if #matches == 0 then
        return ""
    end
 
    local ok, json = pcall(mw.text.jsonEncode, matches)
    if ok and json then
         return json
    end
 
    return ""
end
 
function p.getTplId(frame)
    local args = frame.args or {}
    local searchValue = args[1] or ""
    local pagePath = args[2] or ""
    local keyPath = args[3] or ""
    local tplPath = mw.text.unstripNoWiki(args[4] or "")
 
    if searchValue == "" or pagePath == "" or keyPath == "" or tplPath == "" then
        return ""
     end
     end
     if entry == nil then
 
         entry = data["default"]
    local moduleName = get_module_name(pagePath)
    local data = load_cached_data(moduleName)
     if not data then return "" end
 
    local idsTable = data.id
    if type(idsTable) ~= "table" then
         return ""
     end
     end
     if entry == nil then
 
    local matches = find_matching_ids(idsTable, keyPath, searchValue)
 
     if #matches == 0 then
         return ""
         return ""
     end
     end


     if keyPath == "" then
     local out = {}
         return format_value(entry)
    for _, idKey in ipairs(matches) do
        local tpl = p.getTpl({ args = { idKey, pagePath, tplPath } })
        if tpl ~= "" then
            out[#out + 1] = tpl
        end
    end
 
    if #out == 0 then
         return ""
     end
     end


     local value = get_by_path(entry, keyPath)
     local result = table.concat(out, " ")
     return format_value(value)
     return preprocess_or_return(frame, result)
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 .. "}}"
 
    return preprocess_or_return(frame, tplStr)
end
 
function p.getTplGenerator(frame)
    local args = frame.args or {}
    local searchId = args[1] or ""
    local kind = (args[2] or ""):lower()
    local generatorId = args[3] or ""
    local tplPath = args[4] or ""


     local baseUser = "IanComradeBot/"
     if searchId == "" or generatorId == "" 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
     if kind ~= "prototype" and kind ~= "component" then
         return ""
         return ""
     end
     end


     local entry = data[id] or data["default"] or {}
     local dir = (kind == "prototype") and "prototype/" or "component/"
    local pagePath = dir .. generatorId .. ".json"
 
    local idsJson = p.findInGenerator({ args = { searchId, kind, generatorId } })
    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
        if prefix ~= "" then
            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
     end
     end


     walk(entry, "")
     local result = table.concat(out, " ")
    return preprocess_or_return(frame, result)
end
 
function p.getGenerator(frame)
    local args = frame.args or {}
    local searchId = args[1] or ""
    local kind = (args[2] or ""):lower()
    local generatorId = args[3] or ""
 
    if searchId == "" or generatorId == "" then
        return ""
    end
    if kind ~= "prototype" and kind ~= "component" then
        return ""
    end
 
    local idsJson = p.findInGenerator({ args = { searchId, kind, generatorId } })
    local ok, ids = pcall(mw.text.jsonDecode, idsJson or "")
    if not ok or type(ids) ~= "table" or #ids == 0 then
        return ""
    end
 
    local okOut, outJson = pcall(mw.text.jsonEncode, ids)
    if okOut and outJson then
        return outJson
    end
 
    return ""
end
 
function p.hasComp(frame)
    local args = frame.args or {}
    local entityId = args[1] or ""
    local compName = args[2] or ""
 
    if entityId == "" or compName == "" then
        return "false"
    end
 
    local moduleName = get_module_name("component.json")
    local data = load_cached_data(moduleName)
    if not data then
        return "false"
    end
 
    if type(data) ~= "table" then
        return "false"
    end
 
    local entry = data[entityId]
    if type(entry) ~= "table" then
        return "false"
    end


     local parts = {}
     local target = tostring(compName)
     for k, v in pairs(flat) do
     for _, v in ipairs(entry) do
         table.insert(parts, tostring(k) .. "=" .. tostring(v))
         if tostring(v) == target then
            return "true"
        end
     end
     end
     return table.concat(parts, "|")
 
     return "false"
end
end


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