@fugood/bricks-project 2.24.0-beta.9 → 2.24.0

package/tools/mcp-tools/compile.ts

@@ -0,0 +1,91 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
+import { z } from 'zod'
+import { $ } from 'bun'
+
+export function register(server: McpServer, projectDir: string) {
+  const { dirname } = import.meta
+
+  server.tool('compile', {}, async () => {
+    let log = ''
+    try {
+      log += 'Type checking & Compiling...'
+      log += await $`bun compile`.cwd(projectDir).text()
+    } catch (err) {
+      log += `${err.stdout.toString()}\n${err.stderr.toString()}`
+    }
+    return {
+      content: [{ type: 'text', text: log }],
+    }
+  })
+
+  server.tool(
+    'preview',
+    {
+      delay: z
+        .number()
+        .describe('Delay in milliseconds before taking screenshot')
+        .optional()
+        .default(3000),
+      width: z.number().describe('Width of the screenshot').optional().default(600),
+      height: z.number().optional().default(480),
+      responseImage: z
+        .boolean()
+        .describe(
+          'Whether to response image content (base64 encoded jpeg). If false, only saved path will be responded as text.',
+        )
+        .optional()
+        .default(false),
+      testId: z.string().describe('Automation test ID to trigger').optional(),
+      testTitleLike: z
+        .string()
+        .describe('Find automation test by partial title match (case-insensitive)')
+        .optional(),
+    } as any,
+    async ({ delay, width, height, responseImage, testId, testTitleLike }: any) => {
+      let log = ''
+      let error = false
+      try {
+        const toolsDir = `${dirname}/..`
+        const args = [
+          '--no-keep-open',
+          '--take-screenshot',
+          JSON.stringify({
+            delay,
+            width,
+            height,
+            path: `${toolsDir}/screenshot.jpg`,
+            closeAfter: true,
+            headless: true,
+          }),
+        ]
+        if (testId) args.push('--test-id', testId)
+        if (testTitleLike) args.push('--test-title-like', testTitleLike)
+        log = await $`bunx --bun electron ${toolsDir}/preview-main.mjs ${args}`
+          .cwd(projectDir)
+          .text()
+      } catch (err) {
+        log = `${err.stdout.toString()}\n${err.stderr.toString()}`
+        error = true
+      }
+      let screenshotBase64: string | null = null
+      if (!error && responseImage) {
+        const toolsDir = `${dirname}/..`
+        const screenshot = await Bun.file(`${toolsDir}/screenshot.jpg`).arrayBuffer()
+        screenshotBase64 = Buffer.from(screenshot).toString('base64')
+      }
+      const content: Array<
+        { type: 'text'; text: string } | { type: 'image'; data: string; mimeType: string }
+      > = [{ type: 'text', text: log }]
+      if (screenshotBase64) {
+        content.push({
+          type: 'image',
+          data: screenshotBase64,
+          mimeType: 'image/jpeg',
+        })
+      }
+      return {
+        content,
+      }
+    },
+  )
+}