Модуль:Research

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

local p = {}
local boardIndex = {}  -- Индекс по ID плат
local machineIndex = {}  -- Индекс по имени машин

-- Функция для загрузки данных плат из JSON-файла и создания индексов
local function loadData()
	local data = mw.text.jsonDecode(mw.title.new("Участник:IanComradeBot/entity_prototypes.json"):getContent())

	-- Создание индекса плат по ID и машин по имени
	for id, entity in pairs(data) 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
	return data
end

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

-- Функция для перевода ID плат в ID машин с использованием индексов
local function translateBoardIDToMachineID(boardID)
	-- Используем индекс для быстрого поиска платы
	local board = boardIndex[boardID]
	if not board then
		return nil
	end

	-- Проверка на машинные или консольные платы
	local machineName = board.name:gsub(" %(машинная плата%)", ""):gsub(" %(консольная плата%)", "")
	return machineIndex[machineName]
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 = {}
	for i = 1, 10 do
		local customIndex = frame.args["customRecipeUnlocksIndex" .. i]
		if customIndex then
			table.insert(customRecipeUnlocksIndexes, {index = i, recipe = customIndex})
		end
	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