--[[
    SetupRemotes.lua
    Run this ONCE in Studio command bar to create all RemoteEvents and folder structure
    
    Copy/paste into command bar:
    require(game.ServerStorage.SetupRemotes)()
]]

local ReplicatedStorage = game:GetService("ReplicatedStorage")

local function setup()
    print("[Setup] Creating admin panel structure...")
    
    -- Create Remotes folder
    local remotes = ReplicatedStorage:FindFirstChild("Remotes")
    if not remotes then
        remotes = Instance.new("Folder")
        remotes.Name = "Remotes"
        remotes.Parent = ReplicatedStorage
    end
    
    -- Create all RemoteEvents
    local remoteNames = {
        "OpenPanel",
        "RunAdminAction",
        "ServerAnnouncement",
        "ServerBanner",
        "RequestPlayerList",
        "UserAdminChange",
        "LuckMultiplierChanged",
        "SendAnnouncement",
        "DoubleLuck"
    }
    
    for _, remoteName in ipairs(remoteNames) do
        if not remotes:FindFirstChild(remoteName) then
            local remote = Instance.new("RemoteEvent")
            remote.Name = remoteName
            remote.Parent = remotes
            print("[Setup] Created RemoteEvent:", remoteName)
        end
    end
    
    -- Create ServerState folder
    local serverState = ReplicatedStorage:FindFirstChild("ServerState")
    if not serverState then
        serverState = Instance.new("Folder")
        serverState.Name = "ServerState"
        serverState.Parent = ReplicatedStorage
    end
    
    -- Create LuckMultiplier value
    local luckMultiplier = serverState:FindFirstChild("LuckMultiplier")
    if not luckMultiplier then
        luckMultiplier = Instance.new("NumberValue")
        luckMultiplier.Name = "LuckMultiplier"
        luckMultiplier.Value = 1
        luckMultiplier.Parent = serverState
        print("[Setup] Created LuckMultiplier NumberValue")
    end
    
    print("[Setup] ✓ All remotes and values created successfully!")
    print("[Setup] Next steps:")
    print("  1. Place AdminConfig in ServerStorage")
    print("  2. Place AdminActions in ServerStorage")
    print("  3. Place ActionControllers in ServerStorage")
    print("  4. Place UIThemes in ServerStorage")
    print("  5. Place AdminCore in ServerScriptService")
    print("  6. Place AdminPanelUI in StarterPlayer.StarterPlayerScripts")
    print("  7. Place ClientEffects in StarterPlayer.StarterPlayerScripts")
    print("  8. Set your OWNER_USERID in AdminConfig.lua")
end

return setup

--[[
    AdminConfig.lua
    Manages admin allowlist via DataStore
    Owner-only editing; provides IsAdmin/IsOwner checks
]]

local DataStoreService = game:GetService("DataStoreService")
local Players = game:GetService("Players")

local AdminConfig = {}

-- IMPORTANT: Set your Roblox UserId here as the owner
local OWNER_USERID = 123456789 -- CHANGE THIS TO YOUR USERID

local AdminStore = DataStoreService:GetDataStore("AdminAllowlist_v1")
local ADMIN_KEY = "AllowedAdmins"

-- Cache of admin UserIds
local adminCache = {}

-- Load admins from DataStore on server start
local function loadAdmins()
    local success, data = pcall(function()
        return AdminStore:GetAsync(ADMIN_KEY)
    end)
    
    if success and data then
        adminCache = data
        print("[AdminConfig] Loaded admins:", table.concat(adminCache, ", "))
    else
        adminCache = {}
        print("[AdminConfig] No admins found in DataStore, starting fresh")
    end
end

-- Save admins to DataStore
local function saveAdmins()
    local success, err = pcall(function()
        AdminStore:SetAsync(ADMIN_KEY, adminCache)
    end)
    
    if not success then
        warn("[AdminConfig] Failed to save admins:", err)
    end
end

-- Check if userId is the owner
function AdminConfig.IsOwner(userId)
    return userId == OWNER_USERID
end

-- Check if userId is an admin (or owner)
function AdminConfig.IsAdmin(userId)
    if AdminConfig.IsOwner(userId) then
        return true
    end
    
    return table.find(adminCache, userId) ~= nil
end

-- Add admin by username (owner-only)
function AdminConfig.AddAdmin(operatorUserId, username)
    if not AdminConfig.IsOwner(operatorUserId) then
        return false, "Only the owner can add admins"
    end
    
    -- Get UserId from username
    local success, userId = pcall(function()
        return Players:GetUserIdFromNameAsync(username)
    end)
    
    if not success or not userId then
        return false, "User not found: " .. username
    end
    
    -- Check if already admin
    if table.find(adminCache, userId) then
        return false, username .. " is already an admin"
    end
    
    -- Add to cache and save
    table.insert(adminCache, userId)
    saveAdmins()
    
    return true, username .. " added as admin"
end

-- Remove admin by username (owner-only)
function AdminConfig.RemoveAdmin(operatorUserId, username)
    if not AdminConfig.IsOwner(operatorUserId) then
        return false, "Only the owner can remove admins"
    end
    
    -- Get UserId from username
    local success, userId = pcall(function()
        return Players:GetUserIdFromNameAsync(username)
    end)
    
    if not success or not userId then
        return false, "User not found: " .. username
    end
    
    -- Check if in cache
    local index = table.find(adminCache, userId)
    if not index then
        return false, username .. " is not an admin"
    end
    
    -- Remove from cache and save
    table.remove(adminCache, index)
    saveAdmins()
    
    return true, username .. " removed from admins"
end

-- Get all admin UserIds
function AdminConfig.GetAdmins()
    return adminCache
end

-- Initialize
loadAdmins()

return AdminConfig

--[[
    AdminActions.lua
    Server-side implementations for admin actions
    Includes: Fling, Freeze, ToggleFly, ToggleInvincibility, ToggleInvisibility
]]

local AdminActions = {}

-- Ragdoll helper for R15 rigs
local function ragdollCharacter(character, duration)
    local humanoid = character:FindFirstChildOfClass("Humanoid")
    if not humanoid then return end
    
    -- Store original state
    local originalState = humanoid:GetState()
    
    -- Switch to physics state for ragdoll effect
    humanoid:ChangeState(Enum.HumanoidStateType.Physics)
    humanoid.PlatformStand = true
    
    -- Disable Motor6Ds temporarily for true ragdoll
    local motors = {}
    for _, desc in pairs(character:GetDescendants()) do
        if desc:IsA("Motor6D") then
            table.insert(motors, {motor = desc, enabled = desc.Enabled})
            desc.Enabled = false
        end
    end
    
    -- Restore after duration
    task.delay(duration, function()
        if character and character.Parent then
            -- Re-enable motors
            for _, data in pairs(motors) do
                if data.motor and data.motor.Parent then
                    data.motor.Enabled = data.enabled
                end
            end
            
            -- Restore humanoid state
            if humanoid and humanoid.Parent then
                humanoid.PlatformStand = false
                humanoid:ChangeState(Enum.HumanoidStateType.GettingUp)
            end
        end
    end)
end

-- FLING: Throw player in random direction
function AdminActions.Fling(targetPlayer)
    local character = targetPlayer.Character
    if not character then return end
    
    local hrp = character:FindFirstChild("HumanoidRootPart")
    if not hrp then return end
    
    -- Generate random direction
    local randomX = math.random(-150, 150)
    local randomY = math.random(100, 200)
    local randomZ = math.random(-150, 150)
    
    -- Apply velocity
    hrp.AssemblyLinearVelocity = Vector3.new(randomX, randomY, randomZ)
end

-- FREEZE: Anchor character for 5 seconds
function AdminActions.Freeze5s(targetPlayer)
    local character = targetPlayer.Character
    if not character then return end
    
    local hrp = character:FindFirstChild("HumanoidRootPart")
    if not hrp then return end
    
    -- Anchor
    hrp.Anchored = true
    
    -- Unanchor after 5 seconds
    task.delay(5, function()
        if hrp and hrp.Parent then
            hrp.Anchored = false
        end
    end)
end

-- TOGGLE FLY: Set attribute; ActionControllers handles the physics
function AdminActions.ToggleFly(targetPlayer)
    local currentState = targetPlayer:GetAttribute("Admin_Fly") or false
    targetPlayer:SetAttribute("Admin_Fly", not currentState)
    return not currentState -- Return new state
end

-- TOGGLE INVINCIBILITY: Set attribute; ActionControllers handles damage immunity
function AdminActions.ToggleInvincibility(targetPlayer)
    local currentState = targetPlayer:GetAttribute("Admin_Invincible") or false
    targetPlayer:SetAttribute("Admin_Invincible", not currentState)
    return not currentState -- Return new state
end

-- TOGGLE INVISIBILITY: Set attribute; ActionControllers handles transparency
function AdminActions.ToggleInvisibility(targetPlayer)
    local currentState = targetPlayer:GetAttribute("Admin_Invisible") or false
    targetPlayer:SetAttribute("Admin_Invisible", not currentState)
    return not currentState -- Return new state
end

return AdminActions

--[[
    ActionControllers.lua
    Applies and maintains fly/invincible/invisible state based on player attributes
    Handles respawns and state persistence
]]

local Players = game:GetService("Players")

local ActionControllers = {}

-- Store original transparency values for invisibility restoration
local invisibilityData = {}

-- FLY CONTROLLER: Client-controlled WASD flight with server physics
local function setupFlyController(player)
    local character = player.Character
    if not character then return end
    
    local hrp = character:FindFirstChild("HumanoidRootPart")
    local humanoid = character:FindFirstChildOfClass("Humanoid")
    if not hrp or not humanoid then return end
    
    -- Check if fly is enabled
    local function updateFly()
        local flyEnabled = player:GetAttribute("Admin_Fly") or false
        
        if flyEnabled then
            -- Create BodyVelocity for flight control
            if not hrp:FindFirstChild("AdminFlyVelocity") then
                local bodyVel = Instance.new("BodyVelocity")
                bodyVel.Name = "AdminFlyVelocity"
                bodyVel.Velocity = Vector3.new(0, 0, 0)
                bodyVel.MaxForce = Vector3.new(100000, 100000, 100000)
                bodyVel.Parent = hrp
                
                local bodyGyro = Instance.new("BodyGyro")
                bodyGyro.Name = "AdminFlyGyro"
                bodyGyro.MaxTorque = Vector3.new(100000, 100000, 100000)
                bodyGyro.P = 10000
                bodyGyro.CFrame = hrp.CFrame
                bodyGyro.Parent = hrp
            end
            
            -- Set humanoid to seated state (prevents walking animations)
            humanoid.Sit = true
            humanoid.PlatformStand = true
            
        else
            -- Remove fly physics
            local bodyVel = hrp:FindFirstChild("AdminFlyVelocity")
            local bodyGyro = hrp:FindFirstChild("AdminFlyGyro")
            if bodyVel then bodyVel:Destroy() end
            if bodyGyro then bodyGyro:Destroy() end
            
            -- Restore normal humanoid states
            humanoid.Sit = false
            humanoid.PlatformStand = false
        end
    end
    
    -- Listen for attribute changes
    local conn = player:GetAttributeChangedSignal("Admin_Fly"):Connect(updateFly)
    
    -- Initial update
    updateFly()
    
    -- Cleanup on character removal
    character.Destroying:Once(function()
        conn:Disconnect()
    end)
end

-- INVINCIBILITY CONTROLLER: Prevent damage via HealthChanged
local function setupInvincibilityController(player)
    local character = player.Character
    if not character then return end
    
    local humanoid = character:FindFirstChildOfClass("Humanoid")
    if not humanoid then return end
    
    -- Monitor health changes and restore if invincible
    local conn = humanoid.HealthChanged:Connect(function(health)
        local invincible = player:GetAttribute("Admin_Invincible") or false
        
        if invincible and health < humanoid.MaxHealth then
            humanoid.Health = humanoid.MaxHealth
        end
    end)
    
    -- Also prevent death state
    local function updateInvincibility()
        local invincible = player:GetAttribute("Admin_Invincible") or false
        
        if invincible then
            humanoid.Health = humanoid.MaxHealth
            humanoid:SetStateEnabled(Enum.HumanoidStateType.Dead, false)
        else
            humanoid:SetStateEnabled(Enum.HumanoidStateType.Dead, true)
        end
    end
    
    local attrConn = player:GetAttributeChangedSignal("Admin_Invincible"):Connect(updateInvincibility)
    updateInvincibility()
    
    -- Cleanup
    character.Destroying:Once(function()
        conn:Disconnect()
        attrConn:Disconnect()
    end)
end

-- INVISIBILITY CONTROLLER: Set transparency on all parts
local function setupInvisibilityController(player)
    local character = player.Character
    if not character then return end
    
    -- Store original transparency values
    local originalTransparency = {}
    
    local function applyInvisibility(invisible)
        for _, desc in pairs(character:GetDescendants()) do
            if desc:IsA("BasePart") then
                if invisible then
                    -- Store and hide
                    if not originalTransparency[desc] then
                        originalTransparency[desc] = desc.Transparency
                    end
                    desc.Transparency = 1
                else
                    -- Restore
                    if originalTransparency[desc] then
                        desc.Transparency = originalTransparency[desc]
                    end
                end
            elseif desc:IsA("Decal") or desc:IsA("Texture") then
                if invisible then
                    if not originalTransparency[desc] then
                        originalTransparency[desc] = desc.Transparency
                    end
                    desc.Transparency = 1
                else
                    if originalTransparency[desc] then
                        desc.Transparency = originalTransparency[desc]
                    end
                end
            end
        end
        
        -- Handle face specifically
        local head = character:FindFirstChild("Head")
        if head then
            local face = head:FindFirstChildOfClass("Decal")
            if face then
                if invisible then
                    originalTransparency[face] = face.Transparency
                    face.Transparency = 1
                else
                    if originalTransparency[face] then
                        face.Transparency = originalTransparency[face]
                    end
                end
            end
        end
    end
    
    -- Listen for attribute changes
    local function updateInvisibility()
        local invisible = player:GetAttribute("Admin_Invisible") or false
        applyInvisibility(invisible)
    end
    
    local conn = player:GetAttributeChangedSignal("Admin_Invisible"):Connect(updateInvisibility)
    updateInvisibility()
    
    -- Cleanup
    character.Destroying:Once(function()
        conn:Disconnect()
    end)
end

-- Setup all controllers for a character
local function onCharacterAdded(player, character)
    setupFlyController(player)
    setupInvincibilityController(player)
    setupInvisibilityController(player)
end

-- Initialize for all players
function ActionControllers.Init()
    Players.PlayerAdded:Connect(function(player)
        player.CharacterAdded:Connect(function(character)
            onCharacterAdded(player, character)
        end)
        
        -- Handle if character already exists
        if player.Character then
            onCharacterAdded(player, player.Character)
        end
    end)
    
    -- Setup for existing players
    for _, player in pairs(Players:GetPlayers()) do
        if player.Character then
            onCharacterAdded(player, player.Character)
        end
        
        player.CharacterAdded:Connect(function(character)
            onCharacterAdded(player, character)
        end)
    end
end

return ActionControllers

--[[
    UIThemes.lua
    Design tokens for consistent UI styling
]]

local UIThemes = {}

-- Color palette
UIThemes.Colors = {
    -- Main panel
    Background = Color3.fromRGB(25, 25, 30),
    BackgroundSecondary = Color3.fromRGB(35, 35, 40),
    Border = Color3.fromRGB(60, 60, 70),
    
    -- Accent colors
    Primary = Color3.fromRGB(88, 166, 255),
    Success = Color3.fromRGB(76, 209, 55),
    Warning = Color3.fromRGB(255, 193, 7),
    Danger = Color3.fromRGB(231, 76, 60),
    
    -- Text
    TextPrimary = Color3.fromRGB(255, 255, 255),
    TextSecondary = Color3.fromRGB(180, 180, 190),
    TextMuted = Color3.fromRGB(120, 120, 130),
    
    -- Interactive
    ButtonNormal = Color3.fromRGB(50, 50, 60),
    ButtonHover = Color3.fromRGB(70, 70, 80),
    ButtonActive = Color3.fromRGB(88, 166, 255),
    
    -- Status
    StatusOn = Color3.fromRGB(76, 209, 55),
    StatusOff = Color3.fromRGB(120, 120, 130),
}

-- Typography
UIThemes.Fonts = {
    Primary = Enum.Font.GothamBold,
    Secondary = Enum.Font.Gotham,
    Monospace = Enum.Font.RobotoMono,
}

-- Spacing
UIThemes.Spacing = {
    Small = 4,
    Medium = 8,
    Large = 12,
    XLarge = 16,
}

-- Sizing
UIThemes.Sizes = {
    ButtonHeight = 40,
    InputHeight = 36,
    MinButtonWidth = 100,
    IconSize = 20,
}

return UIThemes

--[[
    ClientEffects.lua
    Handles banner/announcement animations
    Only shows banners for: Announcements and Luck multiplier changes (not toggle spam)
]]

local Players = game:GetService("Players")
local ReplicatedStorage = game:GetService("ReplicatedStorage")
local TweenService = game:GetService("TweenService")

local player = Players.LocalPlayer
local playerGui = player:WaitForChild("PlayerGui")

-- Remotes
local Remotes = ReplicatedStorage:WaitForChild("Remotes")
local ServerAnnouncement = Remotes:WaitForChild("ServerAnnouncement")
local ServerBanner = Remotes:WaitForChild("ServerBanner")
local LuckMultiplierChanged = Remotes:WaitForChild("LuckMultiplierChanged")

-- Banner queue
local bannerQueue = {}
local isShowingBanner = false

-- ========================================
-- BANNER DISPLAY
-- ========================================

local function createBannerGui()
    local screenGui = Instance.new("ScreenGui")
    screenGui.Name = "BannerEffects"
    screenGui.ResetOnSpawn = false
    screenGui.ZIndexBehavior = Enum.ZIndexBehavior.Sibling
    screenGui.DisplayOrder = 10
    screenGui.Parent = playerGui
    
    return screenGui
end

local bannerGui = createBannerGui()

local function showBanner(text, isAnnouncement)
    -- Create banner frame
    local banner = Instance.new("Frame")
    banner.Name = "Banner"
    banner.BackgroundColor3 = Color3.fromRGB(0, 0, 0)
    banner.BackgroundTransparency = 0.3
    banner.BorderSizePixel = 0
    banner.AnchorPoint = Vector2.new(0.5, 0)
    banner.Position = UDim2.new(0.5, 0, 0, -60)
    banner.Size = UDim2.new(0, 500, 0, 50)
    banner.Parent = bannerGui
    
    local corner = Instance.new("UICorner")
    corner.CornerRadius = UDim.new(0, 8)
    corner.Parent = banner
    
    -- Black border
    local stroke = Instance.new("UIStroke")
    stroke.Color = Color3.fromRGB(0, 0, 0)
    stroke.Thickness = 2
    stroke.Parent = banner
    
    -- Banner text (white)
    local label = Instance.new("TextLabel")
    label.Text = text
    label.Font = Enum.Font.GothamBold
    label.TextSize = 16
    label.TextColor3 = Color3.fromRGB(255, 255, 255)
    label.BackgroundTransparency = 1
    label.Size = UDim2.new(1, -20, 1, 0)
    label.Position = UDim2.new(0, 10, 0, 0)
    label.TextWrapped = true
    label.TextXAlignment = Enum.TextXAlignment.Center
    label.Parent = banner
    
    -- Animate in
    local tweenIn = TweenService:Create(banner, TweenInfo.new(0.4, Enum.EasingStyle.Back, Enum.EasingDirection.Out), {
        Position = UDim2.new(0.5, 0, 0, 20)
    })
    
    tweenIn:Play()
    
    -- Hold for duration
    task.wait(3)
    
    -- Animate out
    local tweenOut = TweenService:Create(banner, TweenInfo.new(0.3, Enum.EasingStyle.Back, Enum.EasingDirection.In), {
        Position = UDim2.new(0.5, 0, 0, -60)
    })
    
    tweenOut:Play()
    tweenOut.Completed:Wait()
    
    banner:Destroy()
end

local function processBannerQueue()
    if isShowingBanner or #bannerQueue == 0 then
        return
    end
    
    isShowingBanner = true
    local bannerData = table.remove(bannerQueue, 1)
    
    showBanner(bannerData.text, bannerData.isAnnouncement)
    
    isShowingBanner = false
    
    -- Process next in queue
    if #bannerQueue > 0 then
        processBannerQueue()
    end
end

-- ========================================
-- REMOTE LISTENERS
-- ========================================

-- Global announcements (high priority)
ServerAnnouncement.OnClientEvent:Connect(function(text)
    table.insert(bannerQueue, {text = text, isAnnouncement = true})
    processBannerQueue()
end)

-- Generic banners (lower priority, used sparingly)
ServerBanner.OnClientEvent:Connect(function(text)
    -- This is now only used for non-toggle feedback (like errors)
    -- We don't queue these, just show immediately and briefly
    local banner = Instance.new("Frame")
    banner.BackgroundColor3 = Color3.fromRGB(50, 50, 60)
    banner.BorderSizePixel = 0
    banner.AnchorPoint = Vector2.new(0.5, 1)
    banner.Position = UDim2.new(0.5, 0, 1, -20)
    banner.Size = UDim2.new(0, 300, 0, 40)
    banner.Parent = bannerGui
    
    local corner = Instance.new("UICorner")
    corner.CornerRadius = UDim.new(0, 6)
    corner.Parent = banner
    
    local label = Instance.new("TextLabel")
    label.Text = text
    label.Font = Enum.Font.Gotham
    label.TextSize = 14
    label.TextColor3 = Color3.fromRGB(255, 255, 255)
    label.BackgroundTransparency = 1
    label.Size = UDim2.new(1, 0, 1, 0)
    label.TextWrapped = true
    label.Parent = banner
    
    -- Fade out after 2 seconds
    task.wait(2)
    
    local fadeOut = TweenService:Create(banner, TweenInfo.new(0.3), {
        BackgroundTransparency = 1
    })
    local labelFade = TweenService:Create(label, TweenInfo.new(0.3), {
        TextTransparency = 1
    })
    
    fadeOut:Play()
    labelFade:Play()
    fadeOut.Completed:Wait()
    
    banner:Destroy()
end)

print("[ClientEffects] Banner system loaded")

--[[
    ClientFlightController.lua
    Client-side WASD flight controls when Admin_Fly is enabled
    Place in StarterPlayer > StarterPlayerScripts
]]

local Players = game:GetService("Players")
local UserInputService = game:GetService("UserInputService")
local RunService = game:GetService("RunService")

local player = Players.LocalPlayer
local camera = workspace.CurrentCamera

local flySpeed = 50
local flyConnection = nil

-- Track which keys are pressed
local keys = {
    W = false,
    A = false,
    S = false,
    D = false,
    Space = false,
    Shift = false,
    Q = false,
    E = false
}

-- Update flight velocity based on input
local function updateFlyVelocity(character)
    local hrp = character:FindFirstChild("HumanoidRootPart")
    if not hrp then return end
    
    local bodyVel = hrp:FindFirstChild("AdminFlyVelocity")
    local bodyGyro = hrp:FindFirstChild("AdminFlyGyro")
    if not bodyVel or not bodyGyro then return end
    
    -- Calculate movement direction
    local moveDirection = Vector3.new(0, 0, 0)
    
    if keys.W then
        moveDirection = moveDirection + camera.CFrame.LookVector
    end
    if keys.S then
        moveDirection = moveDirection - camera.CFrame.LookVector
    end
    if keys.A then
        moveDirection = moveDirection - camera.CFrame.RightVector
    end
    if keys.D then
        moveDirection = moveDirection + camera.CFrame.RightVector
    end
    if keys.Space then
        moveDirection = moveDirection + Vector3.new(0, 1, 0)
    end
    if keys.Shift then
        moveDirection = moveDirection - Vector3.new(0, 1, 0)
    end
    
    -- Normalize and apply speed
    if moveDirection.Magnitude > 0 then
        moveDirection = moveDirection.Unit * flySpeed
    end
    
    -- Update BodyVelocity
    bodyVel.Velocity = moveDirection
    
    -- Update BodyGyro to match camera orientation
    bodyGyro.CFrame = CFrame.new(hrp.Position, hrp.Position + camera.CFrame.LookVector)
end

-- Start flying
local function startFlying(character)
    if flyConnection then return end -- Already flying
    
    flyConnection = RunService.Heartbeat:Connect(function()
        updateFlyVelocity(character)
    end)
end

-- Stop flying
local function stopFlying()
    if flyConnection then
        flyConnection:Disconnect()
        flyConnection = nil
    end
    
    -- Reset all keys
    for key, _ in pairs(keys) do
        keys[key] = false
    end
end

-- Monitor fly attribute changes
local function onCharacterAdded(character)
    local humanoid = character:WaitForChild("Humanoid", 5)
    if not humanoid then return end
    
    local function checkFlyState()
        local flyEnabled = player:GetAttribute("Admin_Fly") or false
        
        if flyEnabled then
            startFlying(character)
        else
            stopFlying()
        end
    end
    
    -- Initial check
    checkFlyState()
    
    -- Listen for changes
    player:GetAttributeChangedSignal("Admin_Fly"):Connect(checkFlyState)
end

-- Handle input
UserInputService.InputBegan:Connect(function(input, gameProcessed)
    if gameProcessed then return end
    
    if input.KeyCode == Enum.KeyCode.W then
        keys.W = true
    elseif input.KeyCode == Enum.KeyCode.A then
        keys.A = true
    elseif input.KeyCode == Enum.KeyCode.S then
        keys.S = true
    elseif input.KeyCode == Enum.KeyCode.D then
        keys.D = true
    elseif input.KeyCode == Enum.KeyCode.Space then
        keys.Space = true
    elseif input.KeyCode == Enum.KeyCode.LeftShift then
        keys.Shift = true
    elseif input.KeyCode == Enum.KeyCode.Q then
        keys.Q = true
    elseif input.KeyCode == Enum.KeyCode.E then
        keys.E = true
    end
end)

UserInputService.InputEnded:Connect(function(input)
    if input.KeyCode == Enum.KeyCode.W then
        keys.W = false
    elseif input.KeyCode == Enum.KeyCode.A then
        keys.A = false
    elseif input.KeyCode == Enum.KeyCode.S then
        keys.S = false
    elseif input.KeyCode == Enum.KeyCode.D then
        keys.D = false
    elseif input.KeyCode == Enum.KeyCode.Space then
        keys.Space = false
    elseif input.KeyCode == Enum.KeyCode.LeftShift then
        keys.Shift = false
    elseif input.KeyCode == Enum.KeyCode.Q then
        keys.Q = false
    elseif input.KeyCode == Enum.KeyCode.E then
        keys.E = false
    end
end)

-- Initialize
if player.Character then
    onCharacterAdded(player.Character)
end

player.CharacterAdded:Connect(onCharacterAdded)

print("[ClientFlightController] WASD flight system loaded")