fix: broken LinOps for entity and block NBT

Author: PierreSchwang

worldedit-bukkit/adapters/adapter-1_21_11/src/main/java/com/sk89q/worldedit/bukkit/adapter/impl/fawe/v1_21_11/LinOps.java

@@ -54,14 +54,13 @@
 public class LinOps implements DynamicOps<LinTag<?>> {
 
     static final DynamicOps<LinTag<?>> INSTANCE = new LinOps();
-    private static final LinTag<?> EMPTY_TAG = LinCompoundTag.of(Map.of());
 
     private LinOps() {
     }
 
     @Override
     public LinTag<?> empty() {
-        return EMPTY_TAG;
+        return LinEndTag.instance();
     }
 
     @Override
@@ -71,7 +70,7 @@ public LinTag<?> emptyList() {
 
     @Override
     public LinTag<?> emptyMap() {
-        return EMPTY_TAG;
+        return LinCompoundTag.builder().build();
     }
 
     @Override
@@ -177,11 +176,68 @@ public DataResult<LinTag<?>> mergeToMap(final LinTag<?> map, final LinTag<?> key
             return DataResult.error(() -> "key is not a string: " + key, map);
         }
         if (map instanceof LinCompoundTag mapCompoundTag) {
+            if (value == empty()) {
+                return DataResult.success(mapCompoundTag);
+            }
             return DataResult.success(mapCompoundTag.toBuilder().put(keyStringTag.value(), value).build());
         }
+        if (value == empty()) {
+            return DataResult.success(emptyMap());
+        }
         return DataResult.success(LinCompoundTag.builder().put(keyStringTag.value(), value).build());
     }
 
+    @Override
+    public DataResult<LinTag<?>> mergeToMap(final LinTag<?> map, final MapLike<LinTag<?>> values) {
+        if (!(map instanceof LinCompoundTag) && !(map instanceof LinEndTag)) {
+            return DataResult.error(() -> "mergeToMap called with not a map: " + map, map);
+        }
+        final Iterator<Pair<LinTag<?>, LinTag<?>>> iterator = values.entries().iterator();
+        if (!iterator.hasNext()) {
+            // if map is LinEndTag (returned by empty()) return a functional empty map
+            return map == empty() ? DataResult.success(this.emptyMap()) : DataResult.success(map);
+        }
+        LinCompoundTag.Builder resultBuilder = map instanceof LinCompoundTag compoundTag ?
+                compoundTag.toBuilder() : LinCompoundTag.builder();
+        List<LinTag<?>> nonStringKeys = new ArrayList<>();
+        iterator.forEachRemaining(entry -> {
+            LinTag<?> key = entry.getFirst();
+            if (key instanceof LinStringTag keyStringTag) {
+                if (entry.getSecond() != empty()) {
+                    resultBuilder.put(keyStringTag.value(), entry.getSecond());
+                }
+            } else {
+                nonStringKeys.add(key);
+            }
+        });
+        return nonStringKeys.isEmpty() ? DataResult.success(resultBuilder.build()) :
+                DataResult.error(() -> "some keys are not strings: " + nonStringKeys);
+    }
+
+    @Override
+    public DataResult<LinTag<?>> mergeToMap(final LinTag<?> map, final Map<LinTag<?>, LinTag<?>> values) {
+        if (!(map instanceof LinCompoundTag) && !(map instanceof LinEndTag)) {
+            return DataResult.error(() -> "mergeToMap called with not a map: " + map, map);
+        }
+        if (values.isEmpty()) {
+            return map == empty() ? DataResult.success(this.emptyMap()) : DataResult.success(map);
+        }
+        LinCompoundTag.Builder resultBuilder = map instanceof LinCompoundTag compoundTag ?
+                compoundTag.toBuilder() : LinCompoundTag.builder();
+        List<LinTag<?>> nonStringKeys = new ArrayList<>();
+        values.forEach((key, value) -> {
+            if (key instanceof LinStringTag keyStringTag) {
+                if (value != empty()) {
+                    resultBuilder.put(keyStringTag.value(), value);
+                }
+            } else {
+                nonStringKeys.add(key);
+            }
+        });
+        return nonStringKeys.isEmpty() ? DataResult.success(resultBuilder.build()) :
+                DataResult.error(() -> "some keys are not strings: " + nonStringKeys);
+    }
+
     @Override
     public DataResult<Stream<Pair<LinTag<?>, LinTag<?>>>> getMapValues(final LinTag<?> input) {
         if (input instanceof LinCompoundTag tag) {
@@ -241,7 +297,7 @@ public String toString() {
     @Override
     public LinTag<?> createMap(final Stream<Pair<LinTag<?>, LinTag<?>>> map) {
         LinCompoundTag.Builder builder = LinCompoundTag.builder();
-        map.forEach(pair -> {
+        map.filter(pair -> pair.getSecond() != empty()).forEach(pair -> {
             if (pair.getFirst() instanceof LinStringTag key) {
                 builder.put(key.value(), pair.getSecond());
                 return;
@@ -570,7 +626,10 @@ protected LinCompoundTag.Builder initBuilder() {
 
         @Override
         protected LinCompoundTag.Builder append(final String key, final LinTag<?> value, final LinCompoundTag.Builder builder) {
-            return builder.put(key, value);
+            if (value != ops().empty()) {
+                return builder.put(key, value);
+            }
+            return builder;
         }
 
         @Override

worldedit-bukkit/adapters/adapter-1_21_11/src/main/java/com/sk89q/worldedit/bukkit/adapter/impl/fawe/v1_21_11/LinValueOutput.java

@@ -198,6 +198,9 @@ public boolean isEmpty() {
     }
 
     public LinCompoundTag buildResult() {
+        if (this.isEmpty()) {
+            return LinCompoundTag.builder().build();
+        }
         return LinCompoundTag.of(this.collector.entrySet().stream()
                 .collect(Collectors.toMap(Map.Entry::getKey, entry -> unwrapPendingEntry(entry.getValue())))
         );