src/lib/state/token.svelte.ts

import { env } from "$env/dynamic/public";

class Token {
	#value = $state("");

	constructor() {
		if (env.PUBLIC_HF_TOKEN) {
			this.#value = env.PUBLIC_HF_TOKEN;
			return;
		}

		this.requestTokenFromParent();
	}

	get value() {
		return this.#value;
	}

	set value(token: string) {
		this.#value = token;
	}

	requestTokenFromParent = (): Promise<boolean> => {
		if (typeof window === "undefined") return Promise.resolve(false);

		return new Promise(resolve => {
			const timeout = window.setTimeout(() => {
				window.removeEventListener("message", handleMessage);
				resolve(false);
			}, 5000);

			const handleMessage = (event: MessageEvent) => {
				if (event.data.type === "INFERENCE_JWT_RESPONSE") {
					const token = event.data.token;
					if (!token || typeof token !== "string") return resolve(false);
					this.value = token;
					window.removeEventListener("message", handleMessage);
					resolve(true);
					window.clearTimeout(timeout);
				}
			};

			window.addEventListener("message", handleMessage);
			window.parent?.postMessage({ type: "INFERENCE_JWT_REQUEST" }, "*");
		});
	};
}

export const token = new Token();