@agimon-ai/po-style-system-tooling 0.2.14-alpha.7 → 0.2.14-alpha.9

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-DdX1dqxV.cjs");
+const require_stdio = require("./stdio-DKWqsPkv.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-OeP9ajK2.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-8KmKB4TW.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-DdX1dqxV.cjs");
+const require_stdio = require("./stdio-DKWqsPkv.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,4 +1,5 @@
 import { Container, ContainerModule } from "inversify";
+import { Browser } from "playwright";
 import "zod";
 import { CallToolResult, Server } from "@modelcontextprotocol/server";
 import { AliasOptions, Plugin, UserConfig } from "vite";
@@ -6,6 +7,8 @@ import { AliasOptions, Plugin, UserConfig } from "vite";
 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
@@ -16,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;
@@ -866,6 +868,7 @@ export declare class ComponentRendererService {
   private bundlerFactory;
   private log;
   private readonly preserveOutput;
+  private readonly sharedBrowser;
   /**
    * Creates a new ComponentRendererService instance
    * @param designSystemConfig - Design system configuration
@@ -875,6 +878,7 @@ export declare class ComponentRendererService {
   constructor(designSystemConfig: DesignSystemConfig, appPath: string, options?: {
     bundlerFactory?: BundlerFactory;
     temporaryDirectory?: string;
+    sharedBrowser?: () => Promise<Browser>;
   });
   /**
    * Get the bundler service instance.
@@ -1985,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 {
@@ -2028,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 };

package/dist/index.d.mts

@@ -1,11 +1,14 @@
 import { Container, ContainerModule } from "inversify";
 import { z } from "zod";
+import { Browser } from "playwright";
 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
@@ -16,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;
@@ -866,6 +868,7 @@ export declare class ComponentRendererService {
   private bundlerFactory;
   private log;
   private readonly preserveOutput;
+  private readonly sharedBrowser;
   /**
    * Creates a new ComponentRendererService instance
    * @param designSystemConfig - Design system configuration
@@ -875,6 +878,7 @@ export declare class ComponentRendererService {
   constructor(designSystemConfig: DesignSystemConfig, appPath: string, options?: {
     bundlerFactory?: BundlerFactory;
     temporaryDirectory?: string;
+    sharedBrowser?: () => Promise<Browser>;
   });
   /**
    * Get the bundler service instance.
@@ -1985,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 {
@@ -2028,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 };

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-OeP9ajK2.mjs";
+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 launchBrowserWithFallback, T as hasStyleSystemProjectConfig, _ as createDefaultBundlerService, b as ViteReactBundlerService, c as createStyleSystemModule, d as GetComponentVisualTool, f as GetCSSClassesTool, g as ThemeService, h as ComponentRendererService, i as createServer, j as StoriesIndexService, k as resolveStyleSystemThemePath, 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 getSharedComponentTags, x as BaseBundlerService, y as getBundlerServiceFromConfig } from "./stdio-8KmKB4TW.mjs";
 import { promises } from "node:fs";
 import path from "node:path";
 import { createHash, randomUUID } from "node:crypto";
@@ -85,4 +85,74 @@ 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/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 = {}) {
+	let browser;
+	const sharedBrowser = () => {
+		browser ??= launchBrowserWithFallback().catch((error) => {
+			browser = void 0;
+			throw error;
+		});
+		return browser;
+	};
+	const container = createContainer({
+		sharedBrowser,
+		temporaryDirectory: options.temporaryDirectory
+	});
+	const bundler = createDefaultBundlerService();
+	let queue = Promise.resolve();
+	let disposed = false;
+	const ensureDevServer = async (appPath) => {
+		const resolved = path.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 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
+export { BaseBundlerService, BaseCSSClassesService, CSSClassesServiceFactory, ComponentRendererService, DEFAULT_STYLE_SYSTEM_CONFIG, GetCSSClassesTool, GetComponentVisualTool, GetUiComponentService, ListAppComponentsTool, ListSharedComponentsTool, ListThemesTool, STYLE_SYSTEM_TYPES, StdioTransportHandler, StoriesIndexService, StoryPreviewService, StyleSystemConfigLoader, TailwindCSSClassesService, ThemeService, ViteReactBundlerService, WARM_RENDERER_TOOLS, createContainer, createDefaultBundlerService, createServer, createStyleSystemModule, createWarmStoryRenderer, extractCustomTokenNames, generateRestrictedClassesConfig, getAppDesignSystemConfig, getBundlerService, getBundlerServiceFromConfig, getSharedComponentTags, hasStyleSystemProjectConfig, launchBrowserWithFallback, resetStyleSystemConfigCache, resolveAppDesignSystemConfig, resolveStyleSystemThemePath };