Модуль:Chemistry Lookup

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

local prototypes = mw.loadData("Module:Chemistry Lookup/data")

local p = {}
p.chem = prototypes.chem or {}
p.react = prototypes.react or {}
p.status = prototypes.status or {}
p.groupDirection = prototypes.groupDirection or {}

-- ===== Хелперы =====
local function tablelength(T)
	local count = 0
	for _ in pairs(T or {}) do count = count + 1 end
	return count
end

local function hchangelist(l, frame)
	local out = ""
	local len = tablelength(l)
	local i = 0
	for k, v in pairs(l) do
		i = i + 1
		if len == i and i ~= 1 then
			out = out .. ", and "
		elseif i ~= 1 then
			out = out .. ", "
		end
		out = out .. frame:expandTemplate{ title = "HealthModifier", args = { adj = v, kind = k } }
	end
	return out
end

function p.readscalar(frame)
	local a, b = frame.args[1], frame.args[2]
	if not (a and b) or not p.chem[a] then return "" end
	local v = p.chem[a][b]
	if v == nil then return "" end
	return mw.text.nowiki(tostring(v))
end

function p.readscalarreact(frame)
	local a, b = frame.args[1], frame.args[2]
	if not (a and b) or not p.react[a] then return "" end
	local v = p.react[a][b]
	if v == nil then return "" end
	return mw.text.nowiki(tostring(v))
end

function p.getcolor(frame)
	local a = frame.args[1]
	if not a or not p.chem[a] or not p.chem[a].color then return "" end
	return mw.text.nowiki(p.chem[a].color:sub(1, 7))
end

function p.gettextcolor(frame)
	local a = frame.args[1]
	if not a or not p.chem[a] or not p.chem[a].color then return mw.text.nowiki("#000") end
	local basecol = p.chem[a].color
	local red = tonumber(basecol:sub(2, 3), 16) or 0
	local grn = tonumber(basecol:sub(4, 5), 16) or 0
	local blu = tonumber(basecol:sub(6, 7), 16) or 0
	local luminance = math.sqrt(0.241*red*red + 0.691*grn*grn + 0.068*blu*blu)
	if luminance > 100 then
		return mw.text.nowiki("#000")
	else
		return mw.text.nowiki("#FFF")
	end
end

function p.hasrecipe(frame)
	local chem = frame.args[1]
	if not chem or not p.chem[chem] then return "" end
	return (p.chem[chem].recipes and p.chem[chem].recipes[1]) and "true" or ""
end

function p.buildboxes(frame)
	local out = ""
	local group = frame.args[1]
	for k, v in pairs(p.chem) do
		if type(v) == "table" then
			if (group == nil) or (v.group == group) then
				out = out .. frame:expandTemplate{ title = "Chembox", args = { prototype = k }}
			end
		end
	end
	return out
end

function p.buildreactboxes(frame)
	local out = ""
	for k, v in pairs(p.react) do
		if type(v) == "table" and v.effects and tablelength(v.effects) ~= 0 then
			out = out .. frame:expandTemplate{ title = "Reactionbox", args = { reaction = k }}
		end
	end
	return out
end

function p.buildrecipes(frame)
	local chem = frame.args[1]
	if not chem or not p.chem[chem] then return "" end
	local out = ""
	for _, recipe in pairs(p.chem[chem].recipes or {}) do
		out = out .. p.buildreaction(frame, recipe)
	end
	return out
end

-- Версия для консоли отладки без frame
function p.buildrecipes_debug(chem)
	if not chem or not p.chem[chem] then return "Химикат не найден" end
	local out = "=== Рецепты для " .. chem .. " ===\n"
	for i, recipe in ipairs(p.chem[chem].recipes or {}) do
		out = out .. "Рецепт " .. i .. ": " .. recipe .. "\n"
		if p.react[recipe] then
			local data = p.react[recipe]
			out = out .. "  Реактанты:\n"
			for k, v in pairs(data.reactants or {}) do
				out = out .. "    - " .. k .. ": " .. (v.amount or "?") .. "\n"
			end
			out = out .. "  Продукты:\n"
			for k, v in pairs(data.products or {}) do
				out = out .. "    - " .. k .. ": " .. tostring(v) .. "\n"
			end
		end
		out = out .. "\n"
	end
	return out
end

function p.buildreactionext(frame)
	local react = frame.args[1]
	return p.buildreaction(frame, react)
end

function p.buildreaction(frame, react)
	if not react or not p.react[react] then return "" end
	
	-- Проверка для консоли отладки
	if not frame or not frame.expandTemplate then
		return "Рецепт: " .. react .. " (для полного отображения используйте в MediaWiki)"
	end
	
	local data = p.react[react]
	local args = {}

	-- Реактанты
	local i = 0
	for k, v in pairs(data.reactants or {}) do
		i = i + 1
		local dest = p.groupDirection[(p.chem[k] or {}).group] or "Chemistry"
		args["component-" .. i] = frame:expandTemplate{
			title = "Chem Recipe Component",
			args = { reagent = k, amount = v.amount, dest = dest }
		}
	end

	-- Продукты
	i = 0
	for k, v in pairs(data.products or {}) do
		i = i + 1
		local dest = p.groupDirection[(p.chem[k] or {}).group] or "Chemistry"
		args["result-" .. i] = frame:expandTemplate{
			title = "Chem Recipe Component",
			args = { reagent = k, amount = v, dest = dest }
		}
	end

	-- Эффекты реакции (не метаболизм)
	if data.effects ~= nil then
		args.effects = p.geneffectlist(data.effects, frame, 1)
	end

	return frame:expandTemplate{ title = "Chem Box Recipe", args = args }
end

local function scan_metab(chem)
	local met = (p.chem[chem] or {}).metabolisms
	if not met then return nil end
	return met
end

function p.checksatiatesthirst(frame)
	local chem = frame.args[1]
	local met = scan_metab(chem)
	if not met then return "" end
	for _, v in pairs(met) do
		for _, w in pairs(v.effects or {}) do
			-- Исправляем проверку для JSON структуры
			local effectName = (type(w.type) == "table" and w.type.name) or w.type or ""
			if effectName == "SatiateThirst" then return "1" end
		end
	end
	return ""
end

function p.checksatiateshunger(frame)
	local chem = frame.args[1]
	local met = scan_metab(chem)
	if not met then return "" end
	for _, v in pairs(met) do
		for _, w in pairs(v.effects or {}) do
			-- Исправляем проверку для JSON структуры
			local effectName = (type(w.type) == "table" and w.type.name) or w.type or ""
			if effectName == "SatiateHunger" then return "1" end
		end
	end
	return ""
end

function p.haseffects(frame)
	local chem = frame.args[1]
	local met = scan_metab(chem)
	if not met then return "" end
	for _, v in pairs(met) do
		for _, w in pairs(v.effects or {}) do
			-- Исправляем проверку для JSON структуры
			local effectName = (type(w.type) == "table" and w.type.name) or w.type or ""
			if effectName ~= "SatiateHunger" and effectName ~= "SatiateThirst" then
				return "1"
			end
		end
	end
	return ""
end

function p.geneffects(frame, chem)
	if chem == nil then chem = frame.args[1] end
	local met = scan_metab(chem)
	if not met then return "" end
	local out = ""
	for k, v in pairs(met) do
		-- Исправляем вывод rate - теперь используем правильное поле
		local rate = v.rate or 1.0
		
		-- Получаем локализованное название группы метаболизма
		local groupName = k
		if p.status and p.status[k] and p.status[k].name then
			groupName = p.status[k].name
		end
		
		out = out .. "<b>" .. groupName .. "</b> (" .. tostring(rate) .. " единиц в секунду)\n"
		
		-- Генерируем эффекты для этой группы
		local effectsText = p.geneffectlist(v.effects or {}, frame, rate)
		out = out .. effectsText
		out = out .. "\n"
	end
	return out
end

function p.geneffectlist(effects, frame, rate)
	local out = ""
	for _, w in pairs(effects or {}) do
		-- Исправляем получение ID эффекта для JSON структуры
		local effectId = (type(w.type) == "table" and w.type.name) or w.type or w.id or ""
		
		if effectId == "HealthChange" then
			out = out .. p.genhealthchange(w, rate, frame) .. "\n"
		elseif effectId == "AdjustReagent" then
			out = out .. p.genadjustreagent(w, rate, frame) .. "\n"
		elseif effectId == "FlammableReaction" then
			out = out .. p.genflammablereaction(w, frame) .. "\n"
		elseif effectId == "AdjustTemperature" then
			out = out .. p.genadjusttemperature(w, frame) .. "\n"
		elseif effectId == "GenericStatusEffect" then
			out = out .. p.gengenericstatuseffect(w, frame) .. "\n"
		elseif effectId == "ExplosionReactionEffect" then
			out = out .. p.genexplosionreactioneffect(w, frame) .. "\n"
		elseif effectId == "FoamAreaReactionEffect" then
			out = out .. p.genfoamareareactioneffect(w, frame) .. "\n"
		elseif effectId == "SmokeAreaReactionEffect" then
			out = out .. p.gensmokeareareactioneffect(w, frame) .. "\n"
		elseif effectId == "ModifyBleedAmount" then
			out = out .. p.genmodifybleedamount(w, frame) .. "\n"
		elseif effectId == "SatiateThirst" then
			local text = (type(w.type) == "table" and w.type.text) or "Утоляет жажду"
			out = out .. text .. ".\n"
		elseif effectId == "SatiateHunger" then
			local text = (type(w.type) == "table" and w.type.text) or "Утоляет голод"
			out = out .. text .. ".\n"
		end
	end
	return out
end

function p.gensmokeareareactioneffect(r, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "SmokeAreaReactionEffect", args = { when = conds, prob = r.probability }}
end

function p.genfoamareareactioneffect(r, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "FoamAreaReactionEffect", args = { when = conds, prob = r.probability }}
end

function p.genexplosionreactioneffect(r, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "ExplosionReactionEffect", args = { when = conds, prob = r.probability }}
end

function p.gengenericstatuseffect(r, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{
		title = "GenericStatusEffect",
		args = { key = r.Key or r.key, type = r.Type, time = r.Time or r.time, refresh = r.Refresh, when = conds, prob = r.probability }
	}
end

function p.genadjusttemperature(r, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "AdjustTemperature", args = { amount = r.Amount or r.amount, when = conds, prob = r.probability }}
end

function p.genflammablereaction(r, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "FlammableReaction", args = { multiplier = r.Multiplier, when = conds, prob = r.probability }}
end

function p.genadjustreagent(r, rate, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "AdjustReagent", args = { amount = r.Amount or r.amount, reagent = r.Reagent or r.reagent, when = conds, prob = r.probability }}
end

function p.genmodifybleedamount(r, frame)
	local conds = nil
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "ModifyBleedAmount", args = { amount = r.Amount or r.amount, when = conds, prob = r.probability }}
end

function p.genhealthchange(h, rate, frame)
	local healst = {}
	local dealst = {}
	local r = 1.0 / (rate or 1)

	if h.damage and h.damage.types then
		for k, v in pairs(h.damage.types) do
			if v < 0 then healst[k] = v * r else dealst[k] = v * r end
		end
	end
	if h.damage and h.damage.groups then
		for k, v in pairs(h.damage.groups) do
			if v < 0 then healst[k] = v * r else dealst[k] = v * r end
		end
	end

	local heals = hchangelist(healst, frame)
	local deals = hchangelist(dealst, frame)
	local conds = nil
	if h.conditions ~= nil then 
		conds = p.genconds(h.conditions, frame)
	end
	return frame:expandTemplate{ title = "HealthChange", args = { heals = heals, deals = deals, when = conds, prob = h.probability }}
end

function p.genconds(conds, frame)
	local out = ""
	local len = tablelength(conds or {})
	local i = 0
	for _, c in pairs(conds or {}) do
		i = i + 1
		if len == i and i ~= 1 then
			out = out .. ", and "
		elseif i ~= 1 then
			out = out .. ", "
		end
		out = out .. p.gencond(c, frame)
	end
	return out
end

function p.gencond(c, frame)
	if not c or not c.type then return "" end
	if c.type == "TotalDamage" then
		return frame:expandTemplate{ title = "TotalDamage", args = { min = c.Min or c.min, max = c.Max or c.max } }
	elseif c.type == "ReagentThreshold" then
		return frame:expandTemplate{ title = "ReagentThreshold", args = { min = c.Min or c.min, max = c.Max or c.max, reagent = c.Reagent or c.reagent } }
	elseif c.type == "Temperature" then
		return frame:expandTemplate{ title = "Temperature", args = { min = c.Min or c.min, max = c.Max or c.max } }
	else
		-- Новая логика: если type не является стандартным условием, то это тип органа
		local organTypeName = c.type
		local shouldHave = c.shouldHave
		if shouldHave == nil then shouldHave = true end
		
		-- Получаем локализованное название типа органа из status данных
		if p.status and p.status[organTypeName] and p.status[organTypeName].name then
			organTypeName = p.status[organTypeName].name
		end
		
		return frame:expandTemplate{ title = "OrganType", args = { shouldhave = shouldHave, type = organTypeName } }
	end
	return ""
end

-- Отладочная функция для проверки рецептов
function p.debugrecipes(frame)
	local chem = frame.args[1]
	if not chem then return "Не указан химикат" end
	
	local out = "=== Отладка рецептов для " .. chem .. " ===\n"
	
	-- Проверяем существует ли химикат
	if not p.chem[chem] then
		out = out .. "ОШИБКА: Химикат не найден в базе данных\n"
		return out
	end
	
	-- Проверяем есть ли рецепты
	local recipes = p.chem[chem].recipes
	if not recipes then
		out = out .. "Рецепты не найдены (recipes = nil)\n"
		-- Попробуем найти рецепты вручную
		out = out .. "Поиск рецептов в реакциях:\n"
		for reactId, reactData in pairs(p.react) do
			if reactData.products and reactData.products[chem] then
				out = out .. "- Найдена реакция: " .. reactId .. " (производит " .. tostring(reactData.products[chem]) .. ")\n"
			end
		end
	else
		-- Подсчитываем рецепты правильно
		local recipeCount = 0
		for _ in pairs(recipes) do
			recipeCount = recipeCount + 1
		end
		out = out .. "Найдено рецептов: " .. recipeCount .. "\n"
		
		for i, recipe in ipairs(recipes) do
			out = out .. "- Рецепт " .. i .. ": " .. recipe .. "\n"
			if p.react[recipe] then
				out = out .. "  Реакция существует: да\n"
				if p.react[recipe].products and p.react[recipe].products[chem] then
					out = out .. "  Производит " .. chem .. ": " .. tostring(p.react[recipe].products[chem]) .. "\n"
				else
					out = out .. "  ОШИБКА: Не производит " .. chem .. "\n"
				end
			else
				out = out .. "  ОШИБКА: Реакция не существует\n"
			end
		end
	end
	
	return out
end

return p