db-extra.lua

--[[--
	by ALA
	CREDIT shagu/pfQuest(MIT LICENSE) @ https://github.com/shagu
--]]--
local __addon, __private = ...;
local MT = __private.MT;
local CT = __private.CT;
local VT = __private.VT;
local DT = __private.DT;

-->		upvalue
	local collectgarbage = collectgarbage;
	local date = date;
	local setmetatable = setmetatable;
	local next = next;
	local strfind = string.find;
	local _G = _G;

-->
	local DataAgent = DT.DB;
	local l10n = CT.l10n;

-->
MT.BuildEnv("db-extra");
-->		predef
-->
	local chain_prev_quest = {  };
	DataAgent.chain_prev_quest = chain_prev_quest;

	local item_related_quest = {  };
	DataAgent.item_related_quest = item_related_quest;

	local avl_quest_list = {  };
	local avl_quest_hash = {  };
	DataAgent.avl_quest_list = avl_quest_list;
	DataAgent.avl_quest_hash = avl_quest_hash;

-->
-->		db-extra
	local function check(tbl, key, str)
		if tbl.U then
			for _, unit in next, tbl.U do
				local uinfo = DataAgent.unit[unit];
				if uinfo ~= nil and uinfo[key] ~= -1 then
					return true;
				end
			end
		end
		if tbl.O then
			for _, obj in next, tbl.O do
				local oinfo = DataAgent.object[obj];
				if oinfo ~= nil then
					local fac = oinfo['fac'];
					if fac == nil or strfind(fac, str) then
						return true;
					end
				end
			end
		end
		return false;
	end

	local EmptyInfo = {
		unit = { ["coords"] = {  }, },
		item = {},
		object = { ["coords"] = {  }, },
		refloot = {},
		event = { ["coords"] = {  }, },
	};
	local function MarkItemRelation(quest, item)
		local v = item_related_quest[item];
		if v == nil then
			v = {  };
			item_related_quest[item] = v;
		end
		local num = #v;
		for i = 1, num do
			if v[i] == quest then
				return;
			end
		end
		v[num + 1] = quest;
	end
	local MarkUnit, MarkItem, MarkObject, MarkRefloot, MarkEvent;
	local HashUnit, HashItem, HashObject, HashRefloot, HashEvent = {  }, {  }, {  }, {  }, {  };
	local Hash = { unit = HashUnit, item = HashItem, object = HashObject, refloot = HashRefloot, event = HashEvent, };
	function MarkUnit(quest, unit)
		-- if HashUnit[unit] == nil then
			HashUnit[unit] = 1;
			local info = DataAgent.unit[unit];
			if info ~= nil then
				local spawn = info.spawn;
				if spawn ~= nil then
					if spawn.U ~= nil then
						for unit, _ in next, spawn.U do
							MarkUnit(quest, unit);
						end
					end
					if spawn.O ~= nil then
						for object, _ in next, spawn.O do
							MarkObject(quest, object);
						end
					end
					if spawn.I ~= nil then
						for item, _ in next, spawn.I do
							MarkItem(quest, item);
						end
					end
				end
			end
		-- end
	end
	function MarkItem(quest, item, spawned)
		-- if HashItem[item] == nil then
			HashItem[item] = 1;
			local info = DataAgent.item[item];
			if info ~= nil then
				if info.U ~= nil then
					for unit, _ in next, info.U do
						MarkUnit(quest, unit);
					end
				end
				if info.O ~= nil then
					for object, _ in next, info.O do
						MarkObject(quest, object);
					end
				end
				if info.I ~= nil then
					for item, _ in next, info.I do
						MarkItem(quest, item);
					end
				end
				if info.R ~= nil then
					for ref, _ in next, info.R do
						MarkRefloot(quest, ref);
					end
				end
				if info.V ~= nil then
					for unit, _ in next, info.V do
						MarkUnit(quest, unit);
					end
				end
			end
		-- end
		if spawned == nil or spawned < 2 then
			MarkItemRelation(quest, item);
		end
	end
	function MarkObject(quest, object)
		-- if HashObject[object] == nil then
			HashObject[object] = 1;
			local info = DataAgent.object[object];
			if info ~= nil then
				local spawn = info.spawn;
				if spawn ~= nil then
					if spawn.U ~= nil then
						for unit, _ in next, spawn.U do
							MarkUnit(quest, unit);
						end
					end
					if spawn.O ~= nil then
						for object, _ in next, spawn.O do
							MarkObject(quest, object);
						end
					end
					if spawn.I ~= nil then
						for item, _ in next, spawn.I do
							MarkItem(quest, item);
						end
					end
				end
			end
		-- end
	end
	function MarkRefloot(quest, ref)
		-- if HashRefloot[ref] == nil then
			local info = DataAgent.refloot[ref];
			if info ~= nil then
				if info.U ~= nil then
					for unit, _ in next, info.U do
						MarkUnit(quest, unit);
					end
				end
				if info.O ~= nil then
					for object, _ in next, info.O do
						MarkObject(quest, object);
					end
				end
			end
		-- end
	end
	function MarkEvent(quest, event)
		HashEvent[event] = 1;
			local info = DataAgent.event[event];
			if info ~= nil then
				local spawn = info.spawn;
				if spawn ~= nil then
					if spawn.U ~= nil then
						for unit, _ in next, spawn.U do
							MarkUnit(quest, unit);
						end
					end
					if spawn.O ~= nil then
						for object, _ in next, spawn.O do
							MarkObject(quest, object);
						end
					end
					if spawn.I ~= nil then
						for item, _ in next, spawn.I do
							MarkItem(quest, item);
						end
					end
				end
			end
	end

	local function VerifyData()
		MT.Debug("|cffff7f00Start|r |cffff0000VerifyData|r");
		local vitem, vobject, vunit, vrefloot;
		local VI, VO, VU, VR = {  }, {  }, {  }, {  };
		function vitem(t, k, id, hu, hi, ho, path)
			if DataAgent.blacklist_item[id] or VI[id] then
				return;
			end
			path = path .. " item:" .. id;
			local spawn = DataAgent.item[id];
				if spawn ~= nil then
					if hi[id] then
						return MT.Print("|cffff0000SpawnError|r", path, hi[id]);
					end
					hi[id] = true;
					if spawn.U ~= nil then
						for unit, _ in next, spawn.U do
							vunit(t, k, unit, hu, hi, ho, path);
						end
					end
					if spawn.O ~= nil then
						for object, _ in next, spawn.O do
							vobject(t, k, object, hu, hi, ho, path);
						end
					end
					if spawn.I ~= nil then
						for item, _ in next, spawn.I do
							vitem(t, k, item, hu, hi, ho, path);
						end
					end
					hi[id] = nil;
				end
			VI[id] = true;
		end
		function vobject(t, k, id, hu, hi, ho, path)
			if VO[id] then
				return;
			end
			path = path .. " object:" .. id;
			local info = DataAgent.object[id];
			if info ~= nil then
				local spawn = info.spawn;
				if spawn ~= nil then
					if ho[id] then
						return MT.Print("|cffff0000SpawnError|r", path, ho[id]);
					end
					ho[id] = true;
					if spawn.U ~= nil then
						for unit, _ in next, spawn.U do
							vunit(t, k, unit, hu, hi, ho, path);
						end
					end
					if spawn.O ~= nil then
						for object, _ in next, spawn.O do
							vobject(t, k, object, hu, hi, ho, path);
						end
					end
					if spawn.I ~= nil then
						for item, _ in next, spawn.I do
							vitem(t, k, item, hu, hi, ho, path);
						end
					end
					ho[id] = nil;
				end
			end
			VO[id] = true;
		end
		function vunit(t, k, id, hu, hi, ho, path)
			if VU[id] then
				return;
			end
			path = path .. " unit:" .. id;
			local info = DataAgent.unit[id];
			if info ~= nil then
				local spawn = info.spawn;
				if spawn ~= nil then
					if hu[id] then
						return MT.Print("|cffff0000SpawnError|r", path, hu[id]);
					end
					hu[id] = true;
					if spawn.U ~= nil then
						for unit, _ in next, spawn.U do
							vunit(t, k, unit, hu, hi, ho, path);
						end
					end
					if spawn.O ~= nil then
						for object, _ in next, spawn.O do
							vobject(t, k, object, hu, hi, ho, path);
						end
					end
					if spawn.I ~= nil then
						for item, _ in next, spawn.I do
							vitem(t, k, item, hu, hi, ho, path);
						end
					end
					hu[id] = nil;
				end
			end
			VU[id] = true;
		end
		function vrefloot(t, k, id, hu, hi, ho, path)
			if VR[id] then
				return;
			end
			path = path .. " refloot:" .. id;
			local info = DataAgent.refloot[id];
			if info ~= nil then
				if info.U ~= nil then
					for unit, _ in next, info.U do
						vunit(t, k, unit, hu, hi, ho, path);
					end
				end
				if info.O ~= nil then
					for object, _ in next, info.O do
						vobject(t, k, object, hu, hi, ho, path);
					end
				end
			end
			VR[id] = true;
		end
		for unit, info in next, DataAgent.unit do
			vunit('unit', unit, unit, {  }, {  }, {  }, "");
		end
		for item, info in next, DataAgent.item do
			vitem('item', item, item, {  }, {  }, {  }, "");
		end
		for object, info in next, DataAgent.object do
			vobject('object', object, object, {  }, {  }, {  }, "");
		end
		MT.Debug("|cff00ff00Finish|r |cffff0000VerifyData|r");
	end

	MT.RegisterOnLogin("db-extra", function(LoggedIn)
	-->		patch
		local function patchDB(fix)
			for key, patch in next, fix do
				local db = DataAgent[key];
				if db ~= nil then
					for id, val in next, patch do
						local t = db[id];
						if t ~= nil then
							for k, v in next, val do
								if v == "_NIL" then
									t[k] = nil;
								else
									t[k] = v;
								end
							end
						else
							db[id] = val;
						end
					end
				end
			end
		end
		patchDB(DataAgent.fix);
		if CT.SELFFACTION == "Alliance" then
			patchDB(DataAgent.fix_alliance);
		else
			patchDB(DataAgent.fix_horde);
		end
		if DataAgent.override ~= nil then
			for key, patch in next, DataAgent.override do
				local db = DataAgent[key];
				if db ~= nil then
					for id, val in next, patch do
						db[id] = val;
					end
				end
			end
		end
		for id, val in next, DataAgent.waypoints do
			local waypoints = {  };
			for _, tbl in next, val do
				for _, p in next, tbl do
					waypoints[#waypoints + 1] = p;
				end
			end
			DataAgent.unit[id] = DataAgent.unit[id] or {  };
			DataAgent.unit[id].waypoints = waypoints;
		end
	-->		Misc
		if VT.__is_dev then
			VerifyData();
		end
		for quest, info in next, DataAgent.quest do
			local _next = info.next;
			if _next ~= nil then
				chain_prev_quest[_next] = quest;
			end
		end
		local today = date("*t");
		local year, month, day, wday = today.year, today.month, today.day, today.wday;
		-- if CT.TOCVERSION >= 30000 and C_Calendar and C_Calendar.GetHolidayInfo then
		-- else
		-- end
		for event, limits in next, DataAgent.worldeventperiod do
			local limit = limits[year] or limits["*"];
			if limit == nil or
				(limit[1] <= limit[3] and (month < limit[1] or month > limit[3])) or
				(month < limit[1] and month > limit[3]) or
				(month == limit[1] and day < limit[2]) or
				(month == limit[3] and day > limit[4])
			then
				local eventquests = DataAgent.worldevent[event];
				for _, quest in next, eventquests do
					DataAgent.blacklist_quest[quest] = true;
				end
			else
				local eventquests = DataAgent.worldevent[event];
				for _, quest in next, eventquests do
					DataAgent.blacklist_quest[quest] = nil;
				end
			end
		end
		-->		faction quest list
			local key = CT.SELFFACTION == "Alliance" and "facA" or "facH";
			local str = CT.SELFFACTION == "Alliance" and "A" or "H";
			for quest, info in next, DataAgent.quest do
				if DataAgent.blacklist_quest[quest] == nil then
					local info = DataAgent.quest[quest];
					local race = info.race;
					local class = info.class;
					if (race == nil or MT.CheckSelfRace(race)) and (class == nil or MT.CheckSelfClass(class)) then
						local _start = info['start'];
						local _end = info['end'];
						if (_start and check(_start, key, str)) or (_end and check(_end, key, str)) then
							avl_quest_list[#avl_quest_list + 1] = quest;
							avl_quest_hash[quest] = 1;
						end
					end
				end
			end
		-->
		-->		Delete Unused
			-->		cache
			for quest, info in next, DataAgent.quest do
				local _obj = info['obj'];
				if _obj ~= nil then
					if _obj.U ~= nil then
						for _, unit in next, _obj.U do
							MarkUnit(quest, unit);
						end
					end
					if _obj.I ~= nil then
						for _, item in next, _obj.I do
							MarkItem(quest, item);
						end
					end
					if _obj.IR ~= nil then
						for _, item in next, _obj.IR do
							-- HashItem[item] = 1;
							MarkItemRelation(quest, item);
						end
					end
					if _obj.O ~= nil then
						for _, object in next, _obj.O do
							MarkObject(quest, object);
						end
					end
					if _obj.E ~= nil then
						for _, event in next, _obj.E do
							MarkEvent(quest, event);
						end
					end
				end
				local _start = info['start'];
				if _start ~= nil then
					if _start.U ~= nil then
						for _, unit in next, _start.U do
							MarkUnit(quest, unit);
						end
					end
					if _start.I ~= nil then
						for _, item in next, _start.I do
							MarkItem(quest, item);
						end
					end
					if _start.O ~= nil then
						for _, object in next, _start.O do
							MarkObject(quest, object);
						end
					end
				end
				local _end = info['end'];
				if _end ~= nil then
					if _end.U ~= nil then
						for _, unit in next, _end.U do
							MarkUnit(quest, unit);
						end
					end
					if _end.O ~= nil then
						for _, object in next, _end.O do
							MarkObject(quest, object);
						end
					end
				end
				local _extra = info['extra'];
				if _extra ~= nil then
					if _extra.U ~= nil then
						for unit in next, _extra.U do
							MarkUnit(quest, unit);
						end
					end
					if _extra.I ~= nil then
						for item in next, _extra.I do
							MarkItem(quest, item);
						end
					end
					if _extra.IR ~= nil then
						for item in next, _extra.IR do
							-- HashItem[item] = 1;
							MarkItemRelation(quest, item);
						end
					end
					if _extra.O ~= nil then
						for object in next, _extra.O do
							MarkObject(quest, object);
						end
					end
					if _extra.E ~= nil then
						for event in next, _extra.E do
							MarkEvent(quest, event);
						end
					end
				end
			end
			-->		proc db
			-- local M = {};
			-- local LO = l10n.object;
			-- for object in next, HashObject do
			-- 	if LO[object] == nil then
			-- 		M[#M + 1] = object;
			-- 	end
			-- end
			-- _G.table.sort(M);
			-- VT.ObjectMissLocale = M;
			-- CodexLiteSnerrorSV.ObjectMissLocale=__ala_meta__.questsnerror.VT.ObjectMissLocale
			-->
			local i18n = CT.i18n;
			for which, hash in next, Hash do
				local info = EmptyInfo[which];
				local db = DataAgent[which];
				local loc = l10n[which];
				local def = i18n and i18n[which] or {  };
				if loc == nil then
					for id, _ in next, hash do
						db[id] = db[id] or info;
					end
				else
					for id, _ in next, loc do
						if hash[id] == nil then
							loc[id] = nil;
						end
					end
					for id, _ in next, hash do
						db[id] = db[id] or info;
					end
					if which == "quest" then
						for id, _ in next, hash do
							local v = loc[id];
							if v == nil then
								v = {  };
								loc[id] = v;
							end
							local d = def[id];
							v[1] = v[1] or d[1] or "quest:" .. id;
							v[2] = nil;
							v[3] = v[3] or v[2] or d[3] or d[2];
							-- def[id] = nil;
						end
					else
						for id, _ in next, hash do
							loc[id] = loc[id] or def[id] or which .. ":" .. id;
							-- def[id] = nil;
						end
					end
				end
				for id, _ in next, db do
					if hash[id] == nil then
						db[id] = nil;
					end
				end
				if loc ~= def then
					setmetatable(def, { __mode = "kv", });
				end
			end
			if l10n ~= i18n then
				setmetatable(i18n, { __mode = "kv", });
			end
			-->
			MarkUnit, MarkItem, MarkObject, MarkRefloot, MarkEvent = nil;
			HashUnit, HashItem, HashObject, HashRefloot, HashEvent = nil;
			Hash = nil;
			-->
			CT.i18n = nil;
		-->
		--[==[
		local i18n = CT.i18n;
		if i18nn then
			for _, which in next, { "unit", "item", "object", "refloot", "event", } do
				local loc = l10n[which];
				local def = i18n[which] or {  };
				for key, val in next, def do
					if loc[key] == nil then
						loc[key] = val;
					end
				end
			end
			CT.i18n = nil;
		end
		--]==]
		-->		item-drop
			for iid, info in next, DataAgent.item do
				if info.U ~= nil then
					local throttle = false;
					local U = info.U;
					local O = info.O;
					if O ~= nil then
						for oid, r in next, O do
							if r >= 25 then
								throttle = true;
								break;
							end
						end
					end
					if not throttle then
						for uid, r in next, U do
							if r >= 25 then
								throttle = true;
								break;
							end
						end
					end
					if throttle then
						for uid, r in next, U do
							if r <= 2 then
								U[uid] = nil;
							end
						end
						if O ~= nil then
							for oid, r in next, O do
								if r <= 2 then
									O[oid] = nil;
								end
							end
						end
					end
				end
			end
		-->
		-->		Large Pin
			for quest, info in next, DataAgent.quest do
				local _obj = info['obj'];
				if _obj ~= nil then
					if _obj.U ~= nil then
						for _, unit in next, _obj.U do
							local uinfo = DataAgent.unit[unit];
							if uinfo ~= nil and uinfo.waypoints == nil and uinfo.coords ~= nil and #uinfo.coords == 1 then
								DataAgent.large_pin[quest] = DataAgent.large_pin[quest] or {  };
								DataAgent.large_pin[quest].unit = DataAgent.large_pin[quest].unit or {  };
								DataAgent.large_pin[quest].unit[unit] = 1;
							end
						end
					end
					if _obj.O ~= nil then
						for _, object in next, _obj.O do
							local oinfo = DataAgent.object[object];
							if oinfo ~= nil and oinfo.coords ~= nil and #oinfo.coords == 1 then
								DataAgent.large_pin[quest] = DataAgent.large_pin[quest] or {  };
								DataAgent.large_pin[quest].object = DataAgent.large_pin[quest].object or {  };
								DataAgent.large_pin[quest].object[object] = 1;
							end
						end
					end
					if _obj.I ~= nil then
						for _, item in next, _obj.I do
							local iinfo = DataAgent.item[item];
							if iinfo ~= nil then
								local num = 0;
								local type = nil;
								local id = nil;
								if iinfo.U ~= nil then
									for unit, rate in next, iinfo.U do
										if rate >= 1 then
											local uinfo = DataAgent.unit[unit];
											if uinfo ~= nil and uinfo.coords ~= nil then
												local n = #uinfo.coords;
												if n == 1 then
													type = 'unit';
													id = unit;
												end
												num = num +n;
												if num > 1 then break; end
											end
										end
									end
								end
								if num <= 1 then
								if iinfo.V ~= nil then
									for unit, _ in next, iinfo.V do
										local uinfo = DataAgent.unit[unit];
										if uinfo ~= nil and uinfo.coords ~= nil then
											local n = #uinfo.coords;
											if n == 1 then
												type = 'unit';
												id = unit;
											end
											num = num + n;
											if num > 1 then break; end
										end
									end
								end
								end
								if num <= 1 then
								if iinfo.O ~= nil then
									for object, rate in next, iinfo.O do
										if rate >= 1 then
											local oinfo = DataAgent.object[object];
											if oinfo ~= nil and oinfo.coords ~= nil then
												local n = #oinfo.coords;
												if n == 1 then
													type = 'object';
													id = object;
												end
												num = num + n;
												if num > 1 then break; end
											end
										end
									end
								end
								end
								if num == 1 then
									DataAgent.large_pin[quest] = DataAgent.large_pin[quest] or {  };
									DataAgent.large_pin[quest].item = DataAgent.large_pin[quest].item or {  };
									DataAgent.large_pin[quest].item[item] = 1;
									DataAgent.large_pin[quest][type] = DataAgent.large_pin[quest][type] or {  };
									DataAgent.large_pin[quest][type][id] = 1;
								end
							end
						end
					end
				end
			end
		-->
	-->		Locale
	-->
		collectgarbage('collect');
	end);

-->