TON client api factory (#1153)

LayerZero-Labs · Jan 10, 2025 · de8f24f · de8f24f
1 parent e455d72
commit de8f24f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/.changeset/purple-cows-boil.md b/.changeset/purple-cows-boil.md
@@ -0,0 +1,5 @@
+---
+"@layerzerolabs/devtools-ton": minor
+---
+
+Added API key factory param to TON client factory
diff --git a/packages/devtools-ton/src/omnigraph/coordinates.ts b/packages/devtools-ton/src/omnigraph/coordinates.ts
@@ -1,7 +1,7 @@
 import { TonClient } from '@ton/ton'
 import pMemoize from 'p-memoize'
 
-import { formatEid, OmniPoint, type RpcUrlFactory } from '@layerzerolabs/devtools'
+import { EndpointBasedFactory, formatEid, OmniPoint, type RpcUrlFactory } from '@layerzerolabs/devtools'
 import { ChainType, EndpointId, endpointIdToChainType } from '@layerzerolabs/lz-definitions'
 
 import { TonClientFactory } from './types'
@@ -38,5 +38,14 @@ export const createRpcUrlFactory =
     (eid) =>
         overrides[eid] ?? defaultRpcUrlFactory(eid)
 
-export const createTonClientFactory = (urlFactory: RpcUrlFactory = defaultRpcUrlFactory): TonClientFactory =>
-    pMemoize(async (eid) => new TonClient({ endpoint: await urlFactory(eid) }))
+export const createTonClientFactory = (
+    urlFactory: RpcUrlFactory = defaultRpcUrlFactory,
+    apiKeyFactory?: EndpointBasedFactory<string>
+): TonClientFactory =>
+    pMemoize(
+        async (eid) =>
+            new TonClient({
+                endpoint: await urlFactory(eid),
+                ...(apiKeyFactory ? { apiKey: await apiKeyFactory(eid) } : {}),
+            })
+    )