Move command functions around

2020-02-01 13:03:02 +01:00
parent a60be64551
commit 125ea81ec3
20 changed files with 950 additions and 807 deletions
--- a/PluralKit.Bot/Commands/SystemFront.cs
+++ b/PluralKit.Bot/Commands/SystemFront.cs
@@ -0,0 +1,118 @@
+using System.Linq;
+using System.Threading.Tasks;
+
+using Discord;
+
+using NodaTime;
+
+using PluralKit.Bot.CommandSystem;
+
+namespace PluralKit.Bot.Commands
+{
+    public class SystemFront
+    {
+        private IDataStore _data;
+        private EmbedService _embeds;
+
+        public SystemFront(IDataStore data, EmbedService embeds)
+        {
+            _data = data;
+            _embeds = embeds;
+        }
+        
+        struct FrontHistoryEntry
+        {
+            public Instant? LastTime;
+            public PKSwitch ThisSwitch;
+
+            public FrontHistoryEntry(Instant? lastTime, PKSwitch thisSwitch)
+            {
+                LastTime = lastTime;
+                ThisSwitch = thisSwitch;
+            }
+        }
+
+        public async Task SystemFronter(Context ctx, PKSystem system)
+        {
+            if (system == null) throw Errors.NoSystemError;
+            ctx.CheckSystemPrivacy(system, system.FrontPrivacy);
+            
+            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;
+            ctx.CheckSystemPrivacy(system, system.FrontHistoryPrivacy);
+
+            var sws = _data.GetSwitches(system)
+                .Scan(new FrontHistoryEntry(null, null), (lastEntry, newSwitch) => new FrontHistoryEntry(lastEntry.ThisSwitch?.Timestamp, newSwitch));
+            var totalSwitches = await _data.GetSwitchCount(system);
+            if (totalSwitches == 0) throw Errors.NoRegisteredSwitches;
+            
+            var embedTitle = system.Name != null ? $"Front history of {system.Name} (`{system.Hid}`)" : $"Front history of `{system.Hid}`";
+
+            await ctx.Paginate(
+                sws,
+                totalSwitches,
+                10,
+                embedTitle,
+                async (builder, switches) =>
+                {
+                    var outputStr = "";
+                    foreach (var entry in switches)
+                    {
+                        var lastSw = entry.LastTime;
+
+                        var sw = entry.ThisSwitch;
+                        // Fetch member list and format
+                        var members = await _data.GetSwitchMembers(sw).ToListAsync();
+                        var membersStr = members.Any() ? string.Join(", ", members.Select(m => m.Name)) : "no fronter";
+
+                        var switchSince = SystemClock.Instance.GetCurrentInstant() - sw.Timestamp;
+
+                        // If this isn't the latest switch, we also show duration
+                        string stringToAdd;
+                        if (lastSw != null)
+                        {
+                            // Calculate the time between the last switch (that we iterated - ie. the next one on the timeline) and the current one
+                            var switchDuration = lastSw.Value - sw.Timestamp;
+                            stringToAdd =
+                                $"**{membersStr}** ({Formats.ZonedDateTimeFormat.Format(sw.Timestamp.InZone(system.Zone))}, {Formats.DurationFormat.Format(switchSince)} ago, for {Formats.DurationFormat.Format(switchDuration)})\n";
+                        }
+                        else
+                        {
+                            stringToAdd =
+                                $"**{membersStr}** ({Formats.ZonedDateTimeFormat.Format(sw.Timestamp.InZone(system.Zone))}, {Formats.DurationFormat.Format(switchSince)} ago)\n";
+                        }
+
+                        if (outputStr.Length + stringToAdd.Length > EmbedBuilder.MaxDescriptionLength) break;
+                        outputStr += stringToAdd;
+                    }
+
+                    builder.Description = outputStr;
+                }
+            );
+        }
+        
+        public async Task SystemFrontPercent(Context ctx, PKSystem system)
+        {
+            if (system == null) throw Errors.NoSystemError;
+            ctx.CheckSystemPrivacy(system, system.FrontHistoryPrivacy);
+
+            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));
+        }
+    }
+}