-- 禁用不必要的if诊断警告
---@diagnostic disable: unnecessary-if

-- 导入基类模块
local PWClass = require("Core.PWClass")

--- @class StorageClass
--- @field max_capacity number 最大容量（格子总数）
--- @field cur_capacity number 当前已用容量
--- @field grids_list table[] 存储格子列表
local StorageClass = PWClass.derive("StorageClass")

--- 创建新物品格子
--- @param item_id any 物品唯一标识
--- @return table 新创建的物品格子
local function CreateGrid(item_id)
    return {
        item_id = item_id,  -- 物品ID
        cur_cnt = 0,        -- 当前数量
        max_cnt = 1         -- 最大堆叠数（可扩展为配置项）
    }
end

--- 查找或创建可用物品格子
--- @param storage StorageClass 存储实例
--- @param item_id any 目标物品ID
--- @return table 可用格子（找不到时创建新格子）
local function FindOrCreateAvailableGrid(storage, item_id)
    -- 优先查找同类型且未满的格子
    for _, grid in ipairs(storage.grids_list) do
        if grid.item_id == item_id and grid.cur_cnt < grid.max_cnt then
            return grid
        end
    end

    -- 无可用格子时创建新的物品类型格子
    local new_grid = CreateGrid(item_id)
    table.insert(storage.grids_list, new_grid)
    return new_grid
end

--- 构造函数
function StorageClass:ctor()
    self.max_capacity = 10     -- 默认最大容量
    self.cur_capacity = 0      -- 当前使用容量
    self.grids_list = {}       -- 格子容器
end

--- 设置存储容量上限
--- @param capacity number 新的最大容量
function StorageClass:SetMaxCapacity(capacity)
    self.max_capacity = capacity
end

--- 存储物品
--- @param item_id any 要存储的物品ID
function StorageClass:Store(item_id)
    -- 容量检查
    if self.cur_capacity >= self.max_capacity then
        return false  -- 建议返回操作结果
    end

    local grid = FindOrCreateAvailableGrid(self, item_id)
    grid.cur_cnt = grid.cur_cnt + 1
    self.cur_capacity = self.cur_capacity + 1
    return true  -- 建议添加返回值
end

--- 取出物品
--- @param item_id any 目标物品ID
--- @return boolean 是否成功取出
function StorageClass:Withdraw(item_id)
    -- 逆序遍历提高取出效率（通常新物品在末尾）
    for i = #self.grids_list, 1, -1 do
        local grid = self.grids_list[i]
        if grid ~= nil and grid.item_id == item_id and grid.cur_cnt > 0 then
            grid.cur_cnt = grid.cur_cnt - 1
            self.cur_capacity = self.cur_capacity - 1

            -- 清空空格子
            if grid.cur_cnt == 0 then
                table.remove(self.grids_list, i)
            end
            return true
        end
    end
    return false
end

function StorageClass:Visit(vistor)
    for _, grid in ipairs(self.grids_list) do vistor(_, grid) end
end

--- 查询物品（满足条件的物品及数量）
--- @param query_function fun(item_id:any):boolean 物品过滤函数
--- @return table 物品ID到数量的映射表
function StorageClass:QueryItem(query_function)
    local items = {}
    for _, grid in ipairs(self.grids_list) do
        -- 仅统计有物品且满足查询条件的格子
        if grid.cur_cnt > 0 and query_function(grid.item_id) then
            items[grid.item_id] = (items[grid.item_id] or 0) + grid.cur_cnt
        end
    end
    return items
end

return StorageClass