333 lines
12 KiB
C#
333 lines
12 KiB
C#
using DSharpPlus;
|
|
using DSharpPlus.CommandsNext;
|
|
using DSharpPlus.CommandsNext.Attributes;
|
|
using DSharpPlus.Entities;
|
|
using Newtonsoft.Json;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Text.RegularExpressions;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace StarsiegeBot
|
|
{
|
|
[Aliases("qc"), Group("quickchat")]
|
|
class QuickchatHandler : BaseCommandModule
|
|
{
|
|
private Dictionary<string, Quickchat> quickchats;
|
|
private bool IsEnabled;
|
|
private readonly string fileName = "Json/quickchats.json";
|
|
|
|
|
|
public QuickchatHandler()
|
|
{
|
|
Console.WriteLine("Quick Chat Handler Loaded");
|
|
IsEnabled = LoadQuickChatsImpl();
|
|
}
|
|
|
|
/**
|
|
* LoadQuickChatsImpl loads the quickchats.json file from the current path,
|
|
* parses the file, and populates the private quickchats dictionary.
|
|
**/
|
|
private bool LoadQuickChatsImpl()
|
|
{
|
|
// Check to see if the file exists...
|
|
if (File.Exists(fileName))
|
|
{
|
|
// Load the JSON file.
|
|
var json = "";
|
|
using (var fs = File.OpenRead(fileName))
|
|
using (var sr = new StreamReader(fs, new UTF8Encoding(false)))
|
|
json = sr.ReadToEnd();
|
|
quickchats = JsonConvert.DeserializeObject<Dictionary<string, Quickchat>>(json);
|
|
// enable the commands.
|
|
return true;
|
|
}
|
|
|
|
// else, file not found. disable commands, leave a comment in console.
|
|
Console.WriteLine(" -- --- --- Quick Chat JSON file not found.");
|
|
return false;
|
|
}
|
|
|
|
[Command("load"), RequireOwner]
|
|
[Description("[Owner Only] Reloads the QuickChat files, if they exist. Also enables QC's if the file is found.")]
|
|
public async Task LoadQuickChats(CommandContext ctx)
|
|
{
|
|
await ctx.TriggerTypingAsync();
|
|
string output;
|
|
IsEnabled = LoadQuickChatsImpl();
|
|
if (IsEnabled)
|
|
{
|
|
output = "Loading Quick Chats successful.";
|
|
}
|
|
else
|
|
{
|
|
output = "Loading Quick Chats failed. File does not exist.";
|
|
}
|
|
await ctx.RespondAsync(StartEmbed(output));
|
|
}
|
|
|
|
[Command("script")]
|
|
[Description("Gives the `say()` for use in the Quickchat.cs file.")]
|
|
public async Task GetQuickChatScript(CommandContext ctx, [Description("ID of Quick chat to get script for.")] int index)
|
|
{
|
|
// If we're disabled, let them know. And exit out.
|
|
if (!IsEnabled)
|
|
{
|
|
return;
|
|
}
|
|
// trigger some typing on discord's side.
|
|
await ctx.TriggerTypingAsync();
|
|
|
|
// We're good to go... If we have an index, pull it up.
|
|
index = Math.Abs(index);
|
|
|
|
// is our index within bounds?
|
|
if (index < quickchats.Count)
|
|
{
|
|
// yes it is, give the script to them.
|
|
Quickchat chat = quickchats[index.ToString()];
|
|
await ctx.RespondAsync("```" + chat.ToString() + "```");
|
|
return;
|
|
}
|
|
|
|
// we couldn't find it... Oh well, let the end user know.
|
|
await ctx.RespondAsync("The ID specified is out of bounds. 0-" + (quickchats.Count - 1));
|
|
}
|
|
|
|
[Command("check")]
|
|
[Description("Runs a check to see how many WAV files are missing from all QC's.")]
|
|
public async Task QuickChatCheck(CommandContext ctx)
|
|
{
|
|
// If we're disabled, let them know. And exit out.
|
|
if (!IsEnabled)
|
|
{
|
|
return;
|
|
}
|
|
await ctx.TriggerTypingAsync();
|
|
|
|
// Start a count
|
|
int missingFiles = 0;
|
|
for (int i = 0; i < quickchats.Count; i++)
|
|
{
|
|
// If this file doesnt exist, log the ID in the console, and add to the count.
|
|
if (!File.Exists($"./qc/{quickchats[i.ToString()].SoundFile}"))
|
|
{
|
|
missingFiles++;
|
|
Console.Write($"{i}, ");
|
|
}
|
|
}
|
|
|
|
// Reply to the message and let the user know we're missing X files.
|
|
await ctx.RespondAsync(StartEmbed($"Missing files: {missingFiles}"));
|
|
}
|
|
|
|
[GroupCommand]
|
|
[Description("Gives a specific or random Quickchat. If that QC has a sound file that the bot knows, uploads the sound file too.")]
|
|
public async Task QuickChat(CommandContext ctx, [Description("The ID of the quick Chat.")] int id = -1)
|
|
{
|
|
// If we're disabled, let them know. And exit out.
|
|
if (!IsEnabled)
|
|
{
|
|
return;
|
|
}
|
|
await ctx.TriggerTypingAsync();
|
|
string output;
|
|
|
|
|
|
// Start a new Message Build.
|
|
DiscordMessageBuilder msg = new DiscordMessageBuilder();
|
|
Quickchat chat;
|
|
|
|
// try to get the quick chat, if it doesn't exist, return a random one
|
|
try
|
|
{
|
|
if (id < -1)
|
|
{
|
|
id = Math.Abs(id);
|
|
}
|
|
chat = quickchats[id.ToString()];
|
|
output = id + ": " + chat.Text;
|
|
}
|
|
catch (Exception)
|
|
{
|
|
// find a better quick chat.
|
|
while (true)
|
|
{
|
|
try
|
|
{
|
|
id = Program.rnd.Next(quickchats.Count);
|
|
chat = quickchats[id.ToString()];
|
|
output = $"[{id}] " + chat.Text;
|
|
}
|
|
catch (Exception)
|
|
{
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
// If the sound file to the select QC exists, upload it as well.
|
|
msg.Embed = StartEmbed(output);
|
|
if (File.Exists($"./qc/{chat.SoundFile}"))
|
|
{
|
|
// Open the file.
|
|
FileStream sound = new FileStream($"./qc/{chat.SoundFile}", FileMode.Open, FileAccess.Read);
|
|
|
|
// feed it to the Message Builder.
|
|
msg.WithFile(chat.SoundFile, sound);
|
|
}
|
|
|
|
// Send either the text QC or sound file if we have it
|
|
// msg.Content = output;
|
|
await ctx.RespondAsync(msg);
|
|
}
|
|
|
|
[Command("search"), Aliases("s")]
|
|
[Cooldown(1, 5, CooldownBucketType.Global)]
|
|
[Description("Searches all the [T]ext and [S]ound file names for the Search input.")]
|
|
public async Task SearchQuickChats(CommandContext ctx, [RemainingText, Description("What to search for.")] string toSearch)
|
|
{
|
|
await ctx.TriggerTypingAsync();
|
|
string output;
|
|
|
|
// If we're disabled, let them know. And exit out.
|
|
if (!IsEnabled)
|
|
{
|
|
await ctx.RespondAsync("Quick Chat Commands have been disabled. Please contact the bot owners.");
|
|
return;
|
|
}
|
|
|
|
// We're going to build a message.
|
|
StringBuilder sb = new StringBuilder();
|
|
DiscordMessageBuilder msg = new DiscordMessageBuilder();
|
|
int count = 0;
|
|
|
|
// We're going to search each QC for the stuff in toSearch.
|
|
foreach (KeyValuePair<string, Quickchat> qc in quickchats)
|
|
{
|
|
// do we have a match?
|
|
if (qc.Value.Text.ToLower().Contains(toSearch.ToLower()))
|
|
{
|
|
// add it to the list!
|
|
sb.Append($"[{qc.Key}][T] {quickchats[qc.Key].Text}\r\n");
|
|
count++;
|
|
}
|
|
if (qc.Value.SoundFile.ToLower().Contains(toSearch.ToLower()))
|
|
{
|
|
// add it to the list!
|
|
sb.Append($"[{qc.Key}][S] {quickchats[qc.Key].Text}\r\n");
|
|
count++;
|
|
}
|
|
}
|
|
|
|
// If we have a message return it
|
|
if (sb.Length > 0)
|
|
{
|
|
output = $"{ctx.Member.Mention}, there were {count} total matches for `{toSearch}`.";
|
|
MemoryStream mr = new MemoryStream(Encoding.UTF8.GetBytes(sb.ToString()));
|
|
msg.WithFile($"{ctx.Member.DisplayName}_results.txt", mr, true);
|
|
}
|
|
else
|
|
{
|
|
// If message content is still nothing, report we found nothing.
|
|
output = $"Sorry {ctx.Member.Mention}, no results were found for `{toSearch}`.";
|
|
}
|
|
|
|
msg.Content = output;
|
|
await ctx.RespondAsync(msg);
|
|
}
|
|
|
|
[Command("flag")]
|
|
[Description("Flags a Quick Chat for moderator checking.")]
|
|
public async Task FlagQC(CommandContext ctx, [Description("ID of QC to flag.")] int id, [RemainingText, Description("The reason you believe it needs flagged.")] string flagRerason = "")
|
|
{
|
|
await ctx.TriggerTypingAsync();
|
|
Quickchat chat = quickchats[id.ToString()];
|
|
DiscordEmbedBuilder embed;
|
|
chat.FlagCount++;
|
|
embed = StartEmbed($"Flagging QC ID `{id}` with reason of `{flagRerason}`. (F{chat.FlagCount})");
|
|
chat.IsFlagged = true;
|
|
string output = JsonConvert.SerializeObject(quickchats);
|
|
await File.WriteAllTextAsync(fileName, output);
|
|
await ctx.RespondAsync(embed);
|
|
}
|
|
|
|
[Command("flag-count")]
|
|
public async Task FlagCount(CommandContext ctx)
|
|
{
|
|
await ctx.TriggerTypingAsync();
|
|
int count = 0;
|
|
foreach (var chat in quickchats)
|
|
{
|
|
if (chat.Value.IsFlagged)
|
|
{
|
|
count++;
|
|
}
|
|
}
|
|
await ctx.RespondAsync(StartEmbed($"There are {count} flagged Quick Chats."));
|
|
}
|
|
|
|
|
|
[Command("toggle"), Aliases("t")]
|
|
[RequireOwner]
|
|
[Description("[Owner only] Enables or disables, or checks status of QuickChats.\r\nTo turn the feature on, use `TRUE`, `ON`, or `1`\r\nTo turn off, use `FALSE`, `OFF`, or `0`")]
|
|
public async Task ToggleQC(CommandContext ctx, [RemainingText] string isEnabled = null)
|
|
{
|
|
await ctx.TriggerTypingAsync();
|
|
isEnabled = isEnabled.ToLower();
|
|
string output;
|
|
string[] turnOn = { "on", "true", "1" };
|
|
string[] turnOff = { "off", "false", "0" };
|
|
|
|
if (File.Exists(fileName))
|
|
{
|
|
if (turnOn.Contains(isEnabled))
|
|
{
|
|
IsEnabled = true;
|
|
}
|
|
if (turnOff.Contains(isEnabled))
|
|
{
|
|
IsEnabled = false;
|
|
}
|
|
|
|
output = $"Quickchats {(IsEnabled ? "Enabled" : "Disabled")}";
|
|
}
|
|
else
|
|
{
|
|
IsEnabled = false;
|
|
output = $"{fileName} file is missing, and it can not be enabled.";
|
|
}
|
|
await ctx.RespondAsync(StartEmbed(output));
|
|
}
|
|
|
|
private DiscordEmbedBuilder StartEmbed(string description)
|
|
{
|
|
DiscordEmbedBuilder embed = new DiscordEmbedBuilder
|
|
{
|
|
Description = description,
|
|
Color = Program.colours[Program.rnd.Next(0, Program.colours.Length)]
|
|
};
|
|
return embed;
|
|
}
|
|
}
|
|
|
|
public class Quickchat
|
|
{
|
|
[JsonProperty("text")]
|
|
public string Text { get; protected set; }
|
|
[JsonProperty("soundFile")]
|
|
public string SoundFile { get; protected set; }
|
|
public bool IsFlagged { get; set; }
|
|
public int FlagCount { get; set; }
|
|
public override string ToString()
|
|
{
|
|
return $"say(0,0, \"{Text}\", \"{SoundFile}\");";
|
|
}
|
|
}
|
|
|
|
}
|