PokeBot/storage/inventory.lua

local inventory = {}

local pokemon = require "storage.pokemon"

local input = require "util.input"
local memory = require "util.memory"
local menu = require "util.menu"
local utils = require "util.utils"

local items = {
	pokeball = 4,
	bicycle = 6,
	moon_stone = 10,
	antidote = 11,
	paralyze_heal = 15,
	full_restore = 16,
	super_potion = 19,
	potion = 20,
	escape_rope = 29,
	carbos = 38,
	repel = 30,

	rare_candy = 40,
	helix_fossil = 42,
	nugget = 49,
	pokedoll = 51,
	super_repel = 56,
	fresh_water = 60,
	soda_pop = 61,
	pokeflute = 73,
	ether = 80,
	max_ether = 81,
	elixer = 82,

	x_accuracy = 46,
	x_speed = 67,
	x_special = 68,

	cut = 196,
	fly = 197,
	surf = 198,
	strength = 199,

	horn_drill = 207,
	bubblebeam = 211,
	water_gun = 212,
	ice_beam = 213,
	thunderbolt = 224,
	earthquake = 226,
	dig = 228,
	tm34 = 234,
	rock_slide = 248,
}

local ITEM_BASE = 0xD31E

-- Data

function inventory.indexOf(name)
	local searchID = items[name]
	for i=0,19 do
		local iidx = ITEM_BASE + i * 2
		if memory.raw(iidx) == searchID then
			return i
		end
	end
	return -1
end

function inventory.count(name)
	local index = inventory.indexOf(name)
	if index ~= -1 then
		return memory.raw(ITEM_BASE + index * 2 + 1)
	end
	return 0
end

function inventory.contains(...)
	for i,name in ipairs(arg) do
		if inventory.count(name) > 0 then
			return name
		end
	end
end

-- Actions

function inventory.teach(item, poke, replaceIdx, altPoke)
	local main = memory.value("menu", "main")
	local column = menu.getCol()
	if main == 144 then
		if column == 5 then
			menu.select(replaceIdx, true)
		else
			input.press("A")
		end
	elseif main == 128 then
		if column == 5 then
			menu.select(inventory.indexOf(item), "accelerate", true)
		elseif column == 11 then
			menu.select(2, true)
		elseif column == 14 then
			menu.select(0, true)
		end
	elseif main == 103 then
		input.press("B")
	elseif main == 64 or main == 96 or main == 192 then
		if column == 5 then
			menu.select(replaceIdx, true)
		elseif column == 14 then
			input.press("A")
		elseif column == 15 then
			menu.select(0, true)
		else
			local idx = 0
			if poke then
				idx = pokemon.indexOf(poke, altPoke)
			end
			menu.select(idx, true)
		end
	else
		return false
	end
	return true
end

function inventory.isFull()
	return memory.raw(0xD345) > 0
end

function inventory.use(item, poke, midfight)
	if midfight then
		local battleMenu = memory.value("battle", "menu")
		if battleMenu == 94 then
			local rowSelected = memory.value("menu", "row")
			if menu.getCol() == 9 then
				if rowSelected == 0 then
					input.press("Down")
				else
					input.press("A")
				end
			else
				input.press("Left")
			end
		elseif battleMenu == 233 then
			menu.select(inventory.indexOf(item), "accelerate", true)
		elseif utils.onPokemonSelect(battleMenu) then
			if poke then
				if type(poke) == "string" then
					poke = pokemon.indexOf(poke)
				end
				menu.select(poke, true)
			else
				input.press("A")
			end
		else
			input.press("B")
		end
		return
	end

	local main = memory.value("menu", "main")
	local column = menu.getCol()
	if main == 144 then
		if memory.value("battle", "menu") == 95 then
			input.press("B")
		else
			local idx = 0
			if poke then
				idx = pokemon.indexOf(poke)
			end
			menu.select(idx, true)
		end
	elseif main == 128 or main == 60 then
		if column == 5 then
			menu.select(inventory.indexOf(item), "accelerate", true)
		elseif column == 11 then
			menu.select(2, true)
		elseif column == 14 then
			menu.select(0, true)
		else
			local index = 0
			if poke then
				index = pokemon.indexOf(poke)
			end
			menu.select(index, true)
		end
	elseif main == 228 then
		if column == 14 and memory.value("battle", "menu") == 95 then
			input.press("B")
		end
	elseif main == 103 then
		input.press("B")
	else
		return false
	end
	return true
end

return inventory