Restructured MessageCache, added notes for implementation

This commit is contained in:
Noikoio 2018-02-17 00:45:40 -08:00
parent 683b852de7
commit e771e09d08

View file

@ -1,64 +1,63 @@
using Discord.WebSocket; using Discord.WebSocket;
using Newtonsoft.Json.Linq;
using Noikoio.RegexBot.ConfigItem;
using Npgsql; using Npgsql;
using NpgsqlTypes; using NpgsqlTypes;
using System;
using System.Threading.Tasks; using System.Threading.Tasks;
namespace Noikoio.RegexBot.Module.DBCache namespace Noikoio.RegexBot.Module.ModLogs
{ {
/// <summary> /// <summary>
/// Caches information regarding all incoming messages. /// Helper class for <see cref="ModLogs"/>. Keeps a database-backed cache of recent messages and assists
/// The function of this feature should be transparent to the user, and thus no configuration is needed. /// in reporting message changes and deletions, if configured to do so.
/// Does not manipulate the moderation log managed by the main class, but rather provides supplemental features.
/// </summary> /// </summary>
class MessageCache : BotFeature class MessageCache
{ {
// TODO Something that clears expired cache items private readonly DiscordSocketClient _dClient;
private readonly DatabaseConfig _db; private readonly AsyncLogger _outLog;
private readonly Func<ulong, object> _outGetConfig;
public override string Name => nameof(MessageCache); public MessageCache(DiscordSocketClient client, AsyncLogger logger, Func<ulong, object> getConfFunc)
public MessageCache(DiscordSocketClient client) : base(client)
{ {
_db = RegexBot.Config.Database; _dClient = client;
_outLog = logger;
_outGetConfig = getConfFunc;
if (_db.Enabled) CreateCacheTables();
{
CreateCacheTables();
client.MessageReceived += Client_MessageReceived; client.MessageReceived += Client_MessageReceived;
//client.MessageUpdated += Client_MessageUpdated; client.MessageUpdated += Client_MessageUpdated;
} client.MessageDeleted += Client_MessageDeleted;
else
{
Log("No database storage available.").Wait();
}
} }
#region Table setup
const string TableMessage = "cache_messages";
public override Task<object> ProcessConfiguration(JToken configSection) => Task.FromResult<object>(null);
#region Event handling #region Event handling
// A new message has been created private async Task Client_MessageReceived(SocketMessage arg) => await CacheMessage(arg);
private async Task Client_MessageReceived(SocketMessage arg)
private Task Client_MessageUpdated(Discord.Cacheable<Discord.IMessage, ulong> arg1, SocketMessage arg2, ISocketMessageChannel arg3)
{ {
await Task.Run(() => CacheMessage(arg)); /*
* TODO:
* Edited messages seem to retain their ID. Need to look into this.
* In any case, the new message must be stored in case of future edits.
* The change must be sent to the reporting channel (if one exists) as if it were
* a typical log entry (even though it's not).
*/
throw new NotImplementedException();
} }
//private Task Client_MessageUpdated(Discord.Cacheable<Discord.IMessage, ulong> arg1, SocketMessage arg2, ISocketMessageChannel arg3) private Task Client_MessageDeleted(Discord.Cacheable<Discord.IMessage, ulong> arg1, ISocketMessageChannel arg2)
/* {
* Edited messages seem to retain their ID. This is a problem. // TODO report message deletion, if reporting channel exists and message is in cache.
* The point of this message cache was to have another feature be able to relay throw new NotImplementedException();
* both the previous and current message at once. }
* For now: Do nothing on updated messages.
*/
#endregion #endregion
#region Database manipulation
const string TableMessage = "cache_messages";
private void CreateCacheTables() private void CreateCacheTables()
{ {
using (var db = _db.GetOpenConnectionAsync().GetAwaiter().GetResult()) using (var db = RegexBot.Config.GetOpenDatabaseConnectionAsync().GetAwaiter().GetResult())
{ {
using (var c = db.CreateCommand()) using (var c = db.CreateCommand())
{ {
@ -70,9 +69,9 @@ namespace Noikoio.RegexBot.Module.DBCache
+ "created_ts timestamptz not null, " + "created_ts timestamptz not null, "
+ "edited_ts timestamptz null, " + "edited_ts timestamptz null, "
+ "message text not null, " + "message text not null, "
+ $"FOREIGN KEY (author_id, guild_id) references {EntityCache.Sql.TableUser} (user_id, guild_id)" + $"FOREIGN KEY (author_id, guild_id) references {EntityCache.SqlHelper.TableUser} (user_id, guild_id)"
+ ")"; + ")";
// TODO figure out how to store message edits // TODO are more columns needed for edit info?
c.ExecuteNonQuery(); c.ExecuteNonQuery();
} }
} }
@ -83,7 +82,7 @@ namespace Noikoio.RegexBot.Module.DBCache
{ {
try try
{ {
using (var db = await _db.GetOpenConnectionAsync()) using (var db = await RegexBot.Config.GetOpenDatabaseConnectionAsync())
{ {
using (var c = db.CreateCommand()) using (var c = db.CreateCommand())
{ {
@ -103,7 +102,7 @@ namespace Noikoio.RegexBot.Module.DBCache
} }
catch (NpgsqlException ex) catch (NpgsqlException ex)
{ {
await Log($"SQL error in {nameof(CacheMessage)}: " + ex.Message); await _outLog($"SQL error in {nameof(CacheMessage)}: " + ex.Message);
} }
} }
} }