daan/geekbot
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity

Remove all redis leftovers

Browse source
This commit is contained in:
runebaas 2020-06-01 01:02:36 +02:00
parent 46fee88f03
commit 520633c590
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
8 changed files with 2 additions and 156 deletions
Download patch file Download diff file Expand all files Collapse all files

79
Geekbot.net/Commands/Admin/Owner/Owner.cs
View file

@ -1,16 +1,12 @@
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Discord;
using Discord.Commands;
using Discord.WebSocket;
using Geekbot.net.Lib.AlmostRedis;
using Geekbot.net.Lib.ErrorHandling;
using Geekbot.net.Lib.GlobalSettings;
using Geekbot.net.Lib.Logger;
using Geekbot.net.Lib.ReactionListener;
using Geekbot.net.Lib.UserRepository;
using StackExchange.Redis;
namespace Geekbot.net.Commands.Admin.Owner
{
@ -21,22 +17,16 @@ namespace Geekbot.net.Commands.Admin.Owner
private readonly DiscordSocketClient _client;
private readonly IErrorHandler _errorHandler;
private readonly IGlobalSettings _globalSettings;
private readonly IDatabase _redis;
private readonly IReactionListener _reactionListener;
private readonly IGeekbotLogger _logger;
private readonly IUserRepository _userRepository;
public Owner(DiscordSocketClient client, IGeekbotLogger logger, IUserRepository userRepositry, IErrorHandler errorHandler, IGlobalSettings globalSettings,
IAlmostRedis redis, IReactionListener reactionListener
)
public Owner(DiscordSocketClient client, IGeekbotLogger logger, IUserRepository userRepositry, IErrorHandler errorHandler, IGlobalSettings globalSettings)
{
_client = client;
_logger = logger;
_userRepository = userRepositry;
_errorHandler = errorHandler;
_globalSettings = globalSettings;
_redis = redis.Db;
_reactionListener = reactionListener;
}
[Command("youtubekey", RunMode = RunMode.Async)]
@ -132,72 +122,5 @@ namespace Geekbot.net.Commands.Admin.Owner
await _errorHandler.HandleCommandException(e, Context);
}
}
[Command("migrate_listeners", RunMode = RunMode.Async)]
public async Task MigrateListeners()
{
try
{
var messageIds = _redis.SetMembers("MessageIds");
var connectedGuilds = _client.Guilds;
var messageGuildAssociation = new Dictionary<ulong, ulong>();
foreach (var messageIdUnparsed in messageIds)
{
var messageId = ulong.Parse(messageIdUnparsed);
var reactions = _redis.HashGetAll($"Messages:{messageIdUnparsed.ToString()}");
foreach (var reaction in reactions)
{
_logger.Information(LogSource.Migration, $"{messageIdUnparsed.ToString()} - Starting");
try
{
ulong guildId = 0;
IRole role = null;
var roleId = ulong.Parse(reaction.Value);
if (messageGuildAssociation.ContainsKey(messageId))
{
guildId = messageGuildAssociation[messageId];
_logger.Information(LogSource.Migration, $"{messageIdUnparsed.ToString()} - known to be in {guildId}");
role = _client.GetGuild(guildId).GetRole(roleId);
}
else
{
_logger.Information(LogSource.Migration, $"{messageIdUnparsed.ToString()} - Attempting to find guild");
IRole foundRole = null;
try
{
foreach (var guild in connectedGuilds)
{
foundRole = guild.GetRole(roleId);
if (foundRole != null)
{
role = _client.GetGuild(foundRole.Guild.Id).GetRole(ulong.Parse(reaction.Value));
messageGuildAssociation.Add(messageId, foundRole.Guild.Id);
}
}
} catch { /* ignore */ }
if (foundRole == null)
{
_logger.Warning(LogSource.Migration, $"{messageIdUnparsed.ToString()} - Could not find guild for message");
continue;
}
}
_logger.Information(LogSource.Migration, $"{messageIdUnparsed.ToString()} - Found Role {roleId.ToString()}");
await _reactionListener.AddRoleToListener(ulong.Parse(messageIdUnparsed), guildId, reaction.Name, role);
}
catch (Exception e)
{
_logger.Error(LogSource.Migration, $"Failed to migrate reaction for {messageIdUnparsed.ToString()}", e);
}
}
}
}
catch (Exception e)
{
await _errorHandler.HandleCommandException(e, Context);
}
}
}
}