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

Нет описания правки
мНет описания правки
 
(не показано 17 промежуточных версий этого же участника)
Строка 4: Строка 4:
local getArgs = require('Module:Arguments').getArgs
local getArgs = require('Module:Arguments').getArgs


local function load_cached_data(moduleName)
function p.loadCachedData(pagePath)
local ok, loaded = pcall(mw.loadData, moduleName)
local titleName = JsonPaths.get(pagePath)
if not ok or not loaded then
if not titleName or titleName == "" then
return nil
return nil
end
end
return loaded
 
local ok, data = pcall(mw.loadData, titleName)
if not ok or type(data) ~= "table" then
return nil
end
 
return data
end
end


Строка 90: Строка 96:
local function format_value(v)
local function format_value(v)
if v == nil then
if v == nil then
return "null"
return ""
end
end


Строка 146: Строка 152:
end
end


local function resolve_entry(data, id)
function p.ucfirst(s)
if type(data) ~= "table" then
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


if id and id ~= "" then
local function get_component_from_entity(entity, componentName)
local direct = data[id]
if type(entity) ~= "table" or type(entity.components) ~= "table" then
if direct ~= nil then
return nil
return direct
end
end


local idsTable = data.id
return entity.components["type:" .. componentName]
if type(idsTable) == "table" then
or entity.components[componentName]
local specific = idsTable[id]
or entity.components["!type:" .. componentName]
if type(specific) == "table" then
end
local base = data["default"]
 
if type(base) == "table" then
local function load_specific_entry(pagePath, id, allowDefault)
local merged = deep_copy(base)
local kind, name = split_data_page_path(pagePath)
deep_merge(merged, specific)
if not kind or not id or id == "" then
return merged
return nil
end
end
return specific
 
end
if kind ~= "component" then
return nil
end
 
local base
local defaultData = p.loadCachedData(kind .. "/" .. name .. "/defaultFields.json")
if type(defaultData) == "table" then
base = deep_copy(defaultData)
end
 
local entityPage = "prototype/Entity/" .. id .. ".json"
local entity = p.loadCachedData(entityPage)
local specific = get_component_from_entity(entity, name)
 
if type(specific) ~= "table" then
if allowDefault == false then
return nil
end
end
return base
end
end


local base = data["default"]
if base then
if type(base) == "table" then
deep_merge(base, specific)
return base
return base
end
end


return nil
return specific
end
end


local function collect_id_keys(data)
local function resolve_entry_from_data(data, id, allowDefault)
if type(data) ~= "table" then
if type(data) ~= "table" then
return {}
return nil
end
end


local idsTable = data.id
if id and id ~= "" then
local ids = {}
local direct = data[id]
if direct ~= nil then
return direct
end


if type(idsTable) == "table" then
local idsTable = data.id
for k in pairs(idsTable) do
if type(idsTable) == "table" then
ids[#ids + 1] = k
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
return ids
end
end


for k in pairs(data) do
if allowDefault == false then
if k ~= "default" and k ~= "id" then
return nil
ids[#ids + 1] = k
end
end
 
local base = data["default"]
if type(base) == "table" then
return base
end
end


return ids
return nil
end
end


local function contains_target(v, target)
local function load_entry(id, pagePath, data, allowDefault)
if type(v) == "string" then
local specificPageEntry = load_specific_entry(pagePath, id, allowDefault)
return v == target
if specificPageEntry ~= nil then
return specificPageEntry
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
data = data or p.loadCachedData(pagePath)
if tostring(item) == target then
if allowDefault == nil then
return true
allowDefault = true
end
end
return resolve_entry_from_data(data, id, allowDefault)
end
 
local function collect_id_keys(data)
if type(data) ~= "table" then
return {}
end
 
local idsTable = data.id
local ids = {}
 
if type(idsTable) == "table" then
for k in pairs(idsTable) do
ids[#ids + 1] = k
end
return ids
end
 
for k in pairs(data) do
if k ~= "default" and k ~= "id" then
ids[#ids + 1] = k
end
end
return false
end
end


return tostring(v) == target
return ids
end
end


local function is_nonempty_value(v)
local function contains_target(v, target)
if v == nil then
if type(v) == "string" then
return false
return v == target
end
end
if type(v) == "table" then
if type(v) == "table" then
return next(v) ~= nil
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
return true
 
return tostring(v) == target
end
end


local function preprocess_or_return(frame, text)
local function is_nonempty_value(v)
if v == nil then
return false
end
if type(v) == "table" then
return next(v) ~= nil
end
return true
end
 
local function preprocess_or_return(frame, text)
if type(frame) == "table" and type(frame.preprocess) == "function" then
if type(frame) == "table" and type(frame.preprocess) == "function" then
return frame:preprocess(text)
return frame:preprocess(text)
Строка 413: Строка 510:
end
end


local function build_tpl(id, pagePath, tplPath, data, tplArgs, preEntry)
local function build_tpl(id, pagePath, tplPath, data, tplArgs, preEntry, strictExact)
if id == "" or pagePath == "" or tplPath == "" then
if id == "" or pagePath == "" or tplPath == "" then
return ""
return ""
end
end


local moduleName = JsonPaths.get(pagePath)
local entry
data = data or load_cached_data(moduleName)
if strictExact then
if not data then
entry = load_entry(id, pagePath, data, false)
else
entry = preEntry or load_entry(id, pagePath, data)
end
 
if entry == nil then
return ""
return ""
end
end


local entry = preEntry or resolve_entry(data, id)
local extra = flatten_entry(entry)
local extra = flatten_entry(entry)
local extraTplArgs = tplArgs or ""
local extraTplArgs = tplArgs or ""
Строка 447: Строка 548:
local searchId = args[1] or ""
local searchId = args[1] or ""
local kind = (args[2] or ""):lower()
local kind = (args[2] or ""):lower()
local fieldId = args[3] or ""
local fieldId = p.ucfirst(args[3] or "")


if searchId == "" or fieldId == "" then
if searchId == "" or fieldId == "" then
Строка 457: Строка 558:


local storeName = (kind == "prototype") and "prototype_store.json" or "component_store.json"
local storeName = (kind == "prototype") and "prototype_store.json" or "component_store.json"
local moduleName = JsonPaths.get(storeName)
local data = p.loadCachedData(storeName)
local data = load_cached_data(moduleName)
if not data then
if not data then
return ""
return ""
Строка 494: Строка 594:
end
end


local moduleName = JsonPaths.get(pagePath)
local entry = load_entry(id, pagePath) or {}
local data = load_cached_data(moduleName)
if not data then
return ""
end
 
local entry = resolve_entry(data, id) or {}
return flatten_entry(entry)
return flatten_entry(entry)
end
end
Строка 518: Строка 612:
end
end


local moduleName = JsonPaths.get(pagePath)
local entry = load_entry(id, pagePath) or {}
local data = load_cached_data(moduleName)
if not data then
return ""
end
 
local entry = resolve_entry(data, id) or {}
local parts = flatten_selected_parts(entry, keys)
local parts = flatten_selected_parts(entry, keys)
if #parts == 0 then
if #parts == 0 then
Строка 543: Строка 631:
end
end


local moduleName = JsonPaths.get(pagePath)
local entry = load_entry(id, pagePath)
local data = load_cached_data(moduleName)
if not data then
return ""
end
 
local entry = resolve_entry(data, id)
if entry == nil then
if entry == nil then
return ""
return ""
Строка 660: Строка 742:


return out
return out
end
local function add_scalar_values(value, set)
if type(value) == "table" then
if is_array(value) then
for _, item in ipairs(value) do
add_scalar_values(item, set)
end
else
for _, item in pairs(value) do
add_scalar_values(item, set)
end
end
elseif value ~= nil then
set[tostring(value)] = true
end
end
local function table_has_all_values(value, targets)
if type(targets) ~= "table" or #targets == 0 then
return false
end
if type(value) ~= "table" then
return #targets == 1 and tostring(value) == targets[1]
end
local set = {}
add_scalar_values(value, set)
for i = 1, #targets do
if not set[targets[i]] then
return false
end
end
return true
end
end


Строка 713: Строка 758:
end
end


local moduleName = JsonPaths.get(pagePath)
local data = p.loadCachedData(pagePath)
local data = load_cached_data(moduleName)
if not data then
if not data then
return "[]"
return "[]"
Строка 733: Строка 777:


for _, idKey in ipairs(ids) do
for _, idKey in ipairs(ids) do
local entry = resolve_entry(data, idKey)
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
Строка 751: Строка 795:
end
end


function p.search(frame)
local function collect_ids_from_entry(entry)
local args = getArgs(frame, { removeBlanks = false })
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
 
function p.search(frame)
local args = getArgs(frame, { removeBlanks = false })
local mode = (args[1] or ""):lower()
local mode = (args[1] or ""):lower()
local rawList = args[2] or ""
local rawList = args[2] or ""


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"
if mode == "component" then
local data = p.loadCachedData(pagePath)
pagePath = "component.json"
if not data then
else
return ""
pagePath = "tag.json"
end
end


local moduleName = JsonPaths.get(pagePath)
local matches = {}
local data = load_cached_data(moduleName)
local seen = {}
if not data or type(data) ~= "table" then
 
return "[]"
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 ids = collect_id_keys(data)
for _, key in ipairs(targets) do
if #ids == 0 then
local entry = resolve_entry_from_data(data, key)
return "[]"
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(ids, function(a, b)
table.sort(matches, function(a, b)
return tostring(a) < tostring(b)
return tostring(a) < tostring(b)
end)
end)


local function entry_has_all_targets_component(entry, wanted)
local ok, json = pcall(mw.text.jsonEncode, matches)
if type(entry) ~= "table" then
if ok and json then
return false
return json
end
 
local set = {}
for _, v in ipairs(entry) do
set[tostring(v)] = true
end
 
for i = 1, #wanted do
if not set[wanted[i]] then
return false
end
end
 
return true
end
end


local function entry_has_all_targets_tag(entry, wanted)
return ""
if type(entry) ~= "table" then
end
return false
end


local tags = entry.tags or entry.tag
function p.searchIdTpl(frame)
if type(tags) ~= "table" then
local args = getArgs(frame, { removeBlanks = false })
return false
local searchValue = args[1] or ""
end
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()


local set = {}
if searchType == "" then
if is_array(tags) then
searchType = "value"
for _, v in ipairs(tags) do
set[tostring(v)] = true
end
else
for _, v in pairs(tags) do
set[tostring(v)] = true
end
end
 
for i = 1, #wanted do
if not set[wanted[i]] then
return false
end
end
 
return true
end
end


local matches = {}
if searchType == "path" then
 
searchValue = ""
for _, idKey in ipairs(ids) do
pagePath = args[1] or ""
local entry = resolve_entry(data, idKey)
keyPath = args[2] or ""
if mode == "component" then
tplPath = mw.text.unstripNoWiki(args[3] or "")
if entry_has_all_targets_component(entry, targets) then
tplArgs = args[4] or args.tplArgs or args.templateArgs or ""
matches[#matches + 1] = idKey
end
else
if entry_has_all_targets_tag(entry, targets) then
matches[#matches + 1] = idKey
end
end
end
end


local ok, json = pcall(mw.text.jsonEncode, matches)
tplPath, tplArgs = split_template_spec(tplPath, tplArgs)
if ok and json then
 
return json
if pagePath == "" or keyPath == "" or tplPath == "" then
return ""
end
if searchType ~= "path" and searchValue == "" then
return ""
end
end


return "[]"
local data = p.loadCachedData(pagePath)
end
if not data then
return ""
end


function p.searchIdTpl(frame)
local parsedPath = parse_path(keyPath)
local args = getArgs(frame, { removeBlanks = false })
if not parsedPath then
local searchValue = args[1] or ""
return ""
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
local ids = collect_id_keys(data)
searchValue = ""
if #ids == 0 then
pagePath = args[1] or ""
return ""
keyPath = args[2] or ""
tplPath = mw.text.unstripNoWiki(args[3] or "")
tplArgs = args[4] or args.tplArgs or args.templateArgs or ""
end
end


tplPath, tplArgs = split_template_spec(tplPath, tplArgs)
local matches = {}
local entryCache = {}


if pagePath == "" or keyPath == "" or tplPath == "" then
if searchType == "path" then
return ""
end
if searchType ~= "path" and searchValue == "" then
return ""
end
 
local moduleName = JsonPaths.get(pagePath)
local data = load_cached_data(moduleName)
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
return ""
end
 
local matches = {}
local entryCache = {}
 
if searchType == "path" then
for _, idKey in ipairs(ids) do
for _, idKey in ipairs(ids) do
local entry = resolve_entry(data, idKey)
local entry = resolve_entry_from_data(data, idKey)
entryCache[idKey] = entry
entryCache[idKey] = entry
if type(entry) == "table" and entry_has_any_nonempty_path(entry, parsedPath) then
if type(entry) == "table" and entry_has_any_nonempty_path(entry, parsedPath) then
Строка 919: Строка 945:
local target = tostring(searchValue)
local target = tostring(searchValue)
for _, idKey in ipairs(ids) do
for _, idKey in ipairs(ids) do
local entry = resolve_entry(data, idKey)
local entry = resolve_entry_from_data(data, idKey)
entryCache[idKey] = entry
entryCache[idKey] = entry
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
Строка 933: Строка 959:
local out = {}
local out = {}
for _, idKey in ipairs(matches) do
for _, idKey in ipairs(matches) do
local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs, entryCache[idKey])
local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs, entryCache[idKey], true)
if tpl ~= "" then
if tpl ~= "" then
out[#out + 1] = tpl
out[#out + 1] = tpl
Строка 959: Строка 985:
end
end


local moduleName = JsonPaths.get(pagePath)
local data = frame.data
local data = frame.data
if not data then
local tplStr = build_tpl(id, pagePath, tplPath, data, tplArgs, nil, true)
data = load_cached_data(moduleName)
return preprocess_or_return(frame, tplStr)
end
if not data then
return ""
end
 
local tplStr = build_tpl(id, pagePath, tplPath, data, tplArgs)
return preprocess_or_return(frame, tplStr)
end
end


Строка 976: Строка 994:
local searchId = args[1] or ""
local searchId = args[1] or ""
local kind = (args[2] or ""):lower()
local kind = (args[2] or ""):lower()
local generatorId = args[3] or ""
local generatorId = p.ucfirst(args[3] or "")
local tplPath = mw.text.unstripNoWiki(args[4] or "")
local tplPath = mw.text.unstripNoWiki(args[4] or "")
local tplArgs = args[5] or args.tplArgs or args.templateArgs or ""
local tplArgs = args[5] or args.tplArgs or args.templateArgs or ""
Строка 997: Строка 1015:
end
end


local moduleName = JsonPaths.get(pagePath)
local data = p.loadCachedData(pagePath)
local data = load_cached_data(moduleName)
if not data then
if not data then
return ""
return ""
Строка 1005: Строка 1022:
local out = {}
local out = {}
for _, id in ipairs(ids) do
for _, id in ipairs(ids) do
local tpl = build_tpl(id, pagePath, tplPath, data, tplArgs)
local entry = resolve_entry_from_data(data, id)
if tpl ~= "" then
if entry ~= nil then
out[#out + 1] = tpl
local tpl = build_tpl(id, pagePath, tplPath, data, tplArgs, entry, true)
if tpl ~= "" then
out[#out + 1] = tpl
end
end
end
end
end
Строка 1023: Строка 1043:
local searchId = args[1] or ""
local searchId = args[1] or ""
local kind = (args[2] or ""):lower()
local kind = (args[2] or ""):lower()
local generatorId = args[3] or ""
local generatorId = p.ucfirst(args[3] or "")


if searchId == "" or generatorId == "" then
if searchId == "" or generatorId == "" then
Строка 1046: Строка 1066:
end
end


function p.hasComp(frame)
function p.loadEntityData(entityId)
local args = getArgs(frame, { removeBlanks = false })
if entityId == "" then
local entityId = args[1] or ""
return nil
local compName = args[2] or ""
 
if entityId == "" or compName == "" then
return "false"
end
end


local moduleName = JsonPaths.get("component.json")
local pagePath = "prototype/Entity/" .. entityId .. ".json"
local data = load_cached_data(moduleName)
return p.loadCachedData(pagePath)
if not data then
end
return "false"
end


if type(data) ~= "table" then
function p.entityHasComponent(entity, compName)
return "false"
if type(entity) ~= "table" or type(entity.components) ~= "table" then
return false
end
end


local entry = data[entityId]
return entity.components[compName] ~= nil
if type(entry) ~= "table" then
or entity.components["type:" .. compName] ~= nil
return "false"
or entity.components["!type:" .. compName] ~= nil
end
 
function p.collectEntityComponents(entity)
if type(entity) ~= "table" or type(entity.components) ~= "table" then
return {}
end
end


for _, v in ipairs(entry) do
local out = {}
if tostring(v) == compName then
local seen = {}
return "true"
local comps = entity.components
 
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
end


return "false"
table.sort(out)
return out
end
end


function p.getComp(frame)
function p.hasComp(frame)
local args = getArgs(frame, { removeBlanks = false })
local args = getArgs(frame, { removeBlanks = false })
local entityId = args[1] or ""
local entityId = args[1] or ""
local compName = args[2] or ""


if entityId == "" then
if entityId == "" or compName == "" then
return "[]"
return "false"
end
end


local moduleName = JsonPaths.get("component.json")
local entity = p.loadEntityData(entityId)
local data = load_cached_data(moduleName)
if not entity then
if not data or type(data) ~= "table" then
return "false"
return "[]"
end
end


local entry = data[entityId]
return p.entityHasComponent(entity, compName) and "true" or "false"
if type(entry) ~= "table" then
end
return "[]"
end


local out = {}
function p.getComp(frame)
for _, v in ipairs(entry) do
local args = getArgs(frame, { removeBlanks = false })
if v ~= nil and v ~= "" then
local entityId = args[1] or ""
out[#out + 1] = v
 
end
if entityId == "" then
return ""
end
 
local entity = p.loadEntityData(entityId)
if not entity then
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 moduleName = JsonPaths.get(pagePath)
local data = p.loadCachedData(pagePath)
local data = load_cached_data(moduleName)
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 ""
Строка 1166: Строка 1207:
local args = getArgs(frame, { removeBlanks = false })
local args = getArgs(frame, { removeBlanks = false })
local pagePath = args[1] or ""
local pagePath = args[1] or ""
local tplPath = args[2] or ""
local tplPath = args[2] or ""
local tplArgs = args[3] or args.tplArgs or args.templateArgs or ""
local tplArgs = args[3] or args.tplArgs or args.templateArgs or ""
tplPath, tplArgs = split_template_spec(tplPath, tplArgs)
tplPath, tplArgs = split_template_spec(tplPath, tplArgs)
 
 
if pagePath == "" or tplPath == "" then
if pagePath == "" or tplPath == "" then
return ""
return ""
end
end
 
 
local moduleName = JsonPaths.get(pagePath)
local data = p.loadCachedData(pagePath)
local data = load_cached_data(moduleName)
if not data then
if not data then
return ""
return ""
end
end
 
 
local idsTable = data.id or data
local idsTable = data.id
if type(idsTable) ~= "table" then
if type(idsTable) ~= "table" then
return ""
return ""
end
end
 
 
local out = {}
local out = {}
 
 
for idKey in pairs(idsTable) do
for idKey in pairs(idsTable) do
local entry = resolve_entry_from_data(data, idKey)
local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs)
if entry ~= nil then
if tpl ~= "" then
local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs, entry, true)
out[#out + 1] = tpl
if tpl ~= "" then
out[#out + 1] = tpl
end
end
end
end
end
Строка 1624: Строка 1667:
end
end


local moduleName = JsonPaths.get(dataPage)
local entry = load_entry(id, dataPage) or {}
local data = load_cached_data(moduleName)
if not data then
return ""
end
 
local entry = resolve_entry(data, id) or {}
local parts = flatten_selected_parts(entry, paramNames)
local parts = flatten_selected_parts(entry, paramNames)
if #parts == 0 then
if #parts == 0 then