--[[
| This file was obtained through the combined efforts
| of Madbluntz & Plymouth Antiquarian Society.
|
| Credits: lifestorm, Gregory Wayne Rossel JR.,
| 	Maloy, DrPepper10 @ RIP, Atle!
|
| Visit for more: https://plymouth.thetwilightzone.ru/
--]]


PLUGIN.name = "Save Items"
PLUGIN.author = "Chessnut"
PLUGIN.description = "Saves items that were dropped."

--[[
	function PLUGIN:OnSavedItemLoaded(items)
		for k, v in ipairs(items) do
			-- do something
		end
	end

	function PLUGIN:ShouldDeleteSavedItems()
		return true
	end
]]--

-- as title says.

function PLUGIN:RegisterSaveEnts()
	ix.saveEnts:RegisterEntity("ix_item", true, true, true, {
		OnSave = function(entity, data) --OnSave
            data.itemID = entity.ixItemID
			if (entity.ixSteamID) then
				data.steamID = entity.ixSteamID
				data.charID = entity.ixCharID
			end
		end,
		OnRestore = function(entity, data) --OnRestore
			local query = mysql:Select("ix_items")
				query:Select("item_id")
				query:Select("unique_id")
				query:Select("data")
				query:Where("item_id", data.itemID)
				query:Callback(function(result)
					if (!IsValid(entity)) then return end

					if (istable(result) and result[1]) then
						local dbData = result[1]
						local itemID = tonumber(dbData.item_id)
						local itemData = util.JSONToTable(dbData.data or "[]")
						local uniqueID = dbData.unique_id
						local itemTable = ix.item.list[uniqueID]

						if (itemTable and itemID) then
							local item = ix.item.New(uniqueID, itemID)
							item.data = itemData or {}

							entity:SetItem(item.id)

							if (data.steamID) then
								entity.ixSteamID = data.steamID
								entity.ixCharID = data.charID
								entity:SetNetVar("owner", entity.ixCharID)
							end

							if (data.pos) then entity:SetPos(data.pos) end
							if (data.angles) then entity:SetAngles(data.angles) end
							if (data.skin) then entity:SetSkin(data.skin) end
							if (data.color) then entity:SetColor(data.color) end
							if (data.material) then entity:SetMaterial(data.material) end

							if (data.motion != nil) then
								local physicsObject = entity:GetPhysicsObject()
								if (IsValid(physicsObject)) then
									physicsObject:EnableMotion(data.motion)
								end
							end

							hook.Run("OnItemSpawned", entity)

							item.invID = 0
							if (item.isBag) then
								local invType = ix.item.inventoryTypes[uniqueID]
								ix.inventory.Restore(item:GetData("id"), invType.w, invType.h)
							end

							hook.Run("OnSavedItemLoaded", {item}, {entity}) -- will cover most compatibility
						else
							entity:Remove()
						end
					end
				end)
			query:Execute()
		end,
		ShouldSave = function(entity)
			return entity.ixItemID and !entity.bTemporary
		end
	})
end

function PLUGIN:LoadData()
	if (ix.saveEnts and !ix.config.Get("SaveEntsOldLoadingEnabled")) then return end

	local items = self:GetData()

	if (items) then
		local idRange = {}
		local info = {}

		for _, v in ipairs(items) do
			idRange[#idRange + 1] = v[1]
			info[v[1]] = {v[2], v[3], v[4]}
		end

		if (#idRange > 0) then
			if (hook.Run("ShouldDeleteSavedItems") == true) then
				-- don't spawn saved item and just delete them.
				local query = mysql:Delete("ix_items")
					query:WhereIn("item_id", idRange)
				query:Execute()

				print("Server Deleted Server Items (does not includes Logical Items)")
			else
				local query = mysql:Select("ix_items")
					query:Select("item_id")
					query:Select("unique_id")
					query:Select("data")
					query:WhereIn("item_id", idRange)
					query:Callback(function(result)
						if (istable(result)) then
							local loadedItems, loadedEntities = {}, {}
							local bagInventories = {}

							for _, v in ipairs(result) do
								local itemID = tonumber(v.item_id)
								local data = util.JSONToTable(v.data or "[]")
								local uniqueID = v.unique_id
								local itemTable = ix.item.list[uniqueID]

								if (itemTable and itemID) then
									local item = ix.item.New(uniqueID, itemID)
									item.data = data or {}

									local itemInfo = info[itemID]
									local position, angles, bMovable = itemInfo[1], itemInfo[2], true

									if (isbool(itemInfo[3])) then
										bMovable = itemInfo[3]
									end

									local itemEntity = item:Spawn(position, angles)
									itemEntity.ixItemID = itemID

									local physicsObject = itemEntity:GetPhysicsObject()

									if (IsValid(physicsObject)) then
										physicsObject:EnableMotion(bMovable)
									end

									item.invID = 0
									loadedItems[#loadedItems + 1] = item
									loadedEntities[#loadedEntities + 1] = itemEntity

									if (item.isBag) then
										local invType = ix.item.inventoryTypes[uniqueID]
										bagInventories[item:GetData("id")] = {invType.w, invType.h}
									end
								end
							end

							-- we need to manually restore bag inventories in the world since they don't have a current owner
							-- that it can automatically restore along with the character when it's loaded
							if (!table.IsEmpty(bagInventories)) then
								ix.inventory.Restore(bagInventories)
							end

							hook.Run("OnSavedItemLoaded", loadedItems, loadedEntities) -- when you have something in the dropped item.
						end
					end)
				query:Execute()
			end
		end
	end
end

function PLUGIN:SaveData()
	local items = {}

	for _, v in ipairs(ents.FindByClass("ix_item")) do
		if (v.ixItemID and !v.bTemporary) then
			local physicsObject = v:GetPhysicsObject()
			local bMovable = nil

			if (IsValid(physicsObject)) then
				bMovable = physicsObject:IsMoveable()
			end

			items[#items + 1] = {
				v.ixItemID, v:GetPos(), v:GetAngles(), bMovable
			}
		end
	end

	self:SetData(items)
end