@agimon-ai/po-style-system-tooling 0.2.14-alpha.1 → 0.2.14-alpha.11

package/LICENSE

@@ -1,21 +1,149 @@
-MIT License
-
-Copyright (c) 2026 Vuong Ngo
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+License text copyright © 2017 MariaDB Corporation Ab, All Rights Reserved.
+"Business Source License" is a trademark of MariaDB Corporation Ab.
+
+Parameters
+
+Licensor:             Agimon Pty Ltd
+Licensed Work:        Po, the software in the Po source repository and the
+                      packages released from it, in each version released by
+                      the Licensor.
+                      The Licensed Work is (c) 2026 Agimon Pty Ltd.
+                      Material identified as third-party in the "Third-party
+                      components" section at the end of this file or in a
+                      NOTICE file, and any package or directory whose own
+                      license file grants a different license, is not part of
+                      the Licensed Work and remains under its own license.
+Additional Use Grant: You may make production use of the Licensed Work if
+  your use is Eligible Use. "Eligible Use" is production use that falls
+  within at least one of the following:
+
+  (a) Personal Use. Use by an individual for their own personal purposes,
+      and not on behalf of, or for the benefit of, an employer, client, or
+      other Organization. Use by an individual in the course of employment
+      or engagement by an Organization is use by that Organization.
+
+  (b) Small Organization Use. Use by or on behalf of an Organization whose
+      Annual Revenue is less than USD 1,000,000.
+
+  (c) Non-profit, Education and Research Use. Use, regardless of Annual
+      Revenue, by (i) an Organization registered or recognized as a
+      non-profit or charitable organization under the laws of its
+      jurisdiction, for its non-profit purposes; (ii) an accredited school,
+      college, or university, for teaching, learning, or academic research;
+      or (iii) a publicly funded research institution, for its research.
+
+  (d) Service Provider Use. Use by a consultant, contractor, agency, or other
+      service provider, regardless of its own Annual Revenue, solely to
+      build, deploy, host, operate, or support the Licensed Work for a client
+      whose own use would be Eligible Use under (a), (b), or (c). A service
+      provider's use for its own purposes is Eligible Use only if the service
+      provider itself qualifies under (b) or (c).
+
+
+  Hosted services. Offering the Licensed Work to third parties as a hosted
+  or managed service whose primary value is the Licensed Work is Eligible
+  Use only if each such third party's own use would be Eligible Use.
+
+  Definitions. "Organization" means any company, partnership, sole
+  proprietorship, trust, association, government body, or other entity, and
+  includes an individual acting in a business or professional capacity.
+  "Affiliate" means any entity that directly or indirectly controls, is
+  controlled by, or is under common control with an Organization, where
+  "control" means ownership of more than 50% of the voting interests of an
+  entity or the power to direct its management and policies by contract or
+  otherwise. "Annual Revenue" means the total gross revenue of an
+  Organization and all of its Affiliates combined, from all sources and
+  whether or not related to the Licensed Work, for its most recently
+  completed fiscal year, or, if it has not completed a fiscal year, for the
+  twelve months before the use (or since its formation, if shorter),
+  converted into US dollars at the average exchange rate for that period.
+  Proceeds of equity or debt financing are not revenue.
+
+  Threshold. If an Organization stops qualifying under (b), its existing
+  production use remains Eligible Use for 90 days after it stops
+  qualifying, after which it must obtain a commercial license or stop
+  production use. Use that was Eligible Use when it occurred never requires
+  a commercial license after the fact.
+Change Date:          Four years from the date a version of the Licensed Work
+                      is first publicly released by the Licensor.
+Change License:       Apache License, Version 2.0
+
+For information about alternative licensing arrangements for the Licensed
+Work, please visit: https://agimon.ai/licensing
+
+Notice
+
+The Business Source License (this document, or the "License") is not an Open
+Source license. However, the Licensed Work will eventually be made available
+under an Open Source License, as stated in this License.
+
+-----------------------------------------------------------------------------
+
+Business Source License 1.1
+
+Terms
+
+The Licensor hereby grants you the right to copy, modify, create derivative
+works, redistribute, and make non-production use of the Licensed Work. The
+Licensor may make an Additional Use Grant, above, permitting limited
+production use.
+
+Effective on the Change Date, or the fourth anniversary of the first publicly
+available distribution of a specific version of the Licensed Work under this
+License, whichever comes first, the Licensor hereby grants you rights under
+the terms of the Change License, and the rights granted in the paragraph
+above terminate.
+
+If your use of the Licensed Work does not comply with the requirements
+currently in effect as described in this License, you must purchase a
+commercial license from the Licensor, its affiliated entities, or authorized
+resellers, or you must refrain from using the Licensed Work.
+
+All copies of the original and modified Licensed Work, and derivative works
+of the Licensed Work, are subject to this License. This License applies
+separately for each version of the Licensed Work and the Change Date may vary
+for each version of the Licensed Work released by Licensor.
+
+You must conspicuously display this License on each original or modified copy
+of the Licensed Work. If you receive the Licensed Work in original or
+modified form from a third party, the terms and conditions set forth in this
+License apply to your use of that work.
+
+Any use of the Licensed Work in violation of this License will automatically
+terminate your rights under this License for the current and all other
+versions of the Licensed Work.
+
+This License does not grant you any right in any trademark or logo of
+Licensor or its affiliates (provided that you may use a trademark or logo of
+Licensor as expressly required by this License).
+
+TO THE EXTENT PERMITTED BY APPLICABLE LAW, THE LICENSED WORK IS PROVIDED ON
+AN “AS IS” BASIS. LICENSOR HEREBY DISCLAIMS ALL WARRANTIES AND CONDITIONS,
+EXPRESS OR IMPLIED, INCLUDING (WITHOUT LIMITATION) WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NON-INFRINGEMENT, AND
+TITLE.
+
+MariaDB hereby grants you permission to use this License’s text to license
+your works, and to refer to it using the trademark “Business Source License”,
+as long as you comply with the Covenants of Licensor below.
+
+Covenants of Licensor
+
+In consideration of the right to use this License’s text and the “Business
+Source License” name and trademark, Licensor covenants to MariaDB, and to all
+other recipients of the licensed work to be provided by Licensor:
+
+1. To specify as the Change License the GPL Version 2.0 or any later version,
+   or a license that is compatible with GPL Version 2.0 or a later version,
+   where “compatible” means that software provided under the Change License can
+   be included in a program with software provided under GPL Version 2.0 or a
+   later version. Licensor may specify additional Change Licenses without
+   limitation.
+
+2. To either: (a) specify an additional grant of rights to use that does not
+   impose any additional restriction on the right granted in this License, as
+   the Additional Use Grant; or (b) insert the text “None”.
+
+3. To specify a Change Date.
+
+4. Not to modify this License in any other way.

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                                         |
 
@@ -146,4 +145,4 @@ these commands.
 
 ## License
 
-MIT
+Business Source License 1.1: free for personal use, organisations under USD 1M revenue, non-profits and education. See [LICENSE](LICENSE) and https://agimon.ai/licensing.

package/dist/cli.cjs

@@ -1,15 +1,15 @@
 #!/usr/bin/env node
-const require_stdio = require("./stdio-BqvjQLSc.cjs");
+const require_stdio = require("./stdio-DkQ5zJDm.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
@@ -138,7 +138,8 @@ const REQUESTABLE_COLOR_SCHEMES = [LIGHT_COLOR_SCHEME, DARK_COLOR_SCHEME];
 const getUiComponentCommand = new commander.Command("get-ui-component").description("Get a screenshot of a UI component rendered with app-specific design system configuration. Returns screenshot path and story file content.").requiredOption("-c, --component-name <name>", "The name of the component to capture (e.g., \"Button\", \"Card\")").requiredOption("-a, --app-path <path>", "The app path (relative or absolute) to load design system config from (e.g., \"apps/agiflow-app\")").option("-s, --story-name <name>", "The story name to render (e.g., \"Playground\", \"Default\")", "Playground").option("-d, --dark-mode", "Request dark mode for automatic-theme apps").option("--color-scheme <scheme>", "Request light or dark mode for automatic-theme apps").option("--width <pixels>", "Override the configured viewport width", Number).option("--height <pixels>", "Override the configured viewport height", Number).action(async (options) => {
 	try {
 		if (options.colorScheme !== void 0 && !REQUESTABLE_COLOR_SCHEMES.includes(options.colorScheme)) throw new Error("--color-scheme must be light or dark");
-		const tool = require_stdio.createContainer().get(require_stdio.STYLE_SYSTEM_TYPES.GetComponentVisualTool);
+		const sessionTemp = process.env.PO_SESSION_TEMP_DIR;
+		const tool = require_stdio.createContainer({ temporaryDirectory: sessionTemp ? (0, node_path.join)(sessionTemp, "previews") : void 0 }).get(require_stdio.STYLE_SYSTEM_TYPES.GetComponentVisualTool);
 		const darkMode = options.colorScheme === DARK_COLOR_SCHEME ? true : options.colorScheme === LIGHT_COLOR_SCHEME ? false : options.darkMode || void 0;
 		const result = await tool.execute({
 			componentName: options.componentName,
@@ -405,7 +406,11 @@ const mcpServeCommand = new commander.Command("mcp-serve").description("Start MC
 			console.error(`Dev server started at ${url} (port: ${port})`);
 		}
 		if (transportType === "stdio") {
-			const container = require_stdio.createContainer({ defaultThemePath: parsed.themePath });
+			const sessionTemp = process.env.PO_SESSION_TEMP_DIR;
+			const container = require_stdio.createContainer({
+				defaultThemePath: parsed.themePath,
+				temporaryDirectory: sessionTemp ? (0, node_path.join)(sessionTemp, "previews") : void 0
+			});
 			await startServer(new require_stdio.StdioTransportHandler(() => require_stdio.createServer(container)), bundlerService, devServerPort);
 		} else if (transportType === "http" && parsed.port !== void 0) {
 			const container = require_stdio.createContainer({ defaultThemePath: parsed.themePath });
@@ -420,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
@@ -488,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-B7ni5Bjf.mjs";
+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, i as createServer$1, k as resolveStyleSystemThemePath, o as STYLE_SYSTEM_VERSION, r as generateRestrictedClassesConfig, s as createContainer, t as StdioTransportHandler, y as getBundlerServiceFromConfig } from "./stdio-B_NECMIh.mjs";
 import { promises } from "node:fs";
-import path from "node:path";
+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
@@ -136,7 +136,8 @@ const REQUESTABLE_COLOR_SCHEMES = [LIGHT_COLOR_SCHEME, DARK_COLOR_SCHEME];
 const getUiComponentCommand = new Command("get-ui-component").description("Get a screenshot of a UI component rendered with app-specific design system configuration. Returns screenshot path and story file content.").requiredOption("-c, --component-name <name>", "The name of the component to capture (e.g., \"Button\", \"Card\")").requiredOption("-a, --app-path <path>", "The app path (relative or absolute) to load design system config from (e.g., \"apps/agiflow-app\")").option("-s, --story-name <name>", "The story name to render (e.g., \"Playground\", \"Default\")", "Playground").option("-d, --dark-mode", "Request dark mode for automatic-theme apps").option("--color-scheme <scheme>", "Request light or dark mode for automatic-theme apps").option("--width <pixels>", "Override the configured viewport width", Number).option("--height <pixels>", "Override the configured viewport height", Number).action(async (options) => {
 	try {
 		if (options.colorScheme !== void 0 && !REQUESTABLE_COLOR_SCHEMES.includes(options.colorScheme)) throw new Error("--color-scheme must be light or dark");
-		const tool = createContainer().get(STYLE_SYSTEM_TYPES.GetComponentVisualTool);
+		const sessionTemp = process.env.PO_SESSION_TEMP_DIR;
+		const tool = createContainer({ temporaryDirectory: sessionTemp ? join(sessionTemp, "previews") : void 0 }).get(STYLE_SYSTEM_TYPES.GetComponentVisualTool);
 		const darkMode = options.colorScheme === DARK_COLOR_SCHEME ? true : options.colorScheme === LIGHT_COLOR_SCHEME ? false : options.darkMode || void 0;
 		const result = await tool.execute({
 			componentName: options.componentName,
@@ -403,7 +404,11 @@ const mcpServeCommand = new Command("mcp-serve").description("Start MCP server w
 			console.error(`Dev server started at ${url} (port: ${port})`);
 		}
 		if (transportType === "stdio") {
-			const container = createContainer({ defaultThemePath: parsed.themePath });
+			const sessionTemp = process.env.PO_SESSION_TEMP_DIR;
+			const container = createContainer({
+				defaultThemePath: parsed.themePath,
+				temporaryDirectory: sessionTemp ? join(sessionTemp, "previews") : void 0
+			});
 			await startServer(new StdioTransportHandler(() => createServer$1(container)), bundlerService, devServerPort);
 		} else if (transportType === "http" && parsed.port !== void 0) {
 			const container = createContainer({ defaultThemePath: parsed.themePath });
@@ -418,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
@@ -468,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-BqvjQLSc.cjs");
+const require_stdio = require("./stdio-DkQ5zJDm.cjs");
 let node_fs = require("node:fs");
 let node_path = require("node:path");
 node_path = require_stdio.__toESM(node_path, 1);
@@ -87,6 +87,76 @@ var StoryPreviewService = class {
 	}
 };
 //#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
+};
+/**
+* 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 = {}) {
+	let browser;
+	const sharedBrowser = () => {
+		browser ??= require_stdio.launchBrowserWithFallback().catch((error) => {
+			browser = void 0;
+			throw error;
+		});
+		return browser;
+	};
+	const container = require_stdio.createContainer({
+		sharedBrowser,
+		temporaryDirectory: options.temporaryDirectory
+	});
+	const bundler = require_stdio.createDefaultBundlerService();
+	let queue = Promise.resolve();
+	let disposed = false;
+	const ensureDevServer = async (appPath) => {
+		const resolved = node_path.default.resolve(appPath);
+		if (bundler.isServerRunning() && bundler.getCurrentAppPath() === resolved) return;
+		if (bundler.isServerRunning()) await bundler.cleanup();
+		await bundler.startDevServer(resolved);
+	};
+	const run = async (tool, input) => {
+		if (disposed) throw new Error("The story renderer is shut down.");
+		if (tool === "get_component_visual" && typeof input.appPath === "string") await ensureDevServer(input.appPath);
+		const result = await container.get(WARM_RENDERER_TOOLS[tool]).execute(input);
+		const text = result.content.flatMap((item) => item.type === "text" && item.text ? [item.text] : []).join("\n\n");
+		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
+		};
+	};
+	return {
+		call(tool, input) {
+			const next = queue.then(() => run(tool, input));
+			queue = next.catch(() => void 0);
+			return next;
+		},
+		async dispose() {
+			disposed = true;
+			await queue;
+			const open = browser;
+			browser = void 0;
+			await Promise.allSettled([open?.then((instance) => instance.close()), bundler.cleanup()]);
+		}
+	};
+}
+//#endregion
 exports.BaseBundlerService = require_stdio.BaseBundlerService;
 exports.BaseCSSClassesService = require_stdio.BaseCSSClassesService;
 Object.defineProperty(exports, "CSSClassesServiceFactory", {
@@ -133,12 +203,6 @@ Object.defineProperty(exports, "ListThemesTool", {
 		return require_stdio.ListThemesTool;
 	}
 });
-Object.defineProperty(exports, "PolishComponentTool", {
-	enumerable: true,
-	get: function() {
-		return require_stdio.PolishComponentTool;
-	}
-});
 exports.STYLE_SYSTEM_TYPES = require_stdio.STYLE_SYSTEM_TYPES;
 exports.StdioTransportHandler = require_stdio.StdioTransportHandler;
 Object.defineProperty(exports, "StoriesIndexService", {
@@ -157,10 +221,12 @@ Object.defineProperty(exports, "StyleSystemConfigLoader", {
 exports.TailwindCSSClassesService = require_stdio.TailwindCSSClassesService;
 exports.ThemeService = require_stdio.ThemeService;
 exports.ViteReactBundlerService = require_stdio.ViteReactBundlerService;
+exports.WARM_RENDERER_TOOLS = WARM_RENDERER_TOOLS;
 exports.createContainer = require_stdio.createContainer;
 exports.createDefaultBundlerService = require_stdio.createDefaultBundlerService;
 exports.createServer = require_stdio.createServer;
 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 +234,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;

package/dist/index.d.cts

@@ -1,10 +1,14 @@
 import { Container, ContainerModule } from "inversify";
+import { Browser } from "playwright";
 import "zod";
 import { CallToolResult, Server } from "@modelcontextprotocol/server";
 import { AliasOptions, Plugin, UserConfig } from "vite";
 //#region src/container/module.d.ts
 interface StyleSystemModuleOptions {
   defaultThemePath?: string;
+  temporaryDirectory?: string;
+  /** A browser a long-lived host keeps warm across renders; one-shot CLI runs leave it unset. */
+  sharedBrowser?: () => Promise<Browser>;
 }
 export declare function createStyleSystemModule(optionsInput?: StyleSystemModuleOptions): ContainerModule;
 //#endregion
@@ -15,7 +19,6 @@ export declare const STYLE_SYSTEM_TYPES: {
   readonly GetComponentVisualTool: symbol;
   readonly ListSharedComponentsTool: symbol;
   readonly ListAppComponentsTool: symbol;
-  readonly PolishComponentTool: symbol;
   readonly ThemeServiceFactory: symbol;
   readonly CSSClassesServiceFactory: symbol;
   readonly StoriesIndexService: symbol;
@@ -863,6 +866,9 @@ export declare class ComponentRendererService {
   private themeService;
   private appPath;
   private bundlerFactory;
+  private log;
+  private readonly preserveOutput;
+  private readonly sharedBrowser;
   /**
    * Creates a new ComponentRendererService instance
    * @param designSystemConfig - Design system configuration
@@ -871,6 +877,8 @@ export declare class ComponentRendererService {
    */
   constructor(designSystemConfig: DesignSystemConfig, appPath: string, options?: {
     bundlerFactory?: BundlerFactory;
+    temporaryDirectory?: string;
+    sharedBrowser?: () => Promise<Browser>;
   });
   /**
    * Get the bundler service instance.
@@ -1981,38 +1989,6 @@ export declare class ListSharedComponentsTool implements Tool<ListSharedComponen
   execute(input: ListSharedComponentsInput): Promise<CallToolResult>;
 }
 //#endregion
-//#region src/tools/PolishComponentTool.d.ts
-interface PolishComponentInput {
-  filePath: string;
-  instructions?: string;
-}
-export declare class PolishComponentTool implements Tool<PolishComponentInput> {
-  static readonly TOOL_NAME = "polish_component";
-  private antigravityService;
-  private getUiComponentService;
-  private cssClassesServiceFactory;
-  private storiesIndexFactory;
-  constructor(getUiComponentService: GetUiComponentService, cssClassesServiceFactory: CSSClassesServiceFactory, storiesIndexFactory: StoriesIndexFactory);
-  getDefinition(): ToolDefinition;
-  execute(input: PolishComponentInput): Promise<CallToolResult>;
-  private detectPlatform;
-  private buildSystemPrompt;
-  /**
-   * Read design tokens from config.themePath.
-   * For web: extracts Tailwind CSS classes. For native: reads the TypeScript types file.
-   */
-  private getDesignTokens;
-  private findStoriesFile;
-  /**
-   * Find the app that owns a file, for style-system context.
-   * Only apps declaring their own style-system.config.yaml are eligible.
-   */
-  private findAppPath;
-  private getComponentVisual;
-  private getDesignPrinciples;
-  private getSharedComponents;
-}
-//#endregion
 //#region src/transports/stdio.d.ts
 /** Modern MCP stdio only: reject legacy initialization explicitly. */
 export declare class StdioTransportHandler {
@@ -2024,4 +2000,43 @@ export declare class StdioTransportHandler {
   stop(): Promise<void>;
 }
 //#endregion
-export type { CSSClassCategory, CSSClassValue, CSSClassesResult, DesignSystemConfig, OxlintRestrictedClassPattern, OxlintRestrictedClassesConfig, ResolvedDesignSystemConfig, StoryPreviewIdentity, StoryPreviewInput, StoryPreviewResult, StoryPreviewServiceOptions, StyleSystemConfig, StyleSystemModuleOptions, Tool, ToolDefinition };
+//#region src/utils/screenshot.d.ts
+/**
+ * Attempts to launch a browser, trying multiple configurations in order of preference.
+ * @returns Launched browser instance
+ * @throws Error if no browser can be launched
+ */
+export declare function launchBrowserWithFallback(): Promise<Browser>;
+//#endregion
+//#region src/services/WarmStoryRenderer/WarmStoryRenderer.d.ts
+/** The style-system tools a warm host exposes, by tool name. */
+export declare const WARM_RENDERER_TOOLS: {
+  readonly get_component_visual: symbol;
+  readonly get_css_classes: symbol;
+  readonly list_themes: symbol;
+  readonly list_shared_components: symbol;
+  readonly list_app_components: symbol;
+};
+type WarmRendererToolName = keyof typeof WARM_RENDERER_TOOLS;
+interface WarmRendererResult {
+  text: string;
+  images: {
+    data: string;
+    mimeType: string;
+  }[];
+  isError: boolean;
+}
+interface WarmStoryRenderer {
+  /** Runs one tool. Calls are serialized: the dev server serves one app at a time. */
+  call(tool: WarmRendererToolName, input: Record<string, unknown>): Promise<WarmRendererResult>;
+  dispose(): Promise<void>;
+}
+/**
+ * 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.
+ */
+export declare function createWarmStoryRenderer(options?: {
+  temporaryDirectory?: string;
+}): WarmStoryRenderer;
+//#endregion
+export type { CSSClassCategory, CSSClassValue, CSSClassesResult, DesignSystemConfig, OxlintRestrictedClassPattern, OxlintRestrictedClassesConfig, ResolvedDesignSystemConfig, StoryPreviewIdentity, StoryPreviewInput, StoryPreviewResult, StoryPreviewServiceOptions, StyleSystemConfig, StyleSystemModuleOptions, Tool, ToolDefinition, WarmRendererResult, WarmRendererToolName, WarmStoryRenderer };