include("autorun/server/mysql_config.lua")
util.AddNetworkString("AdminRequest")
util.AddNetworkString("AdminRequestAction")
util.AddNetworkString("AdminRequestClaim")
util.AddNetworkString("AdminRequestClaimData")
util.AddNetworkString("RequestUpdatedAdminRequests")
util.AddNetworkString("UpdatedAdminRequests")
util.AddNetworkString("AdminRequestClose")
util.AddNetworkString("RequestClaimData")
adminClaimData = {}
local adminRequests = {}
local function GetAdmins()
    local adminGroups = {
        "trialmod", "moderator", "seniormod", "admin", "headadmin", "senioradmin", "council", "superadmin", "manager"
    }
    local admins = {}
    for _, ply in ipairs(player.GetAll()) do
        for _, group in ipairs(adminGroups) do
            if ply:IsUserGroup(group) then
                table.insert(admins, ply)
                break
            end
        end
    end
    return admins
end
local function IsAdminGroup(ply)
    local adminGroups = {
        "trialmod", "moderator", "seniormod", "admin", "headadmin", "senioradmin", "council", "superadmin", "manager"
    }
    for _, group in ipairs(adminGroups) do
        if ply:IsUserGroup(group) then
            return true
        end
    end

    return false
end
hook.Add("PlayerSay", "AdminRequestSystem", function(ply, text, teamChat)
    if string.sub(text, 1, 1) == "@" then
        local requestData = {
            steamID64 = ply:SteamID64(),
            name = ply:Nick(),
            message = string.sub(text, 2),
            requestTime = CurTime()
        }
        table.insert(adminRequests, requestData) -- Save the request data to the adminRequests table
        net.Start("AdminRequest")
        net.WriteTable(requestData)
        net.Send(GetAdmins()) -- GetAdmins will now check for the specified user groups
        return "" -- Hide the message from the chat
    end
end)
local function getUpdatedAdminRequests()
    -- Return the contents of the adminRequests table
    return adminRequests
end
-- Handle admin actions
local playerPreviousPositions = {}
net.Receive("AdminRequestAction", function(len, ply)
    if not IsAdminGroup(ply) then return end -- Check if the player is in one of the admin groups
    local action = net.ReadString()
    local target = net.ReadEntity()
    if action == "goto" then
        playerPreviousPositions[ply:SteamID64()] = ply:GetPos()
        ply:SetPos(target:GetPos() + Vector(0, 0, 50))
    elseif action == "bring" then
        playerPreviousPositions[target:SteamID64()] = target:GetPos()
        target:SetPos(ply:GetPos() + ply:GetForward() * 100 + Vector(0, 0, 50))
    elseif action == "return" then
        local prevPos = playerPreviousPositions[target:SteamID64()]
        if prevPos then
            target:SetPos(prevPos)
            playerPreviousPositions[target:SteamID64()] = nil
        end
    end
end)
-- Initialize admin claim data table
function UpdateAdminClaim(ply)
    local adminID = ply:SteamID64()
    local currentWeek = os.date("%U")
    local query = DBConnection:query("SELECT * FROM claim_data WHERE admin_id = '" .. adminID .. "'")
    query:start()
    query.onSuccess = function(_, data)
        if #data > 0 then
            local weeklyCount = data[1].weekly_count
            local totalCount = data[1].total_count
            local week = data[1].week
            if week ~= currentWeek then
                weeklyCount = 0
                week = currentWeek
            end
            weeklyCount = weeklyCount + 1
            totalCount = totalCount + 1
            local updateQuery = DBConnection:query("UPDATE claim_data SET weekly_count = " .. weeklyCount .. ", total_count = " .. totalCount .. ", week = " .. week .. " WHERE admin_id = '" .. adminID .. "'")
            updateQuery:start()
            updateQuery.onSuccess = function()
                print("Updated admin claim data successfully.")
            end
            updateQuery.onError = function(_, err)
                print("Failed to update admin claim data:", err)
            end
        else
            local insertQuery = DBConnection:query("INSERT INTO claim_data (admin_id, name, weekly_count, total_count, week) VALUES ('" .. adminID .. "', '" .. ply:Nick() .. "', 1, 1, " .. currentWeek .. ")")
            insertQuery:start()
            insertQuery.onSuccess = function()
                print("Inserted admin claim data successfully.")
            end
            insertQuery.onError = function(_, err)
                print("Failed to insert admin claim data:", err)
            end
        end
    end
    query.onError = function(_, err)
        print("Failed to retrieve admin claim data:", err)
    end
end
local adminClaimData = {}
net.Receive("AdminRequestClaim", function(len, ply)
    if not IsAdminGroup(ply) then return end -- Check if the player is in one of the admin groups

    local requestData = net.ReadTable() -- Add this line
    local adminID = ply:SteamID64()
    local currentDate = os.date("*t")
    local currentWeek = os.date("%U")

    if not adminClaimData[adminID] then
        adminClaimData[adminID] = {
            name = ply:Nick(),
            totalCount = 0,
            weeklyCount = 0,
            week = currentWeek
        }
    end
    local data = adminClaimData[adminID]
    if data.week ~= currentWeek then
        data.weeklyCount = 0
        data.week = currentWeek
    end
    data.totalCount = data.totalCount + 1
    data.weeklyCount = data.weeklyCount + 1
    local requestData = net.ReadTable() -- Add this line
    UpdateAdminClaim(ply)
end)

-- Handle claimed cases
local function markCaseAsClaimed(caseIndex, adminID)
    adminRequests[caseIndex].claimedBy = adminID
end
net.Receive("RequestUpdatedAdminRequests", function(len, ply)
    -- Get updated admin requests here
    local updatedRequests = getUpdatedAdminRequests()
    net.Start("UpdatedAdminRequests")
    net.WriteTable(updatedRequests)
    net.Send(ply)
end)
net.Receive("AdminRequestClose", function(len, ply)
    if not IsAdminGroup(ply) then return end
    local requestData = net.ReadTable()
    local index = -1
    for i, request in ipairs(adminRequests) do
        if request.steamID64 == requestData.steamID64 and request.requestTime == requestData.requestTime then
            index = i
            break
        end
    end
    if index > 0 then
        table.remove(adminRequests, index)
    end
end)
net.Receive("RequestClaimData", function(_, ply)
    local fetchDataQuery = "SELECT * FROM claim_data;"
    local query = DBConnection:query(fetchDataQuery)

    query:start()

    query.onSuccess = function(_, data)
        net.Start("ReceiveClaimData")
        net.WriteTable(data)
        net.Send(ply)
    end

    query.onError = function(_, err)
        print("Failed to fetch data from claim_data table:", err)
    end
end)
concommand.Add("show_claim_data", function(ply, cmd, args)
    if not IsAdminGroup(ply) then return end -- Check if the player is in one of the admin groups
    -- Send the claim data to the player who requested it
    net.Start("AdminRequestClaimData")
    net.WriteTable(adminClaimData)
    net.Send(ply)
end)