Added Config class to replace that previous mess
This commit is contained in:
parent
1656e4fa64
commit
d49299e063
2 changed files with 113 additions and 70 deletions
75
Module/ModTools/ConfigItem.cs
Normal file
75
Module/ModTools/ConfigItem.cs
Normal file
|
@ -0,0 +1,75 @@
|
||||||
|
using Newtonsoft.Json.Linq;
|
||||||
|
using Noikoio.RegexBot.ConfigItem;
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Collections.ObjectModel;
|
||||||
|
|
||||||
|
namespace Noikoio.RegexBot.Module.ModTools
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Represents ModTools configuration within one server.
|
||||||
|
/// </summary>
|
||||||
|
class ConfigItem
|
||||||
|
{
|
||||||
|
private EntityName? _petitionReportCh;
|
||||||
|
private readonly ReadOnlyDictionary<string, CommandBase> _cmdInstances;
|
||||||
|
|
||||||
|
public EntityName? PetitionReportingChannel => _petitionReportCh;
|
||||||
|
public ReadOnlyDictionary<string, CommandBase> Commands => _cmdInstances;
|
||||||
|
|
||||||
|
public ConfigItem(ModTools instance, JToken inconf)
|
||||||
|
{
|
||||||
|
if (inconf.Type != JTokenType.Object)
|
||||||
|
{
|
||||||
|
throw new RuleImportException("Configuration for this section is invalid.");
|
||||||
|
}
|
||||||
|
var config = (JObject)inconf;
|
||||||
|
|
||||||
|
// Ban petition reporting channel
|
||||||
|
var petitionstr = config["PetitionRelay"]?.Value<string>();
|
||||||
|
if (string.IsNullOrEmpty(petitionstr)) _petitionReportCh = null;
|
||||||
|
else if (petitionstr.Length > 1 && petitionstr[0] != '#')
|
||||||
|
{
|
||||||
|
// Not a channel.
|
||||||
|
throw new RuleImportException("PetitionRelay value must be set to a channel.");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
_petitionReportCh = new EntityName(petitionstr.Substring(1), EntityType.Channel);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Command instances
|
||||||
|
var commands = new Dictionary<string, CommandBase>(StringComparer.OrdinalIgnoreCase);
|
||||||
|
var commandconf = config["Commands"];
|
||||||
|
if (commandconf != null)
|
||||||
|
{
|
||||||
|
if (commandconf.Type != JTokenType.Object)
|
||||||
|
{
|
||||||
|
throw new RuleImportException("CommandDefs is not properly defined.");
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (var def in commandconf.Children<JProperty>())
|
||||||
|
{
|
||||||
|
string label = def.Name;
|
||||||
|
var cmd = CommandBase.CreateInstance(instance, def);
|
||||||
|
if (commands.ContainsKey(cmd.Command))
|
||||||
|
throw new RuleImportException(
|
||||||
|
$"{label}: 'command' value must not be equal to that of another definition. " +
|
||||||
|
$"Given value is being used for {commands[cmd.Command].Label}.");
|
||||||
|
|
||||||
|
commands.Add(cmd.Command, cmd);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_cmdInstances = new ReadOnlyDictionary<string, CommandBase>(commands);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void UpdatePetitionChannel(ulong id)
|
||||||
|
{
|
||||||
|
if (!PetitionReportingChannel.HasValue) return;
|
||||||
|
if (PetitionReportingChannel.Value.Id.HasValue) return; // nothing to update
|
||||||
|
|
||||||
|
// For lack of a better option - create a new EntityName with ID already provided
|
||||||
|
_petitionReportCh = new EntityName($"{id}::{PetitionReportingChannel.Value.Name}", EntityType.Channel);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -4,7 +4,6 @@ using Newtonsoft.Json.Linq;
|
||||||
using Noikoio.RegexBot.ConfigItem;
|
using Noikoio.RegexBot.ConfigItem;
|
||||||
using System;
|
using System;
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
using System.Collections.ObjectModel;
|
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using System.Threading.Tasks;
|
using System.Threading.Tasks;
|
||||||
|
|
||||||
|
@ -29,70 +28,22 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
else if (arg.Channel is IGuildChannel) await CommandCheckInvoke(arg);
|
else if (arg.Channel is IGuildChannel) await CommandCheckInvoke(arg);
|
||||||
}
|
}
|
||||||
|
|
||||||
#region Config
|
|
||||||
[ConfigSection("modtools")]
|
[ConfigSection("modtools")]
|
||||||
public override async Task<object> ProcessConfiguration(JToken configSection)
|
public override async Task<object> ProcessConfiguration(JToken configSection)
|
||||||
{
|
{
|
||||||
if (configSection.Type != JTokenType.Object)
|
// Constructor throws exception on config errors
|
||||||
{
|
var conf = new ConfigItem(this, configSection);
|
||||||
throw new RuleImportException("Configuration for this section is invalid.");
|
|
||||||
}
|
|
||||||
var config = (JObject)configSection;
|
|
||||||
|
|
||||||
// Ban petition reporting channel
|
// Log results
|
||||||
EntityName? petitionrpt;
|
if (conf.Commands.Count > 0)
|
||||||
var petitionstr = config["PetitionRelay"]?.Value<string>();
|
await Log(conf.Commands.Count + " command definition(s) loaded.");
|
||||||
if (string.IsNullOrEmpty(petitionstr)) petitionrpt = null;
|
if (conf.PetitionReportingChannel.HasValue)
|
||||||
else if (petitionstr.Length > 1 && petitionstr[0] != '#')
|
await Log("Ban petitioning has been enabled.");
|
||||||
{
|
|
||||||
// Not a channel.
|
|
||||||
throw new RuleImportException("PetitionRelay value must be set to a channel.");
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
petitionrpt = new EntityName(petitionstr.Substring(1), EntityType.Channel);
|
|
||||||
}
|
|
||||||
|
|
||||||
// And then the commands
|
return conf;
|
||||||
var commands = new Dictionary<string, CommandBase>(StringComparer.OrdinalIgnoreCase);
|
|
||||||
var commandconf = config["Commands"];
|
|
||||||
if (commandconf != null)
|
|
||||||
{
|
|
||||||
if (commandconf.Type != JTokenType.Object)
|
|
||||||
{
|
|
||||||
throw new RuleImportException("CommandDefs is not properly defined.");
|
|
||||||
}
|
|
||||||
|
|
||||||
foreach (var def in commandconf.Children<JProperty>())
|
|
||||||
{
|
|
||||||
string label = def.Name;
|
|
||||||
var cmd = CommandBase.CreateInstance(this, def);
|
|
||||||
if (commands.ContainsKey(cmd.Command))
|
|
||||||
throw new RuleImportException(
|
|
||||||
$"{label}: 'command' value must not be equal to that of another definition. " +
|
|
||||||
$"Given value is being used for {commands[cmd.Command].Label}.");
|
|
||||||
|
|
||||||
commands.Add(cmd.Command, cmd);
|
|
||||||
}
|
|
||||||
await Log($"Loaded {commands.Count} command definition(s).");
|
|
||||||
}
|
|
||||||
|
|
||||||
return new Tuple<ReadOnlyDictionary<string, CommandBase>, EntityName?>(
|
|
||||||
new ReadOnlyDictionary<string, CommandBase>(commands), petitionrpt);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
private new ConfigItem GetConfig(ulong guildId) => (ConfigItem)base.GetConfig(guildId);
|
||||||
* Config is stored in a tuple. I admit, not the best choice...
|
|
||||||
* Consider a different approach if more data needs to be stored in the future.
|
|
||||||
* Item 1: Command config (Dictionary<string, CommandBase>)
|
|
||||||
* Item 2: Ban petition channel (EntityName)
|
|
||||||
*/
|
|
||||||
|
|
||||||
private new Tuple<ReadOnlyDictionary<string, CommandBase>, EntityName?> GetConfig(ulong guildId)
|
|
||||||
=> (Tuple<ReadOnlyDictionary<string, CommandBase>, EntityName?>)base.GetConfig(guildId);
|
|
||||||
private ReadOnlyDictionary<string, CommandBase> GetCommandConfig(ulong guild) => GetConfig(guild).Item1;
|
|
||||||
private EntityName? GetPetitionConfig(ulong guild) => GetConfig(guild).Item2;
|
|
||||||
#endregion
|
|
||||||
|
|
||||||
public new Task Log(string text) => base.Log(text);
|
public new Task Log(string text) => base.Log(text);
|
||||||
|
|
||||||
|
@ -115,7 +66,7 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
int spc = arg.Content.IndexOf(' ');
|
int spc = arg.Content.IndexOf(' ');
|
||||||
if (spc != -1) cmdchk = arg.Content.Substring(0, spc);
|
if (spc != -1) cmdchk = arg.Content.Substring(0, spc);
|
||||||
else cmdchk = arg.Content;
|
else cmdchk = arg.Content;
|
||||||
if ((GetCommandConfig(g.Id)).TryGetValue(cmdchk, out var c))
|
if (GetConfig(g.Id).Commands.TryGetValue(cmdchk, out var c))
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
|
@ -130,6 +81,7 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#region Ban petitions
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// List of available appeals. Key is user (for quick lookup). Value is guild (for quick config resolution).
|
/// List of available appeals. Key is user (for quick lookup). Value is guild (for quick config resolution).
|
||||||
/// TODO expiration?
|
/// TODO expiration?
|
||||||
|
@ -138,13 +90,17 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
public void AddPetition(ulong guild, ulong user)
|
public void AddPetition(ulong guild, ulong user)
|
||||||
{
|
{
|
||||||
// Do nothing if disabled
|
// Do nothing if disabled
|
||||||
if (GetPetitionConfig(guild) == null) return;
|
if (!GetConfig(guild).PetitionReportingChannel.HasValue) return;
|
||||||
lock (_openPetitions) _openPetitions[user] = guild;
|
lock (_openPetitions) _openPetitions[user] = guild;
|
||||||
}
|
}
|
||||||
private async Task PetitionRelayCheck(SocketMessage msg)
|
private async Task PetitionRelayCheck(SocketMessage msg)
|
||||||
{
|
{
|
||||||
const string PetitionAccepted = "Your petition has been forwarded to the moderators for review.";
|
const string PetitionAccepted = "Your petition has been forwarded to the moderators for review.";
|
||||||
const string PetitionDenied = "You may not submit a ban petition.";
|
const string PetitionDenied = "You may not submit a ban petition.";
|
||||||
|
|
||||||
|
// It's possible the sender may still block messages sent to them,
|
||||||
|
// hence the empty catch blocks you'll see up ahead.
|
||||||
|
|
||||||
if (!msg.Content.StartsWith("!petition ", StringComparison.InvariantCultureIgnoreCase)) return;
|
if (!msg.Content.StartsWith("!petition ", StringComparison.InvariantCultureIgnoreCase)) return;
|
||||||
|
|
||||||
// Input validation
|
// Input validation
|
||||||
|
@ -154,6 +110,13 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
// Just ignore.
|
// Just ignore.
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
if (ptext.Length > 1000)
|
||||||
|
{
|
||||||
|
// Enforce petition length limit.
|
||||||
|
try { await msg.Author.SendMessageAsync("Your petition message is too long. Try again with a shorter message."); }
|
||||||
|
catch (Discord.Net.HttpException) { }
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
ulong targetGuild = 0;
|
ulong targetGuild = 0;
|
||||||
lock (_openPetitions)
|
lock (_openPetitions)
|
||||||
|
@ -164,11 +127,9 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// It's possible the sender may still block messages sent to them,
|
|
||||||
// hence the empty catch blocks you'll see up ahead.
|
|
||||||
|
|
||||||
if (targetGuild == 0)
|
if (targetGuild == 0)
|
||||||
{
|
{
|
||||||
|
// Not in the list. Nothing to do.
|
||||||
try { await msg.Author.SendMessageAsync(PetitionDenied); }
|
try { await msg.Author.SendMessageAsync(PetitionDenied); }
|
||||||
catch (Discord.Net.HttpException) { }
|
catch (Discord.Net.HttpException) { }
|
||||||
return;
|
return;
|
||||||
|
@ -182,9 +143,9 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get petition reporting target if not already known
|
// Get petition reporting target
|
||||||
var pcv = GetPetitionConfig(targetGuild);
|
var pcv = GetConfig(targetGuild).PetitionReportingChannel;
|
||||||
if (!pcv.HasValue) return; // No target. How'd we get here, anyway?
|
if (!pcv.HasValue) return; // No target. This should be logically impossible, but... just in case.
|
||||||
var rch = pcv.Value;
|
var rch = pcv.Value;
|
||||||
ISocketMessageChannel rchObj;
|
ISocketMessageChannel rchObj;
|
||||||
if (!rch.Id.HasValue)
|
if (!rch.Id.HasValue)
|
||||||
|
@ -192,11 +153,17 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
rchObj = gObj.TextChannels
|
rchObj = gObj.TextChannels
|
||||||
.Where(c => c.Name.Equals(rch.Name, StringComparison.InvariantCultureIgnoreCase))
|
.Where(c => c.Name.Equals(rch.Name, StringComparison.InvariantCultureIgnoreCase))
|
||||||
.FirstOrDefault();
|
.FirstOrDefault();
|
||||||
|
// Update value if found
|
||||||
|
if (rchObj != null)
|
||||||
|
{
|
||||||
|
GetConfig(targetGuild).UpdatePetitionChannel(rchObj.Id);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
rchObj = (ISocketMessageChannel)gObj.GetChannel(rch.Id.Value);
|
rchObj = gObj.GetChannel(rch.Id.Value) as ISocketMessageChannel;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (rchObj == null)
|
if (rchObj == null)
|
||||||
{
|
{
|
||||||
// Channel not found.
|
// Channel not found.
|
||||||
|
@ -206,7 +173,7 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ready to relay as embed
|
// Ready to relay
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
await rchObj.SendMessageAsync("", embed: new EmbedBuilder()
|
await rchObj.SendMessageAsync("", embed: new EmbedBuilder()
|
||||||
|
@ -240,5 +207,6 @@ namespace Noikoio.RegexBot.Module.ModTools
|
||||||
try { await msg.Author.SendMessageAsync(PetitionAccepted); }
|
try { await msg.Author.SendMessageAsync(PetitionAccepted); }
|
||||||
catch (Discord.Net.HttpException) { }
|
catch (Discord.Net.HttpException) { }
|
||||||
}
|
}
|
||||||
|
#endregion
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue