fix(common): move kms account to its own export (#2731)

.changeset/perfect-actors-flow.md

@@ -7,6 +7,7 @@ Added `createKmsAccount`, a [viem custom account](https://viem.sh/docs/accounts/
 To use it, you must first install `@aws-sdk/[email protected]` and `[email protected]` dependencies into your project. Then create a KMS account with:
 
 ```ts
+import { createKmsAccount } from "@latticexyz/common/kms";
 const account = createKmsAccount({ keyId: ... });
 ```
 

packages/common/package.json

@@ -17,7 +17,8 @@
     "./errors": "./dist/errors.js",
     "./foundry": "./dist/foundry.js",
     "./type-utils": "./dist/type-utils.js",
-    "./utils": "./dist/utils.js"
+    "./utils": "./dist/utils.js",
+    "./kms": "./dist/kms.js"
   },
   "typesVersions": {
     "*": {
@@ -44,6 +45,9 @@
       ],
       "utils": [
         "./src/utils/index.ts"
+      ],
+      "kms": [
+        "./src/exports/kms.ts"
       ]
     }
   },

packages/common/src/createKmsAccount.ts → packages/common/src/account/kms/createKmsAccount.ts

@@ -1,8 +1,8 @@
 import { KMSClient } from "@aws-sdk/client-kms";
 import { LocalAccount, hashMessage, hashTypedData, keccak256, serializeTransaction } from "viem";
 import { toAccount } from "viem/accounts";
-import { signWithKms } from "./account/kms/signWithKms";
-import { getAddressFromKMS } from "./account/kms/getAddressFromKms";
+import { signWithKms } from "./signWithKms";
+import { getAddressFromKms } from "./getAddressFromKms";
 
 export type CreateKmsAccountOptions = {
   keyId: string;
@@ -20,7 +20,7 @@ export async function createKmsAccount({
   keyId,
   client = new KMSClient(),
 }: CreateKmsAccountOptions): Promise<KMSAccount> {
-  const address = await getAddressFromKMS({ keyId, client });
+  const address = await getAddressFromKms({ keyId, client });
 
   const account = toAccount({
     address,

packages/common/src/account/kms/getAddressFromKms.ts

@@ -1,7 +1,7 @@
 import { Address, toHex } from "viem";
 import { publicKeyToAddress } from "viem/utils";
 import { KMSClient, SignCommandInput } from "@aws-sdk/client-kms";
-import { getPublicKey } from "./commands/getPublicKey";
+import { getPublicKey } from "./getPublicKey";
 // @ts-expect-error Could not find a declaration file for module 'asn1.js'.
 import asn1 from "asn1.js";
 
@@ -21,7 +21,7 @@ function publicKeyKmsToAddress(publicKey: Uint8Array): Address {
   return address;
 }
 
-export async function getAddressFromKMS({
+export async function getAddressFromKms({
   keyId,
   client,
 }: {

packages/common/src/account/kms/signWithKms.ts

@@ -1,7 +1,7 @@
 import { Hex, isAddressEqual, signatureToHex, toHex } from "viem";
 import { recoverAddress } from "viem/utils";
 import { KMSClient, SignCommandInput } from "@aws-sdk/client-kms";
-import { sign } from "./commands/sign";
+import { sign } from "./sign";
 // @ts-expect-error Could not find a declaration file for module 'asn1.js'.
 import asn1 from "asn1.js";
 

packages/common/src/exports/kms.ts

@@ -0,0 +1 @@
+export { createKmsAccount, type CreateKmsAccountOptions, type KMSAccount } from "../account/kms/createKmsAccount";

packages/common/src/index.ts

@@ -1,7 +1,6 @@
 export * from "./common";
 export * from "./createBenchmark";
 export * from "./createBurnerAccount";
-export * from "./createKmsAccount";
 export * from "./createNonceManager";
 export * from "./getContract";
 export * from "./getBurnerPrivateKey";

packages/common/tsup.config.ts

@@ -10,6 +10,7 @@ export default defineConfig({
     foundry: "src/foundry/index.ts",
     "type-utils": "src/type-utils/index.ts",
     utils: "src/utils/index.ts",
+    kms: "src/exports/kms.ts",
   },
   target: "esnext",
   format: ["esm"],