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

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

local p = {}

-- Глобальная переменная для данных
local data 

-- Хранилище для кэша перевода
local boardToMachineCache = {}

-- Функция для перевода ID плат в ID машин
local function translateBoardIDToMachineID(boardID)
	-- Проверяем кэш
	if boardToMachineCache[boardID] ~= nil then
		return boardToMachineCache[boardID]
	end

	-- Если данные еще не загружены, загружаем их
	if not data then
		data = mw.text.jsonDecode(mw.title.new("Участник:IanComradeBot/entity_prototypes.json"):getContent())
	end

	-- Поиск платы по ID и извлечение её названия
	local board = data[boardID]
	if not board or not board.name then
		return nil -- Если платы нет в данных, возвращаем nil
	end

	-- Проверка наличия "(машинная плата)" или "(консольная плата)" в имени
	if not (board.name:find("%(машинная плата%)") or board.name:find("%(консольная плата%)")) then
		return nil -- Если в имени нет одной из фраз, возвращаем nil
	end

	-- Удаление "(машинная плата)" и "(консольная плата)" из имени платы
	local machineName = board.name:gsub(" %(машинная плата%)", ""):gsub(" %(консольная плата%)", ""):trim()

	-- Список исключений
	local excludeWords = {"Unanchored", "Debug", "Admin"}

	-- Оптимизация: создаем таблицу для быстрого поиска по имени
	local machineID = nil
	for _, entity in pairs(data) do
		if entity.name == machineName then
			-- Проверка на наличие исключений в ID
			local shouldExclude = false
			for _, word in ipairs(excludeWords) do
				if entity.id:find(word) then
					shouldExclude = true
					break
				end
			end
			
			if not shouldExclude then
				machineID = entity.id -- Сохраняем ID машины
				break
			end
		end
	end

	-- Кэшируем результат (nil, если не найдено)
	boardToMachineCache[boardID] = machineID
	return machineID -- Возвращаем ID машины или nil
end

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

	local id = frame.args.id or "" -- ID исследования
	local icon = frame.args.icon or "" -- Изобрадения исследования
	local customRecipeUnlocks = frame.args.customRecipeUnlocks or nil -- Ручной ввод в список разблокируемых технологий

	-- Загрузка данных из JSON один раз
	if not data then
		data = mw.text.jsonDecode(mw.title.new("User:IanComradeBot/research_prototypes.json"):getContent())
	end

	local out = cssLink
	local found = false  -- Флаг для отслеживания наличия элементов
	local disciplineName = "" 

	-- Определяем дисциплину на основе переданного id
	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]

				-- Формирование HTML
				out = out .. '<div class="research" id="' .. discipline .. '">'
				out = out .. '<div class="research__images">[[Файл:' .. icon .. '.png|64x64px|центр|link=]]</div>' -- Используем переданный icon
				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
				if customRecipeUnlocks then
					recipeUnlocks = mw.text.split(customRecipeUnlocks, " ") -- Разбиваем по пробелам
				else
					recipeUnlocks = tech.recipeUnlocks
				end

				-- Проверка каждого ID в recipeUnlocks
				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