Модуль:Prototypes/Механика/Исследование

Для документации этого модуля может быть создана страница Модуль:Prototypes/Механика/Исследование/doc

local p = {}
local dataCache = nil  -- Кэш для данных плат
local researchDataCache = nil  -- Кэш для данных исследований
local machineIDCache = {}  -- Кэш для машинных ID
local boardIndex = {}  -- Индекс по ID плат
local machineIndex = {}  -- Индекс по имени машин

-- Функция для загрузки данных плат из JSON-файла и создания индексов
local function loadData()
	if not dataCache then
		dataCache = mw.text.jsonDecode(mw.title.new("User:IanComradeBot/entity_prototypes.json"):getContent())
		
		-- Создание индекса плат по ID и машин по имени
		for id, entity in pairs(dataCache) do
			boardIndex[id] = entity
			-- Индексируем только машинные или консольные платы
			if entity.name and (entity.name:find("%(машинная плата%)") or entity.name:find("%(консольная плата%)")) then
				local machineName = entity.name:gsub(" %(машинная плата%)", ""):gsub(" %(консольная плата%)", "")
				machineIndex[machineName] = entity.id
			end
		end
	end
	return dataCache
end

-- Функция для загрузки данных исследований из JSON-файла
local function loadResearchData()
	if not researchDataCache then
		researchDataCache = mw.text.jsonDecode(mw.title.new("User:IanComradeBot/research_prototypes.json"):getContent())
	end
	return researchDataCache
end

-- Функция для перевода ID плат в ID машин с кэшированием и использованием индексов
local function translateBoardIDToMachineID(boardID)
	if machineIDCache[boardID] then
		return machineIDCache[boardID]
	end

	-- Используем индекс для быстрого поиска платы
	local board = boardIndex[boardID]
	if not board then
		return nil
	end

	-- Проверка на машинные или консольные платы
	local machineName = board.name:gsub(" %(машинная плата%)", ""):gsub(" %(консольная плата%)", "")
	local machineID = machineIndex[machineName]

	-- Кэшируем результат
	machineIDCache[boardID] = machineID or nil
	return machineID
end

function p.main(frame)
	-- Подключение CSS
	local cssLink = frame:extensionTag('templatestyles', '', {
		src = 'Шаблон:Research/styles.css'
	})

	local id = frame.args.id or ""
	local icon = frame.args.icon or ""
	local customRecipeUnlocks = frame.args.customRecipeUnlocks or nil

	-- Обработка индексов для замены рецептов
	local customRecipeUnlocksIndexes = {}
	local i = 1
	
	while true do
		local customIndex = frame.args["customRecipeUnlocksIndex" .. i]
		if not customIndex then
			break  -- Прекращаем цикл, если нет больше индексов
		end
	
		table.insert(customRecipeUnlocksIndexes, {index = i, recipe = customIndex})
		i = i + 1  -- Увеличиваем индекс
	end

	-- Загрузка данных исследований из кэша
	local data = loadResearchData()

	local out = cssLink
	local found = false
	local disciplineName = ""

	-- Определение дисциплины и отображение исследований
	for discipline, technologies in pairs(data) do
		for _, tech in ipairs(technologies) do
			if tech.id == id then
				found = true
				disciplineName = ({
					Arsenal = "Арсенал",
					Industrial = "Промышленность",
					Experimental = "Экспериментальное",
					CivilianServices = "Обслуживание персонала"
				})[discipline]

				local tierColor = ({
					[1] = "#54d554",
					[2] = "#ed9000",
					[3] = "#d72a2a"
				})[tech.tier]

				out = out .. '<div class="research" id="' .. discipline .. '">'
				out = out .. '<div class="research__images">[[Файл:' .. icon .. '.png|64x64px|центр|link=]]</div>'
				out = out .. '<div class="research__name">' .. tech.name .. '[[Файл:' .. discipline .. '.png|16px|link=]]</div>'
				out = out .. '<div class="research__type">'
				out = out .. '<div>Уровень: <span style="color:' .. tierColor .. ';">' .. tech.tier .. '</span></div>'
				out = out .. '<div class="research__technology">' .. disciplineName .. '</div>'
				out = out .. '<div>Стоимость: <span style="color:#DA70D6;">' .. tech.cost .. '</span></div>'
				out = out .. '</div>'
				out = out .. '<div class="research__unblocks">Разблокирует:'
				out = out .. '<ul>'

				-- Используем кастомные recipeUnlocks или значения из json файла 
				local recipeUnlocks = customRecipeUnlocks and mw.text.split(customRecipeUnlocks, " ") or tech.recipeUnlocks

				-- Замена рецептов по индексам
				for _, customIndex in ipairs(customRecipeUnlocksIndexes) do
					if recipeUnlocks[customIndex.index] then
						recipeUnlocks[customIndex.index] = customIndex.recipe
					end
				end

				for _, recipe in ipairs(recipeUnlocks) do
					local machineID = translateBoardIDToMachineID(recipe) or recipe

					out = out .. frame:preprocess('<li>{{#invoke:Entity Lookup|createimagetooltip|Файл:' .. machineID .. '.png|' .. machineID .. '|Мета=32x32px,link=}} {{#invoke:Entity Lookup|getname|' .. machineID .. '}}</li>')
				end

				out = out .. '</ul>'
				out = out .. '</div>'
				out = out .. '</div>'
			end
		end
	end

	if not found then
		out = out .. '<div style="color:red;">Нет доступных исследований.</div>'
	end

	return out
end

return p