Enoki SDK updates for API (#16049)

MystenLabs · Feb 1, 2024 · 4bf3e00 · 4bf3e00
1 parent 6d29a68
commit 4bf3e00
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/.changeset/poor-scissors-repeat.md b/.changeset/poor-scissors-repeat.md
@@ -0,0 +1,5 @@
+---
+"@mysten/enoki": patch
+---
+
+Update to latest Enoki API changes
diff --git a/sdk/enoki/src/EnokiClient/index.ts b/sdk/enoki/src/EnokiClient/index.ts
@@ -60,7 +60,9 @@ export class EnokiClient {
 		return this.#fetch<CreateZkLoginNonceApiResponse>('zklogin/nonce', {
 			method: 'POST',
 			body: JSON.stringify({
+				network: input.network,
 				ephemeralPublicKey: input.ephemeralPublicKey.toSuiPublicKey(),
+				additionalEpochs: input.additionalEpochs,
 			}),
 		});
 	}
@@ -72,6 +74,7 @@ export class EnokiClient {
 				[ZKLOGIN_HEADER]: input.jwt,
 			},
 			body: JSON.stringify({
+				network: input.network,
 				ephemeralPublicKey: input.ephemeralPublicKey.toSuiPublicKey(),
 				maxEpoch: input.maxEpoch,
 				randomness: input.randomness,

diff --git a/sdk/enoki/src/EnokiClient/type.ts b/sdk/enoki/src/EnokiClient/type.ts
@@ -6,6 +6,8 @@ import type { ZkLoginSignatureInputs } from '@mysten/sui.js/zklogin';
 
 import type { AuthProvider } from '../EnokiFlow.js';
 
+export type EnokiNetwork = 'mainnet' | 'testnet' | 'devnet';
+
 export interface GetAppApiInput {}
 export interface GetAppApiResponse {
 	authenticationProviders: {
@@ -23,7 +25,9 @@ export interface GetZkLoginApiResponse {
 }
 
 export interface CreateZkLoginNonceApiInput {
+	network?: EnokiNetwork;
 	ephemeralPublicKey: PublicKey;
+	additionalEpochs?: number;
 }
 export interface CreateZkLoginNonceApiResponse {
 	nonce: string;
@@ -34,6 +38,7 @@ export interface CreateZkLoginNonceApiResponse {
 }
 
 export interface CreateZkLoginZkpApiInput {
+	network?: EnokiNetwork;
 	jwt: string;
 	ephemeralPublicKey: PublicKey;
 	randomness: string;
@@ -42,7 +47,7 @@ export interface CreateZkLoginZkpApiInput {
 export interface CreateZkLoginZkpApiResponse extends ZkLoginSignatureInputs {}
 
 export interface CreateSponsoredTransactionBlockApiInput {
-	network?: 'mainnet' | 'testnet';
+	network?: EnokiNetwork;
 	jwt: string;
 	transactionBlockKindBytes: string;
 }