@cubist-labs/cubesigner-sdk 0.4.239 → 0.4.244

package/src/response.ts

@@ -1,7 +1,7 @@
 import type { MfaVote, EnvInterface, MfaReceipts, MfaRequired } from ".";
 import { CubeSignerClient, isManyMfaReceipts } from ".";
 import { encodeToBase64Url } from "./util";
-import type { AcceptedResponse } from "./schema_types";
+import type { AcceptedResponse, AcceptedValue, BinanceDryRun, SignDryRun } from "./schema_types";
 
 /**
  * Response type, which can be either a value of type {@link U}
@@ -29,46 +29,95 @@ export type MapFn<U, V> = (u: U) => V;
  * @returns Response whose value for status code 200 is mapped from U to V
  */
 export function mapResponse<U, V>(resp: Response<U>, mapFn: MapFn<U, V>): Response<V> {
-  if ((resp as AcceptedResponse).accepted?.MfaRequired) {
+  if (asAccepted(resp) !== undefined) {
     return resp as AcceptedResponse;
   } else {
     return mapFn(resp as U);
   }
 }
 
+/**
+ * @param resp The response to check
+ * @returns The {@link AcceptedValue} if the response status code is 202.
+ */
+function asAccepted<U>(resp: Response<U>): AcceptedValue | undefined {
+  const acceptedResp = resp as AcceptedResponse;
+  return acceptedResp.error_code === "SignDryRun" || acceptedResp.error_code === "MfaRequired"
+    ? (acceptedResp.accepted ?? undefined)
+    : undefined;
+}
+
 /**
  * A response of a CubeSigner request.
  */
 export class CubeSignerResponse<U> {
   readonly #env: EnvInterface;
   readonly #requestFn: RequestFn<U>;
-  readonly #resp: U | AcceptedResponse;
+  readonly #resp: Response<U>;
+
+  /**
+   * @returns The {@link AcceptedValue} if the response status code is 202.
+   */
+  asAccepted(): AcceptedValue | undefined {
+    return asAccepted(this.#resp);
+  }
+
+  /**
+   * @returns The associated {@link MfaRequired} value, if the response status code is 202 and the response indicates that MFA is required.
+   */
+  asMfaRequired(): MfaRequired | undefined {
+    return this.asAccepted()?.MfaRequired ?? undefined;
+  }
+
   /**
-   * Optional MFA id. Only set if there is an MFA request associated with the request
+   * @returns The associated {@link SignDryRun} value, if the response status code is 202 and the response is a dry run of a sign operation.
    */
-  readonly #mfaRequired?: MfaRequired;
+  asSignDryRun(): SignDryRun | undefined {
+    return this.asAccepted()?.SignDryRun ?? undefined;
+  }
+
+  /**
+   * @returns The associated {@link BinanceDryRun} value, if the response status code is 202 and the response is a dry run of a sign operation.
+   */
+  asBinanceDryRun(): BinanceDryRun | undefined {
+    return this.asAccepted()?.BinanceDryRun ?? undefined;
+  }
+
+  /**
+   * @returns Whether this response is a "200 Success" (in which case it is safe to call {@link data})
+   */
+  isSuccess(): boolean {
+    return this.asAccepted() === undefined;
+  }
+
+  /**
+   * @returns The underlying {@link MfaRequired} response (if any).
+   */
+  private get mfaRequired() {
+    return this.asAccepted()?.MfaRequired;
+  }
 
   /** @returns The first MFA id associated with this request (if any) */
   mfaId(): string | undefined {
-    return this.#mfaRequired?.id;
+    return this.mfaRequired?.id;
   }
 
   /** @returns The MFA ids associated with this request (if any) */
   mfaIds(): string[] {
-    return this.#mfaRequired?.ids ?? [];
+    return this.mfaRequired?.ids ?? [];
   }
 
   /** @returns True if this request requires an MFA approval */
   requiresMfa(): boolean {
-    return this.#mfaRequired !== undefined;
+    return this.mfaRequired !== undefined;
   }
 
   /**
    * @returns Session information to use for any MFA approval requests (if any was included in the response).
    */
   async mfaClient(): Promise<CubeSignerClient | undefined> {
-    if (this.#mfaRequired === undefined) return;
-    const session = (this.#resp as AcceptedResponse).accepted?.MfaRequired?.session ?? undefined;
+    if (this.mfaRequired === undefined) return;
+    const session = this.asMfaRequired()?.session ?? undefined;
     if (session === undefined) return;
 
     const env = this.#env;
@@ -76,7 +125,7 @@ export class CubeSignerResponse<U> {
       env: {
         "Dev-CubeSignerStack": env,
       },
-      org_id: this.#mfaRequired.org_id,
+      org_id: this.mfaRequired.org_id,
       session_exp: session.expiration,
       session_info: session.session_info,
       token: session.token,
@@ -86,8 +135,10 @@ export class CubeSignerResponse<U> {
 
   /** @returns The response data, if no MFA is required */
   data(): U {
-    if (this.requiresMfa()) {
-      throw new Error("Cannot call `data()` while MFA is required");
+    if (!this.isSuccess()) {
+      throw new Error(
+        "Cannot call `data()` on a 202 Accepted response; use `asMfaRequired()` or `asSignDryRun()`",
+      );
     }
     return this.#resp as U;
   }
@@ -131,7 +182,7 @@ export class CubeSignerResponse<U> {
       return this;
     }
 
-    const mfaOrgId = this.#mfaRequired!.org_id;
+    const mfaOrgId = this.mfaRequired!.org_id;
     const mfaApproval = await client.apiClient.mfaVoteTotp(mfaId, code, vote);
     const mfaConf = mfaApproval.receipt?.confirmation;
 
@@ -174,7 +225,7 @@ export class CubeSignerResponse<U> {
       return this;
     }
 
-    const mfaOrgId = this.#mfaRequired!.org_id;
+    const mfaOrgId = this.mfaRequired!.org_id;
 
     const mfaApproval = await client.apiClient.mfaVoteCs(mfaId, mfaVote);
     const mfaConf = mfaApproval.receipt?.confirmation;
@@ -211,11 +262,10 @@ export class CubeSignerResponse<U> {
    * @param resp The response as returned by the OpenAPI client.
    * @internal
    */
-  protected constructor(env: EnvInterface, requestFn: RequestFn<U>, resp: U | AcceptedResponse) {
+  protected constructor(env: EnvInterface, requestFn: RequestFn<U>, resp: Response<U>) {
     this.#env = env;
     this.#requestFn = requestFn;
     this.#resp = resp;
-    this.#mfaRequired = (this.#resp as AcceptedResponse).accepted?.MfaRequired;
   }
 
   /**