@mocanetwork/airkit 1.1.0-beta.0 → 1.2.0-beta.0

package/dist/airkit.cjs.js

@@ -120,6 +120,8 @@ const AirAuthMessageTypes = {
   SIGN_SIWE_MESSAGE_RESPONSE: "air_auth_sign_siwe_message_response",
   CROSS_PARTNER_TOKEN_REQUEST: "air_auth_cross_partner_token_request",
   CROSS_PARTNER_TOKEN_RESPONSE: "air_auth_cross_partner_token_response",
+  GET_PARTNER_ACCESS_TOKEN_REQUEST: "air_auth_get_partner_access_token_request",
+  GET_PARTNER_ACCESS_TOKEN_RESPONSE: "air_auth_get_partner_access_token_response",
   LOGOUT_REQUEST: "air_auth_logout_request",
   LOGOUT_RESPONSE: "air_auth_logout_response",
   RESET_WALLET_COMMUNICATION: "air_auth_reset_wallet_communication"
@@ -131,6 +133,10 @@ const AirMessageTypes = {
   INITIALIZATION_RESPONSE: "air_initialization_response",
   INIT_AUTH_COMMUNICATION: "air_init_auth_communication",
   WALLET_INITIALIZED: "air_wallet_initialized",
+  WALLET_LOGIN_REQUEST: "air_wallet_login_request",
+  WALLET_LOGIN_RESPONSE: "air_wallet_login_response",
+  SETUP_MFA_REQUEST: "air_setup_mfa_request",
+  SETUP_MFA_RESPONSE: "air_setup_mfa_response",
   CLAIM_ID_REQUEST: "air_claim_id_request",
   CLAIM_ID_RESPONSE: "air_claim_id_response",
   DEPLOY_SMART_ACCOUNT_REQUEST: "air_deploy_smart_account_request",
@@ -141,12 +147,13 @@ const AirMessageTypes = {
   EXECUTE_ACTION_RESPONSE: "air_execute_action_response",
   REVOKE_PERMISSIONS_REQUEST: "air_revoke_permissions_request",
   REVOKE_PERMISSIONS_RESPONSE: "air_revoke_permissions_response",
+  LIST_ALL_SESSION_KEY_SCOPES_REQUEST: "air_list_all_session_key_scopes_request",
+  LIST_ALL_SESSION_KEY_SCOPES_RESPONSE: "air_list_all_session_key_scopes_response",
   WALLET_IFRAME_VISIBILITY_REQUEST: "air_wallet_iframe_visibility_request",
   IS_SMART_ACCOUNT_DEPLOYED_REQUEST: "air_is_smart_account_deployed_request",
   IS_SMART_ACCOUNT_DEPLOYED_RESPONSE: "air_is_smart_account_deployed_response",
   LOGOUT_REQUEST: "air_logout_request",
-  LOGOUT_RESPONSE: "air_logout_response",
-  SETUP_MFA_REQUEST: "air_setup_mfa_request"
+  LOGOUT_RESPONSE: "air_logout_response"
 };
 
 class BaseError extends Error {
@@ -1365,12 +1372,15 @@ var AirWalletProviderMessageTypes;
 const ACCOUNT_MESSAGES = [
     AirMessageTypes.INITIALIZATION_RESPONSE,
     AirMessageTypes.WALLET_INITIALIZED,
+    AirMessageTypes.WALLET_LOGIN_RESPONSE,
+    AirMessageTypes.SETUP_MFA_RESPONSE,
     AirMessageTypes.CLAIM_ID_RESPONSE,
     AirMessageTypes.IS_SMART_ACCOUNT_DEPLOYED_RESPONSE,
     AirMessageTypes.DEPLOY_SMART_ACCOUNT_RESPONSE,
     AirMessageTypes.EXECUTE_ACTION_RESPONSE,
     AirMessageTypes.REVOKE_PERMISSIONS_RESPONSE,
     AirMessageTypes.GRANT_PERMISSIONS_RESPONSE,
+    AirMessageTypes.LIST_ALL_SESSION_KEY_SCOPES_RESPONSE,
     AirMessageTypes.WALLET_INITIALIZED,
     AirMessageTypes.WALLET_IFRAME_VISIBILITY_REQUEST,
     AirMessageTypes.LOGOUT_RESPONSE,
@@ -1383,6 +1393,7 @@ const AUTH_MESSAGES = [
     AirAuthMessageTypes.LOGOUT_RESPONSE,
     AirAuthMessageTypes.PARTNER_USER_INFO_RESPONSE,
     AirAuthMessageTypes.CROSS_PARTNER_TOKEN_RESPONSE,
+    AirAuthMessageTypes.GET_PARTNER_ACCESS_TOKEN_RESPONSE,
     AirAuthMessageTypes.IFRAME_VISIBILITY_REQUEST,
 ];
 class AirMessageService {
@@ -1596,9 +1607,27 @@ class AirMessageService {
         walletIframe.contentWindow.postMessage({ type: AirMessageTypes.GRANT_PERMISSIONS_REQUEST, payload }, origin);
         return response;
     }
+    async sendListAllSessionKeyScopesRequest(walletIframe) {
+        const response = firstValueFrom(this.messages$.pipe(filter((msg) => msg.type === AirMessageTypes.LIST_ALL_SESSION_KEY_SCOPES_RESPONSE)));
+        const { origin } = new URL(walletIframe.src);
+        walletIframe.contentWindow.postMessage({ type: AirMessageTypes.LIST_ALL_SESSION_KEY_SCOPES_REQUEST }, origin);
+        return response;
+    }
     onWalletInitialized() {
         return firstValueFrom(this.messages$.pipe(filter((msg) => msg.type === AirMessageTypes.WALLET_INITIALIZED)));
     }
+    sendWalletLoginRequest(walletIframe) {
+        const response = firstValueFrom(this.messages$.pipe(filter((msg) => msg.type === AirMessageTypes.WALLET_LOGIN_RESPONSE)));
+        const { origin } = new URL(walletIframe.src);
+        walletIframe.contentWindow.postMessage({ type: AirMessageTypes.WALLET_LOGIN_REQUEST }, origin);
+        return response;
+    }
+    async sendSetupMfaRequest(walletIframe) {
+        const response = firstValueFrom(this.messages$.pipe(filter((msg) => msg.type === AirMessageTypes.SETUP_MFA_RESPONSE)));
+        const { origin } = new URL(walletIframe.src);
+        walletIframe.contentWindow.postMessage({ type: AirMessageTypes.SETUP_MFA_REQUEST }, origin);
+        return response;
+    }
     async sendClaimIdRequest(walletIframe, payload) {
         const response = firstValueFrom(this.messages$.pipe(filter((msg) => msg.type === AirMessageTypes.CLAIM_ID_RESPONSE)));
         const { origin } = new URL(walletIframe.src);
@@ -1645,6 +1674,12 @@ class AirMessageService {
         this._providerMessage$ = new Subject();
         this._providerEvent$ = new Subject();
     }
+    async sendGetPartnerAccessTokenRequest(authIframe) {
+        const response = firstValueFrom(this.authMessage$.pipe(filter((msg) => msg.type === AirAuthMessageTypes.GET_PARTNER_ACCESS_TOKEN_RESPONSE)));
+        const { origin } = new URL(authIframe.src);
+        authIframe.contentWindow.postMessage({ type: AirAuthMessageTypes.GET_PARTNER_ACCESS_TOKEN_REQUEST }, origin);
+        return response;
+    }
 }
 var AirMessageService$1 = AirMessageService.instance;
 
@@ -1682,7 +1717,7 @@ class AirWalletProvider {
         _AirWalletProvider_ensureWallet.set(this, undefined);
         _AirWalletProvider_getWalletIframeController.set(this, undefined);
         _AirWalletProvider_eventListeners.set(this, undefined);
-        this.listenProviderEventMessage = () => {
+        this.startEventMessageListening = () => {
             AirMessageService$1.providerEvent$.subscribe((message) => {
                 __classPrivateFieldGet(this, _AirWalletProvider_instances, "m", _AirWalletProvider_emit).call(this, message.payload.event, ...[message.payload.data]);
             });
@@ -1829,7 +1864,7 @@ IframeController.defaultState = {
     isVisible: false,
 };
 
-var _AirService_instances, _AirService_loginResult, _AirService_buildEnv, _AirService_enableLogging, _AirService_partnerId, _AirService_authIframeController, _AirService_isAuthInitialized, _AirService_airAuthListener, _AirService_walletIframeController, _AirService_walletInitialization, _AirService_isWalletInitialized, _AirService_walletLoggedInResult, _AirService_airWalletProvider, _AirService_ensureWallet, _AirService_initializeWallet, _AirService_triggerEventListeners, _AirService_triggerAirAuthInitialized, _AirService_triggerAirAuthLoggedIn, _AirService_triggerAirAuthLoggedOut, _AirService_triggerWalletInitialized, _AirService_createLoginResult, _AirService_createWalletInitializedResult, _AirService_cleanUpAuth, _AirService_cleanUpWallet, _AirService_cleanUpAuthWalletCommunication;
+var _AirService_instances, _AirService_loginResult, _AirService_buildEnv, _AirService_enableLogging, _AirService_partnerId, _AirService_authIframeController, _AirService_isAuthInitialized, _AirService_airAuthListener, _AirService_walletIframeController, _AirService_walletInitialization, _AirService_walletLoggedInResult, _AirService_airWalletProvider, _AirService_ensureWallet, _AirService_initializeWallet, _AirService_triggerEventListeners, _AirService_triggerAirAuthInitialized, _AirService_triggerAirAuthLoggedIn, _AirService_triggerAirAuthLoggedOut, _AirService_triggerWalletInitialized, _AirService_createLoginResult, _AirService_createWalletInitializedResult, _AirService_cleanUpAuth, _AirService_cleanUpWallet;
 class AirService {
     constructor({ partnerId }) {
         _AirService_instances.add(this);
@@ -1843,7 +1878,6 @@ class AirService {
         _AirService_airAuthListener.set(this, []);
         _AirService_walletIframeController.set(this, undefined);
         _AirService_walletInitialization.set(this, undefined);
-        _AirService_isWalletInitialized.set(this, false);
         _AirService_walletLoggedInResult.set(this, undefined);
         _AirService_airWalletProvider.set(this, undefined);
         __classPrivateFieldSet(this, _AirService_partnerId, partnerId, "f");
@@ -2017,6 +2051,24 @@ class AirService {
             throw AirServiceError.from(error);
         }
     }
+    /**
+     * @experimental This method is experimental and will change in the future.
+     */
+    async listAllSessionKeyScopes() {
+        if (!__classPrivateFieldGet(this, _AirService_isAuthInitialized, "f"))
+            throw new Error("Service is not initialized");
+        await __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_ensureWallet).call(this);
+        try {
+            const { payload } = await AirMessageService$1.sendListAllSessionKeyScopesRequest(__classPrivateFieldGet(this, _AirService_walletIframeController, "f").iframeElement);
+            if (payload.success === false) {
+                throw new AirServiceError(payload.errorName, payload.errorMessage);
+            }
+            return payload.sessionKeyScopes;
+        }
+        catch (error) {
+            throw AirServiceError.from(error);
+        }
+    }
     /**
      * @experimental This method is experimental and will change in the future.
      */
@@ -2071,9 +2123,14 @@ class AirService {
         return this.provider;
     }
     async preloadWallet() {
-        if (__classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f"))
-            return __classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f");
-        return await __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_ensureWallet).call(this);
+        await __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, { skipWalletLogin: true });
+    }
+    async setupOrUpdateMfa() {
+        await __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, { skipWalletLogin: true });
+        const result = await AirMessageService$1.sendSetupMfaRequest(__classPrivateFieldGet(this, _AirService_walletIframeController, "f").iframeElement);
+        if (result.payload.success === false) {
+            throw new AirServiceError(result.payload.errorName, result.payload.errorMessage);
+        }
     }
     async claimAirId(options) {
         await __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_ensureWallet).call(this);
@@ -2142,50 +2199,51 @@ class AirService {
     clearEventListeners() {
         __classPrivateFieldSet(this, _AirService_airAuthListener, [], "f");
     }
-    // This is for internal use only
-    // @ts-expect-error for unused function
-    async _showWalletWidget() {
-        await __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_ensureWallet).call(this);
-        // return (this.#realmEmbed as any)._showWalletWidget();
-        log.info("_showWalletWidget");
-        return null;
+    async getPartnerAccessToken() {
+        if (!__classPrivateFieldGet(this, _AirService_isAuthInitialized, "f"))
+            throw new Error("Service is not initialized");
+        if (!this.isLoggedIn)
+            throw new Error("No active session to get partner access token");
+        const result = await AirMessageService$1.sendGetPartnerAccessTokenRequest(__classPrivateFieldGet(this, _AirService_authIframeController, "f").iframeElement);
+        if (result.payload.success === false) {
+            throw new AirServiceError(result.payload.errorName, result.payload.errorMessage);
+        }
+        if (!result.payload.partnerAccessToken) {
+            throw new Error("Partner access token not found in response");
+        }
+        return result.payload.partnerAccessToken;
     }
 }
-_AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _AirService_enableLogging = new WeakMap(), _AirService_partnerId = new WeakMap(), _AirService_authIframeController = new WeakMap(), _AirService_isAuthInitialized = new WeakMap(), _AirService_airAuthListener = new WeakMap(), _AirService_walletIframeController = new WeakMap(), _AirService_walletInitialization = new WeakMap(), _AirService_isWalletInitialized = new WeakMap(), _AirService_walletLoggedInResult = new WeakMap(), _AirService_airWalletProvider = new WeakMap(), _AirService_instances = new WeakSet(), _AirService_ensureWallet = 
-// Result of this.#walletInitialization:
-// - If skipWalletLogin is true, it returns the result of empty abstractAccountAddress
-// - If skipWalletLogin is false, it returns the result of abstractAccountAddress from wallet initialization
-// - If #ensureWallet is called again, it returns the result of abstractAccountAddress from the second
-//   wallet initialization if this first wallet initialization doesn't have abstractAccountAddress
-async function _AirService_ensureWallet(option) {
+_AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _AirService_enableLogging = new WeakMap(), _AirService_partnerId = new WeakMap(), _AirService_authIframeController = new WeakMap(), _AirService_isAuthInitialized = new WeakMap(), _AirService_airAuthListener = new WeakMap(), _AirService_walletIframeController = new WeakMap(), _AirService_walletInitialization = new WeakMap(), _AirService_walletLoggedInResult = new WeakMap(), _AirService_airWalletProvider = new WeakMap(), _AirService_instances = new WeakSet(), _AirService_ensureWallet = async function _AirService_ensureWallet(option) {
     if (!this.isInitialized)
         throw new Error("Service not initialized");
     if (!this.isLoggedIn && !option?.skipWalletLogin)
         throw new Error("User not logged in");
     if (!__classPrivateFieldGet(this, _AirService_walletInitialization, "f"))
         __classPrivateFieldSet(this, _AirService_walletInitialization, __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_initializeWallet).call(this, option), "f");
+    if (__classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f"))
+        return __classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f");
     try {
         const result = await __classPrivateFieldGet(this, _AirService_walletInitialization, "f");
-        // Note: Checking for abstractAccountAddress from the result and this.#walletLoggedInResult to avoid race condition
-        if (option?.skipWalletLogin ||
-            result?.abstractAccountAddress ||
-            __classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f")?.abstractAccountAddress) {
-            return result || __classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f");
-        }
-        const walletInitResult = await AirMessageService$1.onWalletInitialized();
-        if (walletInitResult.payload.success !== true) {
-            throw new AirServiceError(walletInitResult.payload.errorName, walletInitResult.payload.errorMessage);
+        if (option?.skipWalletLogin || result.isWalletLoggedIn) {
+            return __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_createWalletInitializedResult).call(this, result);
         }
-        return __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_createWalletInitializedResult).call(this, walletInitResult.payload);
     }
     catch (error) {
         __classPrivateFieldSet(this, _AirService_walletInitialization, null, "f");
         log.error("Error initializing wallet", error);
         throw error;
     }
+    if (__classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f"))
+        return __classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f");
+    const walletLoginResult = await AirMessageService$1.sendWalletLoginRequest(__classPrivateFieldGet(this, _AirService_walletIframeController, "f").iframeElement);
+    if (walletLoginResult.payload.success !== true) {
+        throw new AirServiceError(walletLoginResult.payload.errorName, walletLoginResult.payload.errorMessage);
+    }
+    return __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_createWalletInitializedResult).call(this, walletLoginResult.payload);
 }, _AirService_initializeWallet = async function _AirService_initializeWallet(option) {
-    if (__classPrivateFieldGet(this, _AirService_isWalletInitialized, "f"))
-        throw new AirServiceError("ALREADY_INITIALISED", "Already initialized");
+    if (__classPrivateFieldGet(this, _AirService_walletInitialization, "f"))
+        throw new Error("Already initializing");
     const { walletUrl } = AIR_URLS[__classPrivateFieldGet(this, _AirService_buildEnv, "f")];
     log.info(walletUrl, "url loaded");
     const walletIframeOrigin = new URL(walletUrl).origin;
@@ -2201,21 +2259,11 @@ async function _AirService_ensureWallet(option) {
                     walletIframeController.updateIframeState();
                     break;
                 }
-                case AirMessageTypes.WALLET_INITIALIZED: {
-                    if (msg.payload.success === true && msg.payload.addresses?.aa) {
+                case AirMessageTypes.WALLET_LOGIN_RESPONSE: {
+                    if (msg.payload.success === true && !__classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f")) {
                         __classPrivateFieldSet(this, _AirService_walletLoggedInResult, __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_createWalletInitializedResult).call(this, msg.payload), "f");
                         __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_triggerWalletInitialized).call(this, __classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f"));
                     }
-                    if (msg.payload.success === false) {
-                        if (!__classPrivateFieldGet(this, _AirService_isWalletInitialized, "f")) {
-                            // MFA fail case, also reset wallet port
-                            __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_cleanUpAuthWalletCommunication).call(this);
-                        }
-                        else {
-                            // Logout wallet and destroy the wallet iframe
-                            await __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_cleanUpWallet).call(this);
-                        }
-                    }
                     break;
                 }
             }
@@ -2240,15 +2288,18 @@ async function _AirService_ensureWallet(option) {
             };
             window.addEventListener("message", handleAuthMessage);
         });
-        __classPrivateFieldGet(this, _AirService_airWalletProvider, "f").listenProviderEventMessage();
+        __classPrivateFieldGet(this, _AirService_airWalletProvider, "f").startEventMessageListening();
         const walletInitPromise = AirMessageService$1.onWalletInitialized();
         AirMessageService$1.setupIframeCommunication(__classPrivateFieldGet(this, _AirService_authIframeController, "f").iframeElement, __classPrivateFieldGet(this, _AirService_walletIframeController, "f").iframeElement, option?.skipWalletLogin ?? false);
         const walletInitResult = await walletInitPromise;
         if (walletInitResult.payload.success !== true) {
             throw new AirServiceError(walletInitResult.payload.errorName, walletInitResult.payload.errorMessage);
         }
-        __classPrivateFieldSet(this, _AirService_isWalletInitialized, true, "f");
-        return __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_createWalletInitializedResult).call(this, walletInitResult.payload);
+        if (!__classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f") && walletInitResult.payload.isWalletLoggedIn) {
+            __classPrivateFieldSet(this, _AirService_walletLoggedInResult, __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_createWalletInitializedResult).call(this, walletInitResult.payload), "f");
+            __classPrivateFieldGet(this, _AirService_instances, "m", _AirService_triggerWalletInitialized).call(this, __classPrivateFieldGet(this, _AirService_walletLoggedInResult, "f"));
+        }
+        return walletInitResult.payload;
     }
     catch (error) {
         log.error("Error initializing wallet", error);
@@ -2282,8 +2333,14 @@ async function _AirService_ensureWallet(option) {
         isMFASetup: payload.isMFASetup,
     };
 }, _AirService_createWalletInitializedResult = function _AirService_createWalletInitializedResult(payload) {
+    if ("isWalletLoggedIn" in payload && !payload.isWalletLoggedIn) {
+        return {
+            abstractAccountAddress: null,
+            isMFASetup: __classPrivateFieldGet(this, _AirService_loginResult, "f")?.isMFASetup ?? false,
+        };
+    }
     return {
-        abstractAccountAddress: payload.addresses?.aa ?? null,
+        abstractAccountAddress: payload.addresses.aa,
         isMFASetup: payload.isMFASetup,
     };
 }, _AirService_cleanUpAuth = async function _AirService_cleanUpAuth() {
@@ -2308,18 +2365,14 @@ async function _AirService_ensureWallet(option) {
         __classPrivateFieldGet(this, _AirService_walletIframeController, "f").destroy();
         __classPrivateFieldSet(this, _AirService_walletIframeController, undefined, "f");
     }
-    // Close the message service
-    AirMessageService$1.closeAirObservables();
-    __classPrivateFieldSet(this, _AirService_walletLoggedInResult, undefined, "f");
-    __classPrivateFieldSet(this, _AirService_walletInitialization, undefined, "f");
-    __classPrivateFieldSet(this, _AirService_isWalletInitialized, false, "f");
-}, _AirService_cleanUpAuthWalletCommunication = function _AirService_cleanUpAuthWalletCommunication() {
-    if (!__classPrivateFieldGet(this, _AirService_isAuthInitialized, "f"))
-        return;
     const authIframeElement = __classPrivateFieldGet(this, _AirService_authIframeController, "f")?.iframeElement;
     if (isElement(authIframeElement) && window.document.body.contains(authIframeElement)) {
         AirMessageService$1.sendResetAuthServiceWalletCommunication(authIframeElement);
     }
+    // Close the message service
+    AirMessageService$1.closeAirObservables();
+    __classPrivateFieldSet(this, _AirService_walletLoggedInResult, undefined, "f");
+    __classPrivateFieldSet(this, _AirService_walletInitialization, undefined, "f");
 };
 
 exports.AirService = AirService;