Files
tgstation-server/TGS.CommandLine/ChatCommands.cs
T

722 lines
18 KiB
C#

using System;
using System.Collections.Generic;
using TGS.Interface;
using TGS.Interface.Components;
namespace TGS.CommandLine
{
class IRCCommand : InstanceRootCommand
{
public IRCCommand()
{
Keyword = "irc";
Children = new Command[] { new IRCNickCommand(), new IRCAuthCommand(), new IRCDisableAuthCommand(), new IRCServerCommand(), new ChatJoinCommand(ChatProvider.IRC), new ChatPartCommand(ChatProvider.IRC), new ChatListAdminsCommand(ChatProvider.IRC), new ChatReconnectCommand(ChatProvider.IRC), new ChatAddminCommand(ChatProvider.IRC), new ChatDeadminCommand(ChatProvider.IRC), new ChatEnableCommand(ChatProvider.IRC), new ChatDisableCommand(ChatProvider.IRC), new ChatStatusCommand(ChatProvider.IRC), new IRCAuthModeCommand(), new IRCAuthLevelCommand() };
}
public override string GetHelpText()
{
return "Manages the IRC bot";
}
}
class DiscordCommand : InstanceRootCommand
{
public DiscordCommand()
{
Keyword = "discord";
Children = new Command[] { new DiscordSetTokenCommand(), new ChatJoinCommand(ChatProvider.Discord), new ChatPartCommand(ChatProvider.Discord), new ChatListAdminsCommand(ChatProvider.Discord), new ChatReconnectCommand(ChatProvider.Discord), new ChatAddminCommand(ChatProvider.Discord), new ChatDeadminCommand(ChatProvider.Discord), new ChatEnableCommand(ChatProvider.Discord), new ChatDisableCommand(ChatProvider.Discord), new ChatStatusCommand(ChatProvider.Discord) , new DiscordAuthModeCommand() };
}
public override string GetHelpText()
{
return "Manages the Discord bot";
}
}
class IRCNickCommand : ConsoleCommand
{
public IRCNickCommand()
{
Keyword = "nick";
RequiredParameters = 1;
}
public override string GetArgumentString()
{
return "<name>";
}
public override string GetHelpText()
{
return "Sets the IRC nickname";
}
protected override ExitCode Run(IList<string> parameters)
{
var Chat = Interface.GetComponent<ITGChat>();
Chat.SetProviderInfo(new IRCSetupInfo(Chat.ProviderInfos()[(int)ChatProvider.IRC])
{
Nickname = parameters[0],
});
return ExitCode.Normal;
}
}
class ChatJoinCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatJoinCommand(ChatProvider pI)
{
Keyword = "join";
RequiredParameters = 2;
providerIndex = (int)pI;
}
public override string GetArgumentString()
{
return "<channel> <dev|wd|game|admin>";
}
public override string GetHelpText()
{
return "Joins a channel for listening and broadcasting of the specified message type (Developer, Watchdog, Game, Admin)";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = IRC.ProviderInfos()[providerIndex];
List<string> channels;
switch (parameters[1].ToLower())
{
case "dev":
channels = info.DevChannels;
break;
case "wd":
channels = info.WatchdogChannels;
break;
case "game":
channels = info.GameChannels;
break;
case "admin":
channels = info.AdminChannels;
break;
default:
OutputProc("Invalid parameter: " + parameters[1]);
return ExitCode.BadCommand;
}
var lowerParam = parameters[0].ToLower();
foreach (var I in channels)
{
if (I.ToLower() == lowerParam)
{
OutputProc("Already in this channel!");
return ExitCode.BadCommand;
}
}
channels.Add(parameters[0]);
switch (parameters[1].ToLower())
{
case "dev":
info.DevChannels = channels;
break;
case "wd":
info.WatchdogChannels = channels;
break;
case "game":
info.GameChannels = channels;
break;
case "admin":
info.AdminChannels = channels;
break;
}
var res = IRC.SetProviderInfo(info);
if(res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class ChatPartCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatPartCommand(ChatProvider pI)
{
Keyword = "part";
RequiredParameters = 2;
providerIndex = (int)pI;
}
public override string GetArgumentString()
{
return "<channel> <dev|wd|game|admin>";
}
public override string GetHelpText()
{
return "Stops listening and broadcasting on a channel for the specified message type (Developer, Watchdog, Game, Admin)";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = IRC.ProviderInfos()[providerIndex];
List<string> channels;
switch (parameters[1].ToLower())
{
case "dev":
channels = info.DevChannels;
break;
case "wd":
channels = info.WatchdogChannels;
break;
case "game":
channels = info.GameChannels;
break;
case "admin":
channels = info.AdminChannels;
break;
default:
OutputProc("Invalid parameter: " + parameters[1]);
return ExitCode.BadCommand;
}
var lowerParam = parameters[0].ToLower();
if ((ChatProvider)providerIndex == ChatProvider.IRC && lowerParam[0] != '#')
lowerParam = "#" + lowerParam;
channels.Remove(lowerParam);
switch (parameters[1].ToLower())
{
case "dev":
info.DevChannels = channels;
break;
case "wd":
info.WatchdogChannels = channels;
break;
case "game":
info.GameChannels = channels;
break;
case "admin":
info.AdminChannels = channels;
break;
}
var res = IRC.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class ChatListAdminsCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatListAdminsCommand(ChatProvider pI)
{
Keyword = "list-admins";
providerIndex = (int)pI;
}
public override string GetHelpText()
{
return "List users which can use restricted commands in the admin channel";
}
protected override ExitCode Run(IList<string> parameters)
{
var info = Interface.GetComponent<ITGChat>().ProviderInfos()[providerIndex];
string authType;
switch ((ChatProvider)providerIndex)
{
case ChatProvider.IRC:
if (info.AdminsAreSpecial)
authType = "Mode:";
else
authType = "Nicknames:";
break;
case ChatProvider.Discord:
if (info.AdminsAreSpecial)
authType = "Role IDs:";
else
authType = "User IDs:";
break;
default:
OutputProc(String.Format("Invalid provider: {0}!", providerIndex));
return ExitCode.ServerError;
}
OutputProc("Authorized " + authType);
if (info.AdminsAreSpecial && (ChatProvider)providerIndex == ChatProvider.IRC)
switch(new IRCSetupInfo(info).AuthLevel)
{
case IRCMode.Voice:
OutputProc("+");
break;
case IRCMode.Halfop:
OutputProc("%");
break;
case IRCMode.Op:
OutputProc("@");
break;
case IRCMode.Owner:
OutputProc("~");
break;
}
else
foreach (var I in info.AdminList)
OutputProc(I);
return ExitCode.Normal;
}
}
class ChatReconnectCommand : ConsoleCommand
{
readonly ChatProvider providerIndex;
public ChatReconnectCommand(ChatProvider pI)
{
Keyword = "reconnect";
providerIndex = pI;
}
public override string GetHelpText()
{
return "Restablish the chat connection";
}
protected override ExitCode Run(IList<string> parameters)
{
var res = Interface.GetComponent<ITGChat>().Reconnect(providerIndex);
if (res != null)
{
OutputProc("Error: " + res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class ChatAddminCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatAddminCommand(ChatProvider pI)
{
Keyword = "addmin";
RequiredParameters = 1;
providerIndex = (int)pI;
}
public override string GetArgumentString()
{
return "[nick]";
}
public override string GetHelpText()
{
return "Add a user which can use restricted commands in the admin channels";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = IRC.ProviderInfos()[providerIndex];
var newmin = parameters[0].ToLower();
if (info.AdminsAreSpecial && (ChatProvider)providerIndex == ChatProvider.IRC)
{
OutputProc("Invalid auth mode for this command!");
return ExitCode.BadCommand;
}
if (info.AdminList.Contains(newmin))
{
OutputProc(parameters[0] + " is already an admin!");
return ExitCode.BadCommand;
}
var al = info.AdminList;
al.Add(newmin);
info.AdminList = al;
var res = IRC.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class IRCAuthModeCommand : ConsoleCommand
{
public IRCAuthModeCommand()
{
Keyword = "set-auth-mode";
RequiredParameters = 1;
}
public override string GetArgumentString()
{
return "<channel-mode|nickname>";
}
public override string GetHelpText()
{
return "Switch between admin command authorization via user channel mode or nicknames";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = IRC.ProviderInfos()[(int)ChatProvider.IRC];
var lowerparam = parameters[0].ToLower();
if (lowerparam == "channel-mode")
info.AdminsAreSpecial = true;
else if (lowerparam == "nickname")
info.AdminsAreSpecial = false;
else
{
OutputProc("Invalid parameter: " + parameters[0]);
return ExitCode.BadCommand;
}
var res = IRC.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class DiscordAuthModeCommand : ConsoleCommand
{
public DiscordAuthModeCommand()
{
Keyword = "set-auth-mode";
RequiredParameters = 1;
}
public override string GetArgumentString()
{
return "<role-id|user-id>";
}
public override string GetHelpText()
{
return "Switch between admin command authorization via user roles or individual users";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = IRC.ProviderInfos()[(int)ChatProvider.Discord];
var lowerparam = parameters[0].ToLower();
if (lowerparam == "role-id")
info.AdminsAreSpecial = true;
else if (lowerparam == "user-id")
info.AdminsAreSpecial = false;
else
{
OutputProc("Invalid parameter: " + parameters[0]);
return ExitCode.BadCommand;
}
var res = IRC.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class IRCAuthLevelCommand : ConsoleCommand
{
public IRCAuthLevelCommand()
{
Keyword = "set-auth-level";
RequiredParameters = 1;
}
public override string GetArgumentString()
{
return "<+|%|@|~>";
}
public override string GetHelpText()
{
return "Set the required channel mode for users to use admin commands";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = new IRCSetupInfo(IRC.ProviderInfos()[(int)ChatProvider.IRC]);
switch (parameters[0])
{
case "+":
info.AuthLevel = IRCMode.Voice;
break;
case "%":
info.AuthLevel = IRCMode.Halfop;
break;
case "@":
info.AuthLevel = IRCMode.Op;
break;
case "~":
info.AuthLevel = IRCMode.Owner;
break;
default:
OutputProc("Invalid parameter: " + parameters[0]);
return ExitCode.BadCommand;
}
var res = IRC.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class ChatDeadminCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatDeadminCommand(ChatProvider pI)
{
Keyword = "deadmin";
RequiredParameters = 1;
providerIndex = (int)pI;
}
public override string GetArgumentString()
{
return "<nick>";
}
public override string GetHelpText()
{
return "Remove a user which can use restricted commands in the admin channels";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = IRC.ProviderInfos()[providerIndex];
var newmin = parameters[0].ToLower();
if (info.AdminsAreSpecial && (ChatProvider)providerIndex == ChatProvider.IRC)
{
OutputProc("Invalid auth mode for this command!");
return ExitCode.BadCommand;
}
if (!info.AdminList.Contains(newmin))
{
OutputProc(parameters[0] + " is not an admin!");
return ExitCode.BadCommand;
}
var al = info.AdminList;
al.Remove(newmin);
info.AdminList = al;
var res = IRC.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class IRCAuthCommand : ConsoleCommand
{
public IRCAuthCommand()
{
Keyword = "setup-auth";
RequiredParameters = 2;
}
public override string GetArgumentString()
{
return "<target> <message>";
}
public override string GetHelpText()
{
return "Set the authentication message to send to target for identification. e.g. NickServ \"identify hunter2\"";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
IRC.SetProviderInfo(new IRCSetupInfo(IRC.ProviderInfos()[(int)ChatProvider.IRC])
{
AuthTarget = parameters[0],
AuthMessage = parameters[1]
});
return ExitCode.Normal;
}
}
class IRCDisableAuthCommand : ConsoleCommand
{
public IRCDisableAuthCommand()
{
Keyword = "disable-auth";
}
public override string GetHelpText()
{
return "Turns off IRC authentication";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
IRC.SetProviderInfo(new IRCSetupInfo(IRC.ProviderInfos()[(int)ChatProvider.IRC])
{
AuthTarget = null,
AuthMessage = null,
});
return ExitCode.Normal;
}
}
class ChatStatusCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatStatusCommand(ChatProvider pI)
{
Keyword = "status";
providerIndex = (int)pI;
}
public override string GetHelpText()
{
return "Lists channels and connections status";
}
protected override ExitCode Run(IList<string> parameters)
{
var IRC = Interface.GetComponent<ITGChat>();
var info = IRC.ProviderInfos()[providerIndex];
OutputProc("Currently configured channels:");
OutputProc("Admin:");
foreach (var I in info.AdminChannels)
OutputProc("\t" + I);
OutputProc("Watchdog:");
foreach (var I in info.WatchdogChannels)
OutputProc("\t" + I);
OutputProc("Game:");
foreach (var I in info.GameChannels)
OutputProc("\t" + I);
OutputProc("Developer:");
foreach (var I in info.DevChannels)
OutputProc("\t" + I);
OutputProc("Chat bot is: " + (!info.Enabled ? "Disabled" : IRC.Connected(info.Provider) ? "Connected" : "Disconnected"));
return ExitCode.Normal;
}
}
class ChatEnableCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatEnableCommand(ChatProvider pI)
{
Keyword = "enable";
providerIndex = (int)pI;
}
public override string GetHelpText()
{
return "Enables the chat bot";
}
protected override ExitCode Run(IList<string> parameters)
{
var Chat = Interface.GetComponent<ITGChat>();
var info = Chat.ProviderInfos()[providerIndex];
info.Enabled = true;
var res = Chat.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class ChatDisableCommand : ConsoleCommand
{
readonly int providerIndex;
public ChatDisableCommand(ChatProvider pI)
{
Keyword = "disable";
providerIndex = (int)pI;
}
public override string GetHelpText()
{
return "Disables the chat bot";
}
protected override ExitCode Run(IList<string> parameters)
{
var Chat = Interface.GetComponent<ITGChat>();
var info = Chat.ProviderInfos()[providerIndex];
info.Enabled = false;
var res = Chat.SetProviderInfo(info);
if (res != null)
{
OutputProc(res);
return ExitCode.ServerError;
}
return ExitCode.Normal;
}
}
class IRCServerCommand : ConsoleCommand
{
public IRCServerCommand()
{
Keyword = "set-server";
RequiredParameters = 1;
}
public override string GetArgumentString()
{
return "<url>:<port>";
}
public override string GetHelpText()
{
return "Sets the IRC server";
}
protected override ExitCode Run(IList<string> parameters)
{
var splits = parameters[0].Split(':');
if(splits.Length < 2)
{
OutputProc("Invalid parameter!");
return ExitCode.BadCommand;
}
var Chat = Interface.GetComponent<ITGChat>();
var PI = new IRCSetupInfo(Chat.ProviderInfos()[(int)ChatProvider.IRC])
{
URL = splits[0]
};
try
{
PI.Port = Convert.ToUInt16(splits[1]);
}
catch
{
OutputProc("Invalid port number!");
return ExitCode.BadCommand;
}
var res = Chat.SetProviderInfo(PI);
OutputProc(res ?? "Success");
return res == null ? ExitCode.Normal : ExitCode.ServerError;
}
}
class DiscordSetTokenCommand : ConsoleCommand
{
public DiscordSetTokenCommand()
{
Keyword = "set-token";
RequiredParameters = 1;
}
public override string GetArgumentString()
{
return "<bot-token>";
}
public override string GetHelpText()
{
return "Sets the discord API bot token";
}
protected override ExitCode Run(IList<string> parameters)
{
var Chat = Interface.GetComponent<ITGChat>();
var res = Chat.SetProviderInfo(new DiscordSetupInfo(Chat.ProviderInfos()[(int)ChatProvider.Discord]) { BotToken = parameters[0] });
OutputProc(res ?? "Success");
return res == null ? ExitCode.Normal : ExitCode.ServerError;
}
}
}