39 lines
1.3 KiB
C#
39 lines
1.3 KiB
C#
|
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 ?? throw new ArgumentNullException(nameof(waitHandle));
|
|||
|
|
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) {
|
|||
|
|
ctx.User.Send(new LegacyCommandResponse(LCR.COMMAND_NOT_ALLOWED, true, $"/{ctx.Name}"));
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if(!ShutdownCheck())
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
if(ctx.NameEquals("restart"))
|
|||
|
|
lock(ctx.Chat.SessionsAccess)
|
|||
|
|
foreach(ChatUserSession sess in ctx.Chat.Sessions)
|
|||
|
|
sess.PrepareForRestart();
|
|||
|
|
|
|||
|
|
ctx.Chat.Update();
|
|||
|
|
WaitHandle?.Set();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|