feat(protocol-parser): add getFieldIndex to internal exports (#3046)

Co-authored-by: Kevin Ingersoll <[email protected]>
latticexyz · Aug 20, 2024 · 43ba187 · 43ba187
1 parent 2bbec22
commit 43ba187
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/packages/protocol-parser/src/exports/internal.ts b/packages/protocol-parser/src/exports/internal.ts
@@ -26,6 +26,7 @@ export * from "../staticDataLength";
 export * from "../valueSchemaToFieldLayoutHex";
 export * from "../valueSchemaToHex";
 
+export * from "../getFieldIndex";
 export * from "../getKeySchema";
 export * from "../getValueSchema";
 export * from "../getSchemaTypes";

diff --git a/packages/protocol-parser/src/getFieldIndex.ts b/packages/protocol-parser/src/getFieldIndex.ts
@@ -0,0 +1,18 @@
+import { isDynamicAbiType, isStaticAbiType } from "@latticexyz/schema-type/internal";
+import { ValueSchema } from "./common";
+
+export function getFieldIndex<valueSchema extends ValueSchema>(
+  valueSchema: valueSchema,
+  fieldName: keyof valueSchema & string,
+): number {
+  const fieldNames = [
+    ...Object.entries(valueSchema)
+      .filter(([, fieldType]) => isStaticAbiType(fieldType))
+      .map(([fieldName]) => fieldName),
+    ...Object.entries(valueSchema)
+      .filter(([, fieldType]) => isDynamicAbiType(fieldType))
+      .map(([fieldName]) => fieldName),
+  ];
+
+  return fieldNames.indexOf(fieldName);
+}