feat(protocol-parser): add encodeKeyTuple (#1099)

packages/protocol-parser/src/decodeKeyTuple.test.ts

@@ -1,6 +1,5 @@
 import { describe, expect, it } from "vitest";
 import { decodeKeyTuple } from "./decodeKeyTuple";
-import { Schema } from "./common";
 
 describe("decodeKeyTuple", () => {
   it("can decode bool key tuple", () => {

packages/protocol-parser/src/encodeKeyTuple.test.ts

@@ -0,0 +1,39 @@
+import { describe, expect, it } from "vitest";
+import { encodeKeyTuple } from "./encodeKeyTuple";
+
+describe("encodeKeyTuple", () => {
+  it("can encode bool key tuple", () => {
+    expect(encodeKeyTuple({ staticFields: ["bool"], dynamicFields: [] }, [false])).toStrictEqual([
+      "0x0000000000000000000000000000000000000000000000000000000000000000",
+    ]);
+    expect(
+      encodeKeyTuple(
+        {
+          staticFields: ["bool"],
+          dynamicFields: [],
+        },
+        [true]
+      )
+    ).toStrictEqual(["0x0000000000000000000000000000000000000000000000000000000000000001"]);
+  });
+
+  it("can encode complex key tuple", () => {
+    expect(
+      encodeKeyTuple({ staticFields: ["uint256", "int32", "bytes16", "address", "bool", "int8"], dynamicFields: [] }, [
+        42n,
+        -42,
+        "0x12340000000000000000000000000000",
+        "0xFFfFfFffFFfffFFfFFfFFFFFffFFFffffFfFFFfF",
+        true,
+        3,
+      ])
+    ).toStrictEqual([
+      "0x000000000000000000000000000000000000000000000000000000000000002a",
+      "0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffd6",
+      "0x1234000000000000000000000000000000000000000000000000000000000000",
+      "0x000000000000000000000000ffffffffffffffffffffffffffffffffffffffff",
+      "0x0000000000000000000000000000000000000000000000000000000000000001",
+      "0x0000000000000000000000000000000000000000000000000000000000000003",
+    ]);
+  });
+});

packages/protocol-parser/src/encodeKeyTuple.ts

@@ -0,0 +1,7 @@
+import { StaticPrimitiveType } from "@latticexyz/schema-type";
+import { Hex, encodeAbiParameters } from "viem";
+import { Schema } from "./common";
+
+export function encodeKeyTuple(keySchema: Schema, keyTuple: StaticPrimitiveType[]): Hex[] {
+  return keyTuple.map((key, index) => encodeAbiParameters([{ type: keySchema.staticFields[index] }], [key]));
+}

packages/protocol-parser/src/index.ts

@@ -4,6 +4,7 @@ export * from "./decodeKeyTuple";
 export * from "./decodeRecord";
 export * from "./decodeStaticField";
 export * from "./encodeField";
+export * from "./encodeKeyTuple";
 export * from "./encodeRecord";
 export * from "./errors";
 export * from "./hexToPackedCounter";