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

Add posibility to remove quotes

Browse source
This commit is contained in:
runebaas 2018-01-21 16:01:43 +01:00
parent a2eb475201
commit 815b7d6163
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
1 changed files with 47 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

63
Geekbot.net/Commands/Quote.cs
View file

@ -12,13 +12,13 @@ namespace Geekbot.net.Commands
[Group("quote")] [Group("quote")]
public class Quote : ModuleBase public class Quote : ModuleBase
{ {
private readonly IErrorHandler errorHandler; private readonly IErrorHandler _errorHandler;
private readonly IDatabase redis; private readonly IDatabase _redis;
public Quote(IDatabase redis, IErrorHandler errorHandler) public Quote(IDatabase redis, IErrorHandler errorHandler, Random random)
{ {
this.redis = redis; _redis = redis;
this.errorHandler = errorHandler; _errorHandler = errorHandler;
} }
[Command] [Command]
@ -26,16 +26,18 @@ namespace Geekbot.net.Commands
[Summary("Return a random quoute from the database")] [Summary("Return a random quoute from the database")]
public async Task getRandomQuote() public async Task getRandomQuote()
{ {
var randomQuote = redis.SetRandomMember($"{Context.Guild.Id}:Quotes");
try try
{ {
var randomQuotes = _redis.SetMembers($"{Context.Guild.Id}:Quotes");
var randomNumber = new Random().Next(randomQuotes.Length - 1);
var randomQuote = randomQuotes[randomNumber];
var quote = JsonConvert.DeserializeObject<QuoteObject>(randomQuote); var quote = JsonConvert.DeserializeObject<QuoteObject>(randomQuote);
var embed = quoteBuilder(quote); var embed = quoteBuilder(quote, randomNumber + 1);
await ReplyAsync("", false, embed.Build()); await ReplyAsync("", false, embed.Build());
} }
catch (Exception e) catch (Exception e)
{ {
errorHandler.HandleCommandException(e, Context, "Whoops, seems like the quote was to edgy to return"); _errorHandler.HandleCommandException(e, Context, "Whoops, seems like the quote was to edgy to return");
} }
} }
@ -61,13 +63,13 @@ namespace Geekbot.net.Commands
var lastMessage = await getLastMessageByUser(user); var lastMessage = await getLastMessageByUser(user);
var quote = createQuoteObject(lastMessage); var quote = createQuoteObject(lastMessage);
var quoteStore = JsonConvert.SerializeObject(quote); var quoteStore = JsonConvert.SerializeObject(quote);
redis.SetAdd($"{Context.Guild.Id}:Quotes", quoteStore); _redis.SetAdd($"{Context.Guild.Id}:Quotes", quoteStore);
var embed = quoteBuilder(quote); var embed = quoteBuilder(quote);
await ReplyAsync("**Quote Added**", false, embed.Build()); await ReplyAsync("**Quote Added**", false, embed.Build());
} }
catch (Exception e) catch (Exception e)
{ {
errorHandler.HandleCommandException(e, Context, _errorHandler.HandleCommandException(e, Context,
"I counldn't find a quote from that user :disappointed:"); "I counldn't find a quote from that user :disappointed:");
} }
} }
@ -94,13 +96,13 @@ namespace Geekbot.net.Commands
var quote = createQuoteObject(message); var quote = createQuoteObject(message);
var quoteStore = JsonConvert.SerializeObject(quote); var quoteStore = JsonConvert.SerializeObject(quote);
redis.SetAdd($"{Context.Guild.Id}:Quotes", quoteStore); _redis.SetAdd($"{Context.Guild.Id}:Quotes", quoteStore);
var embed = quoteBuilder(quote); var embed = quoteBuilder(quote);
await ReplyAsync("**Quote Added**", false, embed.Build()); await ReplyAsync("**Quote Added**", false, embed.Build());
} }
catch (Exception e) catch (Exception e)
{ {
errorHandler.HandleCommandException(e, Context, _errorHandler.HandleCommandException(e, Context,
"I couldn't find a message with that id :disappointed:"); "I couldn't find a message with that id :disappointed:");
} }
} }
@ -119,7 +121,7 @@ namespace Geekbot.net.Commands
} }
catch (Exception e) catch (Exception e)
{ {
errorHandler.HandleCommandException(e, Context, _errorHandler.HandleCommandException(e, Context,
"I counldn't find a quote from that user :disappointed:"); "I counldn't find a quote from that user :disappointed:");
} }
} }
@ -138,11 +140,39 @@ namespace Geekbot.net.Commands
} }
catch (Exception e) catch (Exception e)
{ {
errorHandler.HandleCommandException(e, Context, _errorHandler.HandleCommandException(e, Context,
"I couldn't find a message with that id :disappointed:"); "I couldn't find a message with that id :disappointed:");
} }
} }
[Command("remove")]
[RequireUserPermission(GuildPermission.KickMembers)]
[RequireUserPermission(GuildPermission.ManageMessages)]
[RequireUserPermission(GuildPermission.ManageRoles)]
[Remarks(CommandCategories.Quotes)]
[Summary("Remove a quote (required mod permissions)")]
public async Task removeQuote([Summary("quoteId")] int id)
{
try
{
var quotes = _redis.SetMembers($"{Context.Guild.Id}:Quotes");
var success = _redis.SetRemove($"{Context.Guild.Id}:Quotes", quotes[id - 1]);
if (success)
{
await ReplyAsync($"Removed quote #{id}");
}
else
{
await ReplyAsync($"I couldn't find a quote with that id :disappointed:");
}
}
catch (Exception e)
{
_errorHandler.HandleCommandException(e, Context,
"I couldn't find a quote with that id :disappointed:");
}
}
private async Task<IMessage> getLastMessageByUser(IUser user) private async Task<IMessage> getLastMessageByUser(IUser user)
{ {
var list = Context.Channel.GetMessagesAsync().Flatten(); var list = Context.Channel.GetMessagesAsync().Flatten();
@ -154,12 +184,13 @@ namespace Geekbot.net.Commands
&& !msg.Content.ToLower().StartsWith("!")); && !msg.Content.ToLower().StartsWith("!"));
} }
private EmbedBuilder quoteBuilder(QuoteObject quote) private EmbedBuilder quoteBuilder(QuoteObject quote, int id = 0)
{ {
var user = Context.Client.GetUserAsync(quote.userId).Result; var user = Context.Client.GetUserAsync(quote.userId).Result;
var eb = new EmbedBuilder(); var eb = new EmbedBuilder();
eb.WithColor(new Color(143, 167, 232)); eb.WithColor(new Color(143, 167, 232));
eb.Title = $"{user.Username} @ {quote.time.Day}.{quote.time.Month}.{quote.time.Year}"; eb.Title = id == 0 ? "" : $"#{id} | ";
eb.Title += $"{user.Username} @ {quote.time.Day}.{quote.time.Month}.{quote.time.Year}";
eb.Description = quote.quote; eb.Description = quote.quote;
eb.ThumbnailUrl = user.GetAvatarUrl(); eb.ThumbnailUrl = user.GetAvatarUrl();
if (quote.image != null) eb.ImageUrl = quote.image; if (quote.image != null) eb.ImageUrl = quote.image;