@nine-lab/nine-mu 0.1.256 → 0.1.258

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nine-lab/nine-mu",
-  "version": "0.1.256",
+  "version": "0.1.258",
   "description": "AI-Driven Full-Stack Code Fabrication Engine",
   "type": "module",
   "main": "./dist/nine-mu.umd.js",

package/src/components/NineChatManager.js

@@ -3,7 +3,8 @@ import { WebSocketClientTransport } from "@modelcontextprotocol/sdk/client/webso
 import { trace } from "@nopeer";
 import { api } from "@nine-lab/nine-util";
 import { z } from "zod";
-import { LoggingMessageNotificationSchema } from "@modelcontextprotocol/sdk/types.js";
+//import { LoggingMessageNotificationSchema } from "@modelcontextprotocol/sdk/types.js";
+import { NotificationHandler } from "./NotificationHandler.js";
 
 export class NineChatManager {
     #mcpClient;
@@ -13,6 +14,7 @@ export class NineChatManager {
     #owner;
     #routeUrl;
     #chatHistory = []; // 💡 이전 대화들을 저장할 메모리 창고
+    #notiHandler;
 
     constructor(owner, callbacks = {}) {
         this.#owner = owner;
@@ -23,6 +25,7 @@ export class NineChatManager {
         this.#onStatus = callbacks.onStatus;
         this.#onMessage = callbacks.onMessage;
 
+        this.#notiHandler = new NotificationHandler(this.#owner);
         // 연결 시작 (비동기)
         this.#connect();
     }
@@ -44,7 +47,7 @@ export class NineChatManager {
 
         //trace.log(this.#mcpClient);
 
-        this.#mcpClient.connect(transport, {timeout: 300000})
+        this.#mcpClient.connect(transport, {timeout: 3000000})
             .then( async () => {
                 this.#updateStatus("✅ MCP Connected");
 
@@ -69,6 +72,14 @@ export class NineChatManager {
                     })
                 });
 //
+                this.#mcpClient.setNotificationHandler(
+                    customNotificationSchema,
+                    (notification) => {
+                        // 날것의 수신 데이터를 그대로 가공 전담 파일로 던짐!
+                        this.#notiHandler.handle(notification);
+                    }
+                );
+                /**
                 this.#mcpClient.setNotificationHandler(
                     customNotificationSchema, // 👈 껍데기 통과용 커스텀 스키마
                     (notification) => {
@@ -78,7 +89,7 @@ export class NineChatManager {
                         this.#owner.addMessage("ai", notification.params.message);
                         this.#owner.addMessage("ing", "");
                     }
-                );
+                ); */
             })
             .catch((err) => {
                 trace.error("❌ MCP 연결 실패", err);
@@ -114,7 +125,7 @@ export class NineChatManager {
         return this.#mcpClient.callTool(
             { name: toolName, arguments: args },
             undefined, // 필요 시 extra 변수 자리 (기본값 undefined)
-            { timeout: 300000 } // 💡 개별 요청 타임아웃을 5분으로 확장
+            { timeout: 3000000 } // 💡 개별 요청 타임아웃을 5분으로 확장
         );
     }
 

package/src/components/NotificationHandler.js

@@ -0,0 +1,66 @@
+import { trace } from "@nopeer";
+
+export class NotificationHandler {
+    #owner;
+
+    constructor(owner) {
+        this.#owner = owner; // 웹 컴포넌트나 UI DOM 인스턴스
+    }
+
+    /**
+     * MCP로부터 수신된 날것의 노티파이 데이터를 정석대로 라우팅하는 메인 메서드
+     */
+    handle(notification) {
+        try {
+            const { method, params } = notification;
+
+            // 로깅 및 진행 상황 메시지 노티인 경우
+            if (method === "notifications/logging/message") {
+                this.#handleLoggingMessage(params);
+            }
+
+            // 추후 다른 프로토콜 메서드가 추가되면 여기에 깔끔하게 분기 확장 가능
+            // else if (method === "notifications/다른거") { ... }
+
+        } catch (error) {
+            trace.error("🔥 노티파이 핸들러 내부 처리 실패:", error);
+        }
+    }
+
+    /**
+     * [분리 비즈니스 로직 1] 로깅 및 UI 메시지 출력 제어
+     */
+    #handleLoggingMessage(params) {
+        const { message, data, logger } = params;
+
+        // 1. 기본 UI 메시지 출력 (화면에 실시간으로 빌드 상태 뿌려주기)
+        if (message) {
+            this.#owner.addMessage("ai", message);
+            this.#owner.addMessage("ing", "");
+        }
+
+        // 2. 🚨 [나중에 100시간짜리 완벽을 채울 핵심 길목]
+        // 데이터에 소스코드 알맹이(data.source)나 메타데이터가 실려 내려왔을 때의 행위를 여기서 격리 처리합니다.
+        if (data && data.status === "PROGRESS") {
+            trace.log(`[시그널 감지] ${data.menu} - ${data.layer} 설계 완료`);
+
+            // 여기서 파일 저장 전 버퍼에 모으거나,
+            // 방어용 헬퍼 객체를 호출하여 소스 파싱 검증을 수행하는 등의 정석 로직이 들어갈 자리입니다.
+            //this.#processSourceData(data);
+        }
+    }
+
+    /**
+     * [분리 비즈니스 로직 2] 수신된 소스코드 데이터 유효성 검증 및 추후 파일 저장 연동
+     */
+    #processSourceData(sourceData) {
+        // 소스코드 데이터가 정상적으로 확보되었는지 1%의 누락 검증
+        if (!sourceData.source || !sourceData.file_name) {
+            trace.warn("⚠️ 소스 또는 파일명 누락 데이터 수신됨", sourceData);
+            return;
+        }
+
+        // TODO: 추후 다른 거 개발 완료 시, 여기에 실제 로컬 물리 파일 라이터 서비스 연결
+        // Example: FileWritableService.write(sourceData.path, sourceData.file_name, sourceData.source);
+    }
+}