Remove FlagStorage and merge into RegionData

BauSystem/BauSystem_Main/src/de/steamwar/bausystem/region/RegionData.java

@@ -19,76 +19,134 @@
 
 package de.steamwar.bausystem.region;
 
-import de.steamwar.bausystem.worlddata.WorldData;
+import de.steamwar.bausystem.region.flags.Flag;
 import de.steamwar.sql.SchematicNode;
+import lombok.NonNull;
 import yapion.hierarchy.types.YAPIONObject;
 
-import java.util.Objects;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.function.Function;
 
-public interface RegionData {
+public abstract class RegionData {
 
-    void clear();
+    private final List<Property<?, ?>> properties = new ArrayList<>();
 
-    SchematicNode getTestblockSchematic();
+    protected final YAPIONObject data;
+    protected final YAPIONObject flagData;
+    protected final Runnable onChange;
+    protected final Map<Flag<?>, Flag.Value<?>> flagMap = new HashMap<>();
 
-    void setTestblockSchematic(SchematicNode schematic);
+    private final class Property<T, K> {
+        private final String field;
+        private final Function<K, T> loader;
+        private final Function<T, K> writer;
 
-    RegionData EMPTY = new RegionData() {
+        private T value;
 
-        @Override
-        public void clear() {
+        public Property(String field, Function<K, T> loader, Function<T, K> writer) {
+            this.field = field;
+            this.loader = loader;
+            this.writer = writer;
+            properties.add(this);
         }
 
-        @Override
-        public SchematicNode getTestblockSchematic() {
-            return null;
-        }
-
-        @Override
-        public void setTestblockSchematic(SchematicNode schematic) {
-        }
-    };
-
-    class RegionDataImpl implements RegionData {
-
-        private final YAPIONObject yapionObject;
-        private final Runnable onChange;
-
-        public RegionDataImpl(YAPIONObject yapionObject, Runnable onChange) {
-            this.yapionObject = yapionObject;
-            this.onChange = onChange;
-
-            if (yapionObject.containsKey("testblockSchematic")) {
-                testblockSchematic = SchematicNode.getSchematicNode(yapionObject.getInt("testblockSchematic"));
-            }
-        }
-
-        @Override
-        public void clear() {
-            testblockSchematic = null;
-            yapionObject.remove("testblockSchematic");
-            onChange.run();
-        }
-
-        private SchematicNode testblockSchematic = null;
-
-        @Override
-        public SchematicNode getTestblockSchematic() {
-            return testblockSchematic;
-        }
-
-        @Override
-        public void setTestblockSchematic(SchematicNode schematic) {
-            if (Objects.equals(this.testblockSchematic, schematic)) {
-                return;
-            }
-            this.testblockSchematic = schematic;
-            if (schematic == null) {
-                yapionObject.remove("testblockSchematic");
+        public void load() {
+            if (flagData.containsKey(field)) {
+                value = loader.apply(flagData.getPlainValue(field));
             } else {
-                yapionObject.put("testblockSchematic", testblockSchematic.getId());
+                value = null;
+            }
+        }
+
+        public T get() {
+            return value;
+        }
+
+        public void set(T value) {
+            this.value = value;
+            if (value == null) {
+                flagData.remove(field);
+            } else {
+                flagData.put(field, writer.apply(value));
             }
-            onChange.run();
         }
     }
+
+    private Property<SchematicNode, Integer> testblockSchematic = new Property<>("testblockSchematic", SchematicNode::byId, SchematicNode::getId);
+
+    protected RegionData(YAPIONObject data, Runnable onChange) {
+        this.data = data;
+        this.flagData = data.getObjectOrSetDefault("flagStorage", new YAPIONObject());
+        this.onChange = onChange;
+        initialize();
+        for (final Flag flag : Flag.getFlags()) {
+            if (!has(flag).isWritable()) continue;
+            try {
+                String s = flagData.getPlainValue(flag.name());
+                flagMap.put(flag, flag.valueOfValue(s));
+            } catch (Exception e) {
+                flagMap.put(flag, (Flag.Value<?>) flag.getDefaultValue());
+            }
+        }
+        properties.forEach(Property::load);
+    }
+
+    protected void initialize() {
+    }
+
+    @NonNull
+    public abstract <T extends Enum<T> & Flag.Value<T>> RegionFlagPolicy has(@NonNull Flag<T> flag);
+
+    /**
+     * Returns true if the flag was changed and did not already contain the provided value
+     */
+    public final <T extends Enum<T> & Flag.Value<T>> boolean set(@NonNull Flag<T> flag, @NonNull T value) {
+        if (has(flag).isWritable()) {
+            if (flagMap.put(flag, value) != value) {
+                flagData.put(flag.name(), value.name());
+                onChange.run();
+                return true;
+            }
+        }
+        return false;
+    }
+
+    @NonNull
+    public final <T extends Enum<T> & Flag.Value<T>> FlagOptional<T> get(@NonNull Flag<T> flag) {
+        return FlagOptional.of(flag, (T) flagMap.get(flag));
+    }
+
+    public final void clear() {
+        for (Flag flag : Flag.getFlags()) {
+            if (has(flag).isWritable()) {
+                flagMap.remove(flag);
+                flagData.remove(flag.name());
+            }
+        }
+        properties.forEach(property -> property.set(null));
+        onChange.run();
+    }
+
+    public final Map<Flag<?>, Flag.Value<?>> getBackedMap() {
+        return flagMap;
+    }
+
+    public SchematicNode getTestblockSchematic() {
+        return testblockSchematic.get();
+    }
+
+    public void setTestblockSchematic(SchematicNode schematic) {
+        testblockSchematic.set(schematic);
+        onChange.run();
+    }
+
+    @Override
+    public final String toString() {
+        return getClass().getSimpleName() + "{" +
+                "flagMap=" + flagMap +
+                '}';
+    }
 }