@andy-liquid-labs/lighter-ts-sdk 1.0.0

package/src/client/clientManager.ts

@@ -0,0 +1,121 @@
+/**
+ * Client Manager for handling multiple TxClients
+ * Similar to the Go SDK's SharedClientManager
+ */
+
+import { TxClient } from './txClient';
+import { MinimalHTTPClient } from './http';
+import { generateApiKey } from '../signer/keyManager';
+
+// Global client storage
+let defaultTxClient: TxClient | null = null;
+const defaultClientPerAccount: Map<bigint, TxClient> = new Map();
+const allTxClients: Map<bigint, Map<number, TxClient>> = new Map();
+
+// Default values for optional parameters
+export const DEFAULT_API_KEY_INDEX = 255;
+export const DEFAULT_ACCOUNT_INDEX = -1n;
+
+/**
+ * Get a client for specific account and API key
+ */
+export function getClient(
+  apiKeyIndex: number = DEFAULT_API_KEY_INDEX,
+  accountIndex: bigint = DEFAULT_ACCOUNT_INDEX
+): TxClient {
+  // If apiKeyIndex==255 && accountIndex!=-1, return default client for that account
+  if (apiKeyIndex === DEFAULT_API_KEY_INDEX && accountIndex !== DEFAULT_ACCOUNT_INDEX) {
+    const client = defaultClientPerAccount.get(accountIndex);
+    if (client) {
+      return client;
+    }
+  }
+
+  // Special case: return default client
+  if (apiKeyIndex === DEFAULT_API_KEY_INDEX && accountIndex === DEFAULT_ACCOUNT_INDEX) {
+    if (!defaultTxClient) {
+      throw new Error('Client is not created, call createClient() first');
+    }
+    return defaultTxClient;
+  }
+
+  // Look up client in double map
+  const accountClients = allTxClients.get(accountIndex);
+  const client = accountClients?.get(apiKeyIndex);
+
+  if (!client) {
+    throw new Error(
+      `Client is not created for apiKeyIndex: ${apiKeyIndex} accountIndex: ${accountIndex}`
+    );
+  }
+
+  return client;
+}
+
+/**
+ * Create a new TxClient and store it
+ */
+export function createClient(
+  httpClient: MinimalHTTPClient | null,
+  privateKey: string,
+  chainId: number,
+  apiKeyIndex: number,
+  accountIndex: bigint
+): TxClient {
+  if (accountIndex <= 0n) {
+    throw new Error('Invalid account index');
+  }
+
+  const txClientInstance = new TxClient(
+    httpClient,
+    privateKey,
+    accountIndex,
+    apiKeyIndex,
+    chainId
+  );
+
+  // Store in allTxClients map
+  if (!allTxClients.has(accountIndex)) {
+    allTxClients.set(accountIndex, new Map());
+  }
+  allTxClients.get(accountIndex)!.set(apiKeyIndex, txClientInstance);
+
+  // Update default client (most recently created becomes default)
+  defaultTxClient = txClientInstance;
+  defaultClientPerAccount.set(accountIndex, txClientInstance);
+
+  return txClientInstance;
+}
+
+/**
+ * Check that a client exists and the API key matches the one on the server
+ */
+export async function checkClient(
+  apiKeyIndex: number = DEFAULT_API_KEY_INDEX,
+  accountIndex: bigint = DEFAULT_ACCOUNT_INDEX
+): Promise<void> {
+  const client = getClient(apiKeyIndex, accountIndex);
+  await client.check();
+}
+
+/**
+ * Generate a new API key pair
+ * Returns [privateKey, publicKey] as hex strings
+ */
+export { generateApiKey };
+
+/**
+ * Clear all stored clients (useful for testing)
+ */
+export function clearClients(): void {
+  defaultTxClient = null;
+  defaultClientPerAccount.clear();
+  allTxClients.clear();
+}
+
+/**
+ * Get the default client
+ */
+export function getDefaultClient(): TxClient | null {
+  return defaultTxClient;
+}