feat: use auth token for IPC calls.

Author: shdwmtr

typescript-packages/loader/src/index.ts

@@ -18,6 +18,7 @@ class Bootstrap {
 	logger: import('@steambrew/client/build/logger').default;
 	startTime: number;
 	ctx: Context;
+	millenniumAuthToken: string | undefined = undefined;
 
 	async init() {
 		const loggerModule = await import('@steambrew/client/build/logger');
@@ -62,8 +63,14 @@ class Bootstrap {
 		}
 
 		this.logger.log('Injecting Millennium frontend library...');
-		Object.assign((window.MILLENNIUM_API ??= {}), await import('@steambrew/client'), await import('./millennium-api'));
 
+		const steambrewClientModule = await import('@steambrew/client');
+		const millenniumApiModule = await import('./millennium-api');
+
+		/** Set Auth Token */
+		millenniumApiModule.setMillenniumAuthToken(this.millenniumAuthToken);
+
+		Object.assign((window.MILLENNIUM_API ??= {}), steambrewClientModule, millenniumApiModule);
 		this.logger.log('Millennium API injected successfully.', window.MILLENNIUM_API);
 	}
 
@@ -81,9 +88,11 @@ class Bootstrap {
 		});
 	}
 
-	async StartPreloader(port: number, shimList?: string[]) {
+	async StartPreloader(port: number, millenniumAuthToken: string, shimList?: string[]) {
 		await this.init();
 
+		this.millenniumAuthToken = millenniumAuthToken;
+
 		/** Setup IPC */
 		window.MILLENNIUM_IPC_PORT = port;
 		window.MILLENNIUM_IPC_SOCKET = await this.connectMillenniumBackend(port);
@@ -101,7 +110,10 @@ class Bootstrap {
 			}
 			case Context.Browser: {
 				this.logger.log('Running in browser context...');
-				window.MILLENNIUM_API = await import('./millennium-api');
+				const millenniumApiModule = await import('./millennium-api');
+				millenniumApiModule.setMillenniumAuthToken(this.millenniumAuthToken);
+
+				window.MILLENNIUM_API = millenniumApiModule;
 
 				const browserUtils = await import('./browser-init');
 				await browserUtils.appendAccentColor();

typescript-packages/loader/src/millennium-api.ts

@@ -14,11 +14,21 @@ type Json = string | number | boolean | null | Json[] | { [key: string]: Json };
 type IPC_types = string | number | boolean | Json;
 declare const g_PopupManager: any;
 
+let millenniumAuthToken: string | undefined = undefined;
+
+export function setMillenniumAuthToken(token: string) {
+	millenniumAuthToken = token;
+}
+
 const backendIPC = {
 	postMessage: (messageId: number, contents: any): Promise<any> =>
 		new Promise((resolve) => {
 			const iteration = window.CURRENT_IPC_CALL_COUNT++;
-			const message = { id: messageId, iteration, data: contents };
+			const message = { id: messageId, iteration, data: contents, millenniumAuthToken };
+
+			if (!millenniumAuthToken) {
+				console.warn('No Millennium auth token set. This will cause issues with IPC calls.');
+			}
 
 			const handler = (event: MessageEvent) => {
 				try {