From 5eea18ea497d4bc11466b2bd875582ba26f188f5 Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Thu, 14 Nov 2024 02:59:11 +0100
Subject: Add lists

---
 src/main/java/moe/nea/pcj/json/JsonCodec.java | 40 +++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

(limited to 'src/main/java/moe/nea/pcj/json/JsonCodec.java')

diff --git a/src/main/java/moe/nea/pcj/json/JsonCodec.java b/src/main/java/moe/nea/pcj/json/JsonCodec.java
index 65f4a5e..6195bee 100644
--- a/src/main/java/moe/nea/pcj/json/JsonCodec.java
+++ b/src/main/java/moe/nea/pcj/json/JsonCodec.java
@@ -2,11 +2,51 @@ package moe.nea.pcj.json;
 
 import moe.nea.pcj.Codec;
 import moe.nea.pcj.Result;
+import moe.nea.pcj.Unit;
+
+import java.util.ArrayList;
+import java.util.List;
 
 public interface JsonCodec<T, Format> extends Codec<
 		T, Format, JsonLikeOperations<Format>,
 		JsonLikeError, JsonLikeError> {
 
+	default JsonCodec<List<T>, Format> listOf() {
+		return new JsonCodec<>() {
+			@Override
+			public Result<Format, JsonLikeError> encode(List<T> data, JsonLikeOperations<Format> ops) {
+				var list = ops.createList();
+				var erros = new ArrayList<JsonLikeError>();
+				for (int i = 0; i < data.size(); i++) {
+					var datum = data.get(i);
+					final var index = i;
+					var result = JsonCodec.this.encode(datum, ops)
+					                           .mapError(it -> new AtIndex(index, it));
+					erros.addAll(result.errors());
+					result.valueOrPartial().ifPresent(list::add);
+				}
+				return Result.<Format, JsonLikeError>ok(list.complete()).appendErrors(erros);
+			}
+
+			@Override
+			public Result<List<T>, JsonLikeError> decode(Format format, JsonLikeOperations<Format> ops) {
+				var view = Result.<ListView<Format>, JsonLikeError>cast(ops.getList(format));
+				return view.flatMap(elements -> {
+					var acc = new ArrayList<T>(elements.length());
+					var errors = new ArrayList<JsonLikeError>();
+					for (int i = 0; i < elements.length(); i++) {
+						final var index = i;
+						var result = JsonCodec.this.decode(elements.getUnsafe(i), ops)
+						                           .mapError(it -> new AtIndex(index, it));
+						errors.addAll(result.errors());
+						result.valueOrPartial().ifPresent(acc::add);
+					}
+					return Result.<List<T>, JsonLikeError>ok(acc).appendErrors(errors);
+				});
+			}
+		};
+	}
+
 	default MapCodec<T, Format> fieldOf(String key) {
 		return new MapCodec<>() {
 			@Override
-- 
cgit