added Tag->equals to compare NBT tags

aternosorg · Feb 28, 2022 · 5cfc1a4 · 5cfc1a4
1 parent 692eb12
commit 5cfc1a4
Show file tree

Hide file tree

Showing 8 changed files with 113 additions and 6 deletions.
diff --git a/src/Tag/ArrayValueTag.php b/src/Tag/ArrayValueTag.php
@@ -171,4 +171,23 @@ public function jsonSerialize()
     {
         return $this->valueArray;
     }
-}
+
+    /**
+     * @inheritDoc
+     */
+    public function equals(Tag $tag): bool
+    {
+        if($tag === $this) {
+            return true;
+        }
+        if(!$tag instanceof ArrayValueTag || $this->getType() !== $tag->getType() || count($tag) !== count($this)) {
+            return false;
+        }
+        foreach ($this as $i => $val) {
+            if($val !== $tag[$i]) {
+                return false;
+            }
+        }
+        return true;
+    }
+}
diff --git a/src/Tag/CompoundTag.php b/src/Tag/CompoundTag.php
@@ -341,4 +341,27 @@ public function getString(string $name): ?StringTag
         $tag = $this->get($name);
         return $tag instanceof StringTag ? $tag : null;
     }
+
+    /**
+     * @inheritDoc
+     */
+    function equals(Tag $tag): bool
+    {
+        if($tag === $this) {
+            return true;
+        }
+        if(!$tag instanceof CompoundTag || $this->getType() !== $tag->getType() || count($tag) !== count($this)) {
+            return false;
+        }
+        /**
+         * @var string $key
+         * @var Tag $val
+         */
+        foreach ($this as $key => $val) {
+            if(!isset($tag[$key]) || !$val->equals($tag[$key])) {
+                return false;
+            }
+        }
+        return true;
+    }
 }
diff --git a/src/Tag/EndTag.php b/src/Tag/EndTag.php
@@ -48,4 +48,12 @@ public function jsonSerialize()
     {
         return null;
     }
-}
+
+    /**
+     * @inheritDoc
+     */
+    function equals(Tag $tag): bool
+    {
+        return $tag->getType() === $this->getType();
+    }
+}
diff --git a/src/Tag/FloatValueTag.php b/src/Tag/FloatValueTag.php
@@ -39,4 +39,13 @@ public function jsonSerialize()
     {
         return $this->value;
     }
-}
+
+    /**
+     * @inheritDoc
+     */
+    public function equals(Tag $tag): bool
+    {
+        return $tag instanceof FloatValueTag && $this->getType() === $tag->getType() &&
+            $tag->getValue() === $this->getValue();
+    }
+}
diff --git a/src/Tag/IntValueTag.php b/src/Tag/IntValueTag.php
@@ -39,4 +39,13 @@ public function jsonSerialize()
     {
         return $this->value;
     }
-}
+
+    /**
+     * @inheritDoc
+     */
+    public function equals(Tag $tag): bool
+    {
+        return $tag instanceof IntValueTag && $this->getType() === $tag->getType() &&
+            $tag->getValue() === $this->getValue();
+    }
+}
diff --git a/src/Tag/ListTag.php b/src/Tag/ListTag.php
@@ -116,4 +116,28 @@ protected function getTagTypeString(): string
     {
         return parent::getTagTypeString() . "<" . TagType::NAMES[$this->contentTagType] . ">";
     }
-}
+
+    /**
+     * @inheritDoc
+     */
+    public function equals(Tag $tag): bool
+    {
+        if($tag === $this) {
+            return true;
+        }
+        if(!$tag instanceof ListTag || $this->getType() !== $tag->getType() ||
+            $this->getContentTag() !== $tag->getContentTag() || count($tag) !== count($this)) {
+            return false;
+        }
+        /**
+         * @var int $i
+         * @var Tag $val
+         */
+        foreach ($this as $i => $val) {
+            if(!$val->equals($tag[$i])) {
+                return false;
+            }
+        }
+        return true;
+    }
+}
diff --git a/src/Tag/StringTag.php b/src/Tag/StringTag.php
@@ -76,4 +76,13 @@ public function jsonSerialize()
     {
         return $this->value;
     }
-}
+
+    /**
+     * @inheritDoc
+     */
+    public function equals(Tag $tag): bool
+    {
+        return $tag instanceof StringTag && $this->getType() === $tag->getType() &&
+            $tag->getValue() === $this->getValue();
+    }
+}
diff --git a/src/Tag/Tag.php b/src/Tag/Tag.php
@@ -169,6 +169,12 @@ public function __toString(): string
         return $this->getTagTypeString() . "('" . ($this->getName() ?: "None") . "'): " . $this->getValueString();
     }
 
+    /**
+     * @param Tag $tag
+     * @return bool
+     */
+    abstract function equals(Tag $tag): bool;
+
     /**
      * @return string
      */