@fugood/bricks-project 2.22.0-beta.2 → 2.22.0-beta.3

package/package.json

@@ -1,11 +1,12 @@
 {
   "name": "@fugood/bricks-project",
-  "version": "2.22.0-beta.2",
+  "version": "2.22.0-beta.3",
   "main": "index.ts",
   "scripts": {
     "build": "node scripts/build.js"
   },
   "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.7.0",
     "@types/escodegen": "^0.0.10",
     "@types/lodash": "^4.17.12",
     "acorn": "^8.13.0",
@@ -13,5 +14,5 @@
     "lodash": "^4.17.4",
     "uuid": "^8.3.1"
   },
-  "gitHead": "a230efafdd33bc1dd02ed6397cf5741b318e172f"
+  "gitHead": "cbb283b7ce73520369f855f7fbe86e8e07b831eb"
 }

package/tools/mcp-server.ts

@@ -0,0 +1,86 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
+import { z } from 'zod'
+import { $ } from 'bun'
+
+const server = new McpServer({
+  name: 'bricks-project',
+  version: '1.0.0',
+})
+
+const dirname = import.meta.dirname
+const projectDir = String(dirname).split('/node_modules/')[0]
+
+server.tool('compile', {}, async () => {
+  let log
+  try {
+    log = await $`bun compile`.cwd(projectDir).text()
+  } catch (err) {
+    log = err.stdout.toString() + '\n' + err.stderr.toString()
+  }
+  return {
+    content: [{ type: 'text', text: log || 'Compiled successfully' }],
+  }
+})
+
+// NOTE: Cursor (Or VSCode) seems set ELECTRON_RUN_AS_NODE to 1, so we need to unset it
+process.env.ELECTRON_RUN_AS_NODE = ''
+
+server.tool(
+  'take-preview-screenshot',
+  {
+    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),
+  } as any,
+  async ({ delay, width, height, responseImage }: any) => {
+    let log = ''
+    let error = false
+    try {
+      const args = [
+        '--take-screenshot',
+        JSON.stringify({
+          delay,
+          width,
+          height,
+          path: `${dirname}/screenshot.jpg`,
+          closeAfter: true,
+          headless: true,
+        }),
+      ]
+      log = await $`bunx --bun electron ${dirname}/preview-main.mjs ${args}`.cwd(projectDir).text()
+    } catch (err) {
+      log = err.stdout.toString() + '\n' + err.stderr.toString()
+      error = true
+    }
+    let screenshotBase64: any = null
+    if (!error && responseImage) {
+      const screenshot = await Bun.file(`${dirname}/screenshot.jpg`).arrayBuffer()
+      screenshotBase64 = Buffer.from(screenshot).toString('base64')
+    }
+    return {
+      content: [
+        { type: 'text', text: log },
+        screenshotBase64 && {
+          type: 'image',
+          data: screenshotBase64,
+          mimeType: 'image/jpeg',
+        },
+      ].filter(Boolean),
+    }
+  },
+)
+
+const transport = new StdioServerTransport()
+await server.connect(transport)

package/tools/postinstall.ts

@@ -1,5 +1,5 @@
 import { $ } from 'bun'
-
+import { stat, readFile, writeFile } from 'fs/promises'
 const cwd = process.cwd()
 
 const libFiles = ['types', 'utils', 'index.ts']
@@ -9,4 +9,56 @@ for (const file of libFiles) {
   await $`cp -r ${__dirname}/../${file} ${cwd}/project`
 }
 
-console.log('Copied files to project/')
+console.log('Copied files to project/')
+
+async function exists(f: string) {
+  try {
+    await stat(f)
+    return true
+  } catch {
+    return false
+  }
+}
+
+const projectMcpServer = {
+  command: 'bun',
+  args: [`${cwd}/node_modules/@fugood/bricks-project/tools/mcp-server.ts`],
+}
+
+const defaultMcpConfig = {
+  mcpServers: {
+    'bricks-project': projectMcpServer,
+  },
+}
+
+const handleMcpConfigOverride = async (mcpConfigPath: string) => {
+  let mcpConfig: { mcpServers: Record<string, typeof projectMcpServer> } | null = null
+  if (await exists(mcpConfigPath)) {
+    const configStr = await readFile(mcpConfigPath, 'utf-8')
+    try {
+      mcpConfig = JSON.parse(configStr)
+      if (!mcpConfig?.mcpServers) throw new Error('mcpServers is not defined')
+      mcpConfig.mcpServers['bricks-project'] = projectMcpServer
+    } catch (e) {
+      mcpConfig = defaultMcpConfig
+    }
+  } else {
+    mcpConfig = defaultMcpConfig
+  }
+
+  await writeFile(mcpConfigPath, `${JSON.stringify(mcpConfig, null, 2)}\n`)
+
+  console.log(`Updated ${mcpConfigPath}`)
+}
+
+if (await exists(`${cwd}/.cursorrules`)) {
+  await $`mkdir -p ${cwd}/.cursor`
+  const cursorMcpConfigPath = `${cwd}/.cursor/mcp.json`
+  await handleMcpConfigOverride(cursorMcpConfigPath)
+}
+
+if (await exists(`${cwd}/CLAUDE.md`)) {
+  await $`mkdir -p ${cwd}/.cursor`
+  const claudeCodeMcpConfigPath = `${cwd}/.mcp.json`
+  await handleMcpConfigOverride(claudeCodeMcpConfigPath)
+}

package/tools/preview-main.mjs

@@ -20,7 +20,7 @@ let takeScreenshotConfig = null
 try {
   if (values['take-screenshot']) {
     takeScreenshotConfig = JSON.parse(values['take-screenshot'])
-    if (!takeScreenshotConfig.path) takeScreenshotConfig.path = `${cwd}/screenshot.png`
+    if (!takeScreenshotConfig.path) takeScreenshotConfig.path = `${cwd}/screenshot.jpg`
     if (!takeScreenshotConfig.width) takeScreenshotConfig.width = 1280
     if (!takeScreenshotConfig.height) takeScreenshotConfig.height = 768
     if (!takeScreenshotConfig.delay) takeScreenshotConfig.delay = 1000
@@ -46,8 +46,13 @@ if (!previewUrl) throw new Error(`Invalid BRICKS_STAGE: ${stage}`)
 
 app.on('ready', () => {
   let show = true
-  if (takeScreenshotConfig?.headless) show = false
-  const mainWindow = new BrowserWindow({ width: 1280, height: 768, show })
+  if (takeScreenshotConfig && !takeScreenshotConfig.noHeadless) show = false
+  const mainWindow = new BrowserWindow({
+    width: takeScreenshotConfig?.width || 1280,
+    height: takeScreenshotConfig?.height || 768,
+    frame: !takeScreenshotConfig,
+    show,
+  })
   mainWindow.setBackgroundColor('#333')
   mainWindow.loadURL(previewUrl)
 
@@ -56,18 +61,19 @@ app.on('ready', () => {
       type: 'config',
       configFile: { _originTitle: 'Test', ...config, title: Math.random().toString() },
       workspace: { billing: { lock: {}, plan: 'free' } },
+      showMenu: false,
     }
     mainWindow.webContents.executeJavaScript(
       `window.postMessage(JSON.stringify(${JSON.stringify(payload)}))`,
     )
     if (takeScreenshotConfig) {
-      const { delay, width, height, path, closeAfter } = takeScreenshotConfig
+      const { delay, width, height, path, keepOpen = false } = takeScreenshotConfig
       setTimeout(() => {
         console.log('Taking screenshot')
-        mainWindow.webContents.capturePage({ width, height }).then((image) => {
+        mainWindow.webContents.capturePage().then((image) => {
           console.log('Writing screenshot to', path)
-          writeFile(path, image.toPNG())
-          if (closeAfter) {
+          writeFile(path, image.resize({ width, height }).toJPEG(75))
+          if (!keepOpen) {
             console.log('Closing app')
             app.quit()
           }

package/tools/preview.ts

@@ -1,5 +1,6 @@
 import { $ } from 'bun'
 import { watch } from 'fs'
+import type { FSWatcher } from 'fs'
 import { parseArgs } from 'util'
 import { debounce } from 'lodash'
 
@@ -8,55 +9,60 @@ const { values } = parseArgs({
   options: {
     'skip-typecheck': { type: 'boolean' },
     'clear-cache': { type: 'boolean' },
-    'screenshot': { type: 'boolean' },
+    screenshot: { type: 'boolean' },
     'screenshot-delay': { type: 'string' },
     'screenshot-width': { type: 'string' },
     'screenshot-height': { type: 'string' },
     'screenshot-path': { type: 'string' },
-    'screenshot-close-after': { type: 'boolean' },
-    'screenshot-headless': { type: 'boolean' },
+    'screenshot-keep-open': { type: 'boolean' },
+    'screenshot-no-headless': { type: 'boolean' },
   },
   strict: true,
   allowPositionals: true,
 })
 
-const useTypecheck = !values['skip-typecheck']
-
-const compile = async () => {
-  if (useTypecheck) await $`bun typecheck`
-  await $`bun compile`
-}
-
-await compile()
-
-const compileDebounced = debounce(compile, 500)
-
 const cwd = process.cwd()
 
-const watcher = watch(`${cwd}/subspaces`, { recursive: true }, async (event, filename) => {
-  console.log(`Detected ${event} in ${filename}`)
-  compileDebounced()
-})
-
 const app = await Bun.file(`${cwd}/application.json`).json()
 
 let args: string[] = []
 if (values['clear-cache']) args.push('--clear-cache')
 
+let needWatcher = true
 if (values['screenshot']) {
   args.push(`--take-screenshot`)
+  const keepOpen = values['screenshot-keep-open'] ?? false
   args.push(
     JSON.stringify({
       delay: Number(values['screenshot-delay']) || 1000,
-      width: Number(values['screenshot-width']) || 1920,
-      height: Number(values['screenshot-height']) || 1080,
-      path: values['screenshot-path'] || `${cwd}/screenshot.png`,
-      closeAfter: values['screenshot-close-after'] ?? true,
-      headless: values['screenshot-headless'] ?? true,
+      width: Number(values['screenshot-width']) || 600,
+      height: Number(values['screenshot-height']) || 480,
+      path: values['screenshot-path'] || `${cwd}/screenshot.jpg`,
+      keepOpen,
+      noHeadless: values['screenshot-no-headless'] ?? false,
     }),
   )
+  needWatcher = keepOpen
+}
+
+const useTypecheck = !values['skip-typecheck']
+
+const compile = async () => {
+  if (useTypecheck) await $`bun typecheck`
+  await $`bun compile`
+}
+
+await compile()
+
+let watcher: FSWatcher | null = null
+if (needWatcher) {
+  const compileDebounced = debounce(compile, 500)
+  watcher = watch(`${cwd}/subspaces`, { recursive: true }, async (event, filename) => {
+    console.log(`Detected ${event} in ${filename}`)
+    compileDebounced()
+  })
 }
 
 await $`BRICKS_STAGE=${app.stage || 'production'} bunx --bun electron ${__dirname}/preview-main.mjs ${args}`
 
-watcher.close()
+if (watcher) watcher.close()