triflux 3.2.0-dev.9 → 3.3.0-dev.3

package/hub/tools.mjs

@@ -8,16 +8,25 @@ import {
   teamTaskUpdate,
   teamSendMessage,
 } from './team/nativeProxy.mjs';
+import {
+  ensurePipelineTable,
+  createPipeline,
+} from './pipeline/index.mjs';
+import {
+  readPipelineState,
+  initPipelineState,
+  listPipelineStates,
+} from './pipeline/state.mjs';
 
 /**
  * MCP 도구 목록 생성
  * @param {object} store  — createStore() 반환
  * @param {object} router — createRouter() 반환
- * @param {object} hitl   — createHitlManager() 반환
- * @param {object} pipe   — createPipeServer() 반환
- * @returns {Array<{name, description, inputSchema, handler}>}
- */
-export function createTools(store, router, hitl, pipe = null) {
+ * @param {object} hitl   — createHitlManager() 반환
+ * @param {object} pipe   — createPipeServer() 반환
+ * @returns {Array<{name, description, inputSchema, handler}>}
+ */
+export function createTools(store, router, hitl, pipe = null) {
   /** 도구 핸들러 래퍼 — 에러 처리 + MCP content 형식 변환 */
   function wrap(code, fn) {
     return async (args) => {
@@ -49,10 +58,10 @@ export function createTools(store, router, hitl, pipe = null) {
           heartbeat_ttl_ms: { type: 'integer', minimum: 10000, maximum: 7200000 },
         },
       },
-      handler: wrap('REGISTER_FAILED', (args) => {
-        const data = router.registerAgent(args);
-        return { ok: true, data };
-      }),
+      handler: wrap('REGISTER_FAILED', (args) => {
+        const data = router.registerAgent(args);
+        return { ok: true, data };
+      }),
     },
 
     // ── 2. status ──
@@ -122,10 +131,10 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 5. poll_messages ──
-    {
-      name: 'poll_messages',
-      description: 'Deprecated. poll_messages 대신 Named Pipe subscribe/publish 채널을 사용합니다',
+    // ── 5. poll_messages ──
+    {
+      name: 'poll_messages',
+      description: 'Deprecated. poll_messages 대신 Named Pipe subscribe/publish 채널을 사용합니다',
       inputSchema: {
         type: 'object',
         required: ['agent_id'],
@@ -137,40 +146,40 @@ export function createTools(store, router, hitl, pipe = null) {
           ack_ids:        { type: 'array', items: { type: 'string' }, maxItems: 100 },
           auto_ack:       { type: 'boolean', default: false },
         },
-      },
-      handler: wrap('POLL_DEPRECATED', async (args) => {
-        const replay = router.drainAgent(args.agent_id, {
-          max_messages: args.max_messages,
-          include_topics: args.include_topics,
-          auto_ack: args.auto_ack,
-        });
-        if (args.ack_ids?.length) {
-          router.ackMessages(args.ack_ids, args.agent_id);
-        }
-        return {
-          ok: false,
-          error: {
-            code: 'POLL_DEPRECATED',
-            message: 'poll_messages는 deprecated 되었습니다. pipe subscribe/publish 채널을 사용하세요.',
-          },
-          data: {
-            pipe_path: pipe?.path || null,
-            delivery_mode: 'pipe_push',
-            protocol: 'ndjson',
-            replay: {
-              messages: replay,
-              count: replay.length,
-            },
-            server_time_ms: Date.now(),
-          },
-        };
-      }),
+      },
+      handler: wrap('POLL_DEPRECATED', async (args) => {
+        const replay = router.drainAgent(args.agent_id, {
+          max_messages: args.max_messages,
+          include_topics: args.include_topics,
+          auto_ack: args.auto_ack,
+        });
+        if (args.ack_ids?.length) {
+          router.ackMessages(args.ack_ids, args.agent_id);
+        }
+        return {
+          ok: false,
+          error: {
+            code: 'POLL_DEPRECATED',
+            message: 'poll_messages는 deprecated 되었습니다. pipe subscribe/publish 채널을 사용하세요.',
+          },
+          data: {
+            pipe_path: pipe?.path || null,
+            delivery_mode: 'pipe_push',
+            protocol: 'ndjson',
+            replay: {
+              messages: replay,
+              count: replay.length,
+            },
+            server_time_ms: Date.now(),
+          },
+        };
+      }),
     },
 
     // ── 6. handoff ──
-    {
-      name: 'handoff',
-      description: '다른 에이전트에게 작업을 인계합니다. acceptance_criteria로 완료 기준 지정 가능',
+    {
+      name: 'handoff',
+      description: '다른 에이전트에게 작업을 인계합니다. acceptance_criteria로 완료 기준 지정 가능',
       inputSchema: {
         type: 'object',
         required: ['from', 'to', 'topic', 'task'],
@@ -187,15 +196,90 @@ export function createTools(store, router, hitl, pipe = null) {
           correlation_id:      { type: 'string' },
         },
       },
-      handler: wrap('HANDOFF_FAILED', (args) => {
-        return router.handleHandoff(args);
-      }),
-    },
-
-    // ── 7. request_human_input ──
-    {
-      name: 'request_human_input',
-      description: '사용자에게 입력을 요청합니다 (CAPTCHA, 승인, 자격증명, 선택, 텍스트)',
+      handler: wrap('HANDOFF_FAILED', (args) => {
+        return router.handleHandoff(args);
+      }),
+    },
+
+    // ── 7. assign_async ──
+    {
+      name: 'assign_async',
+      description: 'AWS CAO 스타일 비차단 assign job을 생성하고 워커에게 실시간 전달합니다',
+      inputSchema: {
+        type: 'object',
+        required: ['supervisor_agent', 'worker_agent', 'task'],
+        properties: {
+          supervisor_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          topic: { type: 'string', pattern: '^[a-zA-Z0-9._:-]+$', default: 'assign.job' },
+          task: { type: 'string', minLength: 1, maxLength: 20000 },
+          payload: { type: 'object' },
+          priority: { type: 'integer', minimum: 1, maximum: 9, default: 5 },
+          ttl_ms: { type: 'integer', minimum: 1000, maximum: 86400000, default: 600000 },
+          timeout_ms: { type: 'integer', minimum: 1000, maximum: 86400000, default: 600000 },
+          max_retries: { type: 'integer', minimum: 0, maximum: 20, default: 0 },
+          trace_id: { type: 'string' },
+          correlation_id: { type: 'string' },
+        },
+      },
+      handler: wrap('ASSIGN_ASYNC_FAILED', (args) => {
+        return router.assignAsync(args);
+      }),
+    },
+
+    // ── 8. assign_result ──
+    {
+      name: 'assign_result',
+      description: 'assign job의 진행/완료 결과를 보고합니다. completed + metadata.result 관례를 지원합니다',
+      inputSchema: {
+        type: 'object',
+        required: ['job_id', 'status'],
+        properties: {
+          job_id: { type: 'string', minLength: 1, maxLength: 128 },
+          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          status: { type: 'string', enum: ['queued', 'running', 'in_progress', 'completed', 'succeeded', 'success', 'failed', 'error', 'timed_out', 'timeout'] },
+          attempt: { type: 'integer', minimum: 1 },
+          result: {},
+          error: {},
+          metadata: { type: 'object' },
+          payload: { type: 'object' },
+        },
+      },
+      handler: wrap('ASSIGN_RESULT_FAILED', (args) => {
+        return router.reportAssignResult(args);
+      }),
+    },
+
+    // ── 9. assign_status ──
+    {
+      name: 'assign_status',
+      description: 'assign job 단건 상태 또는 supervisor/worker/status 기준 목록을 조회합니다',
+      inputSchema: {
+        type: 'object',
+        properties: {
+          job_id: { type: 'string', minLength: 1, maxLength: 128 },
+          supervisor_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          status: { type: 'string', enum: ['queued', 'running', 'succeeded', 'failed', 'timed_out'] },
+          statuses: {
+            type: 'array',
+            items: { type: 'string', enum: ['queued', 'running', 'succeeded', 'failed', 'timed_out'] },
+            maxItems: 8,
+          },
+          trace_id: { type: 'string' },
+          correlation_id: { type: 'string' },
+          limit: { type: 'integer', minimum: 1, maximum: 100, default: 50 },
+        },
+      },
+      handler: wrap('ASSIGN_STATUS_FAILED', (args) => {
+        return router.getAssignStatus(args);
+      }),
+    },
+
+    // ── 10. request_human_input ──
+    {
+      name: 'request_human_input',
+      description: '사용자에게 입력을 요청합니다 (CAPTCHA, 승인, 자격증명, 선택, 텍스트)',
       inputSchema: {
         type: 'object',
         required: ['requester_agent', 'kind', 'prompt', 'requested_schema', 'deadline_ms', 'default_action'],
@@ -211,12 +295,12 @@ export function createTools(store, router, hitl, pipe = null) {
           correlation_id:     { type: 'string' },
         },
       },
-      handler: wrap('HITL_REQUEST_FAILED', (args) => {
-        return hitl.requestHumanInput(args);
-      }),
-    },
-
-    // ── 8. submit_human_input ──
+      handler: wrap('HITL_REQUEST_FAILED', (args) => {
+        return hitl.requestHumanInput(args);
+      }),
+    },
+
+    // ── 11. submit_human_input ──
     {
       name: 'submit_human_input',
       description: '사용자 입력 요청에 응답합니다 (accept, decline, cancel)',
@@ -235,7 +319,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 9. team_info ──
+    // ── 12. team_info ──
     {
       name: 'team_info',
       description: 'Claude Native Teams 메타/멤버/경로 정보를 조회합니다',
@@ -253,7 +337,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 10. team_task_list ──
+    // ── 13. team_task_list ──
     {
       name: 'team_task_list',
       description: 'Claude Native Teams task 목록을 owner/status 조건으로 조회합니다',
@@ -277,7 +361,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 11. team_task_update ──
+    // ── 14. team_task_update ──
     {
       name: 'team_task_update',
       description: 'Claude Native Teams task를 claim/update 합니다',
@@ -305,7 +389,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 12. team_send_message ──
+    // ── 15. team_send_message ──
     {
       name: 'team_send_message',
       description: 'Claude Native Teams inbox에 메시지를 append 합니다',
@@ -325,5 +409,81 @@ export function createTools(store, router, hitl, pipe = null) {
         return teamSendMessage(args);
       }),
     },
+
+    // ── 16. pipeline_state ──
+    {
+      name: 'pipeline_state',
+      description: '파이프라인 상태를 조회합니다 (--thorough 모드)',
+      inputSchema: {
+        type: 'object',
+        required: ['team_name'],
+        properties: {
+          team_name: { type: 'string', pattern: '^[a-z0-9][a-z0-9-]*$' },
+        },
+      },
+      handler: wrap('PIPELINE_STATE_FAILED', (args) => {
+        ensurePipelineTable(store.db);
+        const state = readPipelineState(store.db, args.team_name);
+        return state
+          ? { ok: true, data: state }
+          : { ok: false, error: { code: 'PIPELINE_NOT_FOUND', message: `파이프라인 없음: ${args.team_name}` } };
+      }),
+    },
+
+    // ── 17. pipeline_advance ──
+    {
+      name: 'pipeline_advance',
+      description: '파이프라인을 다음 단계로 전이합니다 (전이 규칙 + fix loop 바운딩 적용)',
+      inputSchema: {
+        type: 'object',
+        required: ['team_name', 'phase'],
+        properties: {
+          team_name: { type: 'string', pattern: '^[a-z0-9][a-z0-9-]*$' },
+          phase: { type: 'string', enum: ['plan', 'prd', 'exec', 'verify', 'fix', 'complete', 'failed'] },
+        },
+      },
+      handler: wrap('PIPELINE_ADVANCE_FAILED', (args) => {
+        ensurePipelineTable(store.db);
+        const pipeline = createPipeline(store.db, args.team_name);
+        return pipeline.advance(args.phase);
+      }),
+    },
+
+    // ── 18. pipeline_init ──
+    {
+      name: 'pipeline_init',
+      description: '새 파이프라인을 초기화합니다 (기존 상태 덮어쓰기)',
+      inputSchema: {
+        type: 'object',
+        required: ['team_name'],
+        properties: {
+          team_name: { type: 'string', pattern: '^[a-z0-9][a-z0-9-]*$' },
+          fix_max: { type: 'integer', minimum: 1, maximum: 20, default: 3 },
+          ralph_max: { type: 'integer', minimum: 1, maximum: 100, default: 10 },
+        },
+      },
+      handler: wrap('PIPELINE_INIT_FAILED', (args) => {
+        ensurePipelineTable(store.db);
+        const state = initPipelineState(store.db, args.team_name, {
+          fix_max: args.fix_max,
+          ralph_max: args.ralph_max,
+        });
+        return { ok: true, data: state };
+      }),
+    },
+
+    // ── 19. pipeline_list ──
+    {
+      name: 'pipeline_list',
+      description: '활성 파이프라인 목록을 조회합니다',
+      inputSchema: {
+        type: 'object',
+        properties: {},
+      },
+      handler: wrap('PIPELINE_LIST_FAILED', () => {
+        ensurePipelineTable(store.db);
+        return { ok: true, data: listPipelineStates(store.db) };
+      }),
+    },
   ];
 }