@vibe-forge/mcp 0.9.2 → 0.10.1

package/__tests__/task-tool.spec.ts

@@ -0,0 +1,85 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+import { createToolTester } from './mcp-test-utils.js'
+
+const mocks = vi.hoisted(() => {
+  return {
+    callHook: vi.fn(),
+    createChildSession: vi.fn(),
+    getParentSessionId: vi.fn(),
+    startTask: vi.fn(),
+    getTask: vi.fn(),
+    stopTask: vi.fn(),
+    getAllTasks: vi.fn(),
+    uuid: vi.fn()
+  }
+})
+
+vi.mock('@vibe-forge/hooks', () => ({
+  callHook: mocks.callHook
+}))
+
+vi.mock('@vibe-forge/utils/uuid', () => ({
+  uuid: mocks.uuid
+}))
+
+vi.mock('#~/sync.js', () => ({
+  createChildSession: mocks.createChildSession,
+  getParentSessionId: mocks.getParentSessionId
+}))
+
+vi.mock('#~/tools/task/manager.js', () => ({
+  TaskManager: class {
+    startTask = mocks.startTask
+    getTask = mocks.getTask
+    stopTask = mocks.stopTask
+    getAllTasks = mocks.getAllTasks
+  }
+}))
+
+describe('task tool integration', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    process.env.__VF_PROJECT_AI_SESSION_ID__ = 'sess-1'
+    let nextTaskId = 1
+    mocks.uuid.mockImplementation(() => `task-${nextTaskId++}`)
+    mocks.callHook.mockResolvedValue({ continue: true })
+    mocks.getParentSessionId.mockReturnValue(undefined)
+    mocks.startTask.mockResolvedValue(undefined)
+    mocks.getTask.mockImplementation((taskId: string) => ({
+      taskId,
+      status: 'completed',
+      logs: []
+    }))
+    mocks.stopTask.mockReturnValue(true)
+    mocks.getAllTasks.mockReturnValue([])
+  })
+
+  it('passes resolved task ids to the StartTasks hook', async () => {
+    const { createTaskRegister } = await import('#~/tools/task/index.js')
+
+    const tester = createToolTester()
+    createTaskRegister()(tester.mockRegister)
+
+    await tester.callTool('StartTasks', {
+      tasks: [{
+        description: 'only output ok',
+        type: 'default',
+        background: false
+      }]
+    })
+
+    expect(mocks.callHook).toHaveBeenCalledWith('StartTasks', expect.objectContaining({
+      sessionId: 'sess-1',
+      tasks: [expect.objectContaining({
+        taskId: 'task-1',
+        description: 'only output ok',
+        type: 'default',
+        background: false
+      })]
+    }))
+    expect(mocks.startTask).toHaveBeenCalledWith(expect.objectContaining({
+      taskId: 'task-1'
+    }))
+  })
+})

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-forge/mcp",
-  "version": "0.9.2",
+  "version": "0.10.1",
   "description": "Vibe Forge MCP server",
   "imports": {
     "#~/*.js": {
@@ -33,13 +33,13 @@
     "@modelcontextprotocol/sdk": "^1.25.3",
     "commander": "^12.1.0",
     "zod": "^3.24.1",
-    "@vibe-forge/config": "^0.9.0",
-    "@vibe-forge/cli-helper": "^0.9.0",
-    "@vibe-forge/task": "^0.9.0",
-    "@vibe-forge/register": "^0.9.0",
-    "@vibe-forge/hooks": "^0.9.1",
-    "@vibe-forge/utils": "^0.9.0",
-    "@vibe-forge/types": "^0.9.0"
+    "@vibe-forge/hooks": "^0.10.2",
+    "@vibe-forge/config": "^0.10.0",
+    "@vibe-forge/cli-helper": "^0.10.1",
+    "@vibe-forge/register": "^0.10.1",
+    "@vibe-forge/task": "^0.10.1",
+    "@vibe-forge/types": "^0.10.1",
+    "@vibe-forge/utils": "^0.10.1"
   },
   "scripts": {
     "test": "pnpm -C ../.. exec vitest run --workspace vitest.workspace.ts --project bundler packages/mcp/__tests__"

package/src/command.ts

@@ -59,5 +59,5 @@ export const configureMcpCommand = (command: Command, version: string) => (
 )
 
 export function registerMcpCommand(program: Command, version: string) {
-  return configureMcpCommand(program, version)
+  return configureMcpCommand(program.command('mcp'), version)
 }

package/src/tools/task/index.ts

@@ -65,7 +65,7 @@ export const createTaskRegister = () => {
         await callHook('StartTasks', {
           cwd: process.cwd(),
           sessionId: process.env.__VF_PROJECT_AI_SESSION_ID__!,
-          tasks
+          tasks: resolvedTasks
         })
         const syncResults = parentSessionId
           ? await Promise.allSettled(resolvedTasks.map(task =>

package/src/tools/task/manager.ts

@@ -78,7 +78,10 @@ export class TaskManager {
       const [data, resolvedConfig] = await generateAdapterQueryOptions(
         promptType,
         promptName,
-        promptCWD
+        promptCWD,
+        {
+          adapter
+        }
       )
       const env = {
         ...process.env,