feat(protocol-parser): add abiTypesToSchema

latticexyz · Jul 5, 2023 · dd564e0 · dd564e0
1 parent ca50fef
commit dd564e0
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 0 deletions.
diff --git a/packages/protocol-parser/src/abiTypesToSchema.test.ts b/packages/protocol-parser/src/abiTypesToSchema.test.ts
@@ -0,0 +1,39 @@
+import { describe, expect, it } from "vitest";
+import { abiTypesToSchema } from "./abiTypesToSchema";
+
+describe("hexToSchema", () => {
+  it("converts hex to schema", () => {
+    expect(abiTypesToSchema(["bool"])).toMatchInlineSnapshot(`
+      {
+        "dynamicFields": [],
+        "staticFields": [
+          "bool",
+        ],
+      }
+    `);
+    expect(abiTypesToSchema(["bool", "bool[]"])).toMatchInlineSnapshot(`
+      {
+        "dynamicFields": [
+          "bool[]",
+        ],
+        "staticFields": [
+          "bool",
+        ],
+      }
+    `);
+    expect(abiTypesToSchema(["bytes32", "int32", "uint256[]", "address[]", "bytes", "string"])).toMatchInlineSnapshot(`
+      {
+        "dynamicFields": [
+          "uint256[]",
+          "address[]",
+          "bytes",
+          "string",
+        ],
+        "staticFields": [
+          "bytes32",
+          "int32",
+        ],
+      }
+    `);
+  });
+});
diff --git a/packages/protocol-parser/src/abiTypesToSchema.ts b/packages/protocol-parser/src/abiTypesToSchema.ts
@@ -0,0 +1,12 @@
+import { DynamicAbiType, SchemaAbiType, StaticAbiType, isDynamicAbiType } from "@latticexyz/schema-type";
+import { Schema } from "./common";
+
+export function abiTypesToSchema(abiTypes: SchemaAbiType[]): Schema {
+  const staticFields: StaticAbiType[] = [];
+  const dynamicFields: DynamicAbiType[] = [];
+  for (const abiType of abiTypes) {
+    if (isDynamicAbiType(abiType)) dynamicFields.push(abiType);
+    else staticFields.push(abiType);
+  }
+  return { staticFields, dynamicFields };
+}