Модуль:GetField: различия между версиями
Pok (обсуждение | вклад) Нет описания правки Метка: отменено |
Pok (обсуждение | вклад) мНет описания правки |
||
| (не показано 14 промежуточных версий этого же участника) | |||
| Строка 4: | Строка 4: | ||
local getArgs = require('Module:Arguments').getArgs | local getArgs = require('Module:Arguments').getArgs | ||
function p.loadCachedData(pagePath) | |||
local | local titleName = JsonPaths.get(pagePath) | ||
if not titleName or titleName == "" then | |||
return nil | |||
if | |||
return | |||
end | end | ||
local ok, | local ok, data = pcall(mw.loadData, titleName) | ||
if not ok or type(data) ~= "table" then | |||
return nil | |||
if ok | |||
end | end | ||
return data | |||
return | |||
end | end | ||
| Строка 180: | Строка 152: | ||
end | end | ||
function p.ucfirst(s) | |||
if type( | if not s or s == "" then return s end | ||
return string.upper(s:sub(1, 1)) .. (s:sub(2) or "") | |||
end | |||
function p.normalizeComponentName(name) | |||
if type(name) ~= "string" or name == "" then | |||
return nil | |||
end | |||
name = mw.text.trim(name) | |||
if name:sub(1, 5) == "type:" then | |||
name = name:sub(6) | |||
elseif name:sub(1, 6) == "!type:" then | |||
name = name:sub(7) | |||
end | |||
if name == "" then | |||
return nil | return nil | ||
end | end | ||
return name | |||
end | |||
local function split_data_page_path(pagePath) | |||
local kind, name = string.match(pagePath or "", "^(component)/([^/]+)%.json$") | |||
if not kind then | |||
return nil, nil | |||
end | |||
return kind, p.ucfirst(name) | |||
end | |||
local function get_component_from_entity(entity, componentName) | |||
if type(entity) ~= "table" or type(entity.components) ~= "table" then | |||
return nil | |||
end | end | ||
return entity.components["type:" .. componentName] | |||
or entity.components[componentName] | |||
return | or entity.components["!type:" .. componentName] | ||
end | |||
local function load_specific_entry(pagePath, id, allowDefault) | |||
local kind, name = split_data_page_path(pagePath) | |||
if not kind or not id or id == "" then | |||
return nil | |||
end | end | ||
return nil | if kind ~= "component" then | ||
end | return nil | ||
end | |||
local | local base | ||
if type( | local defaultData = p.loadCachedData(kind .. "/" .. name .. "/defaultFields.json") | ||
if type(defaultData) == "table" then | |||
base = deep_copy(defaultData) | |||
end | end | ||
local | local entityPage = "prototype/Entity/" .. id .. ".json" | ||
local | local entity = p.loadCachedData(entityPage) | ||
local specific = get_component_from_entity(entity, name) | |||
if type( | if type(specific) ~= "table" then | ||
if allowDefault == false then | |||
return nil | |||
end | end | ||
return | return base | ||
end | end | ||
if base then | |||
deep_merge(base, specific) | |||
return base | |||
end | end | ||
return | return specific | ||
end | end | ||
local function | local function resolve_entry_from_data(data, id, allowDefault) | ||
if type( | if type(data) ~= "table" then | ||
return | return nil | ||
end | end | ||
if | |||
if id and id ~= "" then | |||
local direct = data[id] | |||
if direct ~= nil then | |||
return direct | |||
return | |||
end | end | ||
local idsTable = data.id | |||
if | if type(idsTable) == "table" then | ||
return | 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 specific | |||
end | end | ||
end | end | ||
end | end | ||
return | if allowDefault == false then | ||
return nil | |||
end | |||
local base = data["default"] | |||
if type(base) == "table" then | |||
return base | |||
end | |||
return nil | |||
end | end | ||
local function | local function load_entry(id, pagePath, data, allowDefault) | ||
if | local specificPageEntry = load_specific_entry(pagePath, id, allowDefault) | ||
return | if specificPageEntry ~= nil then | ||
return specificPageEntry | |||
end | end | ||
data = data or p.loadCachedData(pagePath) | |||
if allowDefault == nil then | |||
allowDefault = true | |||
end | end | ||
return | return resolve_entry_from_data(data, id, allowDefault) | ||
end | end | ||
local function | local function collect_id_keys(data) | ||
if type( | if type(data) ~= "table" then | ||
return | return {} | ||
end | end | ||
local idsTable = data.id | |||
local | local ids = {} | ||
if type(idsTable) == "table" then | |||
for k in pairs(idsTable) do | |||
if | ids[#ids + 1] = k | ||
return | end | ||
return ids | |||
end | end | ||
for k in pairs(data) do | |||
if k ~= "default" and k ~= "id" then | |||
ids[#ids + 1] = k | |||
end | |||
end | end | ||
return | return ids | ||
end | end | ||
local function | local function contains_target(v, target) | ||
if | if type(v) == "string" then | ||
return | return v == target | ||
end | end | ||
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 | end | ||
local | return tostring(v) == target | ||
if | end | ||
local function is_nonempty_value(v) | |||
if v == nil then | |||
return false | |||
end | |||
if type(v) == "table" then | |||
return next(v) ~= nil | |||
end | end | ||
return true | |||
return | |||
end | end | ||
local function | local function preprocess_or_return(frame, text) | ||
if type( | if type(frame) == "table" and type(frame.preprocess) == "function" then | ||
return | return frame:preprocess(text) | ||
end | end | ||
return text | |||
end | |||
local | 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 | 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 | end | ||
local function | local function apply_pattern(s, pattern, repl) | ||
if not pattern or pattern == "" or not s then | |||
if | return s | ||
return | |||
end | end | ||
local | local text = tostring(s) | ||
if | local replacement | ||
if repl and repl ~= "" then | |||
replacement = tostring(repl) | |||
replacement = replacement:gsub("\\(%d)", "%%%1") | |||
else | |||
replacement = "%1" | |||
end | end | ||
if | local patt = pattern | ||
if not patt:find("%^") and not patt:find("%$") then | |||
patt = "^" .. patt .. "$" | |||
end | |||
return (text:gsub(patt, replacement)) | |||
end | end | ||
local function | local function flatten_parts(entry) | ||
if type(entry | if type(entry) ~= "table" then | ||
return {} | return {} | ||
end | end | ||
local parts = {} | local parts = {} | ||
local function walk(tbl, prefix) | |||
local | local keys = {} | ||
for k in pairs(tbl) do | |||
keys[#keys + 1] = k | |||
end | 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 | |||
local ok, json = pcall(mw.text.jsonEncode, v) | |||
if ok and json then | |||
parts[#parts + 1] = key .. "=<nowiki>" .. json .. "</nowiki>" | |||
end | |||
if is_array(v) then | |||
local first = v[1] | |||
if type(first) == "table" then | |||
walk(first, key) | |||
end | |||
else | |||
walk(v, key) | |||
end | |||
end | |||
else | |||
parts[#parts + 1] = key .. "=" .. tostring(v) | |||
end | |||
end | |||
end | |||
walk(entry, "") | |||
return parts | return parts | ||
end | end | ||
local function | local function flatten_entry(entry) | ||
local | local parts = flatten_parts(entry) | ||
if #parts == 0 then | |||
if | return "" | ||
end | end | ||
return table.concat(parts, "|") | |||
return | |||
end | end | ||
local function | local function append_flattened_part(parts, key, value) | ||
if value == nil then | |||
return | |||
if | |||
end | end | ||
if | if type(value) == "table" then | ||
if next(value) == nil then | |||
return | |||
end | |||
local ok, json = pcall(mw.text.jsonEncode, value) | |||
if ok and json then | |||
parts[#parts + 1] = key .. "=<nowiki>" .. json .. "</nowiki>" | |||
end | end | ||
return | |||
end | end | ||
parts[#parts + 1] = key .. "=" .. tostring(value) | |||
end | end | ||
local function | local function flatten_selected_parts(entry, keys) | ||
if | if type(entry) ~= "table" or type(keys) ~= "table" then | ||
return | return {} | ||
end | end | ||
local | local parts = {} | ||
local | local seen = {} | ||
if | for i = 1, #keys do | ||
local key = keys[i] | |||
if type(key) == "string" and key ~= "" and not seen[key] then | |||
seen[key] = true | |||
end | append_flattened_part(parts, key, navigate_path(entry, key)) | ||
end | |||
end | end | ||
return parts | |||
end | |||
local | local function resolve_template_path(tplPath) | ||
if | local templatePath = tplPath | ||
local project = JsonPaths.project() | |||
if project ~= nil and project ~= "" then | |||
templatePath = tplPath .. "/" .. project | |||
templatePath = "{{#ifexist:Шаблон:" .. templatePath .. "|" .. templatePath .. "|" .. tplPath .. "}}" | |||
end | end | ||
return | return templatePath | ||
end | end | ||
function | local function split_template_spec(tplPath, tplArgs) | ||
tplPath = mw.text.unstripNoWiki(tplPath or "") | |||
tplArgs = mw.text.unstripNoWiki(tplArgs or "") | |||
if tplArgs ~= "" and string.sub(tplArgs, 1, 1) == "|" then | |||
tplArgs = string.sub(tplArgs, 2) | |||
end | |||
if | if tplArgs == "" then | ||
local pipePos = string.find(tplPath, "|", 1, true) | |||
if pipePos then | |||
tplArgs = mw.text.unstripNoWiki(string.sub(tplPath, pipePos + 1)) | |||
if tplArgs ~= "" and string.sub(tplArgs, 1, 1) == "|" then | |||
tplArgs = string.sub(tplArgs, 2) | |||
end | |||
tplPath = string.sub(tplPath, 1, pipePos - 1) | |||
end | |||
end | end | ||
local | return tplPath, tplArgs | ||
end | |||
local function build_tpl(id, pagePath, tplPath, data, tplArgs, preEntry, strictExact) | |||
if id == "" or pagePath == "" or tplPath == "" then | |||
return "" | return "" | ||
end | end | ||
local entry | local entry | ||
if | if strictExact then | ||
entry = load_entry(id, pagePath, data, false) | |||
else | |||
entry = preEntry or load_entry(id, pagePath, data) | |||
end | end | ||
if entry == nil then | |||
return "" | return "" | ||
end | end | ||
local | local extra = flatten_entry(entry) | ||
local extraTplArgs = tplArgs or "" | |||
local templatePath = resolve_template_path(tplPath) | |||
local tplStr = "{{Шаблон:" .. templatePath | |||
local | if extraTplArgs ~= "" then | ||
tplStr = tplStr .. "|" .. extraTplArgs | |||
end | |||
tplStr = tplStr .. "|id=" .. tostring(id) | |||
if extra ~= "" then | |||
tplStr = tplStr .. "|" .. extra | |||
end | end | ||
tplStr = tplStr .. "}}" | |||
return tplStr | |||
return | |||
end | end | ||
function p. | function p.findInGenerator(frame) | ||
local args = frame.args or {} | local args = frame.args or {} | ||
local | local searchId = args[1] or "" | ||
local | local kind = (args[2] or ""):lower() | ||
local | local fieldId = p.ucfirst(args[3] or "") | ||
if | |||
return "" | if searchId == "" or fieldId == "" then | ||
return "" | |||
end | end | ||
if kind ~= "prototype" and kind ~= "component" then | |||
if | |||
return "" | return "" | ||
end | end | ||
local | local storeName = (kind == "prototype") and "prototype_store.json" or "component_store.json" | ||
local data = | local data = p.loadCachedData(storeName) | ||
if not data then | if not data then | ||
return "" | return "" | ||
end | end | ||
local entry = | local entry = data[searchId] | ||
if type(entry) ~= "table" then | |||
return "" | return "" | ||
end | end | ||
local value = get_field_loose(entry, fieldId) | |||
if value == nil then | |||
local | |||
if | |||
return "" | return "" | ||
end | end | ||
local | local out = {} | ||
local | local t = type(value) | ||
if | if t == "table" then | ||
for _, v in ipairs(value) do | |||
out[#out + 1] = v | |||
end | |||
else | |||
out[1] = value | |||
end | end | ||
local | return mw.text.jsonEncode(out) | ||
if | 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 "" | return "" | ||
end | end | ||
local entry = load_entry(id, pagePath) or {} | |||
return flatten_entry(entry) | |||
return | |||
end | end | ||
function p.flattenFieldSelective(frame) | |||
local args = frame.args or {} | |||
local id = args[1] or "" | |||
return | local pagePath = args[2] or "" | ||
local keysJson = mw.text.unstripNoWiki(args[3] or args.keys or "") | |||
if id == "" or pagePath == "" or keysJson == "" then | |||
return "" | |||
end | end | ||
if type( | local okKeys, keys = pcall(mw.text.jsonDecode, keysJson) | ||
return | if not okKeys or type(keys) ~= "table" or #keys == 0 then | ||
return "" | |||
end | end | ||
local | local entry = load_entry(id, pagePath) or {} | ||
local | local parts = flatten_selected_parts(entry, keys) | ||
if #parts == 0 then | |||
return "" | |||
end | |||
return table.concat(parts, "|") | |||
end | |||
function p.get(frame) | |||
local args = getArgs(frame, { removeBlanks = false }) | |||
local id = args[1] or "" | |||
local pagePath = args[2] or "" | |||
local keyPath = args[3] or "" | |||
if pagePath == "" then | |||
return "" | |||
return | |||
end | end | ||
local | local entry = load_entry(id, pagePath) | ||
if entry == nil then | |||
if | return "" | ||
end | end | ||
if | if keyPath == "" then | ||
return format_value(entry) | |||
end | end | ||
return format_value(navigate_path(entry, keyPath)) | |||
end | end | ||
local function | local function collect_by_parsed_path(tbl, parsedPath, pos, out) | ||
if pos > #parsedPath then | |||
out[#out + 1] = tbl | |||
return | |||
end | end | ||
if type(tbl) ~= "table" then | |||
return | |||
end | |||
for _, | local token = parsedPath[pos] | ||
local key = token[1] | |||
local idx = token[2] | |||
if key == "*" then | |||
for _, child in pairs(tbl) do | |||
local nextCur = child | |||
if idx then | |||
if type(nextCur) ~= "table" then | |||
nextCur = nil | |||
else | |||
nextCur = nextCur[idx] | |||
end | |||
end | end | ||
collect_by_parsed_path(nextCur, parsedPath, pos + 1, out) | |||
end | end | ||
return | |||
end | end | ||
local nextCur | |||
if key and key ~= "" then | |||
nextCur = tbl[key] | |||
if nextCur == nil then nextCur = tbl["!type:" .. key] end | |||
else | |||
nextCur = tbl | |||
end | |||
if idx then | |||
if | if type(nextCur) ~= "table" then | ||
return | return | ||
end | end | ||
nextCur = nextCur[idx] | |||
end | end | ||
collect_by_parsed_path(nextCur, parsedPath, pos + 1, out) | |||
end | end | ||
local function | local function get_by_parsed_path_multi(tbl, parsedPath) | ||
local out = {} | local out = {} | ||
collect_by_parsed_path(tbl, parsedPath, 1, out) | |||
return out | |||
end | |||
local function entry_matches_path(entry, parsedPath, searchValue, searchType) | |||
local values = get_by_parsed_path_multi(entry, parsedPath) | |||
if | local target = tostring(searchValue) | ||
for _, v in ipairs(values) do | |||
if searchType == "key" then | |||
if type(v) == "table" and v[target] ~= nil then | |||
return true | |||
end | |||
else | |||
if contains_target(v, target) then | |||
return true | |||
end | |||
end | end | ||
end | end | ||
return | return false | ||
end | end | ||
function | local function entry_has_any_nonempty_path(entry, parsedPath) | ||
local | local values = get_by_parsed_path_multi(entry, parsedPath) | ||
if | for _, v in ipairs(values) do | ||
if is_nonempty_value(v) then | |||
return true | |||
end | |||
end | end | ||
local | return false | ||
end | |||
local function split_csv_list(text) | |||
text = mw.text.unstripNoWiki(text or "") | |||
local out = {} | |||
for part in string.gmatch(text, "([^,]+)") do | |||
part = mw.text.trim(part) | |||
if part ~= "" then | |||
out[#out + 1] = part | |||
end | |||
end | end | ||
local ids = collect_id_keys(data) | return out | ||
end | |||
function p.searchId(frame) | |||
local args = getArgs(frame, { removeBlanks = false }) | |||
local searchValue = args[1] or "" | |||
local pagePath = args[2] or "" | |||
local keyPath = args[3] or "" | |||
local searchType = (args.searchType or ""):lower() | |||
if searchValue == "" or pagePath == "" or keyPath == "" then | |||
return "" | |||
end | |||
if searchType == "" then | |||
searchType = "value" | |||
end | |||
local data = p.loadCachedData(pagePath) | |||
if not data then | |||
return "[]" | |||
end | |||
local parsedPath = parse_path(keyPath) | |||
if not parsedPath then | |||
return "" | |||
end | |||
local ids = collect_id_keys(data) | |||
if #ids == 0 then | if #ids == 0 then | ||
return "" | return "" | ||
| Строка 730: | Строка 777: | ||
for _, idKey in ipairs(ids) do | for _, idKey in ipairs(ids) do | ||
local entry = | local entry = resolve_entry_from_data(data, idKey) | ||
if type(entry) == "table" and entry_matches_path(entry, parsedPath, target, searchType) then | if type(entry) == "table" and entry_matches_path(entry, parsedPath, target, searchType) then | ||
matches[#matches + 1] = idKey | matches[#matches + 1] = idKey | ||
| Строка 746: | Строка 793: | ||
return "" | return "" | ||
end | |||
local function collect_ids_from_entry(entry) | |||
local out = {} | |||
local seen = {} | |||
local function add(v) | |||
if v == nil then | |||
return | |||
end | |||
local s = tostring(v) | |||
if s ~= "" and not seen[s] then | |||
seen[s] = true | |||
out[#out + 1] = s | |||
end | |||
end | |||
if type(entry) == "table" then | |||
if is_array(entry) then | |||
for _, v in ipairs(entry) do | |||
add(v) | |||
end | |||
else | |||
for _, v in pairs(entry) do | |||
if type(v) == "table" and is_array(v) then | |||
for _, item in ipairs(v) do | |||
add(item) | |||
end | |||
elseif type(v) ~= "table" then | |||
add(v) | |||
end | |||
end | |||
end | |||
else | |||
add(entry) | |||
end | |||
return out | |||
end | end | ||
| Строка 754: | Строка 839: | ||
if mode ~= "component" and mode ~= "tag" then | if mode ~= "component" and mode ~= "tag" then | ||
return " | return "" | ||
end | end | ||
local targets = split_csv_list(rawList) | local targets = split_csv_list(rawList) | ||
if #targets == 0 then | if #targets == 0 then | ||
return " | return "" | ||
end | end | ||
local pagePath | local pagePath = (mode == "component") and "component.json" or "tag.json" | ||
local data = p.loadCachedData(pagePath) | |||
if not data then | |||
return "" | |||
end | end | ||
local | local matches = {} | ||
local seen = {} | |||
local function add_id(id) | |||
local s = tostring(id) | |||
if s ~= "" and not seen[s] then | |||
seen[s] = true | |||
matches[#matches + 1] = s | |||
end | |||
end | end | ||
local | for _, key in ipairs(targets) do | ||
local entry = resolve_entry_from_data(data, key) | |||
if entry ~= nil then | |||
local ids = collect_ids_from_entry(entry) | |||
for _, id in ipairs(ids) do | |||
add_id(id) | |||
end | |||
end | |||
end | end | ||
table.sort( | table.sort(matches, function(a, b) | ||
return tostring(a) < tostring(b) | return tostring(a) < tostring(b) | ||
end) | end) | ||
local | local ok, json = pcall(mw.text.jsonEncode, matches) | ||
if ok and json then | |||
return json | |||
end | |||
return "" | |||
end | |||
function p.searchIdTpl(frame) | |||
local args = getArgs(frame, { removeBlanks = false }) | |||
local searchValue = args[1] or "" | |||
local pagePath = args[2] or "" | |||
local keyPath = args[3] or "" | |||
local tplPath = mw.text.unstripNoWiki(args[4] or "") | |||
local tplArgs = args.tplArgs or args.templateArgs or "" | |||
local searchType = (args.searchType or ""):lower() | |||
if searchType == "" then | |||
searchType = "value" | |||
end | end | ||
if searchType == "path" then | |||
searchValue = "" | |||
pagePath = args[1] or "" | |||
keyPath = args[2] or "" | |||
tplPath = mw.text.unstripNoWiki(args[3] or "") | |||
tplArgs = args[4] or args.tplArgs or args.templateArgs or "" | |||
end | |||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | |||
if pagePath == "" or keyPath == "" or tplPath == "" then | |||
if | return "" | ||
end | |||
if searchType ~= "path" and searchValue == "" then | |||
return "" | |||
end | |||
local data = p.loadCachedData(pagePath) | |||
if not data then | |||
return "" | |||
end | |||
local parsedPath = parse_path(keyPath) | |||
if not parsedPath then | |||
return "" | |||
end | |||
return | local ids = collect_id_keys(data) | ||
if #ids == 0 then | |||
return "" | |||
end | end | ||
local matches = {} | local matches = {} | ||
local entryCache = {} | |||
for _, idKey in ipairs(ids) do | if searchType == "path" then | ||
for _, idKey in ipairs(ids) do | |||
local entry = resolve_entry_from_data(data, idKey) | |||
entryCache[idKey] = entry | |||
if type(entry) == "table" and entry_has_any_nonempty_path(entry, parsedPath) then | |||
matches[#matches + 1] = idKey | matches[#matches + 1] = idKey | ||
end | end | ||
else | end | ||
if | else | ||
local target = tostring(searchValue) | |||
for _, idKey in ipairs(ids) do | |||
local entry = resolve_entry_from_data(data, idKey) | |||
entryCache[idKey] = entry | |||
if type(entry) == "table" and entry_matches_path(entry, parsedPath, target, searchType) then | |||
matches[#matches + 1] = idKey | matches[#matches + 1] = idKey | ||
end | end | ||
| Строка 848: | Строка 953: | ||
end | end | ||
local | if #matches == 0 then | ||
if | return "" | ||
return | end | ||
local out = {} | |||
for _, idKey in ipairs(matches) do | |||
local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs, entryCache[idKey], true) | |||
if tpl ~= "" then | |||
out[#out + 1] = tpl | |||
end | |||
end | |||
if #out == 0 then | |||
return "" | |||
end | end | ||
local result = table.concat(out, " ") | |||
return preprocess_or_return(frame, result) | |||
end | end | ||
function p. | function p.getTpl(frame) | ||
local args = getArgs(frame, { removeBlanks = false }) | local args = getArgs(frame, { removeBlanks = false }) | ||
local | local id = args[1] or "" | ||
local pagePath = args[2] or "" | local pagePath = args[2] or "" | ||
local tplPath = mw.text.unstripNoWiki(args[3] or "") | |||
local tplPath = mw.text.unstripNoWiki(args[ | local tplArgs = args[4] or args.tplArgs or args.templateArgs or "" | ||
local | |||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | ||
if | if id == "" or pagePath == "" or tplPath == "" then | ||
return "" | return "" | ||
end | end | ||
local | local data = frame.data | ||
local | local tplStr = build_tpl(id, pagePath, tplPath, data, tplArgs, nil, true) | ||
if | return preprocess_or_return(frame, tplStr) | ||
end | |||
function p.searchStoreTpl(frame) | |||
local args = getArgs(frame, { removeBlanks = false }) | |||
local searchId = args[1] or "" | |||
local kind = (args[2] or ""):lower() | |||
local generatorId = p.ucfirst(args[3] or "") | |||
local tplPath = mw.text.unstripNoWiki(args[4] or "") | |||
local tplArgs = args[5] or args.tplArgs or args.templateArgs or "" | |||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | |||
if searchId == "" or generatorId == "" or tplPath == "" then | |||
return "" | return "" | ||
end | end | ||
if kind ~= "prototype" and kind ~= "component" then | |||
return "" | return "" | ||
end | end | ||
local ids = | local dir = (kind == "prototype") and "prototype/" or "component/" | ||
if #ids == 0 then | 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 "" | return "" | ||
end | end | ||
local | local data = p.loadCachedData(pagePath) | ||
if not data then | |||
return "" | |||
end | |||
local out = {} | |||
for _, id in ipairs(ids) do | |||
local entry = resolve_entry_from_data(data, id) | |||
if entry ~= nil then | |||
local tpl = build_tpl(id, pagePath, tplPath, data, tplArgs, entry, true) | |||
if tpl ~= "" then | |||
out[#out + 1] = tpl | |||
if | |||
end | end | ||
end | end | ||
end | end | ||
local result = table.concat(out, " ") | |||
return preprocess_or_return(frame, result) | |||
end | end | ||
function p.flattenParams(entry) | |||
return flatten_parts(entry) | |||
return | |||
end | end | ||
function p. | function p.searchStore(frame) | ||
local args = getArgs(frame, { removeBlanks = false }) | local args = getArgs(frame, { removeBlanks = false }) | ||
local | local searchId = args[1] or "" | ||
local | local kind = (args[2] or ""):lower() | ||
local | local generatorId = p.ucfirst(args[3] or "") | ||
if searchId == "" or generatorId == "" then | |||
if searchId == "" or generatorId | |||
return "" | return "" | ||
end | end | ||
| Строка 987: | Строка 1051: | ||
return "" | return "" | ||
end | end | ||
local idsJson = p.findInGenerator({ args = { searchId, kind, generatorId } }) | local idsJson = p.findInGenerator({ args = { searchId, kind, generatorId } }) | ||
| Строка 997: | Строка 1058: | ||
end | end | ||
local | local okOut, outJson = pcall(mw.text.jsonEncode, ids) | ||
if okOut and outJson then | |||
if | return outJson | ||
return | |||
end | end | ||
return "" | |||
end | |||
function p.loadEntityData(entityId) | |||
if entityId == "" then | |||
return nil | |||
end | end | ||
local | local pagePath = "prototype/Entity/" .. entityId .. ".json" | ||
return | return p.loadCachedData(pagePath) | ||
end | end | ||
function p. | function p.entityHasComponent(entity, compName) | ||
if type(entity) ~= "table" or type(entity.components) ~= "table" then | |||
return false | |||
end | |||
return entity.components[compName] ~= nil | |||
or entity.components["type:" .. compName] ~= nil | |||
or entity.components["!type:" .. compName] ~= nil | |||
end | end | ||
function p. | function p.collectEntityComponents(entity) | ||
if type(entity) ~= "table" or type(entity.components) ~= "table" then | |||
local | return {} | ||
local | end | ||
local | |||
local out = {} | |||
local seen = {} | |||
local comps = entity.components | |||
if | if #comps > 0 then | ||
for _, v in ipairs(comps) do | |||
local name = p.normalizeComponentName(v) | |||
if name and not seen[name] then | |||
seen[name] = true | |||
out[#out + 1] = name | |||
end | |||
end | |||
else | |||
for k in pairs(comps) do | |||
local name = p.normalizeComponentName(k) | |||
if name and not seen[name] then | |||
seen[name] = true | |||
out[#out + 1] = name | |||
end | |||
end | |||
end | end | ||
table.sort(out) | |||
return out | |||
end | |||
function p.hasComp(frame) | |||
local args = getArgs(frame, { removeBlanks = false }) | |||
local entityId = args[1] or "" | |||
function p.hasComp(frame) | |||
local args = getArgs(frame, { removeBlanks = false }) | |||
local entityId = args[1] or "" | |||
local compName = args[2] or "" | local compName = args[2] or "" | ||
| Строка 1055: | Строка 1125: | ||
end | end | ||
local | local entity = p.loadEntityData(entityId) | ||
if not entity then | |||
if not | |||
return "false" | return "false" | ||
end | end | ||
return p.entityHasComponent(entity, compName) and "true" or "false" | |||
end | end | ||
| Строка 1084: | Строка 1138: | ||
if entityId == "" then | if entityId == "" then | ||
return " | return "" | ||
end | end | ||
local | local entity = p.loadEntityData(entityId) | ||
if not entity then | |||
if not | return "" | ||
return " | |||
end | end | ||
local out = p.collectEntityComponents(entity) | |||
local ok, json = pcall(mw.text.jsonEncode, out) | local ok, json = pcall(mw.text.jsonEncode, out) | ||
if ok and json then | if ok and json then | ||
| Строка 1110: | Строка 1152: | ||
end | end | ||
return " | return "" | ||
end | end | ||
| Строка 1123: | Строка 1165: | ||
end | end | ||
local | local data = p.loadCachedData(pagePath) | ||
if not data then | if not data then | ||
return "" | return "" | ||
end | end | ||
local idsTable = data.id | local idsTable = data.id or data | ||
if type(idsTable) ~= "table" then | if type(idsTable) ~= "table" then | ||
return "" | return "" | ||
| Строка 1174: | Строка 1215: | ||
end | end | ||
local | local data = p.loadCachedData(pagePath) | ||
if not data then | if not data then | ||
return "" | return "" | ||
end | end | ||
local idsTable = data.id | local idsTable = data.id or data | ||
if type(idsTable) ~= "table" then | if type(idsTable) ~= "table" then | ||
return "" | return "" | ||
| Строка 1188: | Строка 1228: | ||
for idKey in pairs(idsTable) do | for idKey in pairs(idsTable) do | ||
local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs) | local entry = resolve_entry_from_data(data, idKey) | ||
if entry ~= nil then | |||
local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs, entry, true) | |||
if tpl ~= "" then | |||
out[#out + 1] = tpl | |||
end | |||
end | end | ||
end | end | ||
| Строка 1624: | Строка 1667: | ||
end | end | ||
local entry = load_entry(id, dataPage) or {} | |||
local entry = | |||
local parts = flatten_selected_parts(entry, paramNames) | local parts = flatten_selected_parts(entry, paramNames) | ||
if #parts == 0 then | if #parts == 0 then | ||