equals/hashcode for modes

KittehOrg · Sep 26, 2016 · 9970b24 · 9970b24
1 parent 91753b3
commit 9970b24
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/src/main/java/org/kitteh/irc/client/library/implementation/ModeData.java b/src/main/java/org/kitteh/irc/client/library/implementation/ModeData.java
@@ -34,6 +34,7 @@
 
 import javax.annotation.Nonnull;
 import java.time.Instant;
+import java.util.Objects;
 import java.util.Optional;
 
 final class ModeData {
@@ -64,11 +65,25 @@ static class IRCChannelUserMode extends IRCModeBase implements ChannelUserMode {
             this.prefix = prefix;
         }
 
+        @Override
+        public boolean equals(Object o) {
+            if (!(o instanceof IRCChannelUserMode)) {
+                return false;
+            }
+            IRCChannelUserMode other = (IRCChannelUserMode) o;
+            return (other.getNickPrefix() == this.getNickPrefix()) && (other.getType() == this.getType()) && (other.getClient().equals(this.getClient())) && (other.getChar() == this.getChar());
+        }
+
         @Override
         public char getNickPrefix() {
             return this.prefix;
         }
 
+        @Override
+        public int hashCode() {
+            return Objects.hash(this.getClient(), this.getChar(), this.getType(), this.getNickPrefix());
+        }
+
         @Nonnull
         @Override
         public String toString() {
@@ -84,12 +99,26 @@ static class IRCChannelMode extends IRCModeBase implements ChannelMode {
             this.type = type;
         }
 
+        @Override
+        public boolean equals(Object o) {
+            if (!(o instanceof IRCChannelMode)) {
+                return false;
+            }
+            IRCChannelMode other = (IRCChannelMode) o;
+            return (other.getType() == this.getType()) && (other.getClient().equals(this.getClient())) && (other.getChar() == this.getChar());
+        }
+
         @Nonnull
         @Override
         public Type getType() {
             return this.type;
         }
 
+        @Override
+        public int hashCode() {
+            return Objects.hash(this.getClient(), this.getChar(), this.type);
+        }
+
         @Nonnull
         @Override
         public String toString() {
@@ -102,6 +131,20 @@ static class IRCUserMode extends IRCModeBase implements UserMode {
             super(client, mode);
         }
 
+        @Override
+        public boolean equals(Object o) {
+            if (!(o instanceof IRCUserMode)) {
+                return false;
+            }
+            IRCUserMode other = (IRCUserMode) o;
+            return (other.getClient().equals(this.getClient())) && (other.getChar() == this.getChar());
+        }
+
+        @Override
+        public int hashCode() {
+            return Objects.hash(this.getClient(), this.getChar());
+        }
+
         @Nonnull
         @Override
         public String toString() {