sharp-chat/SharpChat/Commands/ShutdownRestartCommand.cs

using SharpChat.Packet;
using System;
using System.Threading;

namespace SharpChat.Commands {
    public class ShutdownRestartCommand : IChatCommand {
        private readonly ManualResetEvent WaitHandle;
        private readonly Func<bool> ShutdownCheck;

        public ShutdownRestartCommand(ManualResetEvent waitHandle, Func<bool> shutdownCheck) {
            WaitHandle = waitHandle;
            ShutdownCheck = shutdownCheck;
        }

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

        public void Dispatch(ChatCommandContext ctx) {
            if(ctx.User.UserId != 1) {
                ctx.Chat.SendTo(ctx.User, new CommandNotAllowedErrorPacket(ctx.Name));
                return;
            }

            if(!ShutdownCheck())
                return;

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

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