mirror of
https://github.com/NoiTheCat/BirthdayBot.git
synced 2024-11-21 13:54:36 +00:00
Noi
c727bead98
It now knows what a SIGTERM is. Additionally, made some small tweaks to the file's structure and removed some unnecessary details.
53 lines
1.8 KiB
C#
53 lines
1.8 KiB
C#
namespace BirthdayBot;
|
|
class Program {
|
|
private static ShardManager _bot = null!;
|
|
private static readonly DateTimeOffset _botStartTime = DateTimeOffset.UtcNow;
|
|
|
|
/// <summary>
|
|
/// Returns the amount of time the program has been running in a human-readable format.
|
|
/// </summary>
|
|
public static string BotUptime => (DateTimeOffset.UtcNow - _botStartTime).ToString("d' days, 'hh':'mm':'ss");
|
|
|
|
static async Task Main() {
|
|
Configuration? cfg = null;
|
|
try {
|
|
cfg = new Configuration();
|
|
} catch (Exception ex) {
|
|
Console.WriteLine(ex);
|
|
Environment.Exit(2);
|
|
}
|
|
|
|
_bot = new ShardManager(cfg);
|
|
AppDomain.CurrentDomain.ProcessExit += OnCancelEvent;
|
|
Console.CancelKeyPress += OnCancelEvent;
|
|
|
|
await Task.Delay(-1);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sends a formatted message to console.
|
|
/// </summary>
|
|
public static void Log(string source, string message) {
|
|
var ts = DateTime.Now;
|
|
var ls = new string[] { "\r\n", "\n" };
|
|
foreach (var item in message.Split(ls, StringSplitOptions.None))
|
|
Console.WriteLine($"{ts:s} [{source}] {item}");
|
|
}
|
|
|
|
private static bool _shutdownRequested = false;
|
|
private static void OnCancelEvent(object? sender, EventArgs e) {
|
|
if (e is ConsoleCancelEventArgs ce) ce.Cancel = true;
|
|
|
|
if (_shutdownRequested) return;
|
|
_shutdownRequested = true;
|
|
Log(nameof(Program), "Shutting down...");
|
|
|
|
var dispose = Task.Run(_bot.Dispose);
|
|
if (!dispose.Wait(15000)) {
|
|
Log(nameof(Program), "Disconnection is taking too long. Will force exit.");
|
|
Environment.ExitCode = 1;
|
|
}
|
|
Log(nameof(Program), $"Uptime: {BotUptime}");
|
|
Environment.Exit(Environment.ExitCode);
|
|
}
|
|
}
|