Implement or() and and() methods in schema classes

Author: Glavo

src/main/java/org/glavo/nbt/internal/schema/IntersectionSchema.java

@@ -22,7 +22,9 @@
 import org.glavo.nbt.validation.NBTValidationException;
 import org.jetbrains.annotations.Unmodifiable;
 
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
 
 public record IntersectionSchema<T extends Tag>(
         @Unmodifiable List<NBTSchema<? extends T>> schemas) implements TagSchema<T> {
@@ -51,4 +53,15 @@ public void validateTag(Tag tag) throws NBTValidationException {
             }
         }
     }
+
+    @Override
+    public NBTSchema<T> and(NBTSchema<? extends T> other) {
+        Objects.requireNonNull(other, "other");
+
+        var list = new ArrayList<NBTSchema<? extends T>>(schemas.size() + 1);
+        list.addAll(schemas);
+        list.add(other);
+
+        return new IntersectionSchema<>(List.copyOf(list));
+    }
 }

src/main/java/org/glavo/nbt/internal/schema/UnionSchema.java

@@ -22,7 +22,9 @@
 import org.glavo.nbt.validation.NBTValidationException;
 import org.jetbrains.annotations.Unmodifiable;
 
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
 
 public record UnionSchema<T extends Tag>(@Unmodifiable List<NBTSchema<? extends T>> schemas) implements TagSchema<T> {
 
@@ -57,4 +59,14 @@ public void validateTag(Tag tag) throws NBTValidationException {
         }
         throw exception;
     }
+
+    @Override
+    public NBTSchema<T> or(NBTSchema<? extends T> other) {
+        Objects.requireNonNull(other, "other");
+
+        var list = new ArrayList<NBTSchema<? extends T>>(schemas.size() + 1);
+        list.addAll(schemas);
+        list.add(other);
+        return new UnionSchema<>(List.copyOf(list));
+    }
 }