@agimon-ai/po-style-system-tooling 0.2.14-alpha.3 → 0.2.14-alpha.30

package/README.md

@@ -21,7 +21,6 @@ style-system --help
 | `get-css-classes`        | Extract the valid CSS classes from a theme CSS file                     |
 | `get-ui-component`       | Render a component to a screenshot using the app's design-system config |
 | `list-app-components`    | List app-specific components and package components                     |
-| `polish-component`       | Design-system-aware component polish                                    |
 | `generate-oxlint-config` | Generate an oxlint `no-restricted-classes` config from theme tokens     |
 | `mcp-serve`              | Run as an MCP server over stdio                                         |
 

package/dist/cli.cjs

@@ -1,15 +1,15 @@
 #!/usr/bin/env node
-const require_stdio = require("./stdio-BiO1Kn_N.cjs");
+const require_stdio = require("./stdio-Bu0NEnig.cjs");
 let node_fs = require("node:fs");
 let node_path = require("node:path");
 node_path = require_stdio.__toESM(node_path, 1);
 let zod = require("zod");
-let _agimon_ai_foundation_exec = require("@agimon-ai/foundation-exec");
 let _modelcontextprotocol_server = require("@modelcontextprotocol/server");
 require("reflect-metadata");
 let commander = require("commander");
 let chalk = require("chalk");
 chalk = require_stdio.__toESM(chalk, 1);
+let _agimon_ai_foundation_exec = require("@agimon-ai/foundation-exec");
 let node_http = require("node:http");
 let _modelcontextprotocol_node = require("@modelcontextprotocol/node");
 //#region src/utils/print.ts
@@ -425,42 +425,6 @@ const mcpServeCommand = new commander.Command("mcp-serve").description("Start MC
 	}
 });
 //#endregion
-//#region src/commands/polish-component.ts
-/**
-* Polish Component Command
-*
-* DESIGN PATTERNS:
-* - Command pattern with Commander for CLI argument parsing
-* - Async/await pattern for asynchronous operations
-* - Error handling pattern with try-catch and proper exit codes
-*
-* CODING STANDARDS:
-* - Use async action handlers for asynchronous operations
-* - Handle errors gracefully with process.exit()
-*
-* AVOID:
-* - Synchronous blocking operations in action handlers
-* - Missing error handling (always use try-catch)
-* - Hardcoded values (use options or environment variables)
-*/
-const polishComponentCommand = new commander.Command("polish-component").description("Polish a UI component using Antigravity with design system context").argument("<file>", "Path to the component .tsx file to polish").option("-i, --instructions <text>", "Specific polishing instructions (e.g., \"improve spacing\")").action(async (file, options) => {
-	try {
-		const result = await require_stdio.createContainer().get(require_stdio.STYLE_SYSTEM_TYPES.PolishComponentTool).execute({
-			filePath: file,
-			instructions: options.instructions
-		});
-		if (result.isError) {
-			const firstContent = result.content[0];
-			const errorText = firstContent?.type === "text" ? firstContent.text : "Unknown error";
-			throw new Error(errorText);
-		}
-		for (const content of result.content) if (content.type === "text") console.error(content.text);
-	} catch (error) {
-		console.error("Error:", error instanceof Error ? error.message : String(error));
-		process.exit(1);
-	}
-});
-//#endregion
 //#region src/cli.ts
 /**
 * MCP Server CLI Entry Point
@@ -493,7 +457,6 @@ async function main() {
 	program.addCommand(listThemesCommand);
 	program.addCommand(listSharedComponentsCommand);
 	program.addCommand(mcpServeCommand);
-	program.addCommand(polishComponentCommand);
 	await program.parseAsync(process.argv);
 }
 main();

package/dist/cli.mjs

@@ -1,13 +1,13 @@
 #!/usr/bin/env node
-import { A as STYLE_SYSTEM_TYPES, C as getAppDesignSystemConfig, N as DEFAULT_STYLE_SYSTEM_CONFIG, P as TailwindCSSClassesService, R as log, a as STYLE_SYSTEM_CLI_NAME, b as getBundlerServiceFromConfig, i as createServer$1, k as resolveStyleSystemThemePath, o as STYLE_SYSTEM_VERSION, r as generateRestrictedClassesConfig, s as createContainer, t as StdioTransportHandler } from "./stdio-B2oCIjrt.mjs";
+import { A as resolveStyleSystemThemePath, B as STYLE_SYSTEM_TYPES, F as TailwindCSSClassesService, P as DEFAULT_STYLE_SYSTEM_CONFIG, a as STYLE_SYSTEM_CLI_NAME, i as createServer$1, o as STYLE_SYSTEM_VERSION, r as generateRestrictedClassesConfig, s as createContainer, t as StdioTransportHandler, w as getAppDesignSystemConfig, y as getBundlerServiceFromConfig, z as log } from "./stdio-CSCT2Hj0.mjs";
 import { promises } from "node:fs";
 import path, { join } from "node:path";
 import { z } from "zod";
-import { registerManagedProcess } from "@agimon-ai/foundation-exec";
 import { createMcpHandler } from "@modelcontextprotocol/server";
 import "reflect-metadata";
 import { Command } from "commander";
 import * as chalkImport from "chalk";
+import { registerManagedProcess } from "@agimon-ai/foundation-exec";
 import { createServer } from "node:http";
 import { localhostHostValidation, localhostOriginValidation, toNodeHandler } from "@modelcontextprotocol/node";
 //#region src/utils/print.ts
@@ -423,42 +423,6 @@ const mcpServeCommand = new Command("mcp-serve").description("Start MCP server w
 	}
 });
 //#endregion
-//#region src/commands/polish-component.ts
-/**
-* Polish Component Command
-*
-* DESIGN PATTERNS:
-* - Command pattern with Commander for CLI argument parsing
-* - Async/await pattern for asynchronous operations
-* - Error handling pattern with try-catch and proper exit codes
-*
-* CODING STANDARDS:
-* - Use async action handlers for asynchronous operations
-* - Handle errors gracefully with process.exit()
-*
-* AVOID:
-* - Synchronous blocking operations in action handlers
-* - Missing error handling (always use try-catch)
-* - Hardcoded values (use options or environment variables)
-*/
-const polishComponentCommand = new Command("polish-component").description("Polish a UI component using Antigravity with design system context").argument("<file>", "Path to the component .tsx file to polish").option("-i, --instructions <text>", "Specific polishing instructions (e.g., \"improve spacing\")").action(async (file, options) => {
-	try {
-		const result = await createContainer().get(STYLE_SYSTEM_TYPES.PolishComponentTool).execute({
-			filePath: file,
-			instructions: options.instructions
-		});
-		if (result.isError) {
-			const firstContent = result.content[0];
-			const errorText = firstContent?.type === "text" ? firstContent.text : "Unknown error";
-			throw new Error(errorText);
-		}
-		for (const content of result.content) if (content.type === "text") console.error(content.text);
-	} catch (error) {
-		console.error("Error:", error instanceof Error ? error.message : String(error));
-		process.exit(1);
-	}
-});
-//#endregion
 //#region src/cli.ts
 /**
 * Main entry point
@@ -473,7 +437,6 @@ async function main() {
 	program.addCommand(listThemesCommand);
 	program.addCommand(listSharedComponentsCommand);
 	program.addCommand(mcpServeCommand);
-	program.addCommand(polishComponentCommand);
 	await program.parseAsync(process.argv);
 }
 main();

package/dist/index.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_stdio = require("./stdio-BiO1Kn_N.cjs");
+const require_stdio = require("./stdio-Bu0NEnig.cjs");
 let node_fs = require("node:fs");
 let node_path = require("node:path");
 node_path = require_stdio.__toESM(node_path, 1);
@@ -16,7 +16,7 @@ var StoryPreviewService = class {
 	bundlerFactory;
 	previews = /* @__PURE__ */ new Map();
 	constructor(options = {}) {
-		this.workspaceRoot = node_path.default.resolve(options.workspaceRoot ?? require_stdio.TemplatesManagerService.getWorkspaceRootSync());
+		this.workspaceRoot = options.workspaceRoot !== void 0 ? Promise.resolve(node_path.default.resolve(options.workspaceRoot)) : void 0;
 		this.storiesIndexFactory = options.storiesIndexFactory ?? ((workspaceRoot, storyFilePath) => new require_stdio.StoriesIndexService({
 			workspaceRoot,
 			storyFiles: [storyFilePath],
@@ -27,7 +27,8 @@ var StoryPreviewService = class {
 	}
 	async build(input) {
 		if (!STORY_EXPORT_PATTERN.test(input.storyExport)) throw new Error("storyExport must be an exact JavaScript named export");
-		const workspaceRoot = await node_fs.promises.realpath(this.workspaceRoot);
+		const root = await (this.workspaceRoot ??= require_stdio.TemplatesManagerService.getWorkspaceRoot());
+		const workspaceRoot = await node_fs.promises.realpath(root);
 		const appPath = await this.resolveContainedPath(workspaceRoot, input.appPath, "appPath");
 		const storyFilePath = await this.resolveContainedPath(workspaceRoot, input.storyFilePath, "storyFilePath");
 		if (!STORY_FILE_PATTERN.test(storyFilePath)) throw new Error("storyFilePath must name a .stories.ts or .stories.tsx file");
@@ -87,6 +88,244 @@ var StoryPreviewService = class {
 	}
 };
 //#endregion
+//#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) => {
+				require_stdio.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 = require_stdio.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: require_stdio.STYLE_SYSTEM_TYPES.GetComponentVisualTool,
+	get_css_classes: require_stdio.STYLE_SYSTEM_TYPES.GetCSSClassesTool,
+	list_themes: require_stdio.STYLE_SYSTEM_TYPES.ListThemesTool,
+	list_shared_components: require_stdio.STYLE_SYSTEM_TYPES.ListSharedComponentsTool,
+	list_app_components: require_stdio.STYLE_SYSTEM_TYPES.ListAppComponentsTool
+};
+/** Long enough to cover an edit-and-render loop; an idle dev server still holds its module graph and watchers. */
+const IDLE_CLOSE_MS = 6e5;
+/**
+* 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.
+* After IDLE_CLOSE_MS without a call it stops them; the next call starts them again.
+*/
+function createWarmStoryRenderer(options = {}) {
+	const workspaceRoot = options.workspaceRoot ? node_path.default.resolve(options.workspaceRoot) : void 0;
+	let browser;
+	const sharedBrowser = () => {
+		browser ??= require_stdio.launchBrowserWithFallback().catch((error) => {
+			browser = void 0;
+			throw error;
+		});
+		return browser;
+	};
+	let activeBundler;
+	const bundlers = /* @__PURE__ */ new Map();
+	const container = require_stdio.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(require_stdio.STYLE_SYSTEM_TYPES.StyleSystemConfigLoader);
+	const createBundler = (config) => config.bundler ? require_stdio.getBundlerService(config.bundler, workspaceRoot) : require_stdio.createDefaultBundlerService(workspaceRoot);
+	let queue = Promise.resolve();
+	let disposed = false;
+	let captures = 0;
+	let inFlight = 0;
+	let idleTimer;
+	const ensureDevServer = async (appPath) => {
+		const resolved = node_path.default.resolve(workspaceRoot ?? ".", appPath);
+		const { config } = await configLoader.resolveProject(resolved);
+		const key = config.bundler ?? "vite-react";
+		let pending = bundlers.get(key);
+		if (!pending) {
+			pending = workspaceRoot ? require_stdio.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([node_fs.promises.realpath(workspaceRoot), node_fs.promises.realpath(node_path.default.resolve(workspaceRoot, input.appPath))]);
+			if (!require_stdio.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 node_fs.promises.readFile(imagePath)).toString("base64"),
+				mimeType: "image/png"
+			});
+		}
+		return {
+			text,
+			images,
+			isError: result.isError === true
+		};
+	};
+	const enqueue = (task) => {
+		const next = queue.then(task);
+		queue = next.catch(() => void 0);
+		return next;
+	};
+	const armIdleClose = () => {
+		if (--inFlight > 0 || disposed) return;
+		idleTimer = setTimeout(() => void enqueue(reset), IDLE_CLOSE_MS);
+		idleTimer.unref();
+	};
+	return {
+		call(tool, input) {
+			clearTimeout(idleTimer);
+			inFlight++;
+			const next = enqueue(() => run(tool, input));
+			next.then(armIdleClose, armIdleClose);
+			return next;
+		},
+		restart() {
+			return enqueue(reset);
+		},
+		async dispose() {
+			disposed = true;
+			clearTimeout(idleTimer);
+			await queue;
+			await reset();
+		}
+	};
+}
+//#endregion
 exports.BaseBundlerService = require_stdio.BaseBundlerService;
 exports.BaseCSSClassesService = require_stdio.BaseCSSClassesService;
 Object.defineProperty(exports, "CSSClassesServiceFactory", {
@@ -133,12 +372,7 @@ Object.defineProperty(exports, "ListThemesTool", {
 		return require_stdio.ListThemesTool;
 	}
 });
-Object.defineProperty(exports, "PolishComponentTool", {
-	enumerable: true,
-	get: function() {
-		return require_stdio.PolishComponentTool;
-	}
-});
+exports.MAX_STYLE_SYSTEM_VIEWPORT_DIMENSION = require_stdio.MAX_STYLE_SYSTEM_VIEWPORT_DIMENSION;
 exports.STYLE_SYSTEM_TYPES = require_stdio.STYLE_SYSTEM_TYPES;
 exports.StdioTransportHandler = require_stdio.StdioTransportHandler;
 Object.defineProperty(exports, "StoriesIndexService", {
@@ -157,10 +391,14 @@ Object.defineProperty(exports, "StyleSystemConfigLoader", {
 exports.TailwindCSSClassesService = require_stdio.TailwindCSSClassesService;
 exports.ThemeService = require_stdio.ThemeService;
 exports.ViteReactBundlerService = require_stdio.ViteReactBundlerService;
+exports.ViteReactNativeBundlerService = require_stdio.ViteReactNativeBundlerService;
+exports.WARM_RENDERER_TOOLS = WARM_RENDERER_TOOLS;
 exports.createContainer = require_stdio.createContainer;
 exports.createDefaultBundlerService = require_stdio.createDefaultBundlerService;
 exports.createServer = require_stdio.createServer;
+exports.createStoryBrowser = createStoryBrowser;
 exports.createStyleSystemModule = require_stdio.createStyleSystemModule;
+exports.createWarmStoryRenderer = createWarmStoryRenderer;
 exports.extractCustomTokenNames = require_stdio.extractCustomTokenNames;
 exports.generateRestrictedClassesConfig = require_stdio.generateRestrictedClassesConfig;
 exports.getAppDesignSystemConfig = require_stdio.getAppDesignSystemConfig;
@@ -168,6 +406,7 @@ exports.getBundlerService = require_stdio.getBundlerService;
 exports.getBundlerServiceFromConfig = require_stdio.getBundlerServiceFromConfig;
 exports.getSharedComponentTags = require_stdio.getSharedComponentTags;
 exports.hasStyleSystemProjectConfig = require_stdio.hasStyleSystemProjectConfig;
+exports.launchBrowserWithFallback = require_stdio.launchBrowserWithFallback;
 exports.resetStyleSystemConfigCache = require_stdio.resetStyleSystemConfigCache;
 exports.resolveAppDesignSystemConfig = require_stdio.resolveAppDesignSystemConfig;
 exports.resolveStyleSystemThemePath = require_stdio.resolveStyleSystemThemePath;