sharp-chat/SharpChat/ClientCommands/ShutdownRestartClientCommand.cs

31 lines
1.1 KiB
C#

using SharpChat.SockChat.S2CPackets;
using ZLogger;
namespace SharpChat.ClientCommands;
public class ShutdownRestartClientCommand(CancellationTokenSource cancellationTokenSource) : ClientCommand {
public bool IsMatch(ClientCommandContext ctx) {
return ctx.NameEquals("shutdown")
|| ctx.NameEquals("restart");
}
public async Task Dispatch(ClientCommandContext ctx) {
if(!ctx.User.UserId.Equals("1")) {
long msgId = ctx.Chat.RandomSnowflake.Next();
await ctx.Chat.SendTo(ctx.User, new CommandResponseS2CPacket(msgId, LCR.COMMAND_NOT_ALLOWED, true, $"/{ctx.Name}"));
return;
}
if(cancellationTokenSource.IsCancellationRequested)
return;
ctx.Connection.Logger.ZLogInformation($"Shutdown requested through Sock Chat command...");
if(ctx.NameEquals("restart"))
foreach(Connection conn in ctx.Chat.Connections)
conn.PrepareForRestart();
await ctx.Chat.Update();
await cancellationTokenSource.CancelAsync();
}
}