InventoryManager/InventoryManager.lua at master · el0ny/InventoryManager · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
-----------------------------------------------------------------------------------
-- Addon Name: InventoryManager
-- Creator: elDrag0n
-- Addon Ideal: Manages resources between bank and inventory
-- Addon Creation Date: July 18, 2022
--
-- File Name: BankManagement.lua
-- File Description: This file manages resources between banks and inventory
-- Load Order Requirements: None
--
-----------------------------------------------------------------------------------

InventoryManager = InventoryManager or {}
InventoryManager.version = "0.2.0"
InventoryManager.name = 'InventoryManager'
InventoryManager.colours = {
	greenCP = "|c557C29",
	author = "|c5959D5"
}
name = "test"


function InventoryManager:buildStorageTable(storage_bag_id)
	storage_table = {}
	local slot_id = ZO_GetNextBagSlotIndex(storage_bag_id)
	while slot_id do
		local item_type = GetItemType(storage_bag_id, slot_id)
		if self.valid_item_types[item_type] then
			local item_id = GetItemId(storage_bag_id, slot_id)
			storage_table[item_id] = slot_id
		end
		--d(GetItemName(storage_bag_id, slot_id))

		slot_id = ZO_GetNextBagSlotIndex(storage_bag_id, slot_id)
	end
	return storage_table
end

function InventoryManager.main(event, storage_bag_id)
	if not InventoryManager.valid_storage_types[storage_bag_id] then
		d("oops: " .. storage_bag_id)
		return
	end

	local inventory_slot = ZO_GetNextBagSlotIndex(BAG_BACKPACK)
	storage_table = InventoryManager:buildStorageTable(storage_bag_id)

	while inventory_slot do
		local item_type = GetItemType(BAG_BACKPACK, inventory_slot)
		local item_id = GetItemId(BAG_BACKPACK, inventory_slot)
		if InventoryManager.valid_item_types[item_type] and storage_table[item_id] then

			local storage_slot = storage_table[item_id]
			--d(GetItemName(BAG_BACKPACK, slot))
			--d(item_type .. GetItemName(storage_bag_id, inventory_slot))
			if storage_slot then
				local bank_slot_size, max_bank_slot_size = GetSlotStackSize(storage_bag_id, storage_slot)
				local bag_slot_size, max_bag_slot_size = GetSlotStackSize(BAG_BACKPACK, inventory_slot)

				if bag_slot_size ~= max_bag_slot_size then
					local available_space = max_bank_slot_size - bank_slot_size

					if available_space > bag_slot_size then
						--d(GetItemName(BAG_BACKPACK, inventory_slot)..' Bank slot size: '..bank_slot_size..' Bank max slot size: '..max_bank_slot_size..' Bag slot size: '..bag_slot_size)
						--d("wow2")
						if IsProtectedFunction("RequestMoveItem") then
							--d(GetItemName(BAG_BACKPACK, inventory_slot))
							CallSecureProtected("RequestMoveItem", BAG_BACKPACK, inventory_slot, storage_bag_id, storage_slot, bag_slot_size)
						else
							RequestMoveItem(BAG_BACKPACK, inventory_slot, storage_bag_id, storage_slot, bag_slot_size)
						end
					else
						local available_bag_space = max_bag_slot_size - bag_slot_size
						local bank_remain = bank_slot_size - available_bag_space
						d(GetItemName(BAG_BACKPACK, inventory_slot)..' x'..max_bag_slot_size..' was withdrawn, x'..bank_remain..' remains in bank ')
						if IsProtectedFunction("RequestMoveItem") then
							CallSecureProtected("RequestMoveItem", storage_bag_id, storage_slot, BAG_BACKPACK, inventory_slot, available_bag_space)
						else
							RequestMoveItem(storage_bag_id, storage_slot, BAG_BACKPACK, inventory_slot, available_bag_space)
						end
					end
				end
			end
		end
		inventory_slot = ZO_GetNextBagSlotIndex(BAG_BACKPACK, inventory_slot)
		--d(inventory_slot)
	end
end

InventoryManager.slot_surveys = slot_surveys


function InventoryManager:Initialize()

	InventoryManager:InitSavedVariables()

	InventoryManager:InitMenu()
	EVENT_MANAGER:UnregisterForEvent(name, EVENT_ADD_ON_LOADED)
end

function InventoryManager.OnAddOnLoaded(event, addonName)
	if addonName == InventoryManager.name then
		InventoryManager:Initialize()
	end
end

EVENT_MANAGER:RegisterForEvent(InventoryManager.name, EVENT_ADD_ON_LOADED, InventoryManager.OnAddOnLoaded)
EVENT_MANAGER:RegisterForEvent(InventoryManager.name, EVENT_OPEN_BANK, InventoryManager.main)


--
local function _onInventoryChanged(eventCode, bagId, slotIndex, isNewItem, itemSoundCategory, updateReason, stackCountChange)
	if InventoryManager.savedVariables.debug then
		 local link = GetItemLink(bagId, slotIndex)
		 local item_type = GetItemType(bagId, slotIndex)
   		 d("Picked up a " .. link .. "." .. item_type .. " Bag: " .. bagId)
	end
end

EVENT_MANAGER:RegisterForEvent(InventoryManager.name, EVENT_INVENTORY_SINGLE_SLOT_UPDATE, _onInventoryChanged)