From f54b4237652cc0a1f0887c338b958950a08dc857 Mon Sep 17 00:00:00 2001
From: Yanbing Zhao <zzzz@mail.ustc.edu.cn>
Date: Thu, 7 Dec 2023 01:44:28 +0800
Subject: [PATCH] Merge two enums

---
 .../slides/network/ProjectorURLSummaryPacket.java  | 14 +++++---------
 .../java/org/teacon/slides/url/ProjectorURL.java   |  2 +-
 2 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/src/main/java/org/teacon/slides/network/ProjectorURLSummaryPacket.java b/src/main/java/org/teacon/slides/network/ProjectorURLSummaryPacket.java
index 040b88a..83df768 100644
--- a/src/main/java/org/teacon/slides/network/ProjectorURLSummaryPacket.java
+++ b/src/main/java/org/teacon/slides/network/ProjectorURLSummaryPacket.java
@@ -39,10 +39,6 @@ public final class ProjectorURLSummaryPacket implements Function<ProjectorURL, P
     private final HashFunction hmacNonceFunction;
     private final Object2BooleanMap<Bits256> hmacUrlToBlockStatus;
 
-    private enum Status {
-        END, BLOCKED, UNBLOCKED
-    }
-
     public ProjectorURLSummaryPacket(BiMap<UUID, ProjectorURL> idToUrl, Set<UUID> blockedIdSet) {
         var hmacNonce = Bits256.random();
         var hmacNonceFunction = Hashing.hmacSha256(hmacNonce.toBytes());
@@ -61,15 +57,15 @@ public ProjectorURLSummaryPacket(FriendlyByteBuf buf) {
         var nonce = Bits256.read(buf);
         var hmacUrlToBlockStatus = new Object2BooleanArrayMap<Bits256>(defaultCapacity);
         while (true) {
-            switch (buf.readEnum(Status.class)) {
-                case END -> {
+            switch (buf.readEnum(ProjectorURL.Status.class)) {
+                case UNKNOWN -> {
                     this.hmacNonce = nonce;
                     this.hmacNonceFunction = Hashing.hmacSha256(nonce.toBytes());
                     this.hmacUrlToBlockStatus = Object2BooleanMaps.unmodifiable(hmacUrlToBlockStatus);
                     return;
                 }
                 case BLOCKED -> hmacUrlToBlockStatus.put(Bits256.read(buf), true);
-                case UNBLOCKED -> hmacUrlToBlockStatus.put(Bits256.read(buf), false);
+                case ALLOWED -> hmacUrlToBlockStatus.put(Bits256.read(buf), false);
             }
         }
     }
@@ -85,10 +81,10 @@ public void sendToClient(ServerPlayer player) {
     public void write(FriendlyByteBuf buf) {
         this.hmacNonce.write(buf);
         for (var entry : this.hmacUrlToBlockStatus.object2BooleanEntrySet()) {
-            buf.writeEnum(entry.getBooleanValue() ? ProjectorURLSummaryPacket.Status.BLOCKED : ProjectorURLSummaryPacket.Status.UNBLOCKED);
+            buf.writeEnum(entry.getBooleanValue() ? ProjectorURL.Status.BLOCKED : ProjectorURL.Status.ALLOWED);
             entry.getKey().write(buf);
         }
-        buf.writeEnum(ProjectorURLSummaryPacket.Status.END);
+        buf.writeEnum(ProjectorURL.Status.UNKNOWN);
     }
 
     @Override
diff --git a/src/main/java/org/teacon/slides/url/ProjectorURL.java b/src/main/java/org/teacon/slides/url/ProjectorURL.java
index 79bf69f..706a4b3 100644
--- a/src/main/java/org/teacon/slides/url/ProjectorURL.java
+++ b/src/main/java/org/teacon/slides/url/ProjectorURL.java
@@ -45,7 +45,7 @@ public boolean equals(Object o) {
     }
 
     public enum Status {
-        BLOCKED, ALLOWED, UNKNOWN;
+        UNKNOWN, BLOCKED, ALLOWED;
 
         public boolean isBlocked() {
             return this == BLOCKED;