@tinycloud/node-sdk 3.2.6-beta.2 → 3.2.6-beta.4

package/dist/index.cjs

@@ -21653,12 +21653,18 @@ var _TinyCloudNode = class _TinyCloudNode {
   }
   async endSession() {
     const leavingAddress = this.address ?? this._restoredTcSession?.address;
-    this.teardownSession();
+    const leavingAuth = this.auth;
     try {
-      await this.accountRegistryTail;
-      if (this.replicationRuntime) await (await this.replicationRuntime).unbind();
+      this.teardownSession();
     } finally {
-      if (leavingAddress) await this.clearPersistedSessionFor(leavingAddress);
+      try {
+        await this.accountRegistryTail;
+        if (this.replicationRuntime) await (await this.replicationRuntime).unbind();
+      } finally {
+        if (leavingAddress) {
+          await (leavingAuth ? leavingAuth.clearPersistedSession(leavingAddress) : this.config.sessionStorage?.clear(leavingAddress));
+        }
+      }
     }
   }
   /**
@@ -21670,19 +21676,21 @@ var _TinyCloudNode = class _TinyCloudNode {
     this.pendingAccountRegistrySync?.controller.abort();
     const oldGraph = this._serviceGraph;
     const oldCore = this.tc;
-    const sessionKey = createDefaultSessionKey(this.wasmBindings);
-    this.sessionManager = sessionKey.manager;
-    this.sessionKeyId = sessionKey.keyId;
-    this.sessionKeyJwk = sessionKey.jwk;
-    this.auth?.resetSession(
-      sessionKey.manager,
-      sessionKey.keyId,
+    const oldAuth = this.auth;
+    this.auth = null;
+    this.tc = null;
+    this._address = void 0;
+    this._restoredTcSession = void 0;
+    this._activeServiceSession = void 0;
+    oldGraph.retire();
+    oldCore?.retireServices();
+    oldAuth?.resetSession(
+      this.sessionManager,
+      this.sessionKeyId,
       this.explicitHost ? [this.explicitHost] : void 0
     );
     this.config.host = this.explicitHost ?? DEFAULT_HOST;
-    this._address = void 0;
     this._chainId = 1;
-    this._restoredTcSession = void 0;
     this._restoredNodeFeatures = void 0;
     this._activeServiceSession = void 0;
     this._serviceContext = void 0;
@@ -21709,20 +21717,25 @@ var _TinyCloudNode = class _TinyCloudNode {
     this._bootstrapStatus = { skipped: false };
     this.replicationDelegateSession = void 0;
     this.replicationScopeError = void 0;
+    const sessionKey = createDefaultSessionKey(this.wasmBindings);
+    this.sessionManager = sessionKey.manager;
+    this.sessionKeyId = sessionKey.keyId;
+    this.sessionKeyJwk = sessionKey.jwk;
+    oldAuth?.resetSession(
+      sessionKey.manager,
+      sessionKey.keyId,
+      this.explicitHost ? [this.explicitHost] : void 0
+    );
     this._serviceGraph = this.createServiceGraphLifetime();
-    oldGraph.retire();
+    this.auth = oldAuth;
     this.tc = this.auth ? new import_sdk_core9.TinyCloud(this.auth, {
       invokeAny: this.invokeAnyWithRuntimePermissions,
       telemetry: this.config.telemetry
     }) : null;
-    oldCore?.retireServices();
     this._capabilityRegistry = new import_sdk_core9.CapabilityKeyRegistry();
     this._keyProvider = new WasmKeyProvider({ sessionManager: this.sessionManager });
     this._sharingService = this.createReceiveOnlySharing();
   }
-  async clearPersistedSessionFor(address) {
-    await (this.auth ? this.auth.clearPersistedSession(address) : this.config.sessionStorage?.clear(address));
-  }
   /** Throws when a sign-out started after `epoch` was read. */
   assertSessionEpoch(epoch, operation) {
     if (epoch !== this.sessionEpoch) {