sharp-chat/SharpChat/Commands/ShutdownRestartCommand.cs

using SharpChat.Packet;

namespace SharpChat.Commands {
    public class ShutdownRestartCommand(ManualResetEvent waitHandle, Func<bool> shutdownCheck) : IChatCommand {
        private readonly ManualResetEvent WaitHandle = waitHandle ?? throw new ArgumentNullException(nameof(waitHandle));
        private readonly Func<bool> ShutdownCheck = shutdownCheck ?? throw new ArgumentNullException(nameof(shutdownCheck));

        public bool IsMatch(ChatCommandContext ctx) {
            return ctx.NameEquals("shutdown")
                || ctx.NameEquals("restart");
        }

        public void Dispatch(ChatCommandContext ctx) {
            if(ctx.User.UserId != 1) {
                long msgId = ctx.Chat.RandomSnowflake.Next();
                ctx.Chat.SendTo(ctx.User, new LegacyCommandResponse(msgId, LCR.COMMAND_NOT_ALLOWED, true, $"/{ctx.Name}"));
                return;
            }

            if(!ShutdownCheck())
                return;

            if(ctx.NameEquals("restart"))
                foreach(ChatConnection conn in ctx.Chat.Connections)
                    conn.PrepareForRestart();

            ctx.Chat.Update();
            WaitHandle?.Set();
        }
    }
}