Re-added StreamCodec extensions

Author: LatvianModder

interfaces.json

@@ -7,5 +7,8 @@
 	],
 	"com/mojang/blaze3d/vertex/PoseStack$Pose": [
 		"dev/latvian/mods/klib/core/KLibPoseStackPose"
+	],
+	"net/minecraft/network/codec/StreamCodec": [
+		"dev/latvian/mods/klib/core/KLibStreamCodec<B, V>"
 	]
 }

src/main/java/dev/latvian/mods/klib/codec/KLibStreamCodecs.java

@@ -39,7 +39,7 @@ static <B extends ByteBuf, V> StreamCodec<B, Set<V>> setOf(StreamCodec<? super B
 		return new SetStreamCodec<>(parent);
 	}
 
-	static <B extends ByteBuf, V> StreamCodec<B, Set<V>> linkedSet(StreamCodec<? super B, V> parent) {
+	static <B extends ByteBuf, V> StreamCodec<B, Set<V>> linkedSetOf(StreamCodec<? super B, V> parent) {
 		return new LinkedSetStreamCodec<>(parent);
 	}
 

src/main/java/dev/latvian/mods/klib/core/KLibStreamCodec.java

@@ -0,0 +1,34 @@
+package dev.latvian.mods.klib.core;
+
+import dev.latvian.mods.klib.codec.KLibStreamCodecs;
+import net.minecraft.network.codec.ByteBufCodecs;
+import net.minecraft.network.codec.StreamCodec;
+
+import java.util.List;
+import java.util.Optional;
+
+public interface KLibStreamCodec<B, V> {
+	default StreamCodec<B, Optional<V>> optional() {
+		return ByteBufCodecs.optional((StreamCodec) this);
+	}
+
+	default StreamCodec<B, V> optional(V defaultValue) {
+		return KLibStreamCodecs.optional((StreamCodec) this, defaultValue);
+	}
+
+	default StreamCodec<B, V> nullable() {
+		return optional(null);
+	}
+
+	default StreamCodec<B, List<V>> listOf() {
+		return KLibStreamCodecs.listOf((StreamCodec) this);
+	}
+
+	default StreamCodec<B, List<V>> setOf() {
+		return KLibStreamCodecs.setOf((StreamCodec) this);
+	}
+
+	default StreamCodec<B, List<V>> linkedSetOf() {
+		return KLibStreamCodecs.linkedSetOf((StreamCodec) this);
+	}
+}

src/main/java/dev/latvian/mods/klib/core/mixin/StreamCodecMixin.java

@@ -0,0 +1,9 @@
+package dev.latvian.mods.klib.core.mixin;
+
+import dev.latvian.mods.klib.core.KLibStreamCodec;
+import net.minecraft.network.codec.StreamCodec;
+import org.spongepowered.asm.mixin.Mixin;
+
+@Mixin(StreamCodec.class)
+public interface StreamCodecMixin<B, V> extends KLibStreamCodec<B, V> {
+}

src/main/java/dev/latvian/mods/klib/util/Comparison.java

@@ -3,6 +3,8 @@
 import dev.latvian.mods.klib.data.DataType;
 import net.minecraft.util.StringRepresentable;
 
+import java.util.function.IntFunction;
+
 public enum Comparison implements StringRepresentable {
 	EQUALS("equals", "=="),
 	NOT_EQUALS("not_equals", "!="),
@@ -12,6 +14,7 @@ public enum Comparison implements StringRepresentable {
 	LESS_THAN_OR_EQUALS("less_than_or_equals", "<=");
 
 	public static final Comparison[] VALUES = values();
+	public static final IntFunction<Comparison[]> ARRAY_FACTORY = Comparison[]::new;
 	public static final DataType<Comparison> DATA_TYPE = DataType.of(VALUES);
 
 	public final String name;