329 lines
15 KiB
C#
329 lines
15 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Discord;
|
|
using Humanizer;
|
|
using NodaTime;
|
|
using NodaTime.Text;
|
|
using NodaTime.TimeZones;
|
|
|
|
using PluralKit.Bot.CommandSystem;
|
|
using PluralKit.Core;
|
|
|
|
namespace PluralKit.Bot.Commands
|
|
{
|
|
public class SystemCommands
|
|
{
|
|
private IDataStore _data;
|
|
private EmbedService _embeds;
|
|
|
|
private ProxyCacheService _proxyCache;
|
|
|
|
public SystemCommands(EmbedService embeds, ProxyCacheService proxyCache, IDataStore data)
|
|
{
|
|
_embeds = embeds;
|
|
_proxyCache = proxyCache;
|
|
_data = data;
|
|
}
|
|
|
|
public async Task Query(Context ctx, PKSystem system) {
|
|
if (system == null) throw Errors.NoSystemError;
|
|
|
|
await ctx.Reply(embed: await _embeds.CreateSystemEmbed(system));
|
|
}
|
|
|
|
public async Task New(Context ctx)
|
|
{
|
|
ctx.CheckNoSystem();
|
|
|
|
var system = await _data.CreateSystem(ctx.RemainderOrNull());
|
|
await _data.AddAccount(system, ctx.Author.Id);
|
|
await ctx.Reply($"{Emojis.Success} Your system has been created. Type `pk;system` to view it, and type `pk;help` for more information about commands you can use now.");
|
|
}
|
|
|
|
public async Task Name(Context ctx)
|
|
{
|
|
ctx.CheckSystem();
|
|
|
|
var newSystemName = ctx.RemainderOrNull();
|
|
if (newSystemName != null && newSystemName.Length > Limits.MaxSystemNameLength) throw Errors.SystemNameTooLongError(newSystemName.Length);
|
|
|
|
ctx.System.Name = newSystemName;
|
|
await _data.SaveSystem(ctx.System);
|
|
await ctx.Reply($"{Emojis.Success} System name {(newSystemName != null ? "changed" : "cleared")}.");
|
|
}
|
|
|
|
public async Task Description(Context ctx) {
|
|
ctx.CheckSystem();
|
|
|
|
var newDescription = ctx.RemainderOrNull();
|
|
if (newDescription != null && newDescription.Length > Limits.MaxDescriptionLength) throw Errors.DescriptionTooLongError(newDescription.Length);
|
|
|
|
ctx.System.Description = newDescription;
|
|
await _data.SaveSystem(ctx.System);
|
|
await ctx.Reply($"{Emojis.Success} System description {(newDescription != null ? "changed" : "cleared")}.");
|
|
}
|
|
|
|
public async Task Tag(Context ctx)
|
|
{
|
|
ctx.CheckSystem();
|
|
|
|
var newTag = ctx.RemainderOrNull();
|
|
ctx.System.Tag = newTag;
|
|
|
|
if (newTag != null)
|
|
if (newTag.Length > Limits.MaxSystemTagLength)
|
|
throw Errors.SystemNameTooLongError(newTag.Length);
|
|
|
|
await _data.SaveSystem(ctx.System);
|
|
await ctx.Reply($"{Emojis.Success} System tag {(newTag != null ? $"changed. Member names will now end with `{newTag.SanitizeMentions()}` when proxied." : "cleared")}.");
|
|
|
|
await _proxyCache.InvalidateResultsForSystem(ctx.System);
|
|
}
|
|
|
|
public async Task SystemAvatar(Context ctx)
|
|
{
|
|
ctx.CheckSystem();
|
|
|
|
if (ctx.RemainderOrNull() == null && ctx.Message.Attachments.Count == 0)
|
|
{
|
|
if ((ctx.System.AvatarUrl?.Trim() ?? "").Length > 0)
|
|
{
|
|
var eb = new EmbedBuilder()
|
|
.WithTitle($"System avatar")
|
|
.WithImageUrl(ctx.System.AvatarUrl)
|
|
.WithDescription($"To clear, use `pk;system avatar clear`.");
|
|
await ctx.Reply(embed: eb.Build());
|
|
}
|
|
else
|
|
throw new PKSyntaxError($"This system does not have an avatar set. Set one by attaching an image to this command, or by passing an image URL or @mention.");
|
|
|
|
return;
|
|
}
|
|
|
|
var member = await ctx.MatchUser();
|
|
if (member != null)
|
|
{
|
|
if (member.AvatarId == null) throw Errors.UserHasNoAvatar;
|
|
ctx.System.AvatarUrl = member.GetAvatarUrl(ImageFormat.Png, size: 256);
|
|
await _data.SaveSystem(ctx.System);
|
|
|
|
var embed = new EmbedBuilder().WithImageUrl(ctx.System.AvatarUrl).Build();
|
|
await ctx.Reply(
|
|
$"{Emojis.Success} System avatar changed to {member.Username}'s avatar! {Emojis.Warn} Please note that if {member.Username} changes their avatar, the system's avatar will need to be re-set.", embed: embed);
|
|
}
|
|
else if (ctx.Match("clear"))
|
|
{
|
|
ctx.System.AvatarUrl = null;
|
|
await _data.SaveSystem(ctx.System);
|
|
await ctx.Reply($"{Emojis.Success} System avatar cleared.");
|
|
}
|
|
else
|
|
{
|
|
// They can't both be null - otherwise we would've hit the conditional at the very top
|
|
string url = ctx.RemainderOrNull() ?? ctx.Message.Attachments.FirstOrDefault()?.ProxyUrl;
|
|
await ctx.BusyIndicator(() => Utils.VerifyAvatarOrThrow(url));
|
|
|
|
ctx.System.AvatarUrl = url;
|
|
await _data.SaveSystem(ctx.System);
|
|
|
|
var embed = url != null ? new EmbedBuilder().WithImageUrl(url).Build() : null;
|
|
await ctx.Reply($"{Emojis.Success} System avatar changed.", embed: embed);
|
|
}
|
|
|
|
await _proxyCache.InvalidateResultsForSystem(ctx.System);
|
|
}
|
|
|
|
public async Task Delete(Context ctx) {
|
|
ctx.CheckSystem();
|
|
|
|
var msg = await ctx.Reply($"{Emojis.Warn} Are you sure you want to delete your system? If so, reply to this message with your system's ID (`{ctx.System.Hid}`).\n**Note: this action is permanent.**");
|
|
var reply = await ctx.AwaitMessage(ctx.Channel, ctx.Author, timeout: TimeSpan.FromMinutes(1));
|
|
if (reply.Content != ctx.System.Hid) throw new PKError($"System deletion cancelled. Note that you must reply with your system ID (`{ctx.System.Hid}`) *verbatim*.");
|
|
|
|
await _data.DeleteSystem(ctx.System);
|
|
await ctx.Reply($"{Emojis.Success} System deleted.");
|
|
|
|
await _proxyCache.InvalidateResultsForSystem(ctx.System);
|
|
}
|
|
|
|
public async Task MemberShortList(Context ctx, PKSystem system) {
|
|
if (system == null) throw Errors.NoSystemError;
|
|
|
|
var members = await _data.GetSystemMembers(system);
|
|
var embedTitle = system.Name != null ? $"Members of {system.Name.SanitizeMentions()} (`{system.Hid}`)" : $"Members of `{system.Hid}`";
|
|
await ctx.Paginate<PKMember>(
|
|
members.OrderBy(m => m.Name, StringComparer.InvariantCultureIgnoreCase).ToList(),
|
|
25,
|
|
embedTitle,
|
|
(eb, ms) => eb.Description = string.Join("\n", ms.Select((m) => {
|
|
if (m.HasProxyTags) return $"[`{m.Hid}`] **{m.Name.SanitizeMentions()}** *({m.ProxyTagsString().SanitizeMentions()})*";
|
|
return $"[`{m.Hid}`] **{m.Name.SanitizeMentions()}**";
|
|
}))
|
|
);
|
|
}
|
|
|
|
public async Task MemberLongList(Context ctx, PKSystem system) {
|
|
if (system == null) throw Errors.NoSystemError;
|
|
|
|
var members = await _data.GetSystemMembers(system);
|
|
var embedTitle = system.Name != null ? $"Members of {system.Name} (`{system.Hid}`)" : $"Members of `{system.Hid}`";
|
|
await ctx.Paginate<PKMember>(
|
|
members.OrderBy(m => m.Name, StringComparer.InvariantCultureIgnoreCase).ToList(),
|
|
5,
|
|
embedTitle,
|
|
(eb, ms) => {
|
|
foreach (var m in ms) {
|
|
var profile = $"**ID**: {m.Hid}";
|
|
if (m.Pronouns != null) profile += $"\n**Pronouns**: {m.Pronouns}";
|
|
if (m.Birthday != null) profile += $"\n**Birthdate**: {m.BirthdayString}";
|
|
if (m.ProxyTags.Count > 0) profile += $"\n**Proxy tags:** {m.ProxyTagsString()}";
|
|
if (m.Description != null) profile += $"\n\n{m.Description}";
|
|
eb.AddField(m.Name, profile.Truncate(1024));
|
|
}
|
|
}
|
|
);
|
|
}
|
|
|
|
public async Task SystemFronter(Context ctx, PKSystem system)
|
|
{
|
|
if (system == null) throw Errors.NoSystemError;
|
|
|
|
var sw = await _data.GetLatestSwitch(system);
|
|
if (sw == null) throw Errors.NoRegisteredSwitches;
|
|
|
|
await ctx.Reply(embed: await _embeds.CreateFronterEmbed(sw, system.Zone));
|
|
}
|
|
|
|
public async Task SystemFrontHistory(Context ctx, PKSystem system)
|
|
{
|
|
if (system == null) throw Errors.NoSystemError;
|
|
|
|
var sws = (await _data.GetSwitches(system, 10)).ToList();
|
|
if (sws.Count == 0) throw Errors.NoRegisteredSwitches;
|
|
|
|
await ctx.Reply(embed: await _embeds.CreateFrontHistoryEmbed(sws, system.Zone));
|
|
}
|
|
|
|
public async Task SystemFrontPercent(Context ctx, PKSystem system)
|
|
{
|
|
if (system == null) throw Errors.NoSystemError;
|
|
|
|
string durationStr = ctx.RemainderOrNull() ?? "30d";
|
|
|
|
var now = SystemClock.Instance.GetCurrentInstant();
|
|
|
|
var rangeStart = PluralKit.Utils.ParseDateTime(durationStr, true, system.Zone);
|
|
if (rangeStart == null) throw Errors.InvalidDateTime(durationStr);
|
|
if (rangeStart.Value.ToInstant() > now) throw Errors.FrontPercentTimeInFuture;
|
|
|
|
var frontpercent = await _data.GetFrontBreakdown(system, rangeStart.Value.ToInstant(), now);
|
|
await ctx.Reply(embed: await _embeds.CreateFrontPercentEmbed(frontpercent, system.Zone));
|
|
}
|
|
|
|
public async Task SystemProxy(Context ctx)
|
|
{
|
|
ctx.CheckSystem().CheckGuildContext();
|
|
var gs = await _data.GetSystemGuildSettings(ctx.System, ctx.Guild.Id);
|
|
|
|
bool newValue;
|
|
if (ctx.Match("on", "enabled", "true", "yes")) newValue = true;
|
|
else if (ctx.Match("off", "disabled", "false", "no")) newValue = false;
|
|
else if (ctx.HasNext()) throw new PKSyntaxError("You must pass either \"on\" or \"off\".");
|
|
else newValue = !gs.ProxyEnabled;
|
|
|
|
gs.ProxyEnabled = newValue;
|
|
await _data.SetSystemGuildSettings(ctx.System, ctx.Guild.Id, gs);
|
|
|
|
if (newValue)
|
|
await ctx.Reply($"Message proxying in this server ({ctx.Guild.Name.EscapeMarkdown()}) is now **enabled** for your system.");
|
|
else
|
|
await ctx.Reply($"Message proxying in this server ({ctx.Guild.Name.EscapeMarkdown()}) is now **disabled** for your system.");
|
|
}
|
|
|
|
public async Task SystemTimezone(Context ctx)
|
|
{
|
|
if (ctx.System == null) throw Errors.NoSystemError;
|
|
|
|
var zoneStr = ctx.RemainderOrNull();
|
|
if (zoneStr == null)
|
|
{
|
|
ctx.System.UiTz = "UTC";
|
|
await _data.SaveSystem(ctx.System);
|
|
await ctx.Reply($"{Emojis.Success} System time zone cleared.");
|
|
return;
|
|
}
|
|
|
|
var zone = await FindTimeZone(ctx, zoneStr);
|
|
if (zone == null) throw Errors.InvalidTimeZone(zoneStr);
|
|
|
|
var currentTime = SystemClock.Instance.GetCurrentInstant().InZone(zone);
|
|
var msg = await ctx.Reply(
|
|
$"This will change the system time zone to {zone.Id}. The current time is {Formats.ZonedDateTimeFormat.Format(currentTime)}. Is this correct?");
|
|
if (!await ctx.PromptYesNo(msg)) throw Errors.TimezoneChangeCancelled;
|
|
ctx.System.UiTz = zone.Id;
|
|
await _data.SaveSystem(ctx.System);
|
|
|
|
await ctx.Reply($"System time zone changed to {zone.Id}.");
|
|
}
|
|
|
|
public async Task<DateTimeZone> FindTimeZone(Context ctx, string zoneStr) {
|
|
// First, if we're given a flag emoji, we extract the flag emoji code from it.
|
|
zoneStr = PluralKit.Utils.ExtractCountryFlag(zoneStr) ?? zoneStr;
|
|
|
|
// Then, we find all *locations* matching either the given country code or the country name.
|
|
var locations = TzdbDateTimeZoneSource.Default.Zone1970Locations;
|
|
var matchingLocations = locations.Where(l => l.Countries.Any(c =>
|
|
string.Equals(c.Code, zoneStr, StringComparison.InvariantCultureIgnoreCase) ||
|
|
string.Equals(c.Name, zoneStr, StringComparison.InvariantCultureIgnoreCase)));
|
|
|
|
// Then, we find all (unique) time zone IDs that match.
|
|
var matchingZones = matchingLocations.Select(l => DateTimeZoneProviders.Tzdb.GetZoneOrNull(l.ZoneId))
|
|
.Distinct().ToList();
|
|
|
|
// If the set of matching zones is empty (ie. we didn't find anything), we try a few other things.
|
|
if (matchingZones.Count == 0)
|
|
{
|
|
// First, we try to just find the time zone given directly and return that.
|
|
var givenZone = DateTimeZoneProviders.Tzdb.GetZoneOrNull(zoneStr);
|
|
if (givenZone != null) return givenZone;
|
|
|
|
// If we didn't find anything there either, we try parsing the string as an offset, then
|
|
// find all possible zones that match that offset. For an offset like UTC+2, this doesn't *quite*
|
|
// work, since there are 57(!) matching zones (as of 2019-06-13) - but for less populated time zones
|
|
// this could work nicely.
|
|
var inputWithoutUtc = zoneStr.Replace("UTC", "").Replace("GMT", "");
|
|
|
|
var res = OffsetPattern.CreateWithInvariantCulture("+H").Parse(inputWithoutUtc);
|
|
if (!res.Success) res = OffsetPattern.CreateWithInvariantCulture("+H:mm").Parse(inputWithoutUtc);
|
|
|
|
// If *this* didn't parse correctly, fuck it, bail.
|
|
if (!res.Success) return null;
|
|
var offset = res.Value;
|
|
|
|
// To try to reduce the count, we go by locations from the 1970+ database instead of just the full database
|
|
// This elides regions that have been identical since 1970, omitting small distinctions due to Ancient History(tm).
|
|
var allZones = TzdbDateTimeZoneSource.Default.Zone1970Locations.Select(l => l.ZoneId).Distinct();
|
|
matchingZones = allZones.Select(z => DateTimeZoneProviders.Tzdb.GetZoneOrNull(z))
|
|
.Where(z => z.GetUtcOffset(SystemClock.Instance.GetCurrentInstant()) == offset).ToList();
|
|
}
|
|
|
|
// If we have a list of viable time zones, we ask the user which is correct.
|
|
|
|
// If we only have one, return that one.
|
|
if (matchingZones.Count == 1)
|
|
return matchingZones.First();
|
|
|
|
// Otherwise, prompt and return!
|
|
return await ctx.Choose("There were multiple matches for your time zone query. Please select the region that matches you the closest:", matchingZones,
|
|
z =>
|
|
{
|
|
if (TzdbDateTimeZoneSource.Default.Aliases.Contains(z.Id))
|
|
return $"**{z.Id}**, {string.Join(", ", TzdbDateTimeZoneSource.Default.Aliases[z.Id])}";
|
|
|
|
return $"**{z.Id}**";
|
|
});
|
|
}
|
|
}
|
|
}
|