@fugood/bricks-ctor 2.25.0-beta.6 → 2.25.0-beta.60

package/tools/mcp-tools/__tests__/huggingface.test.ts

@@ -0,0 +1,49 @@
+import { buildGGUFSplitFiles } from '../huggingface'
+
+declare const describe: (name: string, fn: () => void) => void
+declare const it: (name: string, fn: () => void) => void
+declare const expect: (actual: unknown) => { toEqual: (expected: unknown) => void }
+
+describe('buildGGUFSplitFiles', () => {
+  it('builds split GGUF files with indexed sibling metadata lookup', () => {
+    const siblings = [
+      {
+        rfilename: 'other-file.txt',
+        size: 999,
+      },
+      {
+        rfilename: 'model-00001-of-00003.gguf',
+        size: 1,
+        lfs: { sha256: 'one' },
+      },
+      {
+        rfilename: 'model-00002-of-00003.gguf',
+        size: 2,
+        lfs: { sha256: 'two' },
+      },
+      {
+        rfilename: 'model-00002-of-00003.gguf',
+        size: 22,
+        lfs: { sha256: 'two-late' },
+      },
+    ]
+
+    expect(buildGGUFSplitFiles('model-00001-of-00003.gguf', '00003', siblings)).toEqual([
+      {
+        rfilename: 'model-00001-of-00003.gguf',
+        size: 1,
+        lfs: { sha256: 'one' },
+      },
+      {
+        rfilename: 'model-00002-of-00003.gguf',
+        size: 2,
+        lfs: { sha256: 'two' },
+      },
+      {
+        rfilename: 'model-00003-of-00003.gguf',
+        size: undefined,
+        lfs: undefined,
+      },
+    ])
+  })
+})

package/tools/mcp-tools/__tests__/icons.test.ts

@@ -0,0 +1,21 @@
+import { searchIcons } from '../icons'
+
+declare const describe: (name: string, fn: () => void) => void
+declare const test: (name: string, fn: () => void) => void
+declare const expect: (actual: unknown) => {
+  toBe: (expected: unknown) => void
+  toHaveLength: (expected: number) => void
+}
+
+describe('icon_search helpers', () => {
+  test('searches the selected style corpus directly', () => {
+    const results = searchIcons('x', 10, 'brands')
+
+    expect(results).toHaveLength(10)
+    expect(results.every((result) => result.item.styles.includes('brands'))).toBe(true)
+  })
+
+  test('keeps unstyled searches capped by limit', () => {
+    expect(searchIcons('arrow', 5)).toHaveLength(5)
+  })
+})

package/tools/mcp-tools/__tests__/mcp-env.test.js

@@ -0,0 +1,19 @@
+import { isTruthyEnv, shouldRegisterEditingTools } from '../../mcp-env'
+
+describe('mcp env helpers', () => {
+  test('recognizes explicit truthy env values', () => {
+    for (const value of ['1', 'true', 'TRUE', 'yes', 'on', ' on ']) {
+      expect(isTruthyEnv(value)).toBe(true)
+    }
+  })
+
+  test('does not enable editing tools by default', () => {
+    expect(shouldRegisterEditingTools({})).toBe(false)
+    expect(shouldRegisterEditingTools({ BRICKS_CTOR_ENABLE_EDITING_TOOLS: '0' })).toBe(false)
+    expect(shouldRegisterEditingTools({ BRICKS_CTOR_ENABLE_EDITING_TOOLS: 'false' })).toBe(false)
+  })
+
+  test('enables editing tools when explicitly requested', () => {
+    expect(shouldRegisterEditingTools({ BRICKS_CTOR_ENABLE_EDITING_TOOLS: '1' })).toBe(true)
+  })
+})

package/tools/mcp-tools/_editing-helpers.ts

@@ -0,0 +1,98 @@
+import * as t from '@babel/types'
+import path from 'node:path'
+
+export const oxfmtOptions = {
+  trailingComma: 'all',
+  tabWidth: 2,
+  semi: false,
+  singleQuote: true,
+  printWidth: 100,
+} as const
+
+export const isRecord = (value: unknown): value is Record<string, unknown> =>
+  Boolean(value) && typeof value === 'object' && !Array.isArray(value)
+
+export const isIdentifierName = (value: string) => /^[$A-Z_a-z][$\w]*$/.test(value)
+
+export const normalizeRelPath = (file: string) => file.replace(/\\/g, '/').replace(/^\.\/+/, '')
+
+export const projectRelativePath = (projectDir: string, absPath: string) =>
+  normalizeRelPath(path.relative(projectDir, absPath))
+
+export const getPropertyKeyName = (key: t.Expression | t.PrivateName) => {
+  if (t.isIdentifier(key)) return key.name
+  if (t.isStringLiteral(key) || t.isNumericLiteral(key)) return String(key.value)
+  return null
+}
+
+export const makeObjectKey = (key: string) =>
+  isIdentifierName(key) ? t.identifier(key) : t.stringLiteral(key)
+
+export const getObjectProperty = (object: t.ObjectExpression, key: string) =>
+  object.properties.find((property): property is t.ObjectProperty => {
+    if (!t.isObjectProperty(property)) return false
+    return getPropertyKeyName(property.key) === key
+  })
+
+export const getStringProperty = (object: t.ObjectExpression, key: string) => {
+  const property = getObjectProperty(object, key)
+  if (!property || !t.isStringLiteral(property.value)) return undefined
+  return property.value.value
+}
+
+export const setObjectProperty = (object: t.ObjectExpression, key: string, value: t.Expression) => {
+  const existing = getObjectProperty(object, key)
+  if (existing) {
+    existing.value = value
+    return
+  }
+  object.properties.push(t.objectProperty(makeObjectKey(key), value))
+}
+
+export const removeObjectProperty = (object: t.ObjectExpression, key: string) => {
+  const index = object.properties.findIndex((property) => {
+    if (!t.isObjectProperty(property)) return false
+    return getPropertyKeyName(property.key) === key
+  })
+  if (index >= 0) object.properties.splice(index, 1)
+}
+
+export const insertImport = (ast: t.File, declaration: t.ImportDeclaration) => {
+  const body = ast.program.body
+  const lastImportIndex = body.findLastIndex((statement) => t.isImportDeclaration(statement))
+  body.splice(lastImportIndex + 1, 0, declaration)
+}
+
+export const ensureBricksCtorImport = (
+  ast: t.File,
+  importKind: 'type' | 'value',
+  names: Iterable<string>,
+) => {
+  const missing = new Set(Array.from(names).filter(Boolean))
+  if (missing.size === 0) return
+
+  for (const statement of ast.program.body) {
+    if (!t.isImportDeclaration(statement) || statement.source.value !== 'bricks-ctor') continue
+    const isTypeImport = statement.importKind === 'type'
+    if ((importKind === 'type') !== isTypeImport) continue
+
+    statement.specifiers.forEach((specifier) => {
+      if (!t.isImportSpecifier(specifier)) return
+      const imported = specifier.imported
+      if (t.isIdentifier(imported)) missing.delete(imported.name)
+      if (t.isStringLiteral(imported)) missing.delete(imported.value)
+    })
+
+    missing.forEach((name) => {
+      statement.specifiers.push(t.importSpecifier(t.identifier(name), t.identifier(name)))
+    })
+    return
+  }
+
+  const declaration = t.importDeclaration(
+    Array.from(missing).map((name) => t.importSpecifier(t.identifier(name), t.identifier(name))),
+    t.stringLiteral('bricks-ctor'),
+  )
+  if (importKind === 'type') declaration.importKind = 'type'
+  insertImport(ast, declaration)
+}

package/tools/mcp-tools/_verify.ts

@@ -0,0 +1,50 @@
+import { sh } from '../_shell'
+import { computeConfigChange, readBuildConfig, type ConfigChange } from '../../compile/config-diff'
+
+// Result of an editing tool's compile verification, carrying the config delta the
+// compile produced. Shared by entry-editing and data-calc-editing.
+export type VerificationResult = {
+  status: 'skipped' | 'compile:ok' | 'compile:failed'
+  errors: string[]
+  configChange?: ConfigChange
+}
+
+// Strip ANSI colors from the spawned compile so the captured output stays plain text.
+const noColorEnv = { FORCE_COLOR: '0', NO_COLOR: '1' }
+
+// Recompile the project and diff the result against the prior build artifact.
+export const compileAndDiff = async (projectDir: string): Promise<VerificationResult> => {
+  // Snapshot the prior compiled config before recompiling (the new artifact overwrites it).
+  const before = await readBuildConfig(projectDir)
+  try {
+    // Turn off the editing-tools audit for this spawned compile so compile() doesn't write
+    // a duplicate generic entry — the calling editing tool records its own richer entry.
+    await sh`bun compile`
+      .cwd(projectDir)
+      .env({ ...noColorEnv, BRICKS_CTOR_ENABLE_EDITING_TOOLS: '0' })
+      .text()
+  } catch (err: any) {
+    const stdout = err.stdout?.toString() ?? ''
+    const stderr = err.stderr?.toString() ?? ''
+    const output = [stdout, stderr, err.message].filter(Boolean).join('\n').trim()
+    return { status: 'compile:failed', errors: output ? [output] : ['bun compile failed'] }
+  }
+  const after = await readBuildConfig(projectDir)
+  return { status: 'compile:ok', errors: [], configChange: computeConfigChange(before, after) }
+}
+
+const shouldVerify = (inputVerify?: boolean) => {
+  if (inputVerify !== undefined) return inputVerify
+  const value = process.env.BRICKS_CTOR_MCP_EDIT_VERIFY ?? 'true'
+  return !['0', 'false', 'no', 'off'].includes(value.toLowerCase())
+}
+
+// Compile verification shared by the source-editing tools: per-call `verify` wins,
+// then the BRICKS_CTOR_MCP_EDIT_VERIFY env toggle, defaulting to on.
+export const verifyProject = async (
+  projectDir: string,
+  verify?: boolean,
+): Promise<VerificationResult> => {
+  if (!shouldVerify(verify)) return { status: 'skipped', errors: [] }
+  return compileAndDiff(projectDir)
+}

package/tools/mcp-tools/compile.ts

@@ -3,24 +3,33 @@ import { readFile } from 'node:fs/promises'
 import { z } from 'zod'
 import { sh } from '../_shell'
 
+// Disable ANSI color codes from spawned tools so MCP text output stays readable
+// (the host renders raw text; escape sequences leak into the result otherwise).
+const noColorEnv = { FORCE_COLOR: '0', NO_COLOR: '1' }
+
 export function register(server: McpServer, projectDir: string) {
   const { dirname } = import.meta
 
+  // `bun compile` records the resulting config delta to `.bricks/edits.jsonl` itself
+  // (see compile() in compile/index.ts), so the spawned output below already reflects it.
   server.tool('compile', {}, async () => {
-    let log = ''
+    let log = 'Type checking & Compiling...\n'
     try {
-      log += 'Type checking & Compiling...'
-      log += await sh`bun compile`.cwd(projectDir).text()
-    } catch (err) {
-      log += `${err.stdout.toString()}\n${err.stderr.toString()}`
+      log += await sh`bun compile`.cwd(projectDir).env(noColorEnv).text()
+    } catch (err: any) {
+      const stdout = err.stdout?.toString() ?? ''
+      const stderr = err.stderr?.toString() ?? ''
+      log += [stdout, stderr].filter(Boolean).join('\n')
     }
     return {
       content: [{ type: 'text', text: log }],
     }
   })
 
+  if (process.env.BRICKS_CTOR_MCP_DISABLE_PREVIEW === '1') return
+
   server.tool(
-    'preview',
+    'simulator',
     {
       delay: z
         .number()
@@ -61,11 +70,14 @@ export function register(server: McpServer, projectDir: string) {
         ]
         if (testId) args.push('--test-id', testId)
         if (testTitleLike) args.push('--test-title-like', testTitleLike)
-        log = await sh`bunx --bun electron ${toolsDir}/preview-main.mjs ${args}`
+        log = await sh`bunx --bun electron ${toolsDir}/simulator-main.mjs ${args}`
           .cwd(projectDir)
+          .env(noColorEnv)
           .text()
-      } catch (err) {
-        log = `${err.stdout.toString()}\n${err.stderr.toString()}`
+      } catch (err: any) {
+        const stdout = err.stdout?.toString() ?? ''
+        const stderr = err.stderr?.toString() ?? ''
+        log = [stdout, stderr].filter(Boolean).join('\n')
         error = true
       }
       let screenshotBase64: string | null = null