[endec] Implement EdmMap allowing for easy map operations within EdmE…

…ndec

src/main/java/io/wispforest/owo/serialization/format/edm/EdmElement.java

@@ -5,12 +5,12 @@
 import java.util.Optional;
 import java.util.stream.Collectors;
 
-public final class EdmElement<T> {
+public sealed class EdmElement<T> permits EdmMap {
 
     private final T value;
     private final Type type;
 
-    private EdmElement(T value, Type type) {
+    EdmElement(T value, Type type) {
         this.value = value;
         this.type = type;
     }
@@ -40,6 +40,14 @@ public Object unwrap() {
         }
     }
 
+    public EdmMap asMap() {
+        if(this.type != Type.MAP) {
+            throw new IllegalStateException("Given Element is not a Map Type: " + this);
+        }
+
+        return new EdmMap(this.cast());
+    }
+
     @Override
     public String toString() {
         return "E(" + this.type.name() + ", " + this.value + ")";
@@ -48,10 +56,7 @@ public String toString() {
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
-        if (o == null || getClass() != o.getClass()) return false;
-
-        EdmElement<?> that = (EdmElement<?>) o;
-
+        if (!(o instanceof EdmElement<?> that)) return false;
         if (!this.value.equals(that.value)) return false;
         return this.type == that.type;
     }

src/main/java/io/wispforest/owo/serialization/format/edm/EdmMap.java

@@ -0,0 +1,34 @@
+package io.wispforest.owo.serialization.format.edm;
+
+import io.wispforest.owo.serialization.endec.KeyedEndec;
+import io.wispforest.owo.serialization.util.MapCarrier;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.Map;
+
+public final class EdmMap extends EdmElement<Map<String, EdmElement<?>>> implements MapCarrier {
+
+    EdmMap(EdmElement<Map<String, EdmElement<?>>> mapElement) {
+        super(mapElement.value(), mapElement.type());
+    }
+
+    @Override
+    public <T> T getWithErrors(@NotNull KeyedEndec<T> key) {
+        return key.endec().decodeFully(EdmDeserializer::of, this.value().get(key.key()));
+    }
+
+    @Override
+    public <T> void put(@NotNull KeyedEndec<T> key, @NotNull T value) {
+        this.value().put(key.key(), key.endec().encodeFully(EdmSerializer::of, value));
+    }
+
+    @Override
+    public <T> void delete(@NotNull KeyedEndec<T> key) {
+        this.value().remove(key.key());
+    }
+
+    @Override
+    public <T> boolean has(@NotNull KeyedEndec<T> key) {
+        return this.value().containsKey(key.key());
+    }
+}