ServerMovement

local RunSvc = game:GetService("RunService")
local Players = game:GetService("Players")
local RepStorage = game:GetService("ReplicatedStorage")

local PlayAnimationState = RepStorage:FindFirstChild("PlayAnimation")
if not PlayAnimationState then
    PlayAnimationState = Instance.new("RemoteEvent")
    PlayAnimationState.Name = "PlayAnimationState"
    PlayAnimationState.Parent = RepStorage
end

local MOVEMENT_THRESHOLD = 0.01
local CharacterData = {}

local function getSpecialState(currentHumState)
    if currentHumState == Enum.HumanoidStateType.Jumping then
        return "Jump"
    elseif currentHumState == Enum.HumanoidStateType.Freefall then
        return "Jump"
    elseif currentHumState == Enum.HumanoidStateType.Swimming then
        return "Swim"
    elseif currentHumState == Enum.HumanoidStateType.Climbing then
        return "Climb"
    elseif currentHumState == Enum.HumanoidStateType.Seated then
        return "Sit"
    end
    return nil
end

local function isStateValid(data, newState)
    if not data.Humanoid or not data.RootPart then return false end

    if newState == "Walk" then
        return data.Humanoid.MoveDirection.Magnitude > MOVEMENT_THRESHOLD
        and data.Humanoid.FloorMaterial ~= Enum.Material.Air

    elseif newState == "Idle" then
        return data.Humanoid.MoveDirection.Magnitude <= MOVEMENT_THRESHOLD
        and data.Humanoid.FloorMaterial ~= Enum.Material.Air
    end

    return true
end

local function FireToAllExceptOwner(playerToExclude, newState)
    for _, player in ipairs(Players:GetPlayers()) do
        if player ~= playerToExclude then
            PlayAnimationState:FireClient(player, playerToExclude, newState)
        end
    end
end

local function fireState(player, data, newState)
    if not isStateValid(data, newState) then return end
    if data.CurrentState == newState then return end

    data.CurrentState = newState

    PlayAnimationState:FireClient(player, player, newState)
    FireToAllExceptOwner(player, newState)
end

local function unlockAfterSeat(player, data)
    -- When unseated, instantly unlock & recalc state
    data.SpecialStateLock = false

    local human = data.Humanoid

    if human.MoveDirection.Magnitude > MOVEMENT_THRESHOLD then
        fireState(player, data, "Walk")
    else
        fireState(player, data, "Idle")
    end
end

local function handleCharacterAdded(character)
    local player = Players:GetPlayerFromCharacter(character)
    if not player then return end

    local humanoid = character:WaitForChild("Humanoid")
    local rootPart = character:WaitForChild("HumanoidRootPart")

    humanoid:SetStateEnabled(Enum.HumanoidStateType.Seated, true)

    local data = {
    Character = character,
    Humanoid = humanoid,
    RootPart = rootPart,
    CurrentState = "Idle",
    SpecialStateLock = false
    }
    CharacterData[player] = data

    PlayAnimationState:FireClient(player, player, "Idle")

    -- Detect entering & leaving seat
    humanoid:GetPropertyChangedSignal("Sit"):Connect(function()
        if humanoid.Sit == true then
            -- Player sits
            data.SpecialStateLock = true
            fireState(player, data, "Sit")
        else
            -- Player leaves the seat
            unlockAfterSeat(player, data)
        end
    end)

    humanoid.StateChanged:Connect(function(_, newState)
        local specialState = getSpecialState(newState)

        if specialState == "Sit" then
            data.SpecialStateLock = true
            fireState(player, data, "Sit")
            return
        end

        -- Leaving seated state
        if newState ~= Enum.HumanoidStateType.Seated and data.SpecialStateLock and humanoid.Sit == false then
            unlockAfterSeat(player, data)
            return
        end

        if specialState then
            data.SpecialStateLock = true
            fireState(player, data, specialState)
        else
            data.SpecialStateLock = false
        end
    end)
end

local function onPlayerAdded(player)
    player.CharacterAdded:Connect(handleCharacterAdded)
    player.CharacterRemoving:Connect(function()
        CharacterData[player] = nil
    end)
    if player.Character then
        handleCharacterAdded(player.Character)
    end
end

local function onPlayerRemoving(player)
    CharacterData[player] = nil
end

Players.PlayerAdded:Connect(onPlayerAdded)
Players.PlayerRemoving:Connect(onPlayerRemoving)

for _, player in ipairs(Players:GetPlayers()) do
    onPlayerAdded(player)
end

RunSvc.Heartbeat:Connect(function()
    for player, data in pairs(CharacterData) do
        if data.Character
            and data.Humanoid.Health > 0
            and data.RootPart
            and not data.SpecialStateLock then

            local moveMag = data.Humanoid.MoveDirection.Magnitude
            local nextState = (moveMag > MOVEMENT_THRESHOLD
            and data.Humanoid.FloorMaterial ~= Enum.Material.Air)
            and "Walk"
            or "Idle"

            if data.CurrentState ~= nextState then
                fireState(player, data, nextState)
            end
        end
    end
end)