From 5e4cdbbfa75ce021dfbfd8e1f2d38370a702285e Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Fri, 27 Nov 2020 01:26:23 +0800
Subject: ...

---
 BackEnd/Timeline/Services/UserPermissionService.cs | 33 ++++++++++++++++++++--
 1 file changed, 30 insertions(+), 3 deletions(-)

(limited to 'BackEnd/Timeline/Services/UserPermissionService.cs')

diff --git a/BackEnd/Timeline/Services/UserPermissionService.cs b/BackEnd/Timeline/Services/UserPermissionService.cs
index 9683000a..bd7cd6aa 100644
--- a/BackEnd/Timeline/Services/UserPermissionService.cs
+++ b/BackEnd/Timeline/Services/UserPermissionService.cs
@@ -28,7 +28,7 @@ namespace Timeline.Services
     /// <summary>
     /// Represents a user's permissions.
     /// </summary>
-    public class UserPermissions : IEnumerable<UserPermission>
+    public class UserPermissions : IEnumerable<UserPermission>, IEquatable<UserPermissions>
     {
         public static UserPermissions AllPermissions { get; } = new UserPermissions(Enum.GetValues<UserPermission>());
 
@@ -49,10 +49,10 @@ namespace Timeline.Services
         public UserPermissions(IEnumerable<UserPermission> permissions)
         {
             if (permissions == null) throw new ArgumentNullException(nameof(permissions));
-            _permissions = new HashSet<UserPermission>(permissions);
+            _permissions = new SortedSet<UserPermission>(permissions);
         }
 
-        private readonly HashSet<UserPermission> _permissions = new();
+        private readonly SortedSet<UserPermission> _permissions = new();
 
         /// <summary>
         /// Check if a permission is contained in the list.
@@ -108,6 +108,33 @@ namespace Timeline.Services
         {
             return ((IEnumerable)_permissions).GetEnumerator();
         }
+
+        public bool Equals(UserPermissions? other)
+        {
+            if (other == null)
+                return false;
+
+            return _permissions.SequenceEqual(other._permissions);
+        }
+
+        public override bool Equals(object? obj)
+        {
+            return Equals(obj as UserPermissions);
+        }
+
+        public override int GetHashCode()
+        {
+            int result = 0;
+            foreach (var permission in Enum.GetValues<UserPermission>())
+            {
+                if (_permissions.Contains(permission))
+                {
+                    result += 1;
+                }
+                result <<= 1;
+            }
+            return result;
+        }
     }
 
     public interface IUserPermissionService
-- 
cgit v1.2.3