--// Services & Variables
local ReplicatedStorage, RunService, Players, UserInputService, TweenService, Workspace
pcall(function()
    ReplicatedStorage = game:GetService("ReplicatedStorage")
    RunService = game:GetService("RunService")
    Players = game:GetService("Players")
    UserInputService = game:GetService("UserInputService")
    TweenService = game:GetService("TweenService")
    Workspace = game:GetService("Workspace")
end)

local player = Players.LocalPlayer
local leaderstats = player:WaitForChild("leaderstats", 15)
if not leaderstats then
    warn("[LokasornScript] ERROR: Could not find leaderstats! The script will not run.")
    return
end

--// Simple Logger
local Logger = { Enabled = true }
function Logger:Log(message)
    if not self.Enabled then return end
    print("[LokasornScript] INFO: " .. tostring(message))
end
function Logger:Warn(message)
    if not self.Enabled then return end
    warn("[LokasornScript] WARN: " .. tostring(message))
end
function Logger:Error(message)
    if not self.Enabled then return end
    warn("[LokasornScript] ERROR: " .. tostring(message))
end

Logger:Log("Script initialized successfully.")

--// Configuration
local autoTrainEnabled, fastTrainEnabled, fastBattleEnabled, antiAfkEnabled = false, false, false, false
local trainCoroutine, fastTrainCoroutine, fastBattleMonitorLoop, antiAfkCoroutine = nil, nil, nil, nil
local cooldown = false
local canFirePrompt = pcall(function() return fireproximityprompt end)
local canFireClick = pcall(function() return fireclickdetector end)

--// Game Content References
local WEIGHT_FOLDER = Workspace:FindFirstChild("STR0", true) and Workspace:FindFirstChild("STR0", true).Parent
local FOOD_FOLDER = Workspace:FindFirstChild("Foods")
local WEIGHT_SYSTEM_EVENT = ReplicatedStorage:FindFirstChild("WeightSystem")

if not WEIGHT_FOLDER then Logger:Error("Could not find WEIGHT_FOLDER.") end
if not FOOD_FOLDER then Logger:Error("Could not find FOOD_FOLDER.") end

--// Core Functions
local noclipConnection = nil
local function setNoclip(enabled, character)
    if not character then return end
    if enabled and not noclipConnection then
        Logger:Log("Noclip enabled.")
        noclipConnection = RunService.Stepped:Connect(function()
            for _, part in ipairs(character:GetDescendants()) do
                if part:IsA("BasePart") and part.CanCollide then part.CanCollide = false end
            end
        end)
    elseif not enabled and noclipConnection then
        Logger:Log("Noclip disabled.")
        noclipConnection:Disconnect()
        noclipConnection = nil
    end
end

local function parseRequirementValue(name)
    local num_part = name:match("STR([%d%.]+)")
    if not num_part then return nil end
    local num = tonumber(num_part)
    if name:find("K$") then num = num * 1e3
    elseif name:find("M$") then num = num * 1e6
    elseif name:find("B$") then num = num * 1e9 end
    return num
end

local function findBestStation(stationType)
    local highestLevel, bestStation = -1, nil
    local folder, requiredStat, reqFunc = nil, nil, nil

    if stationType == "Weight" then
        folder, requiredStat, reqFunc = WEIGHT_FOLDER, leaderstats.Strength, function(s) return parseRequirementValue(s.Name) end
    elseif stationType == "Food" then
        folder, requiredStat, reqFunc = FOOD_FOLDER, leaderstats.Wins, function(s) return s:FindFirstChild("Wins") and s.Wins.Value end
    end
    
    if not folder then return nil end
    
    for _, station in ipairs(folder:GetChildren()) do
        if station.Name ~= "Man" then
            local reqVal = reqFunc(station)
            if reqVal and requiredStat.Value >= reqVal and reqVal > highestLevel then
                highestLevel, bestStation = reqVal, station
            end
        end
    end
    return bestStation
end

local function teleportTo(humanoidRootPart, target)
    if not humanoidRootPart or not target then return end
    local targetPosition = target:IsA("Model") and target:GetBoundingBox().Position or target.Position
    if targetPosition then
        humanoidRootPart.CFrame = CFrame.new(targetPosition) * CFrame.new(0, 3, 0)
        task.wait(0.3)
    end
end

local function findPromptInTarget(target)
    if not target then return nil end
    if target:IsA("Model") then
        for _, child in ipairs(target:GetChildren()) do
            if (child.Name == "Part" and child:IsA("BasePart")) or child:IsA("Model") then
                local prompt = child:FindFirstChildOfClass("ProximityPrompt")
                if prompt then return prompt end
            end
        end
    end
    return target:FindFirstChildOfClass("ProximityPrompt", true)
end

local function triggerPrompt(prompt)
    if not prompt or not prompt.Parent then return end
    prompt.Enabled = true
    if canFirePrompt then
        fireproximityprompt(prompt)
    else
        prompt:InputHoldBegin()
        task.wait(prompt.HoldDuration + 0.1)
        prompt:InputHoldEnd()
    end
end

--// Logic Loops
local function antiAfkLogic()
    while antiAfkEnabled do
        local waitTime = math.random(60, 120)
        Logger:Log("Anti-AFK: Waiting for " .. waitTime .. " seconds before next action.")
        task.wait(waitTime)
        if not antiAfkEnabled then break end
        pcall(function()
            local character = player.Character
            local humanoid = character and character:FindFirstChildOfClass("Humanoid")
            if humanoid and humanoid.RootPart then
                Logger:Log("Anti-AFK: Simulating player activity.")
                humanoid:MoveTo(humanoid.RootPart.Position + Vector3.new(math.random(-2, 2), 0, math.random(-2, 2)))
            end
        end)
    end
    Logger:Log("Anti-AFK disabled.")
end

local function fastTrainLogic()
    while fastTrainEnabled do
        pcall(function() if WEIGHT_SYSTEM_EVENT then WEIGHT_SYSTEM_EVENT:FireServer(true) end end)
        task.wait(0.1)
    end
    Logger:Log("Fast Train disabled.")
end

local function autoTrainLogic()
    while autoTrainEnabled do
        local success, err = pcall(function()
            local character = player.Character
            if not character then task.wait(1); return end
            local humanoidRootPart = character:FindFirstChild("HumanoidRootPart")
            if not humanoidRootPart then task.wait(1); return end

            setNoclip(true, character)

            if leaderstats.Energy.Value <= -10 then
                Logger:Log("Energy low. Starting food routine.")
                if WEIGHT_SYSTEM_EVENT then WEIGHT_SYSTEM_EVENT:FireServer(0, false); task.wait(1) end
                local foodSpot = findBestStation("Food")
                if foodSpot then
                    teleportTo(humanoidRootPart, foodSpot)
                    task.wait(1)
                    local foodPrompt = findPromptInTarget(foodSpot)
                    if foodPrompt then
                        local stuckTimer = tick()
                        while leaderstats.Energy.Value < 95 and autoTrainEnabled do
                            local lastEnergy = leaderstats.Energy.Value
                            triggerPrompt(foodPrompt)
                            task.wait(1.5)
                            if leaderstats.Energy.Value > lastEnergy then stuckTimer = tick() end
                            if tick() - stuckTimer > 10 then
                                Logger:Warn("Energy stuck. Re-teleporting.")
                                teleportTo(humanoidRootPart, foodSpot)
                                stuckTimer = tick()
                            end
                        end
                        Logger:Log("Energy restored.")
                    end
                end
            else
                Logger:Log("Energy sufficient. Starting strength routine.")
                local weightStation = findBestStation("Weight")
                if weightStation then
                    teleportTo(humanoidRootPart, weightStation)
                    task.wait(1)
                    local prompt = findPromptInTarget(weightStation)
                    if prompt then
                        local trainConn = RunService.Heartbeat:Connect(function() triggerPrompt(prompt) end)
                        repeat task.wait(0.5) until leaderstats.Energy.Value <= -10 or not autoTrainEnabled
                        trainConn:Disconnect()
                        Logger:Log("Finished training session.")
                    end
                end
            end
        end)
        if not success then Logger:Error("Auto Train Loop: " .. tostring(err)) end
        task.wait(1)
    end
    setNoclip(false, player.Character)
    Logger:Log("Auto Train disabled.")
    if _G.mainFrame then pcall(_G.toggleSwitch, _G.autoTrainButton, _G.autoTrainKnob, _G.autoTrainLabel, false) end
end

--// Other Features
local function activateAllAuras()
    Logger:Log("Attempting to activate all auras...")
    pcall(function()
        local auraSystem = ReplicatedStorage:FindFirstChild("AuraSystem2")
        if not auraSystem then Logger:Warn("AuraSystem2 not found."); return end
        local auras = {"Blue", "Red", "Yellow", "Green", "Pink", "Glitch", "Wind", "Dark Blue", "Fire", "Dark Green", "Dark Red", "Dark", "Venom", "Lightning", "Dark Pink"}
        for _, auraName in ipairs(auras) do auraSystem:FireServer(auraName); task.wait(0.1) end
        Logger:Log("Finished activating auras.")
    end)
end

local function executeBattleWin(battleEvent, battleId)
    if not battleEvent or not battleId or cooldown then return end
    cooldown = true
    Logger:Log("Executing Fast Battle Win with ID: " .. battleId)
    for i = 1, 4 do pcall(function() battleEvent:FireServer(battleId) end); task.wait(0.05) end
    task.delay(1, function() cooldown = false end)
end

local function interceptBattleEvents()
    local originalNamecall
    originalNamecall = hookmetamethod(game, "__namecall", function(self, ...)
        if fastBattleEnabled and getnamecallmethod() == "FireServer" and tostring(self) == "Battle" then
            local args = {...}
            if args[1] and type(args[1]) == "string" and #args[1] > 10 then executeBattleWin(self, args[1]) end
        end
        return originalNamecall(self, ...)
    end)
end

local function solveQTEByTiming()
    pcall(function()
        local qteFrame = player.PlayerGui:FindFirstChild("Wrestling") and player.PlayerGui.Wrestling:FindFirstChild("Battle GUI!")
        if not qteFrame then return end
        local movingFrame = qteFrame:FindFirstChild("ArrowMovement")
        local clickButton = movingFrame and movingFrame:FindFirstChild("TextButton")
        local perfectZone = qteFrame:FindFirstChild("BarFrame") and qteFrame.BarFrame:FindFirstChild("Perfect")
        if not (movingFrame and clickButton and perfectZone) then return end
        if movingFrame.AbsolutePosition.X >= perfectZone.AbsolutePosition.X and movingFrame.AbsolutePosition.X <= (perfectZone.AbsolutePosition.X + perfectZone.AbsoluteSize.X) then
            if canFireClick then fireclickdetector(clickButton) else clickButton.MouseButton1Click:Fire() end
        end
    end)
end

local function toggleFastBattle()
    fastBattleEnabled = not fastBattleEnabled
    _G.toggleSwitch(_G.fastBattleButton, _G.fastBattleKnob, _G.fastBattleLabel, fastBattleEnabled)
    if fastBattleEnabled then
        Logger:Log("Fast Battle enabled!")
        pcall(interceptBattleEvents)
        if fastBattleMonitorLoop then fastBattleMonitorLoop:Disconnect() end
        fastBattleMonitorLoop = RunService.Heartbeat:Connect(solveQTEByTiming)
    else
        Logger:Log("Fast Battle disabled!")
        if fastBattleMonitorLoop then fastBattleMonitorLoop:Disconnect(); fastBattleMonitorLoop = nil end
    end
end

--// GUI Section
pcall(function()
    if player.PlayerGui:FindFirstChild("LokasornScriptGui") then
        player.PlayerGui.LokasornScriptGui:Destroy()
    end

    local screenGui = Instance.new("ScreenGui", player.PlayerGui)
    screenGui.Name = "LokasornScriptGui"
    screenGui.ResetOnSpawn = false
    _G.mainFrame = Instance.new("Frame", screenGui)
    _G.mainFrame.Name = "MainFrame"; _G.mainFrame.Size = UDim2.new(0, 400, 0, 400); _G.mainFrame.BackgroundColor3 = Color3.fromRGB(20, 20, 25); _G.mainFrame.BorderSizePixel = 0; _G.mainFrame.Position = UDim2.new(0.5, -200, 0.5, -200); _G.mainFrame.AnchorPoint = Vector2.new(0.5, 0.5); _G.mainFrame.ClipsDescendants = true
    Instance.new("UICorner", _G.mainFrame).CornerRadius = UDim.new(0, 16)
    Instance.new("UIStroke", _G.mainFrame).Color = Color3.fromRGB(40, 40, 45)
    local titleBar = Instance.new("Frame", _G.mainFrame)
    titleBar.Name = "TitleBar"; titleBar.Size = UDim2.new(1, 0, 0, 50); titleBar.BackgroundColor3 = Color3.fromRGB(30, 30, 35); titleBar.BorderSizePixel = 0
    Instance.new("UICorner", titleBar).CornerRadius = UDim.new(0, 16)
    local titleLabel = Instance.new("TextLabel", titleBar)
    titleLabel.Name = "TitleLabel"; titleLabel.Size = UDim2.new(0.7, 0, 1, 0); titleLabel.Position = UDim2.new(0.05, 0, 0, 0); titleLabel.BackgroundTransparency = 1; titleLabel.Text = "Wrestling Legends"; titleLabel.TextColor3 = Color3.fromRGB(200, 200, 255); titleLabel.TextSize = 24; titleLabel.Font = Enum.Font.GothamBold; titleLabel.TextXAlignment = Enum.TextXAlignment.Left
    local subtitleLabel = Instance.new("TextLabel", titleBar)
    subtitleLabel.Name = "SubtitleLabel"; subtitleLabel.Size = UDim2.new(0.7, 0, 0, 20); subtitleLabel.Position = UDim2.new(0.05, 0, 0, 30); subtitleLabel.BackgroundTransparency = 1; subtitleLabel.Text = "by Lokasorn"; subtitleLabel.TextColor3 = Color3.fromRGB(150, 150, 200); subtitleLabel.TextSize = 14; subtitleLabel.Font = Enum.Font.Gotham; subtitleLabel.TextXAlignment = Enum.TextXAlignment.Left
    local closeButton = Instance.new("TextButton", titleBar)
    closeButton.Name = "CloseButton"; closeButton.Size = UDim2.new(0, 36, 0, 36); closeButton.Position = UDim2.new(1, -46, 0, 7); closeButton.BackgroundColor3 = Color3.fromRGB(40, 40, 45); closeButton.Text = "X"; closeButton.TextColor3 = Color3.fromRGB(255, 150, 150); closeButton.TextSize = 18; closeButton.Font = Enum.Font.GothamBold
    Instance.new("UICorner", closeButton).CornerRadius = UDim.new(0, 12)
    local minimizeButton = Instance.new("TextButton", titleBar)
    minimizeButton.Name = "MinimizeButton"; minimizeButton.Size = UDim2.new(0, 36, 0, 36); minimizeButton.Position = UDim2.new(1, -92, 0, 7); minimizeButton.BackgroundColor3 = Color3.fromRGB(40, 40, 45); minimizeButton.Text = "−"; minimizeButton.TextColor3 = Color3.fromRGB(255, 200, 100); minimizeButton.TextSize = 18; minimizeButton.Font = Enum.Font.GothamBold
    Instance.new("UICorner", minimizeButton).CornerRadius = UDim.new(0, 12)
    local contentFrame = Instance.new("Frame", _G.mainFrame)
    contentFrame.Name = "ContentFrame"; contentFrame.Size = UDim2.new(1, -20, 1, -70); contentFrame.Position = UDim2.new(0, 10, 0, 60); contentFrame.BackgroundTransparency = 1

    local function createSwitch(parent, text, yPos)
        local frame = Instance.new("Frame", parent); frame.Size = UDim2.new(1, -20, 0, 50); frame.Position = UDim2.new(0, 10, 0, yPos); frame.BackgroundTransparency = 1
        local button = Instance.new("TextButton", frame); button.Size = UDim2.new(0, 60, 0, 30); button.Position = UDim2.new(0, 0, 0.5, -15); button.BackgroundColor3 = Color3.fromRGB(30, 30, 35); button.Text = ""
        Instance.new("UICorner", button).CornerRadius = UDim.new(0, 15)
        local knob = Instance.new("Frame", button); knob.Size = UDim2.new(0, 26, 0, 26); knob.Position = UDim2.new(0, 2, 0, 2); knob.BackgroundColor3 = Color3.fromRGB(100, 100, 120)
        Instance.new("UICorner", knob).CornerRadius = UDim.new(0, 13)
        local label = Instance.new("TextLabel", frame); label.Size = UDim2.new(0, 200, 0, 30); label.Position = UDim2.new(0, 70, 0.5, -15); label.BackgroundTransparency = 1; label.Text = text; label.TextColor3 = Color3.fromRGB(200, 200, 255); label.TextSize = 18; label.Font = Enum.Font.Gotham; label.TextXAlignment = Enum.TextXAlignment.Left
        return button, knob, label
    end

    _G.fastTrainButton, _G.fastTrainKnob, _G.fastTrainLabel = createSwitch(contentFrame, "Fast Train: OFF", 10)
    _G.autoTrainButton, _G.autoTrainKnob, _G.autoTrainLabel = createSwitch(contentFrame, "Auto Train: OFF", 70)
    _G.antiAfkButton, _G.antiAfkKnob, _G.antiAfkLabel = createSwitch(contentFrame, "Anti AFK: OFF", 130)
    _G.fastBattleButton, _G.fastBattleKnob, _G.fastBattleLabel = createSwitch(contentFrame, "Fast Battle: OFF", 190)

    local allAurasButton = Instance.new("TextButton", contentFrame); allAurasButton.Size = UDim2.new(1, -20, 0, 50); allAurasButton.Position = UDim2.new(0, 10, 0, 250); allAurasButton.BackgroundColor3 = Color3.fromRGB(40, 40, 45); allAurasButton.Text = "Activate All Auras"; allAurasButton.TextColor3 = Color3.fromRGB(255, 200, 100); allAurasButton.TextSize = 18; allAurasButton.Font = Enum.Font.GothamBold
    Instance.new("UICorner", allAurasButton).CornerRadius = UDim.new(0, 12)
    local hintLabel = Instance.new("TextLabel", contentFrame); hintLabel.Size = UDim2.new(1, -20, 0, 20); hintLabel.Position = UDim2.new(0, 10, 1, -30); hintLabel.BackgroundTransparency = 1; hintLabel.Text = "Press RIGHT CTRL to toggle menu"; hintLabel.TextColor3 = Color3.fromRGB(120, 120, 150); hintLabel.TextSize = 12; hintLabel.Font = Enum.Font.Gotham; hintLabel.TextXAlignment = Enum.TextXAlignment.Center
    
    local function createTween(i, p, d) return TweenService:Create(i, TweenInfo.new(d, Enum.EasingStyle.Sine, Enum.EasingDirection.Out), p) end
    local function applyHoverEffect(b, h, d)
        b.MouseEnter:Connect(function() createTween(b, {BackgroundColor3 = h}, 0.2):Play() end)
        b.MouseLeave:Connect(function() createTween(b, {BackgroundColor3 = d}, 0.2):Play() end)
    end
    applyHoverEffect(closeButton, Color3.fromRGB(60, 30, 30), Color3.fromRGB(40, 40, 45))
    applyHoverEffect(minimizeButton, Color3.fromRGB(60, 50, 20), Color3.fromRGB(40, 40, 45))
    applyHoverEffect(allAurasButton, Color3.fromRGB(60, 50, 20), Color3.fromRGB(40, 40, 45))
    _G.toggleSwitch = function(b, k, l, s)
        local p = s and UDim2.new(0, 32, 0, 2) or UDim2.new(0, 2, 0, 2)
        local c = s and Color3.fromRGB(50, 100, 50) or Color3.fromRGB(30, 30, 35)
        createTween(k, {Position = p}, 0.3):Play()
        createTween(b, {BackgroundColor3 = c}, 0.3):Play()
        l.Text = s and l.Text:gsub("OFF", "ON") or l.Text:gsub("ON", "OFF")
    end

    local function toggleFastTrain()
        fastTrainEnabled = not fastTrainEnabled
        _G.toggleSwitch(_G.fastTrainButton, _G.fastTrainKnob, _G.fastTrainLabel, fastTrainEnabled)
        if fastTrainEnabled then Logger:Log("Fast Train enabled."); fastTrainCoroutine = coroutine.create(fastTrainLogic); coroutine.resume(fastTrainCoroutine) else if fastTrainCoroutine then fastTrainCoroutine = nil end end
    end

    local function toggleAutoTrain()
        autoTrainEnabled = not autoTrainEnabled
        _G.toggleSwitch(_G.autoTrainButton, _G.autoTrainKnob, _G.autoTrainLabel, autoTrainEnabled)
        if autoTrainEnabled then Logger:Log("Auto Train enabled."); trainCoroutine = coroutine.create(autoTrainLogic); coroutine.resume(trainCoroutine) else if trainCoroutine then trainCoroutine = nil end end
    end

    local function toggleAntiAfk()
        antiAfkEnabled = not antiAfkEnabled
        _G.toggleSwitch(_G.antiAfkButton, _G.antiAfkKnob, _G.antiAfkLabel, antiAfkEnabled)
        if antiAfkEnabled then Logger:Log("Anti-AFK enabled."); antiAfkCoroutine = coroutine.create(antiAfkLogic); coroutine.resume(antiAfkCoroutine) else if antiAfkCoroutine then antiAfkCoroutine = nil end end
    end
    
    local isMinimized = false
    local function minimizeGui()
        isMinimized = not isMinimized
        _G.mainFrame.Visible = not isMinimized
    end
    
    closeButton.MouseButton1Click:Connect(function() screenGui:Destroy() end)
    minimizeButton.MouseButton1Click:Connect(minimizeGui)
    _G.fastTrainButton.MouseButton1Click:Connect(toggleFastTrain)
    _G.autoTrainButton.MouseButton1Click:Connect(toggleAutoTrain)
    _G.antiAfkButton.MouseButton1Click:Connect(toggleAntiAfk)
    _G.fastBattleButton.MouseButton1Click:Connect(toggleFastBattle)
    allAurasButton.MouseButton1Click:Connect(activateAllAuras)
    
    local isDragging = false
    local dragStartPos, dragStartFramePos
    titleBar.InputBegan:Connect(function(input)
        if input.UserInputType == Enum.UserInputType.MouseButton1 then
            isDragging = true
            dragStartPos = input.Position
            dragStartFramePos = _G.mainFrame.Position
        end
    end)
    UserInputService.InputChanged:Connect(function(input)
        if isDragging and input.UserInputType == Enum.UserInputType.MouseMovement then
            local delta = input.Position - dragStartPos
            _G.mainFrame.Position = UDim2.new(dragStartFramePos.X.Scale, dragStartFramePos.X.Offset + delta.X, dragStartFramePos.Y.Scale, dragStartFramePos.Y.Offset + delta.Y)
        end
    end)
    UserInputService.InputEnded:Connect(function(input)
        if input.UserInputType == Enum.UserInputType.MouseButton1 then isDragging = false end
    end)

    UserInputService.InputBegan:Connect(function(input, gameProcessedEvent)
        if gameProcessedEvent then return end
        if input.KeyCode == Enum.KeyCode.RightControl then minimizeGui() end
    end)

    _G.mainFrame.Visible = false
    _G.mainFrame.Size = UDim2.new(0, 0, 0, 0)
    _G.mainFrame.Position = UDim2.new(0.5, 0, 0.5, 0)
    task.wait(0.1)
    _G.mainFrame.Visible = true
    TweenService:Create(_G.mainFrame, TweenInfo.new(0.5, Enum.EasingStyle.Back), {Size = UDim2.new(0, 400, 0, 400), Position = UDim2.new(0.5, -200, 0.5, -200)}):Play()

    Logger:Log("GUI created successfully.")
end)