💾 Add Persistence

-- ServerScriptService/Persistence.luau

local ProfileService = require(game:GetService("ServerScriptService").ProfileStore)
local InventoryService = require(game:GetService("ReplicatedStorage"):WaitForChild("Shared"):WaitForChild("StowayServerV1_2"))

-- Create ProfileStore key
local PlayerData = ProfileService.GetProfileStore("PlayerData", {
    Template = {
        Items = {},           -- UUID -> Item data
        ItemsByID = {},       -- ItemId -> UUID array
        Hotbar = {},          -- Slot -> UUID
        Storage = {},         -- UUID array
        EquippedItemUUID = nil,
        Weight = 0,
        Settings = {
            Limit = 15,
            CanStack = true,
            MaxStackSize = 5,
            MaxHotbarSlots = 10,
            Droppable = true,
            UiType = "Default"
        }
    }
})

-- STEP 1: Load profile on player join
local OnPlayerAdded = function(player)
    local profile = PlayerData.LoadProfileAsync("Player_" .. player.UserId)

    if profile then
        profile.AddUserId(player.UserId)  -- Prevent concurrent writes
        profile.Release()               -- Allow player to load into game

        -- Get or create inventory state
        local state = InventoryService.GetState(player)
        if state then
            -- STEP 2: Replace state with loaded data
            if profile.Data and profile.Data.Items then
                state.Items = profile.Data.Items
                state.ItemsByID = profile.Data.ItemsByID or {}
                state.Hotbar = profile.Data.Hotbar or {}
                state.Storage = profile.Data.Storage or {}
                state.EquippedItemUUID = profile.Data.EquippedItemUUID
                state.Weight = profile.Data.Weight or 0

                -- Apply saved settings or use defaults
                if profile.Data.Settings then
                    for k, v in pairs(profile.Data.Settings) do
                        state.Settings[k] = v
                    end
                end

                -- Sync to client
                InventoryService.SyncSettings(player)
            end
        end
    else
        warn("Failed to load profile for", player.Name)
    end
end

-- STEP 3: Save on player leave
local OnPlayerRemoving = function(player)
    local profile = PlayerData.FindProfileByUserId("Player_" .. player.UserId)
    if profile then
        -- Get latest state before saving
        local state = InventoryService.GetState(player)
        if state then
            profile.Data.Items = state.Items
            profile.Data.ItemsByID = state.ItemsByID
            profile.Data.Hotbar = state.Hotbar
            profile.Data.Storage = state.Storage
            profile.Data.EquippedItemUUID = state.EquippedItemUUID
            profile.Data.Weight = state.Weight
            profile.Data.Settings = state.Settings
        end

        -- Save and release
        profile:SaveAsync()
        profile:Release()
    end
end

game:GetService("Players").PlayerAdded:Connect(OnPlayerAdded)
game:GetService("Players").PlayerRemoving:Connect(OnPlayerRemoving)

-- ServerScriptService/Persistence.luau

local DataStoreService = game:GetService("DataStoreService")
local InventoryStore = DataStoreService:GetDataStore("InventoryStore")
local InventoryService = require(game:GetService("ReplicatedStorage"):WaitForChild("Shared"):WaitForChild("StowayServerV1_2"))

local SavePlayerData = function(player)
    local state = InventoryService.GetState(player)
    if not state then return end

    local success, err = pcall(function()
        InventoryStore:SetAsync(
            "Player_" .. player.UserId,
            {
                Items = state.Items,
                ItemsByID = state.ItemsByID,
                Hotbar = state.Hotbar,
                Storage = state.Storage,
                EquippedItemUUID = state.EquippedItemUUID,
                Weight = state.Weight,
                Settings = state.Settings
            }
        )
    end)

    if not success then
        warn("Failed to save inventory for", player.Name, err)
    end
end

local LoadPlayerData = function(player)
    local success, data = pcall(function()
        return InventoryStore:GetAsync("Player_" .. player.UserId)
    end)

    if success and data then
        local state = InventoryService.GetState(player)
        if state then
            state.Items = data.Items or {}
            state.ItemsByID = data.ItemsByID or {}
            state.Hotbar = data.Hotbar or {}
            state.Storage = data.Storage or {}
            state.EquippedItemUUID = data.EquippedItemUUID
            state.Weight = data.Weight or 0

            if data.Settings then
                for k, v in pairs(data.Settings) do
                    state.Settings[k] = v
                end
            end
        end
    end
end

-- Connect to player lifecycle
game:GetService("Players").PlayerAdded:Connect(function(player)
    task.wait(2)  -- Wait for inventory to initialize
    LoadPlayerData(player)
end)

game:GetService("Players").PlayerRemoving:Connect(function(player)
    SavePlayerData(player)
end)

-- Auto-save every 60 seconds
task.spawn(function()
    while true do
        task.wait(60)
        for _, player in ipairs(game:GetService("Players"):GetPlayers()) do
            if player and player:IsA("Player") then
                task.spawn(SavePlayerData, player)
            end
        end
    end
end)