rebuild discord bot to better user interactions

2025-12-12 18:27:24 +01:00 · 2023-09-17 00:05:57 +02:00 · 2023-09-17 00:05:57 +02:00 · 7e90187822
commit 7e90187822
parent 56b27e5ee8
13 changed files with 1409 additions and 7 deletions
--- a/MareSynchronosServer/MareSynchronosServices/Discord/MareWizardModule.Recover.cs
+++ b/MareSynchronosServer/MareSynchronosServices/Discord/MareWizardModule.Recover.cs
@ -0,0 +1,76 @@
+using Discord.Interactions;
+using Discord;
+using MareSynchronosShared.Data;
+using MareSynchronosShared.Models;
+using MareSynchronosShared.Utils;
+using Microsoft.EntityFrameworkCore;
+
+namespace MareSynchronosServices.Discord;
+
+public partial class MareWizardModule
+{
+    [ComponentInteraction("wizard-recover")]
+    public async Task ComponentRecover()
+    {
+        if (!(await ValidateInteraction().ConfigureAwait(false))) return;
+
+        using var mareDb = GetDbContext();
+        EmbedBuilder eb = new();
+        eb.WithColor(Color.Blue);
+        eb.WithTitle("Recover");
+        eb.WithDescription("In case you have lost your secret key you can recover it here." + Environment.NewLine + Environment.NewLine
+            + "Use the selection below to select the user account you want to recover." + Environment.NewLine + Environment.NewLine
+            + "- 1️⃣ is your primary account/UID" + Environment.NewLine
+            + "- 2️⃣ are all your secondary accounts/UIDs" + Environment.NewLine
+            + "If you are using Vanity UIDs the original UID is displayed in the second line of the account selection.");
+        ComponentBuilder cb = new();
+        await AddUserSelection(mareDb, cb, "wizard-recover-select").ConfigureAwait(false);
+        AddHome(cb);
+        await ModifyInteraction(eb, cb).ConfigureAwait(false);
+    }
+
+    [ComponentInteraction("wizard-recover-select")]
+    public async Task SelectionRecovery(string uid)
+    {
+        if (!(await ValidateInteraction().ConfigureAwait(false))) return;
+
+        using var mareDb = GetDbContext();
+        EmbedBuilder eb = new();
+        eb.WithColor(Color.Green);
+        await HandleRecovery(mareDb, eb, uid).ConfigureAwait(false);
+        ComponentBuilder cb = new();
+        AddHome(cb);
+        await ModifyInteraction(eb, cb).ConfigureAwait(false);
+    }
+
+    private async Task HandleRecovery(MareDbContext db, EmbedBuilder embed, string uid)
+    {
+        string computedHash = string.Empty;
+        Auth auth;
+        var previousAuth = await db.Auth.Include(u => u.User).FirstOrDefaultAsync(u => u.UserUID == uid).ConfigureAwait(false);
+        if (previousAuth != null)
+        {
+            db.Auth.Remove(previousAuth);
+        }
+
+        computedHash = StringUtils.Sha256String(StringUtils.GenerateRandomString(64) + DateTime.UtcNow.ToString());
+        auth = new Auth()
+        {
+            HashedKey = StringUtils.Sha256String(computedHash),
+            User = previousAuth.User,
+            PrimaryUserUID = previousAuth.PrimaryUserUID
+        };
+
+        await db.Auth.AddAsync(auth).ConfigureAwait(false);
+
+        embed.WithTitle($"Recovery for {uid} complete");
+        embed.WithDescription("This is your new private secret key. Do not share this private secret key with anyone. **If you lose it, it is irrevocably lost.**"
+                              + Environment.NewLine + Environment.NewLine
+                              + $"**{computedHash}**"
+                              + Environment.NewLine + Environment.NewLine
+                              + "Enter this key in the Mare Synchronos Service Settings and reconnect to the service.");
+
+        await db.Auth.AddAsync(auth).ConfigureAwait(false);
+        await db.SaveChangesAsync().ConfigureAwait(false);
+    }
+}