✨ Auto completion in chat

2025-10-12 16:00:32 +08:00
parent f4a659fce5
commit 9631cd3edd
4 changed files with 159 additions and 3 deletions
--- a/DysonNetwork.Sphere/Autocompletion/AutocompletionService.cs
+++ b/DysonNetwork.Sphere/Autocompletion/AutocompletionService.cs
@@ -0,0 +1,121 @@
+using Microsoft.EntityFrameworkCore;
+
+namespace DysonNetwork.Sphere.Autocompletion;
+
+public class AutocompletionService(AppDatabase db)
+{
+    public async Task<List<DysonNetwork.Shared.Models.Autocompletion>> GetAutocompletion(string content, int limit = 10)
+    {
+        if (string.IsNullOrWhiteSpace(content))
+            return [];
+
+        if (content.StartsWith('@'))
+        {
+            var afterAt = content[1..];
+            string type;
+            string query;
+            if (afterAt.Contains('/'))
+            {
+                var parts = afterAt.Split('/', 2);
+                type = parts[0];
+                query = parts.Length > 1 ? parts[1] : "";
+            }
+            else
+            {
+                type = "u";
+                query = afterAt;
+            }
+            return await AutocompleteAt(type, query, limit);
+        }
+
+        if (!content.StartsWith(':')) return [];
+        {
+            var query = content[1..];
+            return await AutocompleteSticker(query, limit);
+        }
+
+    }
+
+    private async Task<List<DysonNetwork.Shared.Models.Autocompletion>> AutocompleteAt(string type, string query, int limit)
+    {
+        var results = new List<DysonNetwork.Shared.Models.Autocompletion>();
+
+        switch (type)
+        {
+            case "p":
+                var publishers = await db.Publishers
+                    .Where(p => EF.Functions.Like(p.Name, $"{query}%") || EF.Functions.Like(p.Nick, $"{query}%"))
+                    .Take(limit)
+                    .Select(p => new DysonNetwork.Shared.Models.Autocompletion
+                    {
+                        Type = "publisher",
+                        Keyword = p.Name,
+                        Data = p
+                    })
+                    .ToListAsync();
+                results.AddRange(publishers);
+                break;
+
+            case "r":
+                var realms = await db.Realms
+                    .Where(r => EF.Functions.Like(r.Slug, $"{query}%") || EF.Functions.Like(r.Name, $"{query}%"))
+                    .Take(limit)
+                    .Select(r => new DysonNetwork.Shared.Models.Autocompletion
+                    {
+                        Type = "realm",
+                        Keyword = r.Slug,
+                        Data = r
+                    })
+                    .ToListAsync();
+                results.AddRange(realms);
+                break;
+
+            case "c":
+                var chats = await db.ChatRooms
+                    .Where(c => c.Name != null && EF.Functions.Like(c.Name, $"{query}%"))
+                    .Take(limit)
+                    .Select(c => new DysonNetwork.Shared.Models.Autocompletion
+                    {
+                        Type = "chat",
+                        Keyword = c.Name!,
+                        Data = c
+                    })
+                    .ToListAsync();
+                results.AddRange(chats);
+                break;
+        }
+
+        return results;
+    }
+
+    private async Task<List<DysonNetwork.Shared.Models.Autocompletion>> AutocompleteSticker(string query, int limit)
+    {
+        var stickers = await db.Stickers
+            .Include(s => s.Pack)
+            .Where(s => EF.Functions.Like(s.Slug, $"{query}%"))
+            .Take(limit)
+            .Select(s => new DysonNetwork.Shared.Models.Autocompletion
+            {
+                Type = "sticker",
+                Keyword = s.Slug,
+                Data = s
+            })
+            .ToListAsync();
+
+        // Also possibly search by pack prefix? But user said slug after :
+        // Perhaps combine or search packs
+        var packs = await db.StickerPacks
+            .Where(p => EF.Functions.Like(p.Prefix, $"{query}%"))
+            .Take(limit)
+            .Select(p => new DysonNetwork.Shared.Models.Autocompletion
+            {
+                Type = "sticker_pack",
+                Keyword = p.Prefix,
+                Data = p
+            })
+            .ToListAsync();
+
+        var results = stickers.Concat(packs).Take(limit).ToList();
+        return results;
+    }
+}