Aliases now support %target% placeholder. For example 'alias .bft .bf %target% t'

src/NadekoBot/Modules/Utility/Alias/AliasCommands.cs

@@ -0,0 +1,137 @@
+#nullable disable
+using Microsoft.EntityFrameworkCore;
+using NadekoBot.Db;
+using NadekoBot.Modules.Utility.Services;
+using NadekoBot.Services.Database.Models;
+
+namespace NadekoBot.Modules.Utility;
+
+public partial class Utility
+{
+    [Group]
+    public partial class CommandMapCommands : NadekoModule<AliasService>
+    {
+        private readonly DbService _db;
+        private readonly DiscordSocketClient _client;
+
+        public CommandMapCommands(DbService db, DiscordSocketClient client)
+        {
+            _db = db;
+            _client = client;
+        }
+
+        [Cmd]
+        [RequireContext(ContextType.Guild)]
+        [UserPerm(GuildPerm.Administrator)]
+        public async Task AliasesClear()
+        {
+            var count = _service.ClearAliases(ctx.Guild.Id);
+            await ReplyConfirmLocalizedAsync(strs.aliases_cleared(count));
+        }
+
+        [Cmd]
+        [UserPerm(GuildPerm.Administrator)]
+        [RequireContext(ContextType.Guild)]
+        public async Task Alias(string trigger, [Leftover] string mapping = null)
+        {
+            if (string.IsNullOrWhiteSpace(trigger))
+                return;
+
+            trigger = trigger.Trim().ToLowerInvariant();
+
+            if (string.IsNullOrWhiteSpace(mapping))
+            {
+                if (!_service.AliasMaps.TryGetValue(ctx.Guild.Id, out var maps) || !maps.TryRemove(trigger, out _))
+                {
+                    await ReplyErrorLocalizedAsync(strs.alias_remove_fail(Format.Code(trigger)));
+                    return;
+                }
+
+                await using (var uow = _db.GetDbContext())
+                {
+                    var config = uow.GuildConfigsForId(ctx.Guild.Id, set => set.Include(x => x.CommandAliases));
+                    var tr = config.CommandAliases.FirstOrDefault(x => x.Trigger == trigger);
+                    if (tr is not null)
+                        uow.Set<CommandAlias>().Remove(tr);
+                    uow.SaveChanges();
+                }
+
+                await ReplyConfirmLocalizedAsync(strs.alias_removed(Format.Code(trigger)));
+                return;
+            }
+
+            _service.AliasMaps.AddOrUpdate(ctx.Guild.Id,
+                _ =>
+                {
+                    using (var uow = _db.GetDbContext())
+                    {
+                        var config = uow.GuildConfigsForId(ctx.Guild.Id, set => set.Include(x => x.CommandAliases));
+                        config.CommandAliases.Add(new()
+                        {
+                            Mapping = mapping,
+                            Trigger = trigger
+                        });
+                        uow.SaveChanges();
+                    }
+
+                    return new(new Dictionary<string, string>
+                    {
+                        { trigger.Trim().ToLowerInvariant(), mapping.ToLowerInvariant() }
+                    });
+                },
+                (_, map) =>
+                {
+                    using (var uow = _db.GetDbContext())
+                    {
+                        var config = uow.GuildConfigsForId(ctx.Guild.Id, set => set.Include(x => x.CommandAliases));
+                        var toAdd = new CommandAlias
+                        {
+                            Mapping = mapping,
+                            Trigger = trigger
+                        };
+                        var toRemove = config.CommandAliases.Where(x => x.Trigger == trigger).ToArray();
+                        if (toRemove.Any())
+                            uow.RemoveRange(toRemove);
+                        config.CommandAliases.Add(toAdd);
+                        uow.SaveChanges();
+                    }
+
+                    map.AddOrUpdate(trigger, mapping, (_, _) => mapping);
+                    return map;
+                });
+
+            await ReplyConfirmLocalizedAsync(strs.alias_added(Format.Code(trigger), Format.Code(mapping)));
+        }
+
+
+        [Cmd]
+        [RequireContext(ContextType.Guild)]
+        public async Task AliasList(int page = 1)
+        {
+            page -= 1;
+
+            if (page < 0)
+                return;
+
+            if (!_service.AliasMaps.TryGetValue(ctx.Guild.Id, out var maps) || !maps.Any())
+            {
+                await ReplyErrorLocalizedAsync(strs.aliases_none);
+                return;
+            }
+
+            var arr = maps.ToArray();
+
+            await ctx.SendPaginatedConfirmAsync(page,
+                curPage =>
+                {
+                    return _eb.Create()
+                              .WithOkColor()
+                              .WithTitle(GetText(strs.alias_list))
+                              .WithDescription(string.Join("\n",
+                                  arr.Skip(curPage * 10).Take(10).Select(x => $"`{x.Key}` => `{x.Value}`")));
+                },
+                arr.Length,
+                10);
+        }
+    }
+}