rbxIdleWeapon/src/ReplicatedStorage/Base/UIList.luau

local UIList = {}
UIList.__index = UIList

--> Services
local ReplicatedStorage = game:GetService("ReplicatedStorage")

--> Dependencies
local Utils = require(ReplicatedStorage.Tools.Utils)

-- 子脚本自动注入
local function AutoInjectVariables(self)
    for varName, _ in pairs(self.Variables) do
        if typeof(varName) == "string" then
            local firstChar = string.sub(varName, 1, 1)
            local sixChar = string.sub(varName, 1, 6)
            local target

            if firstChar == "_" then
                if sixChar == "__list" then
                    local prefab = Utils:FindInDescendantsUI(self.UIRoot, varName)
                    target = UIList:Init(prefab)
                else
                    -- _开头，递归查找
                    target = Utils:FindInDescendantsUI(self.UIRoot, varName)
                end

                if not target then
                    error("自动注入失败：未找到UI节点 " .. varName)
                end

                self.Variables[varName] = target
            end
        end
    end
end

function UIList:Init(Prefab: Instance)
    local self = {}
    setmetatable(self, UIList)
    self.Data = {}
    self.Instances = {}
    self.Connections = {}
    self.LayoutOrderKey = nil
    self.Component = nil
    self.UIRoot = Prefab
    self.Org = Utils:FindInDescendantsUI(Prefab, "__org")
    self.Org.Visible = false

    return self
end

function UIList:AddComponent(Component: table)
    self.Component = Component
end

function UIList:SetData(Data: table)
    self.Data = Data
    self:Refresh()
end

function UIList:AddData(data: table)
    self.Data[#self.Data + 1] = data
    self:SetSingleInstance(#self.Data, data)
end

function UIList:RemoveData(data: table)
    for i, v in pairs(self.Data) do
        if Utils:CompareTableJson(v, data) then
            self.Instances[i]:Destroy()
            table.remove(self.Data, i)
            return i
        end
    end
end

function UIList:SetLayoutOrder(keyName: string)
    self.LayoutOrderKey = keyName
    for _, ui in pairs(self.Instances) do
        ui.UIRoot.LayoutOrder = tonumber(ui.Data[keyName])
    end
end

function UIList:SetSingleInstance(index: number, data: table)
    local child = self.Component:Init(data)
    child.TopUI = self.TopUI
    child.ListUI = self

    local uiInstance = self.Org:Clone()
    child.UIRoot = uiInstance
    uiInstance.Name = index
    uiInstance.Parent = self.Org.Parent
    if self.LayoutOrderKey then
        uiInstance.LayoutOrder = tonumber(child.Data[self.LayoutOrderKey])
    end
    self.Instances[index] = child

    AutoInjectVariables(child)
    if child.OnInitFinish then child:OnInitFinish() end
    child:Refresh()
    uiInstance.Visible = true
end

function UIList:GetMinLayoutOrderInstance()
    local minOrder = math.huge
    local minInstance = nil
    for _, ui in pairs(self.Instances) do
        if ui.UIRoot.LayoutOrder < minOrder then
            minOrder = ui.UIRoot.LayoutOrder
            minInstance = ui
        end
    end
    return minInstance
end

function UIList:Refresh()
    for _, ui in pairs(self.Instances) do ui:Destroy() end
    self.Instances = {}

    if not self.Component then warn("UIList:Refresh() Component未设置") return end

    if self.Data then
        for k, v in pairs(self.Data) do
            self:SetSingleInstance(k, v)
        end
    end
end

function UIList:Clean()
    if self.TopUI then self.TopUI = nil end
    if self.ListUI then self.ListUI = nil end
    -- 清除自己的内容
    for _, connection in pairs(self.Connections) do
        connection:Disconnect()
    end
    self.Connections = {}

    -- 清除下面组件的内容
    for _, ui in pairs(self.Instances) do
        if ui.SignalConnections then
            for _, connection in pairs(ui.SignalConnections) do
                connection:DisconnectAll()
            end
            self.SignalConnections = nil
        end

        if ui.Connections then
            for _, connection in pairs(ui.Connections) do
                connection:Disconnect()
            end
            self.Connections = nil
        end
    
        if ui.UIRoot then ui.UIRoot:Destroy() end
        ui:Destroy()
    end
    self.Instances = {}
end

function UIList:Destroy()
    self:Clean()
    self.SignalConnections = nil
    self.Connections = nil
    self.Org:Destroy()
    self = nil
end

return UIList