Модуль:Сущность/data: различия между версиями
Pok (обсуждение | вклад) Отмена версии 213664, сделанной Pok (обсуждение) Метки: отмена отменено |
Pok (обсуждение | вклад) Нет описания правки |
||
| (не показано 16 промежуточных версий этого же участника) | |||
| Строка 2: | Строка 2: | ||
local getArgs = require('Module:Arguments').getArgs | local getArgs = require('Module:Arguments').getArgs | ||
local JsonPaths = require('Module:JsonPaths') | local JsonPaths = require('Module:JsonPaths') | ||
local dp = require( "Module:GetField") | |||
local function load_data_page(page) | |||
local moduleName = JsonPaths.get(page) | |||
local ok, data = pcall(mw.loadData, moduleName) | |||
if not ok or type(data) ~= "table" then | |||
return nil | |||
end | |||
return data | |||
end | |||
local switchModeRegistry = {} | local switchModeRegistry = {} | ||
local switchModeOrder = {} | local switchModeOrder = {} | ||
| Строка 16: | Строка 21: | ||
end | end | ||
local function | local function each_csv_value(str, fn) | ||
if not str or str == "" then | |||
return | |||
return | |||
end | end | ||
for item in string.gmatch(str, "[^,]+") do | |||
local value = trim(item) | |||
if value ~= "" then | |||
fn(value) | |||
end | |||
end | end | ||
end | end | ||
local function load_template_content(path) | local function load_template_content(path) | ||
local title = mw.title.new("Template:" .. path) | local title = mw.title.new("Template:" .. path) | ||
if not title then | if not title then | ||
return nil | return nil | ||
end | end | ||
local ok, content = pcall( | local ok, content = pcall(title.getContent, title) | ||
if not ok then | if not ok then | ||
return nil | return nil | ||
end | end | ||
return content | |||
end | |||
function p.loadEntityComponents(entityId) | |||
return | if not entityId or entityId == "" then | ||
return {} | |||
end | |||
local entity = dp.loadEntityData(entityId) | |||
if entity then | |||
return dp.collectEntityComponents(entity) | |||
end | |||
return {} | |||
end | end | ||
function p.entityHasComponent(entityOrId, compName) | |||
if not | if not compName or compName == "" then | ||
return false | |||
end | |||
local entity | |||
if type(entityOrId) == "string" then | |||
entity = dp and dp.loadEntityData(entityOrId) or load_data_page("prototype/Entity/" .. entityOrId .. ".json") | |||
if not entity then | |||
return false | |||
end | |||
else | |||
entity = entityOrId | |||
end | |||
return dp.entityHasComponent(entity, compName) | |||
end | end | ||
| Строка 96: | Строка 112: | ||
local function get_template_params(tplPath, content) | local function get_template_params(tplPath, content) | ||
return collect_template_params(content) | |||
end | end | ||
| Строка 113: | Строка 122: | ||
end | end | ||
local function make_source(kind, name | local function make_source(kind, name, tplPath) | ||
return { kind = kind, name = name | return { kind = kind, name = name, tplPath = tplPath } | ||
end | end | ||
| Строка 136: | Строка 145: | ||
if not byKey[key] then | if not byKey[key] then | ||
byKey[key] = {} | byKey[key] = {} | ||
state.keyOrder[sw][#state.keyOrder[sw] + 1] = key | |||
end | end | ||
return byKey[key] | return byKey[key] | ||
| Строка 184: | Строка 193: | ||
line = line .. '<div class="ts-Сущность-field">' .. makeSourceLink(src) .. '</div>' | line = line .. '<div class="ts-Сущность-field">' .. makeSourceLink(src) .. '</div>' | ||
end | end | ||
parts[#parts + 1] = '<div class="ts-Сущность">' .. line .. '</div>' | |||
end | end | ||
end | end | ||
| Строка 252: | Строка 261: | ||
return table.concat(out, "\n") | return table.concat(out, "\n") | ||
end | |||
local function normalizeFilterKey(s) | |||
s = trim(s or "") | |||
s = s:gsub("%s*_%s*", "_") | |||
return s | |||
end | end | ||
| Строка 258: | Строка 273: | ||
return false | return false | ||
end | end | ||
callKey = normalizeFilterKey(callKey) | |||
compositeKey = normalizeFilterKey(compositeKey) | |||
return list[callKey] or list[compositeKey] or false | return list[callKey] or list[compositeKey] or false | ||
end | end | ||
| Строка 337: | Строка 354: | ||
local isWhitelisted = cardFilter and matches_card_list(cardFilter.whitelist, callKey, compositeKey) or | local isWhitelisted = cardFilter and matches_card_list(cardFilter.whitelist, callKey, compositeKey) or | ||
false | |||
local isBlacklisted = cardFilter and matches_card_list(cardFilter.blacklist, callKey, compositeKey) or | local isBlacklisted = cardFilter and matches_card_list(cardFilter.blacklist, callKey, compositeKey) or | ||
false | |||
if isWhitelisted and content ~= "" then | if isWhitelisted and content ~= "" then | ||
rawContentParts[#rawContentParts + 1] = content | rawContentParts[#rawContentParts + 1] = content | ||
| Строка 351: | Строка 368: | ||
if not merged.sectionsMap[section] then | if not merged.sectionsMap[section] then | ||
merged.sectionsMap[section] = true | merged.sectionsMap[section] = true | ||
merged.sections[#merged.sections + 1] = section | |||
end | end | ||
if displayLabel ~= "" and (not merged.labelOverrides[compositeKey] or merged.labelOverrides[compositeKey] == "") then | if displayLabel ~= "" and (not merged.labelOverrides[compositeKey] or merged.labelOverrides[compositeKey] == "") then | ||
| Строка 369: | Строка 386: | ||
merged.labelSets[section][compositeKey] = true | merged.labelSets[section][compositeKey] = true | ||
local cur = merged.labelLists[section] or {} | local cur = merged.labelLists[section] or {} | ||
cur[#cur + 1] = compositeKey | |||
merged.labelLists[section] = cur | merged.labelLists[section] = cur | ||
end | end | ||
| Строка 377: | Строка 394: | ||
if not merged.tagSet[tagText] then | if not merged.tagSet[tagText] then | ||
merged.tagSet[tagText] = true | merged.tagSet[tagText] = true | ||
merged.tags[#merged.tags + 1] = tagText | |||
end | end | ||
end | end | ||
| Строка 383: | Строка 400: | ||
end | end | ||
end | end | ||
each_csv_value(frame.args.cardTag or "", function(extraTag) | |||
if not merged.tagSet[extraTag] then | |||
merged.tagSet[extraTag] = true | |||
merged.tags[#merged.tags + 1] = extraTag | |||
end | |||
end) | |||
local out = {} | local out = {} | ||
| Строка 485: | Строка 509: | ||
local function getTemplateMeta(frame, tplPath) | local function getTemplateMeta(frame, tplPath) | ||
local expanded = frame:expandTemplate { | local expanded = frame:expandTemplate { | ||
title = tplPath, | title = tplPath, | ||
| Строка 496: | Строка 516: | ||
local ok, data = pcall(mw.text.jsonDecode, expanded) | local ok, data = pcall(mw.text.jsonDecode, expanded) | ||
if not ok or type(data) ~= "table" then | if not ok or type(data) ~= "table" then | ||
return "" | return "" | ||
end | end | ||
| Строка 508: | Строка 527: | ||
if not seen[lab] then | if not seen[lab] then | ||
seen[lab] = true | seen[lab] = true | ||
cardKeys[#cardKeys + 1] = lab | |||
end | end | ||
end | end | ||
| Строка 516: | Строка 535: | ||
end | end | ||
return data | return data | ||
end | end | ||
| Строка 524: | Строка 542: | ||
if not str or str == "" then return res end | if not str or str == "" then return res end | ||
for item in string.gmatch(str, "[^,]+") do | for item in string.gmatch(str, "[^,]+") do | ||
local s = | local s = normalizeFilterKey(item) | ||
if s ~= "" then | if s ~= "" then | ||
local a, b = s:match("^([^_]+)_(.+)$") | local a, b = s:match("^([^_]+)_(.+)$") | ||
| Строка 576: | Строка 594: | ||
end | end | ||
return not (filter.blacklist[sw] and filter.blacklist[sw][key]) | return not (filter.blacklist[sw] and filter.blacklist[sw][key]) | ||
end | end | ||
| Строка 614: | Строка 620: | ||
end | end | ||
local function collect_entity_sets(id | local function collect_entity_sets(id, prototypeStoreDefs, componentWhitelist, componentBlacklist, prototypeWhitelist, prototypeBlacklist) | ||
local foundComponents, foundPrototypes = {}, {} | local foundComponents, foundPrototypes = {}, {} | ||
local compList = | local compList = p.loadEntityComponents(id) | ||
if type(compList) == "table" then | if type(compList) == "table" then | ||
for _, v in ipairs(compList) do | for _, v in ipairs(compList) do | ||
if type(v) == "string" then | if type(v) == "string" and v ~= "" then | ||
foundComponents[v] = true | foundComponents[v] = true | ||
end | end | ||
| Строка 627: | Строка 632: | ||
end | end | ||
local protoStore = prototypeStoreDefs and prototypeStoreDefs[id] | |||
if type(protoStore) == "table" then | |||
if | for protoName in pairs(protoStore) do | ||
if type(protoName) == "string" and protoName ~= "" then | |||
foundPrototypes[protoName] = true | |||
foundPrototypes[ | |||
end | end | ||
end | end | ||
end | end | ||
apply_entity_set_filters(foundComponents, parse_csv_set(componentWhitelist), parse_csv_set(componentBlacklist)) | apply_entity_set_filters(foundComponents, parse_csv_set(componentWhitelist), parse_csv_set(componentBlacklist)) | ||
| Строка 658: | Строка 662: | ||
end | end | ||
local function get_selective_extra( | local function get_selective_extra(id, dataPage, paramNames) | ||
if not dp or type(dp. | if not dp or type(dp.flattenFieldSelectiveDirect) ~= "function" then | ||
return "" | return "" | ||
end | end | ||
| Строка 666: | Строка 670: | ||
end | end | ||
local | return dp.flattenFieldSelectiveDirect(id, dataPage, paramNames) or "" | ||
if | end | ||
return | |||
local function add_card_tag_value(tags, seen, value) | |||
value = trim(value or "") | |||
if value == "" or seen[value] then | |||
return | |||
end | end | ||
seen[value] = true | |||
tags[#tags + 1] = value | |||
end | |||
local | local function merge_card_tag_text(...) | ||
local tags = {} | |||
local seen = {} | |||
for i = 1, select("#", ...) do | |||
each_csv_value(select(i, ...), function(value) | |||
add_card_tag_value(tags, seen, value) | |||
end) | |||
end | end | ||
return table.concat(tags, ", ") | |||
end | end | ||
| Строка 711: | Строка 725: | ||
end | end | ||
local function | local function extract_whitelist_search_strings(keyFilter) | ||
if not keyFilter or not keyFilter.hasWhitelist then | |||
return nil | |||
end | end | ||
local strings = {} | |||
local | for sw, keys in pairs(keyFilter.whitelist) do | ||
if type(keys) == "table" then | |||
for key in pairs(keys) do | |||
for | strings[#strings + 1] = key | ||
for | |||
end | end | ||
end | end | ||
end | end | ||
return | |||
if #strings == 0 then | |||
return nil | |||
end | |||
return strings | |||
end | end | ||
function | local function content_matches_whitelist(content, searchStrings) | ||
if not searchStrings then | |||
return true | |||
if | end | ||
if not content then | |||
return false | |||
end | |||
for _, s in ipairs(searchStrings) do | |||
if string.find(content, s, 1, true) then | |||
return true | |||
end | |||
end | |||
return false | |||
end | |||
local componentWhitelist = args.componentWhitelist or args.componentwhitelist or "" | local function each_entity_data(frame, id, onEntity, onMissing, keyFilter) | ||
local componentBlacklist = args.componentBlacklist or args.componentblacklist or "" | local componentWhitelist = frame.args.componentWhitelist or frame.args.componentwhitelist or "" | ||
local prototypeWhitelist = args.prototypeWhitelist or args.prototypewhitelist or "" | local componentBlacklist = frame.args.componentBlacklist or frame.args.componentblacklist or "" | ||
local prototypeBlacklist = args.prototypeBlacklist or args.prototypeblacklist or "" | local prototypeWhitelist = frame.args.prototypeWhitelist or frame.args.prototypewhitelist or "" | ||
local prototypeBlacklist = frame.args.prototypeBlacklist or frame.args.prototypeblacklist or "" | |||
local | local prototypeStoreDefs = dp.loadCachedData("prototype_store.json") | ||
if not prototypeStoreDefs then | |||
if | return false | ||
end | |||
local foundComponents, foundPrototypes = collect_entity_sets(id | local foundComponents, foundPrototypes = collect_entity_sets(id, prototypeStoreDefs, | ||
componentWhitelist, componentBlacklist, prototypeWhitelist, prototypeBlacklist) | componentWhitelist, componentBlacklist, prototypeWhitelist, prototypeBlacklist) | ||
local | local compWhitelistSet = parse_csv_set(componentWhitelist) | ||
local | local compBlacklistSet = parse_csv_set(componentBlacklist) | ||
local protoWhitelistSet = parse_csv_set(prototypeWhitelist) | |||
local protoBlacklistSet = parse_csv_set(prototypeBlacklist) | |||
local compHasWhitelist = next(compWhitelistSet) ~= nil | |||
local protoHasWhitelist = next(protoWhitelistSet) ~= nil | |||
local anyEntityWhitelist = compHasWhitelist or protoHasWhitelist | |||
local whitelistSearchStrings = extract_whitelist_search_strings(keyFilter) | |||
local function processEntity(kind, name, isStore) | |||
local tplPath = kind .. "/" .. dp.ucfirst(name) | |||
if isStore then | |||
tplPath = tplPath .. "/store" | |||
end | |||
local content = load_template_content(tplPath) | local content = load_template_content(tplPath) | ||
if not content then | if not content then | ||
if | if onMissing then | ||
onMissing(kind, name, isStore, tplPath) | |||
end | end | ||
return | |||
end | |||
if not content_matches_whitelist(content, whitelistSearchStrings) then | |||
return | return | ||
end | end | ||
| Строка 797: | Строка 813: | ||
local extra = "" | local extra = "" | ||
local paramNames = get_template_params(tplPath, content) | local paramNames = get_template_params(tplPath, content) | ||
if | if dp then | ||
local dataPage = | local dataPage = kind .. "/" .. dp.ucfirst(name) .. ".json" | ||
extra = get_selective_extra( | extra = get_selective_extra(id, dataPage, paramNames) | ||
end | end | ||
onEntity(parsed, { | |||
tplPath = tplPath, | tplPath = tplPath, | ||
id = id, | id = id, | ||
extra = extra, | extra = extra, | ||
source = make_source(kind, name | source = make_source(kind, name, tplPath), | ||
priority = resolve_priority(parsed) | priority = resolve_priority(parsed) | ||
} | }) | ||
end | end | ||
for compName in pairs(foundComponents) do | for compName in pairs(foundComponents) do | ||
processEntity("component", compName, false) | if not anyEntityWhitelist or compHasWhitelist then | ||
processEntity("component", compName, false) | |||
end | |||
end | end | ||
for protoName in pairs(foundPrototypes) do | for protoName in pairs(foundPrototypes) do | ||
processEntity("prototype", protoName, false) | if not anyEntityWhitelist or protoHasWhitelist then | ||
processEntity("prototype", protoName, false) | |||
end | |||
end | end | ||
local componentStoreDefs = | local componentStoreDefs = dp.loadCachedData("component_store.json") | ||
if type(componentStoreDefs) == "table" then | if type(componentStoreDefs) == "table" and (not anyEntityWhitelist or compHasWhitelist) then | ||
local compStore = componentStoreDefs[id] | local compStore = componentStoreDefs[id] | ||
if type(compStore) == "table" then | if type(compStore) == "table" then | ||
for compName in pairs(compStore) do | for compName in pairs(compStore) do | ||
processEntity("component", compName, true) | local allowed = true | ||
if compBlacklistSet[compName] then | |||
allowed = false | |||
elseif compHasWhitelist and not compWhitelistSet[compName] then | |||
allowed = false | |||
end | |||
if allowed then | |||
processEntity("component", compName, true) | |||
end | |||
end | end | ||
end | end | ||
end | end | ||
if type(prototypeStoreDefs) == "table" then | if type(prototypeStoreDefs) == "table" and (not anyEntityWhitelist or protoHasWhitelist) then | ||
local protoStore = prototypeStoreDefs[id] | local protoStore = prototypeStoreDefs[id] | ||
if type(protoStore) == "table" then | if type(protoStore) == "table" then | ||
for protoName in pairs(protoStore) do | for protoName in pairs(protoStore) do | ||
processEntity("prototype", protoName, true) | local allowed = true | ||
if protoBlacklistSet[protoName] then | |||
allowed = false | |||
elseif protoHasWhitelist and not protoWhitelistSet[protoName] then | |||
allowed = false | |||
end | |||
if allowed then | |||
processEntity("prototype", protoName, true) | |||
end | |||
end | |||
end | |||
end | |||
return true | |||
end | |||
local function collect_card_tag_text(frame, args, id) | |||
local filter = build_key_filter(args) | |||
local cardFilter = build_render_options(filter).cardFilter | |||
local entries = {} | |||
local ok = each_entity_data(frame, id, function(parsed, ctx) | |||
local keys = parsed.card or {} | |||
if type(keys) == "table" then | |||
for _, key in ipairs(keys) do | |||
local compositeKey = (key:find("_", 1, true)) and key or ("Сущность_" .. key) | |||
local isWhitelisted = matches_card_list(cardFilter.whitelist, key, compositeKey) | |||
local isBlacklisted = matches_card_list(cardFilter.blacklist, key, compositeKey) | |||
local allowCardEntry = not isWhitelisted and not isBlacklisted and | |||
((not cardFilter.hasWhitelist) or matches_card_list(cardFilter.cardWhitelist, key, compositeKey)) | |||
if allowCardEntry then | |||
entries[#entries + 1] = { | |||
tplTag = makeTplCall(ctx.tplPath, "cardTag", key, ctx.id, ctx.extra), | |||
priority = ctx.priority, | |||
idx = #entries + 1 | |||
} | |||
end | |||
end | |||
end | |||
if parsed.cardTag and parsed.cardTag ~= "" then | |||
entries[#entries + 1] = { | |||
tplTag = makeTplCall(ctx.tplPath, "cardTag", "cardTag", ctx.id, ctx.extra), | |||
priority = ctx.priority, | |||
idx = #entries + 1 | |||
} | |||
end | |||
end) | |||
if not ok then | |||
return trim(args.tag or "") | |||
end | |||
sort_entries_by_priority(entries) | |||
local tags = {} | |||
local seen = {} | |||
for _, entry in ipairs(entries) do | |||
local tagText = trim(frame:preprocess(entry.tplTag or "") or "") | |||
add_card_tag_value(tags, seen, tagText) | |||
end | |||
return merge_card_tag_text(table.concat(tags, ", "), args.tag) | |||
end | |||
p.mergeCardTagText = merge_card_tag_text | |||
p.collectCardTagText = collect_card_tag_text | |||
local function build_missing_template_error(kind, name, isStore, tplPath) | |||
local baseType = (kind and (kind:sub(1, 1):upper() .. kind:sub(2)) or "") | |||
local classType = baseType | |||
if isStore then | |||
classType = classType .. "Store" | |||
end | |||
local className = name .. baseType | |||
local tplLabel = "Template:" .. tplPath | |||
return "{{сущность/infobox/base|тип=" .. classType .. "|название=" .. className .. "|ссылка=" .. tplLabel .. "}}" | |||
end | |||
local function renderBlocks(frame, state, renderOptions, entityId, showSource) | |||
local outLocal = {} | |||
local noHeaders = renderOptions and renderOptions.noHeaders | |||
local cardFilter = renderOptions and renderOptions.cardFilter | |||
for _, sw in ipairs(switchModeOrder) do | |||
local mode = switchModeRegistry[sw] or {} | |||
if mode.full then | |||
local outStr = "" | |||
if type(mode.render_full) == "function" then | |||
outStr = mode.render_full(frame, state.keyOrder[sw], state.keyToTemplates[sw], state.keySources[sw], | |||
entityId, noHeaders, showSource, cardFilter) | |||
end | |||
if outStr and outStr ~= "" then outLocal[#outLocal + 1] = outStr end | |||
else | |||
for _, key in ipairs(state.keyOrder[sw] or {}) do | |||
local entries = state.keyToTemplates[sw][key] or {} | |||
if type(mode.render_key) == "function" then | |||
local outStr = mode.render_key(frame, key, entries, noHeaders, showSource) | |||
if outStr and outStr ~= "" then outLocal[#outLocal + 1] = outStr end | |||
end | |||
end | end | ||
end | end | ||
end | end | ||
return outLocal | |||
end | |||
function p.get(frame) | |||
local args = getArgs(frame, { removeBlanks = false }) | |||
local id = args[1] or "" | |||
if id == "" then return "" end | |||
local showSource = trim(args.showSource or "") == "" | |||
local filter = build_key_filter(args) | |||
local renderOptions = build_render_options(filter) | |||
local state = new_switch_state() | |||
local errors = {} | |||
local ok = each_entity_data(frame, id, function(parsed, ctx) | |||
add_entries_from_meta(state, parsed, ctx, filter, false) | |||
end, function(kind, name, isStore, tplPath) | |||
if not filter.hasWhitelist then | |||
errors[#errors + 1] = build_missing_template_error(kind, name, isStore, tplPath) | |||
end | |||
end, filter) | |||
if not ok then return "" end | |||
local out = {} | local out = {} | ||
if #errors > 0 then | if #errors > 0 then | ||
out[#out + 1] = '{{сущность/infobox|' .. table.concat(errors, "\n") .. '}}' | |||
end | end | ||
| Строка 847: | Строка 998: | ||
local blocks = renderBlocks(frame, state, renderOptions, id, showSource) | local blocks = renderBlocks(frame, state, renderOptions, id, showSource) | ||
for _, b in ipairs(blocks) do | for _, b in ipairs(blocks) do | ||
out[#out + 1] = b | |||
end | end | ||
| Строка 877: | Строка 1028: | ||
id = "", | id = "", | ||
extra = "", | extra = "", | ||
source = make_source("" | source = make_source("", tplPath, tplPath), | ||
priority = 1 | priority = 1 | ||
}, nil, true) | }, nil, true) | ||
local hasWhitelist = previewFilter.hasWhitelist | local hasWhitelist = previewFilter.hasWhitelist | ||
renderOptions.noHeaders = hasWhitelist | renderOptions.noHeaders = hasWhitelist | ||
| Строка 888: | Строка 1038: | ||
local blocks = renderBlocks(frame, state, renderOptions, "", showSource) | local blocks = renderBlocks(frame, state, renderOptions, "", showSource) | ||
for _, b in ipairs(blocks) do | for _, b in ipairs(blocks) do | ||
out[#out + 1] = b | |||
end | end | ||