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.Core/Models/Privacy/SystemPrivacySubject.cs
+++ b/PluralKit.Core/Models/Privacy/SystemPrivacySubject.cs
@@ -0,0 +1,71 @@
+using System;
+
+namespace PluralKit.Core
+{
+    public enum SystemPrivacySubject
+    {
+        Description,
+        MemberList,
+        Front,
+        FrontHistory
+    }
+
+    public static class SystemPrivacyUtils
+    {
+        public static SystemPatch WithPrivacy(this SystemPatch system, SystemPrivacySubject subject, PrivacyLevel level)
+        {
+            // what do you mean switch expressions can't be statements >.>
+            _ = subject switch
+            {
+                SystemPrivacySubject.Description => system.DescriptionPrivacy = level,
+                SystemPrivacySubject.Front => system.FrontPrivacy = level,
+                SystemPrivacySubject.FrontHistory => system.FrontHistoryPrivacy = level,
+                SystemPrivacySubject.MemberList => system.MemberListPrivacy = level,
+                _ => throw new ArgumentOutOfRangeException($"Unknown privacy subject {subject}")
+            };
+            
+            return system;
+        }
+
+        public static SystemPatch WithAllPrivacy(this SystemPatch system, PrivacyLevel level)
+        {
+            foreach (var subject in Enum.GetValues(typeof(SystemPrivacySubject)))
+                WithPrivacy(system, (SystemPrivacySubject) subject, level);
+            return system;
+        }
+        
+        public static bool TryParseSystemPrivacy(string input, out SystemPrivacySubject subject)
+        {
+            switch (input.ToLowerInvariant())
+            {
+                case "description":
+                case "desc":
+                case "text":
+                case "info":
+                    subject = SystemPrivacySubject.Description;
+                    break; 
+                case "members":
+                case "memberlist":
+                case "list":
+                case "mlist":
+                    subject = SystemPrivacySubject.MemberList;
+                    break;
+                case "fronter":
+                case "fronters":
+                case "front":
+                    subject = SystemPrivacySubject.Front;
+                    break;
+                case "switch":
+                case "switches":
+                case "fronthistory":
+                case "fh":
+                    subject = SystemPrivacySubject.FrontHistory;
+                    break;
+                default:
+                    subject = default;
+                    return false;
+            }
+            return true;
+        } 
+    }
+}