Comparator materialdata (#2205)

* Added Comparator Mech/Tag for mode * Registered Comparator Material * Fixed Formatting * Fixed Formatting * Fixed typo * Formatting & Registry Placement Fix
DenizenScript · May 5, 2020 · 301e8d6 · 301e8d6
1 parent 0e3183e
commit 301e8d6
Show file tree

Hide file tree

Showing 2 changed files with 89 additions and 0 deletions.
diff --git a/plugin/src/main/java/com/denizenscript/denizen/objects/properties/PropertyRegistry.java b/plugin/src/main/java/com/denizenscript/denizen/objects/properties/PropertyRegistry.java
@@ -158,6 +158,7 @@ public static void registermainProperties() {
             }
             PropertyParser.registerProperty(MaterialLevel.class, MaterialTag.class);
             PropertyParser.registerProperty(MaterialLightable.class, MaterialTag.class);
+            PropertyParser.registerProperty(MaterialMode.class, MaterialTag.class);
             PropertyParser.registerProperty(MaterialPersistent.class, MaterialTag.class);
             PropertyParser.registerProperty(MaterialCount.class, MaterialTag.class);
             PropertyParser.registerProperty(MaterialSlab.class, MaterialTag.class);

diff --git a/plugin/src/main/java/com/denizenscript/denizen/objects/properties/material/MaterialMode.java b/plugin/src/main/java/com/denizenscript/denizen/objects/properties/material/MaterialMode.java
@@ -0,0 +1,88 @@
+package com.denizenscript.denizen.objects.properties.material;
+
+import com.denizenscript.denizen.objects.MaterialTag;
+import com.denizenscript.denizencore.objects.Mechanism;
+import com.denizenscript.denizencore.objects.ObjectTag;
+import com.denizenscript.denizencore.objects.core.ElementTag;
+import com.denizenscript.denizencore.objects.properties.Property;
+import com.denizenscript.denizencore.objects.properties.PropertyParser;
+import org.bukkit.block.data.type.Comparator;
+
+public class MaterialMode implements Property {
+
+    public static boolean describes(ObjectTag material) {
+        return material instanceof MaterialTag
+                && ((MaterialTag) material).hasModernData()
+                && ((MaterialTag) material).getModernData().data instanceof Comparator;
+    }
+
+    public static MaterialMode getFrom(ObjectTag _material) {
+        if (!describes(_material)) {
+            return null;
+        }
+        else {
+            return new MaterialMode((MaterialTag) _material);
+        }
+    }
+
+    public static final String[] handledMechs = new String[] {
+            "mode"
+    };
+
+    private MaterialMode(MaterialTag _material) {
+        material = _material;
+    }
+
+    MaterialTag material;
+
+    public static void registerTags() {
+
+        // <--[tag]
+        // @attribute <MaterialTag.mode>
+        // @returns ElementTag
+        // @mechanism MaterialTag.mode
+        // @group properties
+        // @description
+        // Returns a comparator's mode.
+        // Output is COMPARE or SUBTRACT.
+        // -->
+        PropertyParser.<MaterialMode>registerTag("mode", (attribute, material) -> {
+            return new ElementTag(material.getComparator().getMode().name());
+        });
+    }
+
+    public Comparator getComparator() {
+        return (Comparator) material.getModernData().data;
+    }
+
+    public void setMode(String mode) {
+        getComparator().setMode(Comparator.Mode.valueOf(mode));
+    }
+
+    @Override
+    public String getPropertyString() {
+        return getComparator().getMode().name();
+    }
+
+    @Override
+    public String getPropertyId() {
+        return "mode";
+    }
+
+    @Override
+    public void adjust(Mechanism mechanism) {
+
+        // <--[mechanism]
+        // @object MaterialTag
+        // @name mode
+        // @input ElementTag
+        // @description
+        // Sets comparator's mode between compare and subtract.
+        // @tags
+        // <MaterialTag.mode>
+        // -->
+        if (mechanism.matches("mode") && mechanism.requireEnum(false, Comparator.Mode.values())) {
+            setMode(mechanism.getValue().asString().toUpperCase());
+        }
+    }
+}