@lifi/perps-sdk 0.1.1-alpha.9 → 0.2.0-alpha.0

package/src/signers/lighter/LighterSigner.ts

@@ -0,0 +1,396 @@
+import { ActionType } from '@lifi/perps-types'
+import {
+  type LighterWasmExports,
+  type LoadLighterWasmOptions,
+  loadLighterWasm,
+  type SignResult,
+} from './wasmLoader.js'
+
+// ---------------------------------------------------------------------------
+// Lighter signing context
+//
+// Every Sign* call needs the Go signer to have been initialized with the API
+// private key for a given (apiKeyIndex, accountIndex) pair via CreateClient.
+// The context is process-global inside the Go runtime; we track which keys we
+// have already initialized so we only call CreateClient once per keypair.
+// ---------------------------------------------------------------------------
+
+export interface LighterSignerContext {
+  /** Lighter-native private key (from GenerateAPIKey, NOT an Ethereum key). */
+  apiKeyPrivateKey: string
+  /** API key slot registered on-chain (0-255). */
+  apiKeyIndex: number
+  /** Lighter account index — looked up from the user's L1 Ethereum address. */
+  accountIndex: number
+}
+
+export interface LighterSignerConfig extends LoadLighterWasmOptions {
+  /** Lighter REST API base URL. */
+  apiUrl?: string
+  /** Lighter chain ID (304 = mainnet). */
+  chainId?: number
+}
+
+export interface LighterSignedBlob {
+  txType: number
+  txInfo: string
+  txHash: string
+}
+
+export interface ChangePubKeyResult extends LighterSignedBlob {
+  /** EIP-191 message the L1 Ethereum wallet must sign to authorize the key rotation. */
+  messageToSign: string
+}
+
+export interface ApiKeyPair {
+  publicKey: string
+  privateKey: string
+}
+
+const DEFAULT_API_URL = 'https://mainnet.zklighter.elliot.ai'
+const DEFAULT_CHAIN_ID = 304
+
+export class LighterSigner {
+  private readonly apiUrl: string
+  private readonly chainId: number
+  private readonly loaderOptions: LoadLighterWasmOptions
+  private wasm: LighterWasmExports | undefined
+  private readonly registeredClients = new Set<string>()
+
+  constructor(config: LighterSignerConfig = {}) {
+    this.apiUrl = config.apiUrl ?? DEFAULT_API_URL
+    this.chainId = config.chainId ?? DEFAULT_CHAIN_ID
+    this.loaderOptions = {
+      wasmBinaryUrl: config.wasmBinaryUrl,
+      wasmExecJsUrl: config.wasmExecJsUrl,
+      wasmExecJsSource: config.wasmExecJsSource,
+    }
+  }
+
+  async initialize(): Promise<void> {
+    if (!this.wasm) {
+      this.wasm = await loadLighterWasm(this.loaderOptions)
+    }
+  }
+
+  /**
+   * Generate a fresh Lighter API keypair. Pass an optional 32-byte hex seed
+   * for deterministic generation (used in tests); omit for a random key.
+   */
+  async generateAPIKey(seed?: string): Promise<ApiKeyPair> {
+    const wasm = await this.ensureLoaded()
+    const result = wasm.GenerateAPIKey(seed ?? '')
+    if (result.error) {
+      throw new Error(`Lighter GenerateAPIKey failed: ${result.error}`)
+    }
+    if (!result.publicKey || !result.privateKey) {
+      throw new Error('Lighter GenerateAPIKey returned an incomplete result')
+    }
+    return { publicKey: result.publicKey, privateKey: result.privateKey }
+  }
+
+  /**
+   * Sign an action blob with the provided (apiKeyPrivateKey, apiKeyIndex,
+   * accountIndex) context. `wasmSignParams` comes straight from the backend's
+   * `WasmBlobActionStep`. Returns the signed `{ txType, txInfo, txHash }`
+   * triple the backend forwards to Lighter's `sendTx` endpoint.
+   *
+   * For REGISTER_API_KEY use `signChangePubKey` instead — it returns an
+   * additional `messageToSign` the L1 wallet must countersign.
+   */
+  async sign(
+    action: ActionType,
+    wasmSignParams: Record<string, unknown>,
+    context: LighterSignerContext
+  ): Promise<LighterSignedBlob> {
+    if (action === ActionType.REGISTER_API_KEY) {
+      throw new Error(
+        'Use signChangePubKey() for REGISTER_API_KEY — the L1 eth_sign hop ' +
+          'must be coordinated by the caller.'
+      )
+    }
+    const wasm = await this.ensureLoaded()
+    await this.ensureClient(context)
+    const result = this.dispatch(wasm, action, wasmSignParams, context)
+    return unwrap(result, action)
+  }
+
+  /**
+   * Step 1 of the REGISTER_API_KEY flow. Generates the WASM blob for a
+   * ChangePubKey tx with `L1Sig` left empty, plus the canonical EIP-191
+   * message the L1 Ethereum wallet must sign next.
+   *
+   * Requires the freshly-generated `privateKey` (returned by
+   * {@link generateAPIKey}) — the Go WASM signer registers a per-slot client
+   * keyed on `(apiKeyIndex, accountIndex)` before it'll sign anything for
+   * that slot, including the ChangePubKey that's about to register the key
+   * on-chain. This is purely client-side bookkeeping; it does not touch the
+   * Lighter API.
+   */
+  async signChangePubKey(
+    pubKeyHex: string,
+    privateKey: string,
+    nonce: number,
+    apiKeyIndex: number,
+    accountIndex: number
+  ): Promise<ChangePubKeyResult> {
+    const wasm = await this.ensureLoaded()
+    await this.ensureClient({
+      apiKeyPrivateKey: privateKey,
+      apiKeyIndex,
+      accountIndex,
+    })
+    // 5-arg call (matches lighter-python's signer wrapper). `skipNonce=0`
+    // means "embed the supplied nonce" — the value we just fetched from
+    // Lighter's authoritative /nextNonce.
+    const SKIP_NONCE_OFF = 0
+    const result = wasm.SignChangePubKey(
+      pubKeyHex,
+      SKIP_NONCE_OFF,
+      nonce,
+      apiKeyIndex,
+      accountIndex
+    )
+    if (result.error) {
+      throw new Error(`Lighter SignChangePubKey failed: ${result.error}`)
+    }
+    if (
+      result.txType === undefined ||
+      result.txInfo === undefined ||
+      result.txHash === undefined ||
+      !result.messageToSign
+    ) {
+      throw new Error('Lighter SignChangePubKey returned incomplete result')
+    }
+    return {
+      txType: result.txType,
+      txInfo: result.txInfo,
+      txHash: result.txHash,
+      messageToSign: result.messageToSign,
+    }
+  }
+
+  /**
+   * Step 2 of REGISTER_API_KEY — inject the L1 signature produced by the
+   * user's Ethereum wallet into the ChangePubKey txInfo JSON. `L1Sig` is the
+   * only field in txInfo that depends on the L1 signature; txHash does NOT
+   * include it (so we do not recompute it).
+   */
+  embedL1Signature(txInfo: string, l1Signature: string): string {
+    let parsed: Record<string, unknown>
+    try {
+      parsed = JSON.parse(txInfo) as Record<string, unknown>
+    } catch (err) {
+      throw new Error(
+        `Failed to parse ChangePubKey txInfo as JSON: ${(err as Error).message}`
+      )
+    }
+    parsed.L1Sig = l1Signature
+    return JSON.stringify(parsed)
+  }
+
+  /**
+   * Create an auth token for authenticated WebSocket subscriptions.
+   * `deadline` is a Unix timestamp in seconds — tokens have an 8h hard cap.
+   */
+  async createAuthToken(
+    deadline: number,
+    context: LighterSignerContext
+  ): Promise<string> {
+    const wasm = await this.ensureLoaded()
+    await this.ensureClient(context)
+    const result = wasm.CreateAuthToken(
+      deadline,
+      context.apiKeyIndex,
+      context.accountIndex
+    )
+    if (result.error) {
+      throw new Error(`Lighter CreateAuthToken failed: ${result.error}`)
+    }
+    if (!result.authToken) {
+      throw new Error('Lighter CreateAuthToken returned no token')
+    }
+    return result.authToken
+  }
+
+  private async ensureLoaded(): Promise<LighterWasmExports> {
+    if (!this.wasm) {
+      await this.initialize()
+    }
+    return this.wasm as LighterWasmExports
+  }
+
+  private async ensureClient(context: LighterSignerContext): Promise<void> {
+    const key = `${context.apiKeyIndex}:${context.accountIndex}:${context.apiKeyPrivateKey}`
+    if (this.registeredClients.has(key)) {
+      return
+    }
+    const wasm = await this.ensureLoaded()
+    const result = wasm.CreateClient(
+      this.apiUrl,
+      context.apiKeyPrivateKey,
+      this.chainId,
+      context.apiKeyIndex,
+      context.accountIndex
+    )
+    if (result.error) {
+      throw new Error(`Lighter CreateClient failed: ${result.error}`)
+    }
+    this.registeredClients.add(key)
+  }
+
+  /**
+   * Map an ActionType + backend-provided params object to the positional-arg
+   * WASM call. The Go signer exports take primitives in order, not an object
+   * — so we pick fields in the exact order the Go side expects. Unknown
+   * fields (e.g. backend's integrator_* hints) are ignored.
+   */
+  private dispatch(
+    wasm: LighterWasmExports,
+    action: ActionType,
+    p: Record<string, unknown>,
+    ctx: LighterSignerContext
+  ): SignResult {
+    const nonce = numberField(p, 'nonce')
+    switch (action) {
+      case ActionType.PLACE_ORDER:
+      case ActionType.PLACE_TRIGGER_ORDER:
+        return wasm.SignCreateOrder(
+          numberField(p, 'market_index'),
+          numberField(p, 'client_order_index'),
+          numberField(p, 'base_amount'),
+          numberField(p, 'price'),
+          numberField(p, 'is_ask'),
+          numberField(p, 'order_type'),
+          numberField(p, 'time_in_force'),
+          // WASM's error message lists "reduceOnly" as bool-like, but Go
+          // actually calls Value.Int() on every arg — passing an actual bool
+          // panics. Coerce to 0/1.
+          numberField(p, 'reduce_only'),
+          numberField(p, 'trigger_price'),
+          numberField(p, 'order_expiry'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      case ActionType.CANCEL_ORDER:
+        return wasm.SignCancelOrder(
+          numberField(p, 'market_index'),
+          numberField(p, 'order_index'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      case ActionType.CANCEL_ALL_ORDERS:
+        return wasm.SignCancelAllOrders(
+          numberField(p, 'time_in_force'),
+          numberField(p, 'timestamp_ms'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      case ActionType.MODIFY_ORDER:
+        return wasm.SignModifyOrder(
+          numberField(p, 'market_index'),
+          numberField(p, 'order_index'),
+          numberField(p, 'base_amount'),
+          numberField(p, 'price'),
+          numberField(p, 'trigger_price'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      case ActionType.UPDATE_LEVERAGE:
+        return wasm.SignUpdateLeverage(
+          numberField(p, 'market_index'),
+          numberField(p, 'fraction'),
+          numberField(p, 'margin_mode'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      case ActionType.UPDATE_POSITION_MARGIN:
+        return wasm.SignUpdateMargin(
+          numberField(p, 'market_index'),
+          numberField(p, 'usdc_amount'),
+          numberField(p, 'direction'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      case ActionType.WITHDRAWAL:
+        return wasm.SignWithdraw(
+          numberField(p, 'amount'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      case ActionType.TRANSFER:
+        // SignTransfer's 7 positional args follow the Go binding order:
+        // `toAccount, usdcAmount, fee, memo, nonce, apiKeyIndex,
+        // accountIndex`. `memo` is copied directly into a `[32]byte`, so the
+        // backend MUST send a JS string whose UTF-8 byte length is exactly
+        // 32 (in practice: 32 ASCII characters) — anything else fails with
+        // "memo expected to be 32 bytes long".
+        return wasm.SignTransfer(
+          numberField(p, 'to_account'),
+          numberField(p, 'usdc_amount'),
+          numberField(p, 'fee'),
+          stringField(p, 'memo'),
+          nonce,
+          ctx.apiKeyIndex,
+          ctx.accountIndex
+        )
+      default:
+        throw new Error(
+          `Lighter WASM signer does not support action: ${action}`
+        )
+    }
+  }
+}
+
+function unwrap(result: SignResult, action: ActionType): LighterSignedBlob {
+  if (result.error) {
+    throw new Error(`Lighter sign(${action}) failed: ${result.error}`)
+  }
+  if (
+    result.txType === undefined ||
+    result.txInfo === undefined ||
+    result.txHash === undefined
+  ) {
+    throw new Error(
+      `Lighter sign(${action}) returned an incomplete signed blob`
+    )
+  }
+  return {
+    txType: result.txType,
+    txInfo: result.txInfo,
+    txHash: result.txHash,
+  }
+}
+
+function numberField(p: Record<string, unknown>, key: string): number {
+  const v = p[key]
+  if (typeof v === 'number') {
+    return v
+  }
+  if (typeof v === 'boolean') {
+    return v ? 1 : 0
+  }
+  if (typeof v === 'string' && v !== '' && !Number.isNaN(Number(v))) {
+    return Number(v)
+  }
+  throw new Error(
+    `Lighter sign params missing numeric field '${key}' (got ${typeof v})`
+  )
+}
+
+function stringField(p: Record<string, unknown>, key: string): string {
+  const v = p[key]
+  if (typeof v === 'string' && v !== '') {
+    return v
+  }
+  throw new Error(
+    `Lighter sign params missing string field '${key}' (got ${typeof v})`
+  )
+}

package/src/signers/lighter/index.ts

@@ -0,0 +1,36 @@
+export type { LighterApiKey } from './LighterKeyStore.js'
+export {
+  DEFAULT_API_KEY_INDEX,
+  LIGHTER_PROVIDER_KEY,
+  LighterKeyStore,
+} from './LighterKeyStore.js'
+export type {
+  ApproveReadOnlyTokenInputs,
+  ApproveReadOnlyTokenResult,
+  LighterCreateTokenResponse,
+  LighterReadOnlyToken,
+  LighterReadOnlyTokenManagerOptions,
+  LighterTokenFetcher,
+  LighterWalletSigner,
+} from './LighterReadOnlyTokenManager.js'
+export {
+  buildReadOnlyTokenMessage,
+  DEFAULT_LIGHTER_API_URL,
+  DEFAULT_READ_ONLY_TOKEN_NAME,
+  defaultLighterTokenFetcher,
+  LighterReadOnlyTokenManager,
+  walletClientSigner,
+} from './LighterReadOnlyTokenManager.js'
+export type {
+  ApiKeyPair,
+  ChangePubKeyResult,
+  LighterSignedBlob,
+  LighterSignerConfig,
+  LighterSignerContext,
+} from './LighterSigner.js'
+export { LighterSigner } from './LighterSigner.js'
+export type {
+  LighterWasmExports,
+  LoadLighterWasmOptions,
+} from './wasmLoader.js'
+export { loadLighterWasm, resetLighterWasmCache } from './wasmLoader.js'