Space Station 14 Вики

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

Интерактивная навигация по истории
← Предыдущая правка
мНет описания правки
мНет описания правки
 
(не показано 18 промежуточных версий этого же участника)
Строка 3: Строка 3:
local cache = {}
local cache = {}
local entryCache = {}
local entryCache = {}
local BASE_USER = "IanComradeBot/"
local function get_module_name(pagePath)
    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
    cache[moduleName] = loaded
    return loaded
end


local function parse_indexed_part(part)
local function parse_indexed_part(part)
Строка 39: Строка 59:


local function format_value(v)
local function format_value(v)
    local okJson, json = pcall(mw.text.jsonEncode, v)
    if okJson and json == "null" then
        return "null"
    end
     if v == nil then return "" end
     if v == nil then return "" end
     local t = type(v)
     local t = type(v)
     if t == "string" or t == "number" or t == "boolean" then
     if t == "string" or t == "number" or t == "boolean" then
         return tostring(v)
         return tostring(v)
     elseif t == "table" then
     elseif t == "table" then
         local max = 0
         local ok, json2 = pcall(mw.text.jsonEncode, v)
        local hasNonNumber = false
         if ok and json2 then
        for k in pairs(v) do
             return json2
            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
         end
        return ""
     else
     else
         return tostring(v)
         return tostring(v)
Строка 73: Строка 81:


local function to_nowiki(v)
local function to_nowiki(v)
    local s = tostring(v)
     return "<nowiki>" .. v .. "</nowiki>"
     return "<nowiki>" .. s .. "</nowiki>"
end
end


Строка 147: Строка 154:
end
end


function p.findInProto(frame)
local function contains_target(v, target)
    if type(v) == "table" then
        if is_array(v) then
            for _, item in ipairs(v) do
                if tostring(item) == target then
                    return true
                end
            end
            return false
        end
 
        for _, item in pairs(v) do
            if tostring(item) == target then
                return true
            end
        end
        return false
    end
 
    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
    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 args = frame.args or {}
     local protoKey = args[1] or ""
     local searchId = args[1] or ""
     local fieldId = args[2] or ""
    local kind = (args[2] or ""):lower()
     local fieldId = args[3] or ""


     if protoKey == "" or fieldId == "" then
     if searchId == "" or fieldId == "" then
        return ""
    end
    if kind ~= "prototype" and kind ~= "component" then
         return ""
         return ""
     end
     end


     local moduleName = "Module:IanComradeBot/prototype.json/data"
     local storeName = (kind == "prototype") and "prototype_store.json" or "component_store.json"
 
    local moduleName = get_module_name(storeName)
     local data = cache[moduleName]
     local data = load_cached_data(moduleName)
     if not data then
     if not data then
         local ok, loaded = pcall(mw.loadData, moduleName)
         return ""
        if not ok or not loaded then
            return ""
        end
        data = loaded
        cache[moduleName] = data
     end
     end


     local proto = data[protoKey]
     local entry = data[searchId]
     if type(proto) ~= "table" then
     if type(entry) ~= "table" then
         return ""
         return ""
     end
     end


     local value = proto[fieldId]
     local value = get_field_loose(entry, fieldId)
     if value == nil then
     if value == nil then
         return ""
         return ""
Строка 197: Строка 261:
     if id == "" or pagePath == "" then return "" end
     if id == "" or pagePath == "" then return "" end


    local baseUser = "IanComradeBot/"
     local moduleName = get_module_name(pagePath)
     local moduleName = "Module:" .. baseUser .. pagePath .. "/data"
     local data = load_cached_data(moduleName)
 
     if not data then return "" end
     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 entry = resolve_entry(data, id) or {}
     local entry = resolve_entry(data, id) or {}
Строка 213: Строка 270:


     local parts = {}
     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 function walk(tbl, prefix)
         local keys = {}
         local keys = {}
Строка 219: Строка 283:
         for _, k in ipairs(keys) do
         for _, k in ipairs(keys) do
             local v = tbl[k]
             local v = tbl[k]
             local key = (prefix == "" and tostring(k) or prefix .. "." .. tostring(k))
            local kStr = tostring(k)
             local key = (prefix == "" and kStr or prefix .. "." .. kStr)
             if type(v) == "table" then
             if type(v) == "table" then
                 if next(v) == nil then
                 if next(v) == nil then
                elseif is_array(v) then
                    local ok, json = pcall(mw.text.jsonEncode, v)
                    if ok and json then
                        parts[#parts + 1] = key .. "=" .. tostring(json)
                    end
                 else
                 else
                     local ok, json = pcall(mw.text.jsonEncode, v)
                     append_table_json(key, v)
                     if ok and json then
                     if not is_array(v) then
                         parts[#parts + 1] = key .. "=" .. to_nowiki(json)
                         walk(v, key)
                     end
                     end
                    walk(v, key)
                 end
                 end
             else
             else
Строка 253: Строка 312:
     if pagePath == "" then return "" end
     if pagePath == "" then return "" end


    local baseUser = "IanComradeBot/"
     local moduleName = get_module_name(pagePath)
     local moduleName = "Module:" .. baseUser .. pagePath .. "/data"
     local data = load_cached_data(moduleName)
 
     if not data then return "" end
     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 entryKey = moduleName .. "|" .. (id ~= "" and id or "default")
     local entryKey = moduleName .. "|" .. (id ~= "" and id or "default")
Строка 278: Строка 330:
     local value = get_by_path(entry, keyPath)
     local value = get_by_path(entry, keyPath)
     return format_value(value)
     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 idsTable = data.id
    if type(idsTable) ~= "table" then
        return ""
    end
    local matches = find_matching_ids(idsTable, keyPath, searchValue)
    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
    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
    local matches = find_matching_ids(idsTable, keyPath, searchValue)
    if #matches == 0 then
        return ""
    end
    local out = {}
    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
    local result = table.concat(out, " ")
    return preprocess_or_return(frame, result)
end
end


Строка 297: Строка 424:
     tplStr = tplStr .. "}}"
     tplStr = tplStr .. "}}"


     if type(frame.preprocess) == "function" then
     return preprocess_or_return(frame, tplStr)
        return frame:preprocess(tplStr)
    end
 
    return tplStr
end
end


function p.getTplProto(frame)
function p.getTplGenerator(frame)
     local args = frame.args or {}
     local args = frame.args or {}
     local searchId = args[1] or ""
     local searchId = args[1] or ""
     local protoId = args[2] or ""
     local kind = (args[2] or ""):lower()
     local tplPath = args[3] or ""
     local generatorId = args[3] or ""
     local pagePath = "prototype/" .. protoId .. ".json"
     local tplPath = args[4] or ""


     if searchId == "" or protoId == "" or tplPath == "" then
     if searchId == "" or generatorId == "" or tplPath == "" then
        return ""
    end
    if kind ~= "prototype" and kind ~= "component" then
         return ""
         return ""
     end
     end


     local idsJson = p.findInProto({ args = { searchId, protoId } })
    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 "")
     local ok, ids = pcall(mw.text.jsonDecode, idsJson or "")
     if not ok or type(ids) ~= "table" or #ids == 0 then
     if not ok or type(ids) ~= "table" or #ids == 0 then
Строка 329: Строка 458:
     end
     end


     local result = table.concat(out, "\n")
     local result = table.concat(out, " ")
     if type(frame.preprocess) == "function" then
    return preprocess_or_return(frame, result)
         return frame:preprocess(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 target = tostring(compName)
    for _, v in ipairs(entry) do
        if tostring(v) == target then
            return "true"
        end
     end
     end


     return result
     return "false"
end
end


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