@gemini-designer/mcp-server 0.1.0

package/dist/tools/create-ui.js

@@ -0,0 +1,167 @@
+/**
+ * create_ui Tool
+ *
+ * Scaffolds new UI components or views.
+ * Generates complete, responsive, accessible frontend code.
+ */
+import { z } from 'zod';
+import { buildContext } from '../context/builder.js';
+import { assertWritablePath } from '../context/guards.js';
+import { generateWithGemini } from '../generation/gemini-client.js';
+import { writeFile } from '../output/file-writer.js';
+import { formatCode } from '../output/formatter.js';
+import { buildRepoHints } from '../context/repo-hints.js';
+const inputSchema = {
+    description: z.string().describe('Description of the UI to create (component or view)'),
+    framework: z.enum(['react', 'vue', 'svelte', 'html']).default('react'),
+    vibe: z.string().optional().describe('Design tokens/vibe from generate_vibes'),
+    contextFiles: z.array(z.string()).optional().describe('Paths to relevant UI files for context'),
+    targetPath: z.string().describe('Target file path to generate (e.g. src/components/Button.tsx)'),
+    includeStyles: z.boolean().default(true).describe('Include styling in the generated code'),
+    writeToFile: z.boolean().default(false).describe('Write the generated code to targetPath'),
+    backup: z.boolean().default(true).describe('Create .bak backup if overwriting an existing file'),
+    format: z.boolean().default(true).describe('Format the code with Prettier before returning/writing'),
+};
+const SYSTEM_PROMPT = `You are a senior frontend engineer and UI designer.
+
+Your task: generate a COMPLETE, production-ready UI implementation.
+
+Hard rules (must follow):
+- Output ONLY the final code. No explanations. No markdown. No code fences.
+- Use the specified framework and keep the code idiomatic for that ecosystem.
+- Do NOT introduce new dependencies unless the user explicitly asked to.
+- Accessibility: meet WCAG AA as a default.
+  - Semantic HTML, proper labels, aria-* where appropriate
+  - Keyboard navigation for all interactive elements
+  - Visible focus states (use :focus-visible patterns if relevant)
+- Responsive by default (mobile-first).
+- If design tokens / vibe are provided, use them (prefer CSS variables) and do not hard-code random colors.
+- Keep logic minimal and UI-focused; avoid fake backend calls.
+
+Quality bar:
+- TypeScript types where applicable
+- No unused imports/vars
+- Clean component structure and clear naming`;
+function buildUserPrompt(args) {
+    return `Create UI code.
+
+${args.repoHints || ''}
+
+Framework: ${args.framework}
+Target file: ${args.targetPath}
+
+Description:
+${args.description}
+
+Design direction / tokens (if provided):
+${args.vibe || '(none)'}
+
+Project UI context (if provided):
+${args.context || '(none)'}
+
+Styling:
+- includeStyles = ${args.includeStyles}
+- If Tailwind classes appear in context, prefer Tailwind.
+- Otherwise use simple CSS variables + className patterns, minimal inline styles.
+
+Return ONLY the code for the target file.`;
+}
+/**
+ * Extract code from a model response that might contain fences.
+ * (We still tell the model not to use fences, but this makes the tool resilient.)
+ */
+function extractCode(response) {
+    const codeBlockMatch = response.match(/```[a-zA-Z]*\n([\s\S]*?)```/);
+    if (codeBlockMatch) {
+        return codeBlockMatch[1].trim();
+    }
+    return response.trim();
+}
+export function registerCreateUI(server, config) {
+    server.registerTool('create_ui', {
+        title: 'Create UI Component/View',
+        description: 'Generate a complete UI component or view from description and design vibe',
+        inputSchema,
+    }, async (args) => {
+        try {
+            const description = args.description;
+            const framework = args.framework || config.defaultFramework;
+            const vibe = args.vibe;
+            const contextFiles = args.contextFiles || [];
+            const targetPath = args.targetPath;
+            const includeStyles = args.includeStyles;
+            const writeToFile = args.writeToFile;
+            const backup = args.backup;
+            const format = args.format;
+            // Build context from files (token-optimized)
+            let context = '';
+            if (contextFiles.length > 0) {
+                context = await buildContext(contextFiles, config);
+            }
+            const userPrompt = buildUserPrompt({
+                description,
+                framework,
+                vibe,
+                context,
+                targetPath,
+                includeStyles,
+                repoHints: buildRepoHints(config),
+            });
+            if (config.debug) {
+                console.error(`[create_ui] Generating UI for: ${description}`);
+                console.error(`[create_ui] Framework: ${framework}`);
+                console.error(`[create_ui] Context files: ${contextFiles.length}`);
+            }
+            // Generate code with Gemini
+            const response = await generateWithGemini(config, SYSTEM_PROMPT, userPrompt, { toolName: 'create_ui' });
+            let code = extractCode(response);
+            // Format if requested
+            if (format) {
+                try {
+                    code = await formatCode(code, { filePath: targetPath });
+                }
+                catch {
+                    // ignore formatting errors
+                }
+            }
+            // Write to file if requested
+            if (writeToFile) {
+                const safePath = assertWritablePath(targetPath, config);
+                const result = await writeFile(safePath, code, { backup, format });
+                if (!result.success) {
+                    return {
+                        content: [
+                            {
+                                type: 'text',
+                                text: `Error writing file: ${result.error || 'Unknown error'}`,
+                            },
+                        ],
+                        isError: true,
+                    };
+                }
+                const msg = [
+                    `✅ UI written successfully.`,
+                    `File: ${result.filePath}`,
+                    result.backupPath ? `Backup: ${result.backupPath}` : undefined,
+                ]
+                    .filter(Boolean)
+                    .join('\n');
+                return {
+                    content: [{ type: 'text', text: msg }],
+                };
+            }
+            // Return code only (no wrappers) for best compatibility with agents
+            return {
+                content: [{ type: 'text', text: code }],
+            };
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            return {
+                content: [{ type: 'text', text: `Error: ${message}` }],
+                isError: true,
+            };
+        }
+    });
+}
+//# sourceMappingURL=create-ui.js.map

package/dist/tools/create-ui.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create-ui.js","sourceRoot":"","sources":["../../src/tools/create-ui.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,MAAM,WAAW,GAAG;IAChB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qDAAqD,CAAC;IACvF,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IACtE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;IAC9E,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;IAC/F,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+DAA+D,CAAC;IAChG,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,uCAAuC,CAAC;IAC1F,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,wCAAwC,CAAC;IAC1F,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,oDAAoD,CAAC;IAChG,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,wDAAwD,CAAC;CACvG,CAAC;AAEF,MAAM,aAAa,GAAG;;;;;;;;;;;;;;;;;;;6CAmBuB,CAAC;AAE9C,SAAS,eAAe,CAAC,IAQxB;IACG,OAAO;;EAET,IAAI,CAAC,SAAS,IAAI,EAAE;;aAET,IAAI,CAAC,SAAS;eACZ,IAAI,CAAC,UAAU;;;EAG5B,IAAI,CAAC,WAAW;;;EAGhB,IAAI,CAAC,IAAI,IAAI,QAAQ;;;EAGrB,IAAI,CAAC,OAAO,IAAI,QAAQ;;;oBAGN,IAAI,CAAC,aAAa;;;;0CAII,CAAC;AAC3C,CAAC;AAED;;;GAGG;AACH,SAAS,WAAW,CAAC,QAAgB;IACjC,MAAM,cAAc,GAAG,QAAQ,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACrE,IAAI,cAAc,EAAE,CAAC;QACjB,OAAO,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACpC,CAAC;IACD,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAiB,EAAE,MAAc;IAC9D,MAAM,CAAC,YAAY,CACf,WAAW,EACX;QACI,KAAK,EAAE,0BAA0B;QACjC,WAAW,EAAE,2EAA2E;QACxF,WAAW;KACd,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACX,IAAI,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,WAAqB,CAAC;YAC/C,MAAM,SAAS,GAAI,IAAI,CAAC,SAAoB,IAAI,MAAM,CAAC,gBAAgB,CAAC;YACxE,MAAM,IAAI,GAAG,IAAI,CAAC,IAA0B,CAAC;YAC7C,MAAM,YAAY,GAAI,IAAI,CAAC,YAAyB,IAAI,EAAE,CAAC;YAC3D,MAAM,UAAU,GAAG,IAAI,CAAC,UAAoB,CAAC;YAC7C,MAAM,aAAa,GAAG,IAAI,CAAC,aAAwB,CAAC;YACpD,MAAM,WAAW,GAAG,IAAI,CAAC,WAAsB,CAAC;YAChD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAiB,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAiB,CAAC;YAEtC,6CAA6C;YAC7C,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,OAAO,GAAG,MAAM,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,UAAU,GAAG,eAAe,CAAC;gBAC/B,WAAW;gBACX,SAAS;gBACT,IAAI;gBACJ,OAAO;gBACP,UAAU;gBACV,aAAa;gBACb,SAAS,EAAE,cAAc,CAAC,MAAM,CAAC;aACpC,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,WAAW,EAAE,CAAC,CAAC;gBAC/D,OAAO,CAAC,KAAK,CAAC,0BAA0B,SAAS,EAAE,CAAC,CAAC;gBACrD,OAAO,CAAC,KAAK,CAAC,8BAA8B,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;YACvE,CAAC;YAED,4BAA4B;YAC5B,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;YACxG,IAAI,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;YAEjC,sBAAsB;YACtB,IAAI,MAAM,EAAE,CAAC;gBACT,IAAI,CAAC;oBACD,IAAI,GAAG,MAAM,UAAU,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC5D,CAAC;gBAAC,MAAM,CAAC;oBACL,2BAA2B;gBAC/B,CAAC;YACL,CAAC;YAED,6BAA6B;YAC7B,IAAI,WAAW,EAAE,CAAC;gBACd,MAAM,QAAQ,GAAG,kBAAkB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;gBACxD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;gBAEnE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAClB,OAAO;wBACH,OAAO,EAAE;4BACL;gCACI,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,uBAAuB,MAAM,CAAC,KAAK,IAAI,eAAe,EAAE;6BACjE;yBACJ;wBACD,OAAO,EAAE,IAAI;qBAChB,CAAC;gBACN,CAAC;gBAED,MAAM,GAAG,GAAG;oBACR,4BAA4B;oBAC5B,SAAS,MAAM,CAAC,QAAQ,EAAE;oBAC1B,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS;iBACjE;qBACI,MAAM,CAAC,OAAO,CAAC;qBACf,IAAI,CAAC,IAAI,CAAC,CAAC;gBAEhB,OAAO;oBACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;iBACzC,CAAC;YACN,CAAC;YAED,oEAAoE;YACpE,OAAO;gBACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aAC1C,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO;gBACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,OAAO,EAAE,EAAE,CAAC;gBACtD,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;IACL,CAAC,CACJ,CAAC;AACN,CAAC"}

package/dist/tools/detect-ui-stack.d.ts

@@ -0,0 +1,15 @@
+/**
+ * detect_ui_stack Tool
+ *
+ * Deterministically inspects the workspace to detect the UI stack:
+ * framework, styling approach, component libraries, storybook, etc.
+ *
+ * This tool is intentionally non-LLM to:
+ * - reduce token usage
+ * - increase reliability
+ * - enable downstream tools to condition prompts/outputs on real repo constraints
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { Config } from '../config/index.js';
+export declare function registerDetectUIStack(server: McpServer, config: Config): void;
+//# sourceMappingURL=detect-ui-stack.d.ts.map

package/dist/tools/detect-ui-stack.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"detect-ui-stack.d.ts","sourceRoot":"","sources":["../../src/tools/detect-ui-stack.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAGpE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAY5C,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAoC7E"}

package/dist/tools/detect-ui-stack.js

@@ -0,0 +1,52 @@
+/**
+ * detect_ui_stack Tool
+ *
+ * Deterministically inspects the workspace to detect the UI stack:
+ * framework, styling approach, component libraries, storybook, etc.
+ *
+ * This tool is intentionally non-LLM to:
+ * - reduce token usage
+ * - increase reliability
+ * - enable downstream tools to condition prompts/outputs on real repo constraints
+ */
+import { z } from 'zod';
+import * as path from 'node:path';
+import { assertReadableDir } from '../context/guards.js';
+import { detectUiStack } from '../stack/detect.js';
+const inputSchema = {
+    root: z
+        .string()
+        .optional()
+        .describe('Project root directory (defaults to current working directory)')
+        .default(process.cwd()),
+};
+export function registerDetectUIStack(server, config) {
+    server.registerTool('detect_ui_stack', {
+        title: 'Detect UI Stack',
+        description: 'Detect framework/styling/component libraries from the repo (no LLM). Useful to ground other tools.',
+        inputSchema,
+    }, async (args) => {
+        const root = args.root || process.cwd();
+        let safeRoot;
+        try {
+            safeRoot = assertReadableDir(root, config);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Invalid root directory';
+            return {
+                content: [{ type: 'text', text: `Error: ${message}` }],
+                isError: true,
+            };
+        }
+        const result = detectUiStack(safeRoot);
+        // Normalize paths for agent readability
+        const normalized = {
+            ...result,
+            root: path.resolve(result.root),
+        };
+        return {
+            content: [{ type: 'text', text: JSON.stringify(normalized, null, 2) }],
+        };
+    });
+}
+//# sourceMappingURL=detect-ui-stack.js.map

package/dist/tools/detect-ui-stack.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"detect-ui-stack.js","sourceRoot":"","sources":["../../src/tools/detect-ui-stack.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,MAAM,WAAW,GAAG;IAChB,IAAI,EAAE,CAAC;SACF,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,gEAAgE,CAAC;SAC1E,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;CAC9B,CAAC;AAEF,MAAM,UAAU,qBAAqB,CAAC,MAAiB,EAAE,MAAc;IACnE,MAAM,CAAC,YAAY,CACf,iBAAiB,EACjB;QACI,KAAK,EAAE,iBAAiB;QACxB,WAAW,EACP,oGAAoG;QACxG,WAAW;KACd,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACX,MAAM,IAAI,GAAI,IAAI,CAAC,IAAe,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;QAEpD,IAAI,QAAgB,CAAC;QACrB,IAAI,CAAC;YACD,QAAQ,GAAG,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC;YAClF,OAAO;gBACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,OAAO,EAAE,EAAE,CAAC;gBAC/D,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;QAED,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QAEvC,wCAAwC;QACxC,MAAM,UAAU,GAAG;YACf,GAAG,MAAM;YACT,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;SAClC,CAAC;QAEF,OAAO;YACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;SAClF,CAAC;IACN,CAAC,CACJ,CAAC;AACN,CAAC"}

package/dist/tools/generate-component-variants.d.ts

@@ -0,0 +1,15 @@
+/**
+ * generate_component_variants Tool
+ *
+ * Given an existing component, generate variants (sizes, intents, states) and
+ * optional Storybook stories.
+ *
+ * Output contract:
+ * - When applyChanges=false: outputs ONE JSON object only:
+ *   {"files": [{"path": string, "content": string}]}
+ * - When applyChanges=true: writes files and returns a short summary string.
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { Config } from '../config/index.js';
+export declare function registerGenerateComponentVariants(server: McpServer, config: Config): void;
+//# sourceMappingURL=generate-component-variants.d.ts.map

package/dist/tools/generate-component-variants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-component-variants.d.ts","sourceRoot":"","sources":["../../src/tools/generate-component-variants.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIpE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AA8D5C,wBAAgB,iCAAiC,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CA2IzF"}

package/dist/tools/generate-component-variants.js

@@ -0,0 +1,199 @@
+/**
+ * generate_component_variants Tool
+ *
+ * Given an existing component, generate variants (sizes, intents, states) and
+ * optional Storybook stories.
+ *
+ * Output contract:
+ * - When applyChanges=false: outputs ONE JSON object only:
+ *   {"files": [{"path": string, "content": string}]}
+ * - When applyChanges=true: writes files and returns a short summary string.
+ */
+import { z } from 'zod';
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { buildContext } from '../context/builder.js';
+import { assertReadablePath, assertWritablePath } from '../context/guards.js';
+import { buildRepoGrounding } from '../context/grounding.js';
+import { generateWithGemini } from '../generation/gemini-client.js';
+import { writeFile } from '../output/file-writer.js';
+const inputSchema = {
+    componentFile: z.string().describe('Path to an existing component file (e.g., src/components/Button.tsx).'),
+    variants: z
+        .object({
+        sizes: z.array(z.string()).optional(),
+        intents: z.array(z.string()).optional(),
+        states: z.array(z.string()).optional(),
+        themes: z.array(z.string()).optional(),
+        custom: z.array(z.string()).optional().describe('Any additional variant requirements.'),
+    })
+        .optional()
+        .describe('Variant dimensions to generate.'),
+    instruction: z.string().optional().describe('Extra guidance about behavior, styling, or API.'),
+    storybook: z.boolean().default(true).describe('Generate a Storybook stories file if true.'),
+    storyFile: z.string().optional().describe('Optional path for the stories file (defaults next to component).'),
+    context: z.array(z.string()).optional().describe('Related files (tokens, theme, utilities).'),
+    applyChanges: z.boolean().default(false).describe('If true, write files to disk.'),
+    backup: z.boolean().default(true).describe('Create .bak backups before overwriting existing files.'),
+};
+const SYSTEM_PROMPT = `You are a senior frontend engineer specializing in reusable component APIs.
+
+Your task:
+- Update an existing component to support the requested variants.
+- Preserve existing behavior and imports unless required.
+- Keep code style consistent with the file.
+- Ensure accessibility: keyboard/focus, aria-labels, disabled states.
+
+Repo alignment (CRITICAL):
+- Prefer reusing existing utilities/components already present in the repo.
+- You will receive an auto-detected stack summary + component catalog; do not invent duplicates.
+
+Output contract (CRITICAL):
+- If storybook=true, output JSON with an array of files: {"files": [{"path": string, "content": string}]}
+- If storybook=false, output JSON with exactly one file (the component file).
+- Output MUST be ONE valid JSON object only. No prose. No markdown. No code fences.
+
+API guidance:
+- Prefer adding/expanding a typed prop such as "size" | "variant" | "intent" and mapping to classes.
+- Avoid breaking changes: keep default props aligned to existing visuals.
+- Do not introduce new dependencies unless explicitly requested.`;
+function cleanJson(response) {
+    let out = response.trim();
+    const fenced = out.match(/```(?:json)?\n([\s\S]*?)```/);
+    if (fenced)
+        out = fenced[1].trim();
+    return out;
+}
+function defaultStoryPath(componentFile) {
+    const dir = path.dirname(componentFile);
+    const base = path.basename(componentFile).replace(/\.(tsx|jsx|ts|js)$/, '');
+    return path.join(dir, `${base}.stories.tsx`);
+}
+export function registerGenerateComponentVariants(server, config) {
+    server.registerTool('generate_component_variants', {
+        title: 'Generate Component Variants',
+        description: 'Enhance an existing component with variants (sizes/intents/states) and optionally generate Storybook stories.',
+        inputSchema,
+    }, async (args) => {
+        const componentFile = args.componentFile;
+        const variants = args.variants || {};
+        const instruction = args.instruction || '';
+        const storybook = args.storybook !== false;
+        const storyFile = args.storyFile || defaultStoryPath(componentFile);
+        const contextPaths = args.context;
+        const applyChanges = args.applyChanges === true;
+        const backup = args.backup !== false;
+        let safeComponent;
+        try {
+            safeComponent = assertReadablePath(componentFile, config);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Invalid componentFile';
+            return { content: [{ type: 'text', text: `Error: ${message}` }], isError: true };
+        }
+        let componentContent;
+        try {
+            componentContent = fs.readFileSync(safeComponent, 'utf-8');
+        }
+        catch {
+            return {
+                content: [{ type: 'text', text: `Error: Could not read ${componentFile}` }],
+                isError: true,
+            };
+        }
+        let contextContent = '';
+        if (contextPaths && contextPaths.length > 0) {
+            contextContent = await buildContext(contextPaths, config);
+        }
+        // Deterministic repo grounding (stack + reusable components)
+        const grounding = await buildRepoGrounding(config, {
+            focusFileAbs: safeComponent,
+            instruction,
+        });
+        const userPrompt = `Enhance the component to support variants.
+
+${grounding}
+
+COMPONENT FILE PATH:
+${componentFile}
+
+VARIANTS (JSON):
+${JSON.stringify(variants, null, 2)}
+
+STORYBOOK:
+${storybook}
+
+STORY FILE PATH (if storybook=true):
+${storyFile}
+
+EXTRA INSTRUCTION:
+${instruction || '(none)'}
+
+CURRENT COMPONENT FILE CONTENT:
+${componentContent}
+
+${contextContent ? `RELATED FILES FOR CONTEXT:\n${contextContent}` : ''}
+
+Remember: Output JSON only: {"files": [{"path": string, "content": string}]}`;
+        try {
+            const resp = await generateWithGemini(config, SYSTEM_PROMPT, userPrompt, {
+                toolName: 'generate_component_variants',
+            });
+            const cleaned = cleanJson(resp);
+            const parsed = JSON.parse(cleaned);
+            if (!parsed || !Array.isArray(parsed.files) || parsed.files.length === 0) {
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: 'Error: Model output did not match expected JSON format: {"files": [...]}',
+                        },
+                    ],
+                    isError: true,
+                };
+            }
+            if (!applyChanges) {
+                return { content: [{ type: 'text', text: JSON.stringify(parsed, null, 2) }] };
+            }
+            const writes = [];
+            for (const f of parsed.files) {
+                const target = f.path;
+                const content = f.content;
+                let safeOut;
+                try {
+                    safeOut = assertWritablePath(target, config);
+                }
+                catch (error) {
+                    const message = error instanceof Error ? error.message : 'Invalid output path';
+                    return { content: [{ type: 'text', text: `Error: ${message}` }], isError: true };
+                }
+                const result = await writeFile(safeOut, content, { format: true, backup });
+                if (!result.success) {
+                    return {
+                        content: [
+                            {
+                                type: 'text',
+                                text: `Error: Failed writing ${target}: ${result.error || 'Unknown error'}`,
+                            },
+                        ],
+                        isError: true,
+                    };
+                }
+                writes.push({ file: result.filePath, backup: result.backupPath });
+            }
+            const summary = [
+                '✅ Generated component variants and wrote files:',
+                ...writes.map((w) => `- ${w.file}${w.backup ? ` (backup: ${w.backup})` : ''}`),
+            ].join('\n');
+            return { content: [{ type: 'text', text: summary }] };
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            return {
+                content: [{ type: 'text', text: `Error generating variants: ${message}` }],
+                isError: true,
+            };
+        }
+    });
+}
+//# sourceMappingURL=generate-component-variants.js.map

package/dist/tools/generate-component-variants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-component-variants.js","sourceRoot":"","sources":["../../src/tools/generate-component-variants.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAErD,MAAM,WAAW,GAAG;IAChB,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uEAAuE,CAAC;IAC3G,QAAQ,EAAE,CAAC;SACN,MAAM,CAAC;QACJ,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;QACrC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;QACvC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;QACtC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;QACtC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;KAC1F,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,iCAAiC,CAAC;IAChD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;IAC9F,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,4CAA4C,CAAC;IAC3F,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kEAAkE,CAAC;IAC7G,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;IAC7F,YAAY,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,+BAA+B,CAAC;IAClF,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,wDAAwD,CAAC;CACvG,CAAC;AAEF,MAAM,aAAa,GAAG;;;;;;;;;;;;;;;;;;;;iEAoB2C,CAAC;AAElE,SAAS,SAAS,CAAC,QAAgB;IAC/B,IAAI,GAAG,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC1B,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACxD,IAAI,MAAM;QAAE,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACnC,OAAO,GAAG,CAAC;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,aAAqB;IAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IACxC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;IAC5E,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,cAAc,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAC,MAAiB,EAAE,MAAc;IAC/E,MAAM,CAAC,YAAY,CACf,6BAA6B,EAC7B;QACI,KAAK,EAAE,6BAA6B;QACpC,WAAW,EACP,+GAA+G;QACnH,WAAW;KACd,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACX,MAAM,aAAa,GAAG,IAAI,CAAC,aAAuB,CAAC;QACnD,MAAM,QAAQ,GAAI,IAAI,CAAC,QAAgB,IAAI,EAAE,CAAC;QAC9C,MAAM,WAAW,GAAI,IAAI,CAAC,WAAkC,IAAI,EAAE,CAAC;QACnE,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,KAAK,KAAK,CAAC;QAC3C,MAAM,SAAS,GAAI,IAAI,CAAC,SAAgC,IAAI,gBAAgB,CAAC,aAAa,CAAC,CAAC;QAC5F,MAAM,YAAY,GAAG,IAAI,CAAC,OAA+B,CAAC;QAC1D,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC;QAChD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,KAAK,KAAK,CAAC;QAErC,IAAI,aAAqB,CAAC;QAC1B,IAAI,CAAC;YACD,aAAa,GAAG,kBAAkB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAAC;YACjF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,OAAO,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC9F,CAAC;QAED,IAAI,gBAAwB,CAAC;QAC7B,IAAI,CAAC;YACD,gBAAgB,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC/D,CAAC;QAAC,MAAM,CAAC;YACL,OAAO;gBACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,yBAAyB,aAAa,EAAE,EAAE,CAAC;gBACpF,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;QAED,IAAI,cAAc,GAAG,EAAE,CAAC;QACxB,IAAI,YAAY,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1C,cAAc,GAAG,MAAM,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAC9D,CAAC;QAED,6DAA6D;QAC7D,MAAM,SAAS,GAAG,MAAM,kBAAkB,CAAC,MAAM,EAAE;YAC/C,YAAY,EAAE,aAAa;YAC3B,WAAW;SACd,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG;;EAE7B,SAAS;;;EAGT,aAAa;;;EAGb,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;;;EAGjC,SAAS;;;EAGT,SAAS;;;EAGT,WAAW,IAAI,QAAQ;;;EAGvB,gBAAgB;;EAEhB,cAAc,CAAC,CAAC,CAAC,+BAA+B,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE;;6EAEM,CAAC;QAElE,IAAI,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,kBAAkB,CAAC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE;gBACrE,QAAQ,EAAE,6BAA6B;aAC1C,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAyD,CAAC;YAE3F,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvE,OAAO;oBACH,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,0EAA0E;yBACnF;qBACJ;oBACD,OAAO,EAAE,IAAI;iBAChB,CAAC;YACN,CAAC;YAED,IAAI,CAAC,YAAY,EAAE,CAAC;gBAChB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAC3F,CAAC;YAED,MAAM,MAAM,GAA6C,EAAE,CAAC;YAC5D,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAC3B,MAAM,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC;gBACtB,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;gBAC1B,IAAI,OAAe,CAAC;gBACpB,IAAI,CAAC;oBACD,OAAO,GAAG,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACjD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC;oBAC/E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,OAAO,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBAC9F,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC3E,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAClB,OAAO;wBACH,OAAO,EAAE;4BACL;gCACI,IAAI,EAAE,MAAe;gCACrB,IAAI,EAAE,yBAAyB,MAAM,KAAK,MAAM,CAAC,KAAK,IAAI,eAAe,EAAE;6BAC9E;yBACJ;wBACD,OAAO,EAAE,IAAI;qBAChB,CAAC;gBACN,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;YACtE,CAAC;YAED,MAAM,OAAO,GAAG;gBACZ,iDAAiD;gBACjD,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;aACjF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO;gBACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,8BAA8B,OAAO,EAAE,EAAE,CAAC;gBACnF,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;IACL,CAAC,CACJ,CAAC;AACN,CAAC"}

package/dist/tools/generate-vibes.d.ts

@@ -0,0 +1,10 @@
+/**
+ * generate_vibes Tool
+ *
+ * Generates design direction options (vibes) before creating UI.
+ * Returns color palettes, typography, spacing, and sample CSS variables.
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { Config } from '../config/index.js';
+export declare function registerGenerateVibes(server: McpServer, config: Config): void;
+//# sourceMappingURL=generate-vibes.d.ts.map

package/dist/tools/generate-vibes.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-vibes.d.ts","sourceRoot":"","sources":["../../src/tools/generate-vibes.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAwE5C,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CA+D7E"}

package/dist/tools/generate-vibes.js

@@ -0,0 +1,145 @@
+/**
+ * generate_vibes Tool
+ *
+ * Generates design direction options (vibes) before creating UI.
+ * Returns color palettes, typography, spacing, and sample CSS variables.
+ */
+import { z } from 'zod';
+import { buildContext } from '../context/builder.js';
+import { generateWithGemini } from '../generation/gemini-client.js';
+const inputSchema = {
+    projectContext: z
+        .string()
+        .optional()
+        .describe('Brief description of the project (e.g., "Modern SaaS dashboard for analytics")'),
+    existingTokens: z
+        .string()
+        .optional()
+        .describe('Path to existing design tokens or CSS file to analyze (optional)'),
+    vibeCount: z
+        .number()
+        .min(1)
+        .max(5)
+        .default(3)
+        .describe('Number of vibe options to generate (1-5)'),
+    preferences: z
+        .object({
+        colorTemp: z.enum(['warm', 'cool', 'neutral']).optional(),
+        density: z.enum(['compact', 'comfortable', 'spacious']).optional(),
+        style: z.enum(['minimal', 'playful', 'corporate', 'brutalist', 'glassmorphism']).optional(),
+    })
+        .optional()
+        .describe('Optional style preferences to guide generation'),
+};
+const SYSTEM_PROMPT = `You are an expert UI/UX designer specializing in modern web design systems.
+
+Your task is to generate design "vibes"—cohesive design directions that include colors, typography, and spacing.
+
+Requirements:
+- Return ONE valid JSON object with a "vibes" array (no markdown, no code fences, no extra commentary).
+- Use semantic color roles (primary, secondary, accent, background, surface, text, textMuted, border, danger, success).
+- Prefer accessible color combinations (aim for WCAG AA contrast for text on background/surface).
+- Provide a sensible type scale and spacing scale (consistent multiplier).
+- Provide a ready-to-paste CSS variable block (light theme). If you can, include a dark theme block too.
+
+Output structure:
+{
+  "vibes": [
+    {
+      "name": "Vibe Name",
+      "description": "Brief aesthetic description",
+      "colors": {
+        "primary": "#hex",
+        "secondary": "#hex",
+        "accent": "#hex",
+        "background": "#hex",
+        "surface": "#hex",
+        "text": "#hex",
+        "textMuted": "#hex",
+        "border": "#hex",
+        "danger": "#hex",
+        "success": "#hex"
+      },
+      "typography": {
+        "fontFamily": "font stack",
+        "headingFamily": "font stack or same",
+        "scale": [12, 14, 16, 18, 20, 24, 30, 36, 48, 60]
+      },
+      "spacing": {
+        "unit": 4,
+        "scale": [0, 4, 8, 12, 16, 24, 32, 48, 64, 96]
+      },
+      "cssVariables": "/* light */\\n:root { --color-primary: #...; }\\n\\n/* dark */\\n[data-theme='dark'] { --color-primary: #...; }"
+    }
+  ]
+}`;
+export function registerGenerateVibes(server, config) {
+    server.registerTool('generate_vibes', {
+        title: 'Generate Design Vibes',
+        description: 'Generate design direction options (vibes) with color palettes, typography, and spacing. Call this before create_ui to establish design consistency.',
+        inputSchema,
+    }, async (args) => {
+        const projectContext = args.projectContext;
+        const vibeCount = args.vibeCount;
+        const preferences = args.preferences;
+        const existingTokensPath = args.existingTokens;
+        let existingTokensContext = '';
+        if (existingTokensPath) {
+            // Use the same context pipeline (path allowlist + sensitive file filtering + truncation)
+            existingTokensContext = await buildContext([existingTokensPath], config);
+        }
+        let userPrompt = buildPrompt({
+            projectContext,
+            vibeCount,
+            preferences,
+        });
+        if (existingTokensContext) {
+            userPrompt += `\n\nExisting design tokens / styles to respect:\n${existingTokensContext}`;
+        }
+        try {
+            const response = await generateWithGemini(config, SYSTEM_PROMPT, userPrompt, { toolName: 'generate_vibes' });
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: response.trim(),
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Error generating vibes: ${message}`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+    });
+}
+function buildPrompt(input) {
+    const vibeCount = input.vibeCount ?? 3;
+    let prompt = `Generate ${vibeCount} distinct design vibes`;
+    if (input.projectContext) {
+        prompt += ` for this project: "${input.projectContext}"`;
+    }
+    if (input.preferences) {
+        const prefs = [];
+        if (input.preferences.colorTemp)
+            prefs.push(`${input.preferences.colorTemp} color temperature`);
+        if (input.preferences.density)
+            prefs.push(`${input.preferences.density} spacing density`);
+        if (input.preferences.style)
+            prefs.push(`${input.preferences.style} style`);
+        if (prefs.length > 0) {
+            prompt += `. Preferences: ${prefs.join(', ')}`;
+        }
+    }
+    prompt += '. Return valid JSON only.';
+    return prompt;
+}
+//# sourceMappingURL=generate-vibes.js.map

package/dist/tools/generate-vibes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-vibes.js","sourceRoot":"","sources":["../../src/tools/generate-vibes.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEpE,MAAM,WAAW,GAAG;IAChB,cAAc,EAAE,CAAC;SACZ,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,gFAAgF,CAAC;IAC/F,cAAc,EAAE,CAAC;SACZ,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,kEAAkE,CAAC;IACjF,SAAS,EAAE,CAAC;SACP,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,CAAC,CAAC;SACN,OAAO,CAAC,CAAC,CAAC;SACV,QAAQ,CAAC,0CAA0C,CAAC;IACzD,WAAW,EAAE,CAAC;SACT,MAAM,CAAC;QACJ,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE;QACzD,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,aAAa,EAAE,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE;QAClE,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC,CAAC,QAAQ,EAAE;KAC9F,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,gDAAgD,CAAC;CAClE,CAAC;AAEF,MAAM,aAAa,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAyCpB,CAAC;AAEH,MAAM,UAAU,qBAAqB,CAAC,MAAiB,EAAE,MAAc;IACnE,MAAM,CAAC,YAAY,CACf,gBAAgB,EAChB;QACI,KAAK,EAAE,uBAAuB;QAC9B,WAAW,EACP,qJAAqJ;QACzJ,WAAW;KACd,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACX,MAAM,cAAc,GAAG,IAAI,CAAC,cAAoC,CAAC;QACjE,MAAM,SAAS,GAAG,IAAI,CAAC,SAA+B,CAAC;QACvD,MAAM,WAAW,GAAG,IAAI,CAAC,WAMV,CAAC;QAEhB,MAAM,kBAAkB,GAAG,IAAI,CAAC,cAAoC,CAAC;QAErE,IAAI,qBAAqB,GAAG,EAAE,CAAC;QAC/B,IAAI,kBAAkB,EAAE,CAAC;YACrB,yFAAyF;YACzF,qBAAqB,GAAG,MAAM,YAAY,CAAC,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,CAAC;QAC7E,CAAC;QAED,IAAI,UAAU,GAAG,WAAW,CAAC;YACzB,cAAc;YACd,SAAS;YACT,WAAW;SACd,CAAC,CAAC;QAEH,IAAI,qBAAqB,EAAE,CAAC;YACxB,UAAU,IAAI,oDAAoD,qBAAqB,EAAE,CAAC;QAC9F,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAE7G,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE;qBACxB;iBACJ;aACJ,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,2BAA2B,OAAO,EAAE;qBAC7C;iBACJ;gBACD,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;IACL,CAAC,CACJ,CAAC;AACN,CAAC;AAED,SAAS,WAAW,CAAC,KAQpB;IACG,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;IACvC,IAAI,MAAM,GAAG,YAAY,SAAS,wBAAwB,CAAC;IAE3D,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;QACvB,MAAM,IAAI,uBAAuB,KAAK,CAAC,cAAc,GAAG,CAAC;IAC7D,CAAC;IAED,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACpB,MAAM,KAAK,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,CAAC,WAAW,CAAC,SAAS;YAAE,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,SAAS,oBAAoB,CAAC,CAAC;QAChG,IAAI,KAAK,CAAC,WAAW,CAAC,OAAO;YAAE,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,OAAO,kBAAkB,CAAC,CAAC;QAC1F,IAAI,KAAK,CAAC,WAAW,CAAC,KAAK;YAAE,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,KAAK,QAAQ,CAAC,CAAC;QAE5E,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,kBAAkB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACnD,CAAC;IACL,CAAC;IAED,MAAM,IAAI,2BAA2B,CAAC;IAEtC,OAAO,MAAM,CAAC;AAClB,CAAC"}