@kvasar/google-stitch 0.1.7 → 0.1.8

package/openclaw.plugin.json

@@ -1,7 +1,7 @@
 {
   "id": "openclaw-google-stitch",
    "name": "Google Stitch MCP",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Integrates Google Stitch MCP services into OpenClaw",
    "skills": ["skills"],
   "configSchema": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kvasar/google-stitch",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "OpenClaw plugin for Google Stitch UI generation, screen design, variants, and design systems",
   "type": "module",
   "main": "./src/index.ts",

package/src/services/stitch-mcp-client.ts

@@ -32,6 +32,14 @@ export class StitchMCPClient {
       this.connected = true;
     }
   }
+    async createProject(title?: string) {
+    await this.connect();
+
+    return this.client.callTool({
+      name: "create_project",
+      arguments: title ? { title } : {},
+    });
+  }
 
   async generateScreen(prompt: string) {
     await this.connect();

package/src/tools/create_project.ts

@@ -1,7 +1,29 @@
-export const createProjectTool = (client:any) => ({
-  name: "create_project",
-  description: "create project",
-  async execute(_: string, params: any) {
-    return await client.request?.("create_project", params);
-  },
-});
+/**
+ * Tool: create_project
+ * Creates a new Stitch project. A project is a container for UI designs and frontend code.
+ */
+
+import { Type } from "@sinclair/typebox";
+import { StitchMCPClient } from "../services/stitch-mcp-client.js";
+
+export interface CreateProjectParams {
+  title?: string;
+}
+
+export function createProjectTool(client: StitchMCPClient) {
+  return {
+    name: "create_project" as const,
+    description: "Creates a new Stitch project. A project is a container for UI designs and frontend code.",
+    parameters: Type.Object({
+      title: Type.Optional(
+        Type.String({ description: "Optional. The title of the project." })
+      ),
+    }),
+    async execute(_id: string, params: CreateProjectParams) {
+      const result = await client.createProject(params.title);
+      return {
+        content: [{ type: "text" as const, text: JSON.stringify(result, null, 2) }],
+      };
+    },
+  };
+}

package/src/tools/generate_screen_from_text_tool.ts

@@ -1,4 +1,7 @@
 import { StitchMCPClient } from "../services/stitch-mcp-client.js";
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
 
 type StitchGenerateResponse = {
   html?: string;
@@ -10,6 +13,16 @@ type StitchGenerateResponse = {
     code?: string;
     jsx?: string;
   };
+  imageUrl?: string;
+  screenshotUrl?: string;
+  previewUrl?: string;
+  data?: {
+    imageUrl?: string;
+    screenshotUrl?: string;
+    previewUrl?: string;
+    base64Image?: string;
+    rawBytes?: string;
+  };
 };
 
 export function generateScreenFromTextTool(client: StitchMCPClient) {
@@ -34,6 +47,59 @@ export function generateScreenFromTextTool(client: StitchMCPClient) {
           params.prompt
         )) as StitchGenerateResponse;
 
+      // Check for image URLs or base64 data
+      const imageUrl = result.imageUrl || result.screenshotUrl || result.previewUrl || 
+                      result.data?.imageUrl || result.data?.screenshotUrl || result.data?.previewUrl;
+      const base64Image = result.data?.base64Image;
+      const rawBytes = result.data?.rawBytes;
+
+      if (imageUrl) {
+        // If we have an image URL, return it as an image block
+        return {
+          content: [
+            {
+              type: "image",
+              url: imageUrl,
+              caption: "Generated by Google Stitch"
+            }
+          ]
+        };
+      } else if (base64Image || rawBytes) {
+        // Decode and save base64 or raw bytes to a temporary file
+        const tempDir = os.tmpdir();
+        const tempFilePath = path.join(tempDir, `stitch-screen-${Date.now()}.png`);
+        
+        try {
+          let imageData: Buffer;
+          if (base64Image) {
+            // Remove data URL prefix if present
+            const cleanBase64 = base64Image.replace(/^data:image\/\w+;base64,/, '');
+            imageData = Buffer.from(cleanBase64, 'base64');
+          } else if (rawBytes) {
+            // Assuming rawBytes is base64 encoded
+            imageData = Buffer.from(rawBytes, 'base64');
+          } else {
+            throw new Error("No image data found");
+          }
+          
+          fs.writeFileSync(tempFilePath, imageData);
+          
+          return {
+            content: [
+              {
+                type: "image",
+                path: tempFilePath,
+                caption: "Generated by Google Stitch"
+              }
+            ]
+          };
+        } catch (error) {
+          console.error("Failed to save image:", error);
+          // Fall back to HTML if image processing fails
+        }
+      }
+
+      // Fallback to HTML if no image data is available
       const html =
         result.html ??
         result.previewHtml ??