Auto-pruning of syncshell, added metrics for pruning, return of count of users in fullgroupdto.

2025-12-05 22:01:48 +01:00
parent 63211b2e8b
commit 091bfbbc29
24 changed files with 1620 additions and 42 deletions
--- a/LightlessSyncServer/LightlessSyncServices/Discord/LightlessWizardModule.Recover.cs
+++ b/LightlessSyncServer/LightlessSyncServices/Discord/LightlessWizardModule.Recover.cs
@@ -0,0 +1,90 @@
+using Discord.Interactions;
+using Discord;
+using LightlessSyncShared.Data;
+using LightlessSyncShared.Models;
+using LightlessSyncShared.Utils;
+using Microsoft.EntityFrameworkCore;
+
+namespace LightlessSyncServices.Discord;
+
+public partial class LightlessWizardModule
+{
+    [ComponentInteraction("wizard-recover")]
+    public async Task ComponentRecover()
+    {
+        if (!(await ValidateInteraction().ConfigureAwait(false))) return;
+
+        _logger.LogInformation("{method}:{userId}", nameof(ComponentRecover), Context.Interaction.User.Id);
+
+        using var lightlessDb = await GetDbContext().ConfigureAwait(false);
+        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
+            + "## ⚠️ **Once you recover your key, the previously used key will be invalidated. If you use Lightless on multiple devices you will have to update the key everywhere you use it.** ⚠️" + 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." + Environment.NewLine
+            + "# Note: instead of recovery and handling secret keys the switch to OAuth2 authentication is strongly suggested.");
+        ComponentBuilder cb = new();
+        await AddUserSelection(lightlessDb, 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;
+
+        _logger.LogInformation("{method}:{userId}:{uid}", nameof(SelectionRecovery), Context.Interaction.User.Id, uid);
+
+        using var lightlessDb = await GetDbContext().ConfigureAwait(false);
+        EmbedBuilder eb = new();
+        eb.WithColor(Color.Green);
+        await HandleRecovery(lightlessDb, eb, uid).ConfigureAwait(false);
+        ComponentBuilder cb = new();
+        AddHome(cb);
+        await ModifyInteraction(eb, cb).ConfigureAwait(false);
+    }
+
+    private async Task HandleRecovery(LightlessDbContext 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());
+        string hashedKey = StringUtils.Sha256String(computedHash);
+        auth = new Auth()
+        {
+            HashedKey = hashedKey,
+            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
+                              + "**__NOTE: Secret keys are considered legacy authentication. If you are using the suggested OAuth2 authentication, you do not need to use the Secret Key or recover ever again.__**"
+                              + Environment.NewLine + Environment.NewLine
+                              + $"||**`{computedHash}`**||"
+                              + Environment.NewLine
+                              + "__NOTE: The Secret Key only contains the letters ABCDEF and numbers 0 - 9.__"
+                              + Environment.NewLine + Environment.NewLine
+                              + "Enter this key in the Lightless Sync Service Settings and reconnect to the service.");
+
+        await db.Auth.AddAsync(auth).ConfigureAwait(false);
+        await db.SaveChangesAsync().ConfigureAwait(false);
+
+        _botServices.Logger.LogInformation("User recovered: {userUID}:{hashedKey}", previousAuth.UserUID, hashedKey);
+        await _botServices.LogToChannel($"{Context.User.Mention} RECOVER SUCCESS: {previousAuth.UserUID}").ConfigureAwait(false);
+    }
+}