92 lines
4.3 KiB
C#
92 lines
4.3 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Discord;
|
|
using Discord.Commands;
|
|
using NodaTime;
|
|
using NodaTime.TimeZones;
|
|
|
|
namespace PluralKit.Bot.Commands
|
|
{
|
|
[Group("switch")]
|
|
public class SwitchCommands: ModuleBase<PKCommandContext>
|
|
{
|
|
public SystemStore Systems { get; set; }
|
|
public SwitchStore Switches { get; set; }
|
|
|
|
[Command]
|
|
[Remarks("switch <member> [member...]")]
|
|
[MustHaveSystem]
|
|
public async Task Switch(params PKMember[] members) => await DoSwitchCommand(members);
|
|
|
|
[Command("out")]
|
|
[MustHaveSystem]
|
|
public async Task SwitchOut() => await DoSwitchCommand(new PKMember[] { });
|
|
|
|
private async Task DoSwitchCommand(ICollection<PKMember> members)
|
|
{
|
|
// Make sure there are no dupes in the list
|
|
// We do this by checking if removing duplicate member IDs results in a list of different length
|
|
if (members.Select(m => m.Id).Distinct().Count() != members.Count) throw Errors.DuplicateSwitchMembers;
|
|
|
|
// Find the last switch and its members if applicable
|
|
var lastSwitch = await Switches.GetLatestSwitch(Context.SenderSystem);
|
|
if (lastSwitch != null)
|
|
{
|
|
var lastSwitchMembers = await Switches.GetSwitchMembers(lastSwitch);
|
|
// Make sure the requested switch isn't identical to the last one
|
|
if (lastSwitchMembers.Select(m => m.Id).SequenceEqual(members.Select(m => m.Id)))
|
|
throw Errors.SameSwitch(members);
|
|
}
|
|
|
|
await Switches.RegisterSwitch(Context.SenderSystem, members);
|
|
|
|
if (members.Count == 0)
|
|
await Context.Channel.SendMessageAsync($"{Emojis.Success} Switch-out registered.");
|
|
else
|
|
await Context.Channel.SendMessageAsync($"{Emojis.Success} Switch registered. Current fronter is now {string.Join(", ", members.Select(m => m.Name))}.");
|
|
}
|
|
|
|
[Command("move")]
|
|
[MustHaveSystem]
|
|
public async Task SwitchMove([Remainder] string str)
|
|
{
|
|
var tz = TzdbDateTimeZoneSource.Default.ForId(Context.SenderSystem.UiTz ?? "UTC");
|
|
|
|
var result = PluralKit.Utils.ParseDateTime(str, true, tz);
|
|
if (result == null) throw Errors.InvalidDateTime(str);
|
|
|
|
var time = result.Value;
|
|
if (time.ToInstant() > SystemClock.Instance.GetCurrentInstant()) throw Errors.SwitchTimeInFuture;
|
|
|
|
// Fetch the last two switches for the system to do bounds checking on
|
|
var lastTwoSwitches = (await Switches.GetSwitches(Context.SenderSystem, 2)).ToArray();
|
|
|
|
// If we don't have a switch to move, don't bother
|
|
if (lastTwoSwitches.Length == 0) throw Errors.NoRegisteredSwitches;
|
|
|
|
// If there's a switch *behind* the one we move, we check to make srue we're not moving the time further back than that
|
|
if (lastTwoSwitches.Length == 2)
|
|
{
|
|
if (lastTwoSwitches[1].Timestamp > time.ToInstant())
|
|
throw Errors.SwitchMoveBeforeSecondLast(lastTwoSwitches[1].Timestamp.InZone(tz));
|
|
}
|
|
|
|
// Now we can actually do the move, yay!
|
|
// But, we do a prompt to confirm.
|
|
var lastSwitchMembers = await Switches.GetSwitchMembers(lastTwoSwitches[0]);
|
|
|
|
// yeet
|
|
var msg = await Context.Channel.SendMessageAsync($"{Emojis.Warn} This will move the latest switch ({string.Join(", ", lastSwitchMembers.Select(m => m.Name))}) from {lastTwoSwitches[0].Timestamp.ToString(Formats.DateTimeFormat, null)} ({SystemClock.Instance.GetCurrentInstant().Minus(lastTwoSwitches[0].Timestamp).ToString(Formats.DurationFormat, null)} ago) to {time.ToString(Formats.DateTimeFormat, null)} ({SystemClock.Instance.GetCurrentInstant().Minus(time.ToInstant()).ToString(Formats.DurationFormat, null)} ago). Is this OK?");
|
|
if (!await Context.PromptYesNo(msg))
|
|
{
|
|
throw Errors.SwitchMoveCancelled;
|
|
}
|
|
|
|
// aaaand *now* we do the move
|
|
await Switches.MoveSwitch(lastTwoSwitches[0], time.ToInstant());
|
|
await Context.Channel.SendMessageAsync($"{Emojis.Success} Switch moved.");
|
|
}
|
|
}
|
|
} |