Codama AccountNode and DiscriminatorNode data structures.

Author: jpe7s

idl-src-gen/src/main/java/software/sava/idl/generator/codama/AccountNode.java

@@ -0,0 +1,95 @@
+package software.sava.idl.generator.codama;
+
+import systems.comodal.jsoniter.JsonIterator;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static systems.comodal.jsoniter.JsonIterator.fieldEquals;
+
+public final class AccountNode extends NamedDocsNode {
+
+  private final TypeNode data;
+  private final LinkNode pda;
+  private final int size;
+  private final List<DiscriminatorNode> discriminators;
+
+  public AccountNode(final String name,
+                     final List<String> docs,
+                     final TypeNode data,
+                     final LinkNode pda,
+                     final int size,
+                     final List<DiscriminatorNode> discriminators) {
+    super(name, docs);
+    this.data = data;
+    this.pda = pda;
+    this.size = size;
+    this.discriminators = discriminators;
+  }
+
+  public TypeNode data() {
+    return data;
+  }
+
+  public LinkNode pda() {
+    return pda;
+  }
+
+  public int size() {
+    return size;
+  }
+
+  public List<DiscriminatorNode> discriminators() {
+    return discriminators;
+  }
+
+  public static AccountNode parse(final JsonIterator ji) {
+    final var parser = new Parser();
+    ji.testObject(parser);
+    return parser.createAccountNode();
+  }
+
+  static final class Parser extends BaseDocsParser {
+
+    private TypeNode data;
+    private LinkNode pda;
+    private int size;
+    private List<DiscriminatorNode> discriminators;
+
+    private Parser() {
+    }
+
+    AccountNode createAccountNode() {
+      return new AccountNode(
+          name,
+          docs == null ? List.of() : docs,
+          data,
+          pda,
+          size,
+          discriminators == null ? List.of() : discriminators
+      );
+    }
+
+    @Override
+    public boolean test(final char[] buf, final int offset, final int len, final JsonIterator ji) {
+      if (fieldEquals("data", buf, offset, len)) {
+        data = TypeNode.parse(ji);
+        return true;
+      } else if (fieldEquals("pda", buf, offset, len)) {
+        pda = LinkNode.parse(ji);
+        return true;
+      } else if (fieldEquals("size", buf, offset, len)) {
+        size = ji.readInt();
+        return true;
+      } else if (fieldEquals("discriminators", buf, offset, len)) {
+        discriminators = new ArrayList<>();
+        while (ji.readArray()) {
+          discriminators.add(DiscriminatorNode.parse(ji));
+        }
+        return true;
+      } else {
+        return super.test(buf, offset, len, ji);
+      }
+    }
+  }
+}

idl-src-gen/src/main/java/software/sava/idl/generator/codama/ConstantDiscriminatorNode.java

@@ -0,0 +1,37 @@
+package software.sava.idl.generator.codama;
+
+import systems.comodal.jsoniter.FieldBufferPredicate;
+import systems.comodal.jsoniter.JsonIterator;
+
+import static systems.comodal.jsoniter.JsonIterator.fieldEquals;
+
+public record ConstantDiscriminatorNode(ValueNode.Constant constant, int offset) implements DiscriminatorNode {
+
+  public static ConstantDiscriminatorNode parse(final JsonIterator ji) {
+    final var parser = new Parser();
+    ji.testObject(parser);
+    return parser.createConstantDiscriminatorNode();
+  }
+
+  static final class Parser implements FieldBufferPredicate {
+
+    private ValueNode.Constant constant;
+    private int offset;
+
+    ConstantDiscriminatorNode createConstantDiscriminatorNode() {
+      return new ConstantDiscriminatorNode(constant, offset);
+    }
+
+    @Override
+    public boolean test(final char[] buf, final int offset, final int len, final JsonIterator ji) {
+      if (fieldEquals("constant", buf, offset, len)) {
+        constant = ValueNode.Constant.parse(ji);
+      } else if (fieldEquals("offset", buf, offset, len)) {
+        this.offset = ji.readInt();
+      } else {
+        throw new IllegalStateException("Unhandled field " + String.valueOf(buf, offset, len));
+      }
+      return true;
+    }
+  }
+}

idl-src-gen/src/main/java/software/sava/idl/generator/codama/DiscriminatorNode.java

@@ -0,0 +1,19 @@
+package software.sava.idl.generator.codama;
+
+import systems.comodal.jsoniter.JsonIterator;
+
+public sealed interface DiscriminatorNode permits
+    ConstantDiscriminatorNode,
+    FieldDiscriminatorNode,
+    SizeDiscriminatorNode {
+
+  static DiscriminatorNode parse(final JsonIterator ji) {
+    final var kind = ji.skipObjField().readString();
+    return switch (kind) {
+      case "constantDiscriminatorNode" -> ConstantDiscriminatorNode.parse(ji);
+      case "fieldDiscriminatorNode" -> FieldDiscriminatorNode.parse(ji);
+      case "sizeDiscriminatorNode" -> SizeDiscriminatorNode.parse(ji);
+      default -> throw new UnsupportedOperationException(kind);
+    };
+  }
+}

idl-src-gen/src/main/java/software/sava/idl/generator/codama/FieldDiscriminatorNode.java

@@ -0,0 +1,44 @@
+package software.sava.idl.generator.codama;
+
+import systems.comodal.jsoniter.JsonIterator;
+
+import static systems.comodal.jsoniter.JsonIterator.fieldEquals;
+
+final class FieldDiscriminatorNode extends NamedNode implements DiscriminatorNode {
+
+  private final int offset;
+
+  public FieldDiscriminatorNode(final String name, final int offset) {
+    super(name);
+    this.offset = offset;
+  }
+
+  public int offset() {
+    return offset;
+  }
+
+  public static FieldDiscriminatorNode parse(final JsonIterator ji) {
+    final var parser = new Parser();
+    ji.testObject(parser);
+    return parser.createFieldDiscriminatorNode();
+  }
+
+  static final class Parser extends BaseParser {
+
+    private int offset;
+
+    FieldDiscriminatorNode createFieldDiscriminatorNode() {
+      return new FieldDiscriminatorNode(name, offset);
+    }
+
+    @Override
+    public boolean test(final char[] buf, final int offset, final int len, final JsonIterator ji) {
+      if (fieldEquals("offset", buf, offset, len)) {
+        this.offset = ji.readInt();
+        return true;
+      } else {
+        return super.test(buf, offset, len, ji);
+      }
+    }
+  }
+}

idl-src-gen/src/main/java/software/sava/idl/generator/codama/SizeDiscriminatorNode.java

@@ -0,0 +1,39 @@
+package software.sava.idl.generator.codama;
+
+import systems.comodal.jsoniter.FieldBufferPredicate;
+import systems.comodal.jsoniter.JsonIterator;
+
+import static systems.comodal.jsoniter.JsonIterator.fieldEquals;
+
+record SizeDiscriminatorNode(int size) implements DiscriminatorNode {
+
+  public static SizeDiscriminatorNode parse(final JsonIterator ji) {
+    final var parser = new Parser();
+    ji.testObject(parser);
+    return parser.createSizeDiscriminatorNode();
+  }
+
+  static final class Parser implements FieldBufferPredicate {
+
+    private int size;
+
+    SizeDiscriminatorNode createSizeDiscriminatorNode() {
+      return new SizeDiscriminatorNode(size);
+    }
+
+    @Override
+    public boolean test(final char[] buf, final int offset, final int len, final JsonIterator ji) {
+      if (fieldEquals("size", buf, offset, len)) {
+        size = ji.readInt();
+      } else if (fieldEquals("kind", buf, offset, len)) {
+        ji.skip();
+      } else {
+        throw new IllegalStateException(String.format(
+            "Unhandled %s field %s.",
+            getClass().getSimpleName(), new String(buf, offset, len)
+        ));
+      }
+      return true;
+    }
+  }
+}