AAEmu.Game/Models/Tasks/ShutdownTask.cs - develop - NLObP/AAEmu

AAEmu.Game/Models/Tasks/ShutdownTask.cs
136 строк4 KB
Roger Barreto
Adding Dictionary, Cleaning all warnings (#1109)
10 дек 2024, 19:51
10 дек 2024, 19:517605fdd
using System;
using System.Collections.Generic;
using System.Drawing;
using AAEmu.Game.Core.Managers;
using AAEmu.Game.Core.Managers.World;
using AAEmu.Game.Core.Packets.G2C;

namespace AAEmu.Game.Models.Tasks;

public class ShutdownTask : Task
{
    // private DateTime _lastCheckTime = DateTime.MinValue;
    private bool _alreadyShuttingDown;
    private DateTime _shutdownTime;
    private DateTime _nextTriggerTime = DateTime.MinValue;
    private readonly int _exitCode;
    // Timings to show a popup, in seconds
    private readonly List<uint> _triggerPoints =
    [
        14400, // 4h
        10800, // 3h
        7200, // 2h
        3600, // 60m
        2700, // 45m
        1800, // 30m
        1200, // 20m
        900, // 15m
        600, // 10m
        300, // 5m
        240, // 4m
        180, // 3m
        120, // 2m
        90,
        60, // 1m
        45,
        30,
        20,
        10
    ];

    public ShutdownTask(DateTime shutdownTime, int exitCode)
    {
        _shutdownTime = shutdownTime;
        _exitCode = exitCode;
        _nextTriggerTime = CalculateLargestNextTrigger();
    }

    private DateTime CalculateLargestNextTrigger()
    {
        var remaining = _shutdownTime - DateTime.UtcNow;
        var remainingDelta = Math.Ceiling(remaining.TotalSeconds);
        foreach (var triggerPoint in _triggerPoints)
        {
            if (remainingDelta > triggerPoint)
                return _shutdownTime - TimeSpan.FromSeconds(triggerPoint);
        }

        return DateTime.MaxValue;
    }

    public void ChangeShutdownTime(DateTime shutdownTime)
    {
        _shutdownTime = shutdownTime;
        var remaining = (int)Math.Ceiling((_shutdownTime - DateTime.UtcNow).TotalMinutes);
        WorldManager.Instance.BroadcastPacketToServer(new SCNoticeMessagePacket(3, Color.Red, 10000, $"Server shutdown has been rescheduled to {remaining} minutes from now!"));
        _nextTriggerTime = CalculateLargestNextTrigger();
    }

    public override void Execute()
    {
        if (_alreadyShuttingDown)
            return;

        if (_shutdownTime <= DateTime.UtcNow)
        {
            _alreadyShuttingDown = true;
            // Do server shut down now

            // Wipe all BuyBack containers manually, as there will be no trigger for disconnected players
            try
            {
                foreach (var character in WorldManager.Instance.GetAllCharacters())
                    character.BuyBackItems.Wipe();
            }
            catch (Exception)
            {
                // Ignore
            }

            if (SaveManager.Instance.DoSave())
            {
                WorldManager.Instance.BroadcastPacketToServer(new SCNoticeMessagePacket(3, Color.Magenta, 15000,
                    "The server is shutting down right now!"));
                Environment.Exit(_exitCode); // Manual Shutdown
            }

            return;
        }

        var remainingTime = _shutdownTime - DateTime.UtcNow;
        var remainingSeconds = (int)Math.Ceiling(remainingTime.TotalSeconds);
        if (DateTime.UtcNow < _nextTriggerTime)
            return;

        _nextTriggerTime = CalculateLargestNextTrigger();

        var showSeconds = remainingSeconds <= 100;
        var showMinutes = remainingSeconds <= 3600 && !showSeconds;
        var showHours = remainingSeconds > 3600 && !showSeconds; ;

        var shutdownText = "The server is shutting down soon!";
        var popupTime = 3000;

        if (showHours)
        {
            shutdownText = $"The server is shutting down in {remainingSeconds / 3600} hours!";
            popupTime = 10000;
        }
        else if (showMinutes)
        {
            shutdownText = $"The server is shutting down in {remainingSeconds / 60} minutes!";
            popupTime = 6000;
        }
        else if (showSeconds)
        {
            shutdownText = $"The server is shutting down in {remainingSeconds} seconds!";
            popupTime = 3000;
        }

        WorldManager.Instance.BroadcastPacketToServer(
            new SCNoticeMessagePacket(3,
                Color.Red,
                popupTime,
                shutdownText));
    }
}