Refactor system/member privacy commands

2020-07-08 00:46:58 +02:00
parent 1449234a84
commit 9f523b3c5f
8 changed files with 249 additions and 196 deletions
--- a/PluralKit.Bot/Commands/Privacy/ContextPrivacyExt.cs
+++ b/PluralKit.Bot/Commands/Privacy/ContextPrivacyExt.cs
@@ -0,0 +1,39 @@
+using PluralKit.Core;
+
+namespace PluralKit.Bot
+{
+    public static class ContextPrivacyExt
+    {
+        public static PrivacyLevel PopPrivacyLevel(this Context ctx)
+        {
+            if (ctx.Match("public", "show", "shown", "visible"))
+                return PrivacyLevel.Public;
+
+            if (ctx.Match("private", "hide", "hidden"))
+                return PrivacyLevel.Private;
+
+            if (!ctx.HasNext())
+                throw new PKSyntaxError("You must pass a privacy level (`public` or `private`)");
+            
+            throw new PKSyntaxError($"Invalid privacy level `{ctx.PopArgument()}` (must be `public` or `private`).");
+        }
+
+        public static SystemPrivacySubject PopSystemPrivacySubject(this Context ctx)
+        {
+            if (!SystemPrivacyUtils.TryParseSystemPrivacy(ctx.PeekArgument(), out var subject))
+                throw new PKSyntaxError($"Invalid privacy subject `{ctx.PopArgument()}` (must be `description`, `members`, `front`, `fronthistory`, or `all`).");
+            
+            ctx.PopArgument();
+            return subject;
+        }
+        
+        public static MemberPrivacySubject PopMemberPrivacySubject(this Context ctx)
+        {
+            if (!MemberPrivacyUtils.TryParseMemberPrivacy(ctx.PeekArgument(), out var subject))
+                throw new PKSyntaxError($"Invalid privacy subject `{ctx.PopArgument()}` (must be `name`, `description`, `avatar`, `birthday`, `pronouns`, `metadata`, `visibility`, or `all).");
+            
+            ctx.PopArgument();
+            return subject;
+        }
+    }
+}