227 lines
9.5 KiB
C#
227 lines
9.5 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Discord;
|
|
using Discord.Commands;
|
|
using Discord.WebSocket;
|
|
using Geekbot.net.Database;
|
|
using Geekbot.net.Lib.Extensions;
|
|
using Geekbot.net.Lib.Logger;
|
|
using Geekbot.net.Lib.ReactionListener;
|
|
using Geekbot.net.Lib.UserRepository;
|
|
using StackExchange.Redis;
|
|
|
|
namespace Geekbot.net
|
|
{
|
|
public class Handlers
|
|
{
|
|
private readonly DatabaseContext _database;
|
|
private readonly IDiscordClient _client;
|
|
private readonly IGeekbotLogger _logger;
|
|
private readonly IDatabase _redis;
|
|
private readonly IServiceProvider _servicesProvider;
|
|
private readonly CommandService _commands;
|
|
private readonly IUserRepository _userRepository;
|
|
private readonly IReactionListener _reactionListener;
|
|
|
|
public Handlers(DatabaseContext database, IDiscordClient client, IGeekbotLogger logger, IDatabase redis,
|
|
IServiceProvider servicesProvider, CommandService commands, IUserRepository userRepository,
|
|
IReactionListener reactionListener)
|
|
{
|
|
_database = database;
|
|
_client = client;
|
|
_logger = logger;
|
|
_redis = redis;
|
|
_servicesProvider = servicesProvider;
|
|
_commands = commands;
|
|
_userRepository = userRepository;
|
|
_reactionListener = reactionListener;
|
|
}
|
|
|
|
//
|
|
// Incoming Messages
|
|
//
|
|
|
|
public Task RunCommand(SocketMessage messageParam)
|
|
{
|
|
try
|
|
{
|
|
if (!(messageParam is SocketUserMessage message)) return Task.CompletedTask;
|
|
if (message.Author.IsBot) return Task.CompletedTask;
|
|
var argPos = 0;
|
|
|
|
// ToDo: remove
|
|
// if (!message.Author.Id.Equals(93061333972455424)) return Task.CompletedTask;
|
|
|
|
var lowCaseMsg = message.ToString().ToLower();
|
|
if (lowCaseMsg.StartsWith("hui"))
|
|
{
|
|
var hasPing = _database.GuildSettings.FirstOrDefault(guild =>
|
|
guild.GuildId.Equals(((SocketGuildChannel) message.Channel).Guild.Id.AsLong()))
|
|
?.Hui ?? false;
|
|
if (hasPing)
|
|
{
|
|
message.Channel.SendMessageAsync("hui!!!");
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|
|
if (lowCaseMsg.StartsWith("ping ") || lowCaseMsg.Equals("ping"))
|
|
{
|
|
var hasPing = _database.GuildSettings.FirstOrDefault(guild =>
|
|
guild.GuildId.Equals(((SocketGuildChannel) message.Channel).Guild.Id.AsLong()))
|
|
?.Ping ?? false;
|
|
if (hasPing)
|
|
{
|
|
message.Channel.SendMessageAsync("pong");
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|
|
if (!(message.HasCharPrefix('!', ref argPos) ||
|
|
message.HasMentionPrefix(_client.CurrentUser, ref argPos))) return Task.CompletedTask;
|
|
var context = new CommandContext(_client, message);
|
|
var commandExec = _commands.ExecuteAsync(context, argPos, _servicesProvider);
|
|
_logger.Information(LogSource.Command,
|
|
context.Message.Content.Split(" ")[0].Replace("!", ""),
|
|
SimpleConextConverter.ConvertContext(context));
|
|
return Task.CompletedTask;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.Error(LogSource.Geekbot, "Failed to Process Message", e);
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|
|
|
|
public Task UpdateStats(SocketMessage message)
|
|
{
|
|
try
|
|
{
|
|
if (message == null) return Task.CompletedTask;
|
|
if (message.Channel.Name.StartsWith('@'))
|
|
{
|
|
_logger.Information(LogSource.Message, $"[DM-Channel] {message.Content}", SimpleConextConverter.ConvertSocketMessage(message));
|
|
return Task.CompletedTask;
|
|
}
|
|
var channel = (SocketGuildChannel) message.Channel;
|
|
|
|
_redis.HashIncrementAsync($"{channel.Guild.Id}:Messages", message.Author.Id.ToString());
|
|
_redis.HashIncrementAsync($"{channel.Guild.Id}:Messages", 0.ToString());
|
|
|
|
if (message.Author.IsBot) return Task.CompletedTask;
|
|
_logger.Information(LogSource.Message, message.Content, SimpleConextConverter.ConvertSocketMessage(message));
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.Error(LogSource.Message, "Could not process message stats", e);
|
|
}
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
//
|
|
// User Stuff
|
|
//
|
|
|
|
public async Task UserJoined(SocketGuildUser user)
|
|
{
|
|
try
|
|
{
|
|
if (!user.IsBot)
|
|
{
|
|
var message = _database.GuildSettings.FirstOrDefault(guild =>
|
|
guild.GuildId.Equals(user.Guild.Id.AsLong()))
|
|
?.WelcomeMessage;
|
|
if (!string.IsNullOrEmpty(message))
|
|
{
|
|
message = message.Replace("$user", user.Mention);
|
|
await user.Guild.DefaultChannel.SendMessageAsync(message);
|
|
}
|
|
}
|
|
await _userRepository.Update(user);
|
|
_logger.Information(LogSource.Geekbot, $"{user.Username} ({user.Id}) joined {user.Guild.Name} ({user.Guild.Id})");
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.Error(LogSource.Geekbot, "Failed to send welcome message", e);
|
|
}
|
|
}
|
|
|
|
public async Task UserUpdated(SocketUser oldUser, SocketUser newUser)
|
|
{
|
|
await _userRepository.Update(newUser);
|
|
}
|
|
|
|
public async Task UserLeft(SocketGuildUser user)
|
|
{
|
|
try
|
|
{
|
|
var guild = _database.GuildSettings.FirstOrDefault(g =>
|
|
g.GuildId.Equals(user.Guild.Id.AsLong()));
|
|
if (guild?.ShowLeave ?? false)
|
|
{
|
|
var modChannelSocket = (ISocketMessageChannel) await _client.GetChannelAsync(guild.ModChannel.AsUlong());
|
|
await modChannelSocket.SendMessageAsync($"{user.Username}#{user.Discriminator} left the server");
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.Error(LogSource.Geekbot, "Failed to send leave message", e);
|
|
}
|
|
_logger.Information(LogSource.Geekbot, $"{user.Username} ({user.Id}) joined {user.Guild.Name} ({user.Guild.Id})");
|
|
}
|
|
|
|
//
|
|
// Message Stuff
|
|
//
|
|
|
|
public async Task MessageDeleted(Cacheable<IMessage, ulong> message, ISocketMessageChannel channel)
|
|
{
|
|
try
|
|
{
|
|
var guildSocketData = ((IGuildChannel) channel).Guild;
|
|
var guild = _database.GuildSettings.FirstOrDefault(g =>
|
|
g.GuildId.Equals(guildSocketData.Id.AsLong()));
|
|
if (guild?.ShowDelete ?? false)
|
|
{
|
|
var modChannelSocket = (ISocketMessageChannel) await _client.GetChannelAsync(guild.ModChannel.AsUlong());
|
|
var sb = new StringBuilder();
|
|
if (message.Value != null)
|
|
{
|
|
sb.AppendLine(
|
|
$"The following message from {message.Value.Author.Username}#{message.Value.Author.Discriminator} was deleted in <#{channel.Id}>");
|
|
sb.AppendLine(message.Value.Content);
|
|
}
|
|
else
|
|
{
|
|
sb.AppendLine("Someone deleted a message, the message was not cached...");
|
|
}
|
|
await modChannelSocket.SendMessageAsync(sb.ToString());
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.Error(LogSource.Geekbot, "Failed to send delete message...", e);
|
|
}
|
|
}
|
|
|
|
//
|
|
// Reactions
|
|
//
|
|
|
|
public Task ReactionAdded(Cacheable<IUserMessage, ulong> cacheable, ISocketMessageChannel socketMessageChannel, SocketReaction reaction)
|
|
{
|
|
if (reaction.User.Value.IsBot) return Task.CompletedTask;
|
|
if (!_reactionListener.IsListener(reaction.MessageId)) return Task.CompletedTask;
|
|
_reactionListener.GiveRole(socketMessageChannel, reaction);
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task ReactionRemoved(Cacheable<IUserMessage, ulong> cacheable, ISocketMessageChannel socketMessageChannel, SocketReaction reaction)
|
|
{
|
|
if (reaction.User.Value.IsBot) return Task.CompletedTask;
|
|
if (!_reactionListener.IsListener(reaction.MessageId)) return Task.CompletedTask;
|
|
_reactionListener.RemoveRole(socketMessageChannel, reaction);
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|
|
}
|