From 2b1581582523bb105203f96823a45380425bb41c Mon Sep 17 00:00:00 2001
From: spiral <spiral@spiral.sh>
Date: Fri, 2 Jul 2021 06:35:25 -0400
Subject: [PATCH] listen for text response in YesNoPrompt interactive

---
 .../Rest/Types/Requests/MessageEditRequest.cs |  3 +
 PluralKit.Bot/Interactive/BaseInteractive.cs  | 27 +++++---
 PluralKit.Bot/Interactive/YesNoPrompt.cs      | 68 ++++++++++++++++++-
 3 files changed, 88 insertions(+), 10 deletions(-)

diff --git a/Myriad/Rest/Types/Requests/MessageEditRequest.cs b/Myriad/Rest/Types/Requests/MessageEditRequest.cs
index bf217c83..3e74f7f9 100644
--- a/Myriad/Rest/Types/Requests/MessageEditRequest.cs
+++ b/Myriad/Rest/Types/Requests/MessageEditRequest.cs
@@ -18,5 +18,8 @@ namespace Myriad.Rest.Types.Requests
         
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingDefault)]
         public Optional<AllowedMentions> AllowedMentions { get; init; }
+
+        [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingDefault)]
+        public Optional<MessageComponent[]?> Components { get; init; }
     }
 }
\ No newline at end of file
diff --git a/PluralKit.Bot/Interactive/BaseInteractive.cs b/PluralKit.Bot/Interactive/BaseInteractive.cs
index 36eb9f57..e5917887 100644
--- a/PluralKit.Bot/Interactive/BaseInteractive.cs
+++ b/PluralKit.Bot/Interactive/BaseInteractive.cs
@@ -6,6 +6,7 @@ using System.Threading.Tasks;
 
 using Autofac;
 
+using Myriad.Rest.Types;
 using Myriad.Rest.Types.Requests;
 using Myriad.Types;
 
@@ -15,10 +16,11 @@ namespace PluralKit.Bot.Interactive
 {
     public abstract class BaseInteractive
     {
-        private readonly Context _ctx;
-        private readonly List<Button> _buttons = new();
-        private readonly TaskCompletionSource _tcs = new();
-        private bool _running;
+        protected readonly Context _ctx;
+        protected readonly List<Button> _buttons = new();
+        protected readonly TaskCompletionSource _tcs = new();
+        protected Message _message { get; private set; }
+        protected bool _running;
         
         protected BaseInteractive(Context ctx)
         {
@@ -52,21 +54,28 @@ namespace PluralKit.Bot.Interactive
                 });
         }
 
-        protected async Task Finish(InteractionContext ctx)
+        protected async Task Finish(InteractionContext? ctx = null)
         {
             foreach (var button in _buttons) 
                 button.Disabled = true;
-            await Update(ctx);
+
+            if (ctx != null)
+                await Update(ctx);
+            else 
+                await _ctx.Rest.EditMessage(_message.ChannelId, _message.Id, new MessageEditRequest {
+                    Components = GetComponents()
+                });
             
             _tcs.TrySetResult();
         }
 
-        protected async Task<Message> Send(string? content = null, Embed? embed = null)
+        protected async Task Send(string? content = null, Embed? embed = null, AllowedMentions? mentions = null)
         {
-            return await _ctx.Rest.CreateMessage(_ctx.Channel.Id, new MessageRequest
+            _message = await _ctx.Rest.CreateMessage(_ctx.Channel.Id, new MessageRequest
             {
                 Content = content,
                 Embed = embed,
+                AllowedMentions = mentions,
                 Components = GetComponents()
             });
         }
@@ -113,7 +122,7 @@ namespace PluralKit.Bot.Interactive
             }
         }
 
-        private void Cleanup()
+        protected void Cleanup()
         {
             var dispatch = _ctx.Services.Resolve<InteractionDispatchService>();
             foreach (var button in _buttons) 
diff --git a/PluralKit.Bot/Interactive/YesNoPrompt.cs b/PluralKit.Bot/Interactive/YesNoPrompt.cs
index 1d90d4e1..8530dcd1 100644
--- a/PluralKit.Bot/Interactive/YesNoPrompt.cs
+++ b/PluralKit.Bot/Interactive/YesNoPrompt.cs
@@ -1,7 +1,15 @@
+using System;
+using System.Threading;
 using System.Threading.Tasks;
 
+using Myriad.Gateway;
+using Myriad.Rest.Types;
 using Myriad.Types;
 
+using PluralKit.Core;
+
+using Autofac;
+
 namespace PluralKit.Bot.Interactive
 {
     public class YesNoPrompt: BaseInteractive
@@ -20,7 +28,13 @@ namespace PluralKit.Bot.Interactive
         {
             AddButton(ctx => OnButtonClick(ctx, true), AcceptLabel, AcceptStyle);
             AddButton(ctx => OnButtonClick(ctx, false), CancelLabel, CancelStyle);
-            await Send(Message);
+
+            AllowedMentions mentions = null;
+
+            if (User != _ctx.Author.Id)
+                mentions = new AllowedMentions {Users = new[] {User!.Value}};
+
+            await Send(Message, mentions: mentions);
         }
 
         private async Task OnButtonClick(InteractionContext ctx, bool result)
@@ -35,6 +49,58 @@ namespace PluralKit.Bot.Interactive
             await Finish(ctx);
         }
 
+        private bool MessagePredicate(MessageCreateEvent e)
+        {
+            if (e.ChannelId != _ctx.Channel.Id) return false;
+            if (e.Author.Id != User) return false;
+
+            var response = e.Content.ToLowerInvariant();
+
+            if (response == "y" || response == "yes")
+            {
+                Result = true;
+                return true;
+            }
+
+            if (response == "n" || response == "no")
+            {
+                Result = false;
+                return true;
+            }
+
+            return false;
+        }
+
+        public new async Task Run()
+        {
+            // todo: can we split this up somehow so it doesn't need to be *completely* copied from BaseInteractive?
+
+            var cts = new CancellationTokenSource(Timeout.ToTimeSpan());
+
+            if (_running)
+                throw new InvalidOperationException("Action is already running");
+            _running = true;
+
+            var queue = _ctx.Services.Resolve<HandlerQueue<MessageCreateEvent>>();
+            
+            var messageDispatch = queue.WaitFor(MessagePredicate, Timeout, cts.Token);
+
+            await Start();
+            
+            cts.Token.Register(() => _tcs.TrySetException(new TimeoutException("Action timed out")));
+
+            try
+            {
+                var doneTask = await Task.WhenAny(_tcs.Task, messageDispatch);
+                if (doneTask == messageDispatch)
+                    await Finish();
+            }
+            finally
+            {
+                Cleanup();
+            }
+        }
+
         public YesNoPrompt(Context ctx): base(ctx)
         {
             User = ctx.Author.Id;