Switch to partial serialized session

Author: aaroncox

src/lib/state/client/wharf.svelte.ts

@@ -12,6 +12,7 @@ import {
 	type AccountCreationPlugin,
 	type CreateAccountOptions,
 	type LoginOptions,
+	type PartialSerializedSession,
 	type SerializedSession,
 	type TransactArgs,
 	type TransactOptions,
@@ -151,7 +152,7 @@ export class WharfState {
 		}
 	}
 
-	public async restore(args?: SerializedSession, options?: LoginOptions) {
+	public async restore(args?: PartialSerializedSession, options?: LoginOptions) {
 		if (!this.sessionKit) {
 			throw new Error('User not initialized');
 		}
@@ -164,12 +165,12 @@ export class WharfState {
 		return session;
 	}
 
-	public async switch(serialized: SerializedSession): Promise<Session> {
+	public async switch(serialized: PartialSerializedSession): Promise<Session> {
 		const session = await this.restore(serialized);
-		this.chainsSession[String(serialized.chain)] = serialized;
 		if (!session) {
 			throw new Error(`Failed to switch session to ${JSON.stringify(serialized)}`);
 		}
+		this.chainsSession[String(session.chain)] = session.serialize();
 		return session;
 	}