Skip to content

Commit

Permalink
Added giveaward command to give users awards
Browse files Browse the repository at this point in the history 
Added listawards command to list all the awards in the database
Added deleteaward command to remove an award from a user
Changed AwardModule base class from ModuleBase to InteractiveBase
  • Loading branch information
@MythicalCuddles
MythicalCuddles committed Jan 18, 2020
1 parent e5db29a commit 967c46c
Showing 1 changed file with 201 additions and 26 deletions.
227 changes: 201 additions & 26 deletions DiscordBot/Modules/AwardModule.cs
View file
Original file line number Diff line number Diff line change
@@ -1,22 +1,27 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

using Discord;
using Discord.Commands;
using Discord.WebSocket;

using Discord.Addons.Interactive;

using DiscordBot.Common;
using DiscordBot.Common.Preconditions;
using DiscordBot.Database;
using DiscordBot.Exceptions;
using DiscordBot.Extensions;
using DiscordBot.Logging;
using DiscordBot.Objects;
using MySql.Data.MySqlClient;

namespace DiscordBot.Modules
{
[RequireContext(ContextType.Guild)]
[Name("Award Commands")]
public class AwardModule : ModuleBase
public class AwardModule : InteractiveBase
{
[MinPermissions(PermissionLevel.User)]
[Command("awards"), Summary("")]
Expand All @@ -41,25 +46,6 @@ public async Task ShowUserAwards(IUser user = null)
.WithColor(userSpecified.GetCustomRGB());

List<Award> awards = new List<Award>();
//
// (MySqlDataReader dr, MySqlConnection conn) reader = DatabaseActivity.ExecuteReader("SELECT * FROM awards WHERE userId=" + userSpecified.Id + ";");
//
// while (reader.dr.Read())
// {
// awards.Add(new Award()
// {
// AwardId = reader.dr.GetInt32("awardID"),
// UserId = (ulong) reader.dr["userID"],
// AwardText = reader.dr["awardText"].ToString(),
// DateAwarded = (DateTime)reader.dr["dateAwarded"]
// });
// }
//
// reader.dr.Close();
// reader.conn.Close();
//
// // Swapping variables x and y in comparison will display oldest awards first - y.dateAwarded.CompareTo(x.dateAwarded)
// awards.Sort((x, y) => x.DateAwarded.CompareTo(y.DateAwarded));

foreach (Award a in Award.Awards)
{
Expand All @@ -70,9 +56,6 @@ public async Task ShowUserAwards(IUser user = null)
}
awards.Sort((x, y) => x.DateAwarded.CompareTo(y.DateAwarded));




StringBuilder sb = new StringBuilder();

if (awards.Count == 0)
Expand All @@ -90,7 +73,6 @@ public async Task ShowUserAwards(IUser user = null)
{
int result = DateTime.Compare(date, a.DateAwarded);
// result : -1:Future Date, 1:Past Date, 1:Today

if (result >= 0)
{
sb.Append(a.AwardText + " - " + a.DateAwarded.ToString("dd/MM/yyyy") + "\n");
Expand All @@ -106,5 +88,198 @@ public async Task ShowUserAwards(IUser user = null)
eb.WithDescription(sb.ToString());
await ReplyAsync("", false, eb.Build());
}

[MinPermissions(PermissionLevel.ServerAdmin)]
[Command("giveaward"), Summary("")]
public async Task AddUserAward(IUser user = null, string aCategory = null, string aText = null)
{
if (user == null || aCategory == null || aText == null)
{
EmbedBuilder eb = new EmbedBuilder
{
Title = "Invalid Syntax",
Description = "**Syntax:** " + Guild.Load(Context.Guild.Id).Prefix + "giveaward [user] [award category] [award name]",
Color = new Color(210, 47, 33)
};

eb.AddField("Notes", "Be sure to use speech marks (\"\") when specifying an award category or name!");
eb.WithCurrentTimestamp();

await ReplyAsync("", false, eb.Build());
return;
}

Award.AddAward(user.Id, aCategory, aText);

AdminLog.Log(Context.User.Id, Context.Message.Content, Context.Guild.Id, user.Id);

EmbedBuilder replyEmbed = new EmbedBuilder()
{
Title = "Congratulations " + user.Username + "!",
Color = new Color(0, 255, 0),
Description = user.Mention + " earned an award!",
Footer = new EmbedFooterBuilder()
{
Text = "Awarded by " + Context.User.Username + " (AID: " + Award.Awards[Award.Awards.Count-1].AwardId + ")"
}
}
.AddField("Award Category", aCategory)
.AddField("Award", aText)
.WithCurrentTimestamp();

await ReplyAsync("", false, replyEmbed.Build());

replyEmbed = new EmbedBuilder()
{
Title = "New Award",
Description = user.Mention + " was given an award by " + Context.User.Mention,
Footer = new EmbedFooterBuilder()
{
Text = "ID: " + Award.Awards[Award.Awards.Count-1].AwardId
}
}
.AddField("Award Category", aCategory)
.AddField("Award", aText)
.WithCurrentTimestamp();

await Configuration.Load().LogChannelId.GetTextChannel().SendMessageAsync("", false, replyEmbed.Build());
}

[MinPermissions(PermissionLevel.ServerAdmin)]
[Command("listawards"), Summary("")]
public async Task ListAllAwards()
{
EmbedBuilder eb;
if (!Award.Awards.Any())
{
eb = new EmbedBuilder
{
Title = "Awards",
Description = "There is no awards in the database.",
Color = new Color(235, 160, 40)
};

await ReplyAsync("", false, eb.Build());
return;
}

AdminLog.Log(Context.User.Id, Context.Message.Content, Context.Guild.Id);

List<string> awardPages = new List<string>();
Award.Awards.Select((v, i) => new { Value = v, Index = i / 10 })
.GroupBy(x => x.Index).ToList()
.ForEach(x => awardPages.Add(String.Join("\n", x.Select(z => z.Value.AwardText + " (ID: " + z.Value.AwardId + ")"))));

await PagedReplyAsync(awardPages);
}

[MinPermissions(PermissionLevel.ServerAdmin)]
[Command("deleteaward", RunMode = RunMode.Async)]
public async Task DeleteUserAward(string id = null)
{
if (id == null)
{
EmbedBuilder eb = new EmbedBuilder
{
Title = "Invalid Syntax",
Description = "**Syntax:** " + Guild.Load(Context.Guild.Id).Prefix + "deleteaward [id]",
Color = new Color(210, 47, 33)
}
.WithCurrentTimestamp();

await ReplyAsync("", false, eb.Build());
return;
}

if (int.TryParse(id, out var aId))
{
Award award;

try
{
award = Award.GetAward(aId);
}
catch (AwardNotFoundException exception)
{
await ReplyAsync(Context.User.Mention + ", I couldn't find an award with that ID. Please try again.");
await new LogMessage(LogSeverity.Warning, "AwardModule", exception.Message).PrintToConsole();
return;
}

AdminLog.Log(Context.User.Id, Context.Message.Content, Context.Guild.Id);

EmbedBuilder awardEmbed = new EmbedBuilder()
{
Title = "Are you sure you wish to delete Award ID: " + award.AwardId,
Description = "You're about to delete the following award:",
Color = new Color(255, 0, 0),
Footer = new EmbedFooterBuilder()
{
Text = "Responding with anything but \"Confirm\" will cancel the operation."
}
}
.AddField("Award Category", award.AwardCategory)
.AddField("Award", award.AwardText)
.WithCurrentTimestamp();

try
{
awardEmbed.AddField("Awarded To", award.UserId.GetUser().Mention);
}
catch (UserNotFoundException exception)
{
awardEmbed.AddField("Awarded To", award.UserId.ToString());
await new LogMessage(LogSeverity.Warning, "AwardModule", exception.Message).PrintToConsole();
}

awardEmbed
.AddField("Date Awarded", award.DateAwarded.ToLongDateString())
.AddField("Confirmation",
"If you wish to delete this award, please type \"Confirm\" into the chat, or type \"Cancel\" if you wish to cancel");

await ReplyAsync("", false, awardEmbed.Build());

var response = await NextMessageAsync();
if (response != null)
{
if (response.Content.ToUpper() == "CONFIRM")
{
AdminLog.Log(Context.User.Id, Context.Message.Content + " confirm", Context.Guild.Id);

Award.DeleteAward(award.AwardId);

await ReplyAsync("Award ID: " + award.AwardId + " has been deleted.");
}
else if (response.Content.ToUpper() == "CANCEL")
{
await ReplyAsync(Context.User.Mention + ", the operation was cancelled successfully.");
}
else
{
await ReplyAsync(Context.User.Mention + ", we've cancelled the deleteaward operation as none of the keywords were mentioned.");
}
}
else
{
await ReplyAsync(Context.User.Mention + ", you didn't reply before the timeout, therefore the operation has been automatically cancelled.");
}
}
else
{
EmbedBuilder eb = new EmbedBuilder
{
Title = "Invalid Syntax",
Description = "**Syntax:** " + Guild.Load(Context.Guild.Id).Prefix + "deleteaward [id]",
Color = new Color(210, 47, 33),
Footer = new EmbedFooterBuilder()
{
Text = "You can use the command listawards to find the ID's of the quotes."
}
}
.WithCurrentTimestamp();

await ReplyAsync("", false, eb.Build());
}
}
}
}

0 comments on commit 967c46c

Please sign in to comment.