public async Task Proxy(Context ctx, PKMember target) { ctx.CheckSystem().CheckOwnMember(target); ProxyTag ParseProxyTags(string exampleProxy) { // // Make sure there's one and only one instance of "text" in the example proxy given var prefixAndSuffix = exampleProxy.Split("text"); if (prefixAndSuffix.Length == 1) { prefixAndSuffix = prefixAndSuffix[0].Split("TEXT"); } if (prefixAndSuffix.Length < 2) { throw Errors.ProxyMustHaveText; } if (prefixAndSuffix.Length > 2) { throw Errors.ProxyMultipleText; } return(new ProxyTag(prefixAndSuffix[0], prefixAndSuffix[1])); } async Task <bool> WarnOnConflict(ProxyTag newTag) { var query = "select * from (select *, (unnest(proxy_tags)).prefix as prefix, (unnest(proxy_tags)).suffix as suffix from members where system = @System) as _ where prefix is not distinct from @Prefix and suffix is not distinct from @Suffix and id != @Existing"; var conflicts = (await _db.Execute(conn => conn.QueryAsync <PKMember>(query, new { Prefix = newTag.Prefix, Suffix = newTag.Suffix, Existing = target.Id, system = target.System }))).ToList(); if (conflicts.Count <= 0) { return(true); } var conflictList = conflicts.Select(m => $"- **{m.NameFor(ctx)}**"); var msg = $"{Emojis.Warn} The following members have conflicting proxy tags:\n{string.Join('\n', conflictList)}\nDo you want to proceed anyway?"; return(await ctx.PromptYesNo(msg, "Proceed")); } // "Sub"command: clear flag if (await ctx.MatchClear()) { // If we already have multiple tags, this would clear everything, so prompt that if (target.ProxyTags.Count > 1) { var msg = $"{Emojis.Warn} You already have multiple proxy tags set: {target.ProxyTagsString()}\nDo you want to clear them all?"; if (!await ctx.PromptYesNo(msg, "Clear")) { throw Errors.GenericCancelled(); } } var patch = new MemberPatch { ProxyTags = Partial <ProxyTag[]> .Present(new ProxyTag[0]) }; await _db.Execute(conn => _repo.UpdateMember(conn, target.Id, patch)); await ctx.Reply($"{Emojis.Success} Proxy tags cleared."); } // "Sub"command: no arguments; will print proxy tags else if (!ctx.HasNext(skipFlags: false)) { if (target.ProxyTags.Count == 0) { await ctx.Reply("This member does not have any proxy tags."); } else { await ctx.Reply($"This member's proxy tags are:\n{target.ProxyTagsString("\n")}"); } } // Subcommand: "add" else if (ctx.Match("add", "append")) { if (!ctx.HasNext(skipFlags: false)) { throw new PKSyntaxError("You must pass an example proxy to add (eg. `[text]` or `J:text`)."); } var tagToAdd = ParseProxyTags(ctx.RemainderOrNull(skipFlags: false)); if (tagToAdd.IsEmpty) { throw Errors.EmptyProxyTags(target); } if (target.ProxyTags.Contains(tagToAdd)) { throw Errors.ProxyTagAlreadyExists(tagToAdd, target); } if (!await WarnOnConflict(tagToAdd)) { throw Errors.GenericCancelled(); } var newTags = target.ProxyTags.ToList(); newTags.Add(tagToAdd); var patch = new MemberPatch { ProxyTags = Partial <ProxyTag[]> .Present(newTags.ToArray()) }; await _db.Execute(conn => _repo.UpdateMember(conn, target.Id, patch)); await ctx.Reply($"{Emojis.Success} Added proxy tags {tagToAdd.ProxyString.AsCode()}."); } // Subcommand: "remove" else if (ctx.Match("remove", "delete")) { if (!ctx.HasNext(skipFlags: false)) { throw new PKSyntaxError("You must pass a proxy tag to remove (eg. `[text]` or `J:text`)."); } var tagToRemove = ParseProxyTags(ctx.RemainderOrNull(skipFlags: false)); if (tagToRemove.IsEmpty) { throw Errors.EmptyProxyTags(target); } if (!target.ProxyTags.Contains(tagToRemove)) { throw Errors.ProxyTagDoesNotExist(tagToRemove, target); } var newTags = target.ProxyTags.ToList(); newTags.Remove(tagToRemove); var patch = new MemberPatch { ProxyTags = Partial <ProxyTag[]> .Present(newTags.ToArray()) }; await _db.Execute(conn => _repo.UpdateMember(conn, target.Id, patch)); await ctx.Reply($"{Emojis.Success} Removed proxy tags {tagToRemove.ProxyString.AsCode()}."); } // Subcommand: bare proxy tag given else { var requestedTag = ParseProxyTags(ctx.RemainderOrNull(skipFlags: false)); if (requestedTag.IsEmpty) { throw Errors.EmptyProxyTags(target); } // This is mostly a legacy command, so it's gonna warn if there's // already more than one proxy tag. if (target.ProxyTags.Count > 1) { var msg = $"This member already has more than one proxy tag set: {target.ProxyTagsString()}\nDo you want to replace them?"; if (!await ctx.PromptYesNo(msg, "Replace")) { throw Errors.GenericCancelled(); } } if (!await WarnOnConflict(requestedTag)) { throw Errors.GenericCancelled(); } var newTags = new[] { requestedTag }; var patch = new MemberPatch { ProxyTags = Partial <ProxyTag[]> .Present(newTags) }; await _db.Execute(conn => _repo.UpdateMember(conn, target.Id, patch)); await ctx.Reply($"{Emojis.Success} Member proxy tags set to {requestedTag.ProxyString.AsCode()}."); } }
public async Task Proxy(Context ctx, PKMember target) { if (ctx.System == null) { throw Errors.NoSystemError; } if (target.System != ctx.System.Id) { throw Errors.NotOwnMemberError; } ProxyTag ParseProxyTags(string exampleProxy) { // // Make sure there's one and only one instance of "text" in the example proxy given var prefixAndSuffix = exampleProxy.Split("text"); if (prefixAndSuffix.Length < 2) { throw Errors.ProxyMustHaveText; } if (prefixAndSuffix.Length > 2) { throw Errors.ProxyMultipleText; } return(new ProxyTag(prefixAndSuffix[0], prefixAndSuffix[1])); } async Task <bool> WarnOnConflict(ProxyTag newTag) { var conflicts = (await _data.GetConflictingProxies(ctx.System, newTag)) .Where(m => m.Id != target.Id) .ToList(); if (conflicts.Count <= 0) { return(true); } var conflictList = conflicts.Select(m => $"- **{m.Name}**"); var msg = await ctx.Reply( $"{Emojis.Warn} The following members have conflicting proxy tags:\n{string.Join('\n', conflictList)}\nDo you want to proceed anyway?"); return(await ctx.PromptYesNo(msg)); } // "Sub"command: no arguments clearing // Also matches the pseudo-subcommand "text" which is equivalent to empty proxy tags on both sides. if (!ctx.HasNext() || ctx.Match("clear", "purge", "clean", "removeall")) { // If we already have multiple tags, this would clear everything, so prompt that if (target.ProxyTags.Count > 1) { var msg = await ctx.Reply( $"{Emojis.Warn} You already have multiple proxy tags set: {target.ProxyTagsString()}\nDo you want to clear them all?"); if (!await ctx.PromptYesNo(msg)) { throw Errors.GenericCancelled(); } } target.ProxyTags = new ProxyTag[] { }; await _data.SaveMember(target); await ctx.Reply($"{Emojis.Success} Proxy tags cleared."); } // Subcommand: "add" else if (ctx.Match("add", "append")) { if (!ctx.HasNext()) { throw new PKSyntaxError("You must pass an example proxy to add (eg. `[text]` or `J:text`)."); } var tagToAdd = ParseProxyTags(ctx.RemainderOrNull()); if (tagToAdd.IsEmpty) { throw Errors.EmptyProxyTags(target); } if (target.ProxyTags.Contains(tagToAdd)) { throw Errors.ProxyTagAlreadyExists(tagToAdd, target); } if (!await WarnOnConflict(tagToAdd)) { throw Errors.GenericCancelled(); } // It's not guaranteed the list's mutable, so we force it to be target.ProxyTags = target.ProxyTags.ToList(); target.ProxyTags.Add(tagToAdd); await _data.SaveMember(target); await ctx.Reply($"{Emojis.Success} Added proxy tags `{tagToAdd.ProxyString.SanitizeMentions()}`."); } // Subcommand: "remove" else if (ctx.Match("remove", "delete")) { if (!ctx.HasNext()) { throw new PKSyntaxError("You must pass a proxy tag to remove (eg. `[text]` or `J:text`)."); } var tagToRemove = ParseProxyTags(ctx.RemainderOrNull()); if (tagToRemove.IsEmpty) { throw Errors.EmptyProxyTags(target); } if (!target.ProxyTags.Contains(tagToRemove)) { throw Errors.ProxyTagDoesNotExist(tagToRemove, target); } // It's not guaranteed the list's mutable, so we force it to be target.ProxyTags = target.ProxyTags.ToList(); target.ProxyTags.Remove(tagToRemove); await _data.SaveMember(target); await ctx.Reply($"{Emojis.Success} Removed proxy tags `{tagToRemove.ProxyString.SanitizeMentions()}`."); } // Subcommand: bare proxy tag given else { if (!ctx.HasNext()) { throw new PKSyntaxError("You must pass an example proxy to set (eg. `[text]` or `J:text`)."); } var requestedTag = ParseProxyTags(ctx.RemainderOrNull()); if (requestedTag.IsEmpty) { throw Errors.EmptyProxyTags(target); } // This is mostly a legacy command, so it's gonna warn if there's // already more than one proxy tag. if (target.ProxyTags.Count > 1) { var msg = await ctx.Reply($"This member already has more than one proxy tag set: {target.ProxyTagsString().SanitizeMentions()}\nDo you want to replace them?"); if (!await ctx.PromptYesNo(msg)) { throw Errors.GenericCancelled(); } } if (!await WarnOnConflict(requestedTag)) { throw Errors.GenericCancelled(); } target.ProxyTags = new[] { requestedTag }; await _data.SaveMember(target); await ctx.Reply($"{Emojis.Success} Member proxy tags set to `{requestedTag.ProxyString.SanitizeMentions()}`."); } }