@agimon-ai/po-style-system-tooling 0.2.14-alpha.2 → 0.2.14-alpha.21

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { A as STYLE_SYSTEM_TYPES, C as getAppDesignSystemConfig, D as resolveAppDesignSystemConfig, E as resetStyleSystemConfigCache, F as BaseCSSClassesService, I as TemplatesManagerService, L as isContained, M as CSSClassesServiceFactory, N as DEFAULT_STYLE_SYSTEM_CONFIG, O as StyleSystemConfigLoader, P as TailwindCSSClassesService, S as BaseBundlerService, T as hasStyleSystemProjectConfig, _ as ThemeService, b as getBundlerServiceFromConfig, c as createStyleSystemModule, d as ListSharedComponentsTool, f as GetComponentVisualTool, g as ComponentRendererService, h as GetUiComponentService, i as createServer, j as StoriesIndexService, k as resolveStyleSystemThemePath, l as PolishComponentTool, m as ListThemesTool, n as extractCustomTokenNames, p as GetCSSClassesTool, r as generateRestrictedClassesConfig, s as createContainer, t as StdioTransportHandler, u as ListAppComponentsTool, v as createDefaultBundlerService, w as getSharedComponentTags, x as ViteReactBundlerService, y as getBundlerService } from "./stdio-Dgaj2-_8.mjs";
+import { A as resolveStyleSystemThemePath, B as STYLE_SYSTEM_TYPES, C as launchBrowserWithFallback, D as resetStyleSystemConfigCache, E as hasStyleSystemProjectConfig, F as TailwindCSSClassesService, I as BaseCSSClassesService, L as TemplatesManagerService, M as StoriesIndexService, N as CSSClassesServiceFactory, O as resolveAppDesignSystemConfig, P as DEFAULT_STYLE_SYSTEM_CONFIG, R as isContained, S as BaseBundlerService, T as getSharedComponentTags, _ as createDefaultBundlerService, b as ViteReactNativeBundlerService, c as createStyleSystemModule, d as GetComponentVisualTool, f as GetCSSClassesTool, g as ThemeService, h as ComponentRendererService, i as createServer, j as MAX_STYLE_SYSTEM_VIEWPORT_DIMENSION, k as StyleSystemConfigLoader, l as ListAppComponentsTool, m as GetUiComponentService, n as extractCustomTokenNames, p as ListThemesTool, r as generateRestrictedClassesConfig, s as createContainer, t as StdioTransportHandler, u as ListSharedComponentsTool, v as getBundlerService, w as getAppDesignSystemConfig, x as ViteReactBundlerService, y as getBundlerServiceFromConfig, z as log } from "./stdio-DBEflKj3.mjs";
 import { promises } from "node:fs";
 import path from "node:path";
 import { createHash, randomUUID } from "node:crypto";
@@ -85,4 +85,226 @@ var StoryPreviewService = class {
 	}
 };
 //#endregion
-export { BaseBundlerService, BaseCSSClassesService, CSSClassesServiceFactory, ComponentRendererService, DEFAULT_STYLE_SYSTEM_CONFIG, GetCSSClassesTool, GetComponentVisualTool, GetUiComponentService, ListAppComponentsTool, ListSharedComponentsTool, ListThemesTool, PolishComponentTool, STYLE_SYSTEM_TYPES, StdioTransportHandler, StoriesIndexService, StoryPreviewService, StyleSystemConfigLoader, TailwindCSSClassesService, ThemeService, ViteReactBundlerService, createContainer, createDefaultBundlerService, createServer, createStyleSystemModule, extractCustomTokenNames, generateRestrictedClassesConfig, getAppDesignSystemConfig, getBundlerService, getBundlerServiceFromConfig, getSharedComponentTags, hasStyleSystemProjectConfig, resetStyleSystemConfigCache, resolveAppDesignSystemConfig, resolveStyleSystemThemePath };
+//#region src/utils/storyBrowser.ts
+const BROWSER_IDLE_MS = 18e5;
+/** Server-owned Chrome. Each capture borrows it until its context cleanup settles. */
+function createStoryBrowser() {
+	let pending;
+	let closing;
+	let timer;
+	let active = 0;
+	let disposed = false;
+	let drained = Promise.resolve();
+	let resolveDrained;
+	const clearIdle = () => {
+		clearTimeout(timer);
+		timer = void 0;
+	};
+	const close = () => {
+		if (closing) return closing;
+		const generation = pending;
+		if (!generation) return Promise.resolve();
+		closing = generation.then((browser) => browser.close()).then(() => {
+			if (pending === generation) pending = void 0;
+		}).finally(() => {
+			closing = void 0;
+		});
+		return closing;
+	};
+	const idle = () => {
+		if (disposed || active !== 0 || !pending) return;
+		clearIdle();
+		const generation = pending;
+		timer = setTimeout(() => {
+			timer = void 0;
+			if (active !== 0 || pending !== generation) return;
+			close().catch((error) => {
+				log.error("[story-browser] Idle close failed:", error);
+				idle();
+			});
+		}, BROWSER_IDLE_MS);
+		timer.unref();
+	};
+	return {
+		async acquire() {
+			if (disposed) throw new Error("The story browser is shut down.");
+			clearIdle();
+			if (active++ === 0) drained = new Promise((resolve) => {
+				resolveDrained = resolve;
+			});
+			let released = false;
+			const release = () => {
+				if (released) return;
+				released = true;
+				if (--active === 0) {
+					resolveDrained?.();
+					idle();
+				}
+			};
+			try {
+				if (closing) await closing;
+				if (disposed) throw new Error("The story browser is shut down.");
+				if (!pending) {
+					const generation = launchBrowserWithFallback().then((browser) => {
+						browser.once("disconnected", () => {
+							if (pending === generation) {
+								pending = void 0;
+								clearIdle();
+							}
+						});
+						return browser;
+					}).catch((error) => {
+						if (pending === generation) pending = void 0;
+						throw error;
+					});
+					pending = generation;
+				}
+				const browser = await pending;
+				if (disposed) throw new Error("The story browser is shut down.");
+				return {
+					browser,
+					release
+				};
+			} catch (error) {
+				release();
+				throw error;
+			}
+		},
+		async dispose() {
+			disposed = true;
+			clearIdle();
+			await drained;
+			await close();
+		}
+	};
+}
+//#endregion
+//#region src/services/WarmStoryRenderer/WarmStoryRenderer.ts
+/** The style-system tools a warm host exposes, by tool name. */
+const WARM_RENDERER_TOOLS = {
+	get_component_visual: STYLE_SYSTEM_TYPES.GetComponentVisualTool,
+	get_css_classes: STYLE_SYSTEM_TYPES.GetCSSClassesTool,
+	list_themes: STYLE_SYSTEM_TYPES.ListThemesTool,
+	list_shared_components: STYLE_SYSTEM_TYPES.ListSharedComponentsTool,
+	list_app_components: STYLE_SYSTEM_TYPES.ListAppComponentsTool
+};
+/**
+* Keeps one browser and one Vite dev server warm across renders, so a render after the first
+* skips the browser launch and the static build. Rendered screenshots come back as base64.
+*/
+function createWarmStoryRenderer(options = {}) {
+	const workspaceRoot = options.workspaceRoot ? path.resolve(options.workspaceRoot) : void 0;
+	let browser;
+	const sharedBrowser = () => {
+		browser ??= launchBrowserWithFallback().catch((error) => {
+			browser = void 0;
+			throw error;
+		});
+		return browser;
+	};
+	let activeBundler;
+	const bundlers = /* @__PURE__ */ new Map();
+	const container = createContainer({
+		...options,
+		workspaceRoot,
+		sharedBrowser: options.sharedBrowser ?? sharedBrowser,
+		bundlerFactory: () => {
+			if (!activeBundler) throw new Error("The story bundler has not been initialized.");
+			return activeBundler;
+		}
+	});
+	const configLoader = container.get(STYLE_SYSTEM_TYPES.StyleSystemConfigLoader);
+	const createBundler = (config) => config.bundler ? getBundlerService(config.bundler, workspaceRoot) : createDefaultBundlerService(workspaceRoot);
+	let queue = Promise.resolve();
+	let disposed = false;
+	let captures = 0;
+	const ensureDevServer = async (appPath) => {
+		const resolved = path.resolve(workspaceRoot ?? ".", appPath);
+		const { config } = await configLoader.resolveProject(resolved);
+		const key = config.bundler ?? "vite-react";
+		let pending = bundlers.get(key);
+		if (!pending) {
+			pending = workspaceRoot ? getBundlerServiceFromConfig({
+				workspaceRoot,
+				configLoader,
+				defaultFactory: () => createBundler(config)
+			}) : Promise.resolve(createBundler(config));
+			bundlers.set(key, pending);
+			pending.catch(() => bundlers.delete(key));
+		}
+		const bundler = await pending;
+		if (activeBundler && activeBundler !== bundler && activeBundler.isServerRunning()) await activeBundler.cleanup();
+		activeBundler = bundler;
+		const recycle = bundler.getBundlerId() === "vite" && captures >= 100;
+		if (bundler.isServerRunning() && bundler.getCurrentAppPath() === resolved && !recycle) return;
+		if (bundler.isServerRunning()) await bundler.cleanup();
+		captures = 0;
+		await bundler.startDevServer(resolved);
+	};
+	const reset = async () => {
+		const open = browser;
+		browser = void 0;
+		captures = 0;
+		await Promise.allSettled([open?.then((instance) => instance.close()), ...[...bundlers.values()].map((pending) => pending.then((bundler) => bundler.cleanup()))]);
+	};
+	const run = async (tool, input) => {
+		if (disposed) throw new Error("The story renderer is shut down.");
+		if (workspaceRoot && typeof input.appPath === "string") {
+			const [root, appPath] = await Promise.all([promises.realpath(workspaceRoot), promises.realpath(path.resolve(workspaceRoot, input.appPath))]);
+			if (!isContained(root, appPath)) throw new Error("appPath must remain inside the configured workspace root.");
+			input = {
+				...input,
+				appPath
+			};
+		}
+		const visualAppPath = tool === "get_component_visual" && typeof input.appPath === "string" ? input.appPath : void 0;
+		const execute = async () => {
+			if (visualAppPath) {
+				await ensureDevServer(visualAppPath);
+				captures++;
+			}
+			const result = await container.get(WARM_RENDERER_TOOLS[tool]).execute(input);
+			return {
+				result,
+				text: result.content.flatMap((item) => item.type === "text" && item.text ? [item.text] : []).join("\n\n")
+			};
+		};
+		let { result, text } = await execute();
+		if (visualAppPath && result.isError && text.includes("Failed to render component")) {
+			await reset();
+			({result, text} = await execute());
+		}
+		const images = [];
+		if (tool === "get_component_visual" && !result.isError) {
+			const imagePath = /"imagePath":\s*"([^"]+)"/u.exec(text)?.[1];
+			if (imagePath) images.push({
+				data: (await promises.readFile(imagePath)).toString("base64"),
+				mimeType: "image/png"
+			});
+		}
+		return {
+			text,
+			images,
+			isError: result.isError === true
+		};
+	};
+	return {
+		call(tool, input) {
+			const next = queue.then(() => run(tool, input));
+			queue = next.catch(() => void 0);
+			return next;
+		},
+		restart() {
+			const next = queue.then(reset);
+			queue = next.catch(() => void 0);
+			return next;
+		},
+		async dispose() {
+			disposed = true;
+			await queue;
+			await reset();
+		}
+	};
+}
+//#endregion
+export { BaseBundlerService, BaseCSSClassesService, CSSClassesServiceFactory, ComponentRendererService, DEFAULT_STYLE_SYSTEM_CONFIG, GetCSSClassesTool, GetComponentVisualTool, GetUiComponentService, ListAppComponentsTool, ListSharedComponentsTool, ListThemesTool, MAX_STYLE_SYSTEM_VIEWPORT_DIMENSION, STYLE_SYSTEM_TYPES, StdioTransportHandler, StoriesIndexService, StoryPreviewService, StyleSystemConfigLoader, TailwindCSSClassesService, ThemeService, ViteReactBundlerService, ViteReactNativeBundlerService, WARM_RENDERER_TOOLS, createContainer, createDefaultBundlerService, createServer, createStoryBrowser, createStyleSystemModule, createWarmStoryRenderer, extractCustomTokenNames, generateRestrictedClassesConfig, getAppDesignSystemConfig, getBundlerService, getBundlerServiceFromConfig, getSharedComponentTags, hasStyleSystemProjectConfig, launchBrowserWithFallback, resetStyleSystemConfigCache, resolveAppDesignSystemConfig, resolveStyleSystemThemePath };