docuking-mcp 2.12.0 → 2.13.0

package/handlers/sync.js

@@ -1227,14 +1227,48 @@ export async function handlePullInternal(args) {
   const localIndex = loadIndex(localPath);
   const syncedFiles = []; // 다운로드/업데이트된 파일 목록
 
-  for (const file of sortedFiles) {
+  // ========================================
+  // 폴더 먼저 처리
+  // ========================================
+  const folders = sortedFiles.filter(f => f.type === 'folder');
+  const files_only = sortedFiles.filter(f => f.type !== 'folder');
+
+  for (const folder of folders) {
     current++;
-    const progress = `${current}/${total}`;
+    const serverMeta = serverPathToMeta[folder.path];
+    let fullPath;
+    if (folder.path.startsWith('xx_') ||
+        folder.path.startsWith('yy_All_Docu/') ||
+        folder.path.startsWith('yy_All_Docu') ||
+        folder.path.startsWith('yy_Coworker_') ||
+        folder.path.startsWith('zz_ai_')) {
+      fullPath = path.join(localPath, folder.path);
+    } else {
+      fullPath = path.join(mainFolderPath, folder.path);
+    }
+
+    if (fs.existsSync(fullPath)) {
+      results.push({ type: 'skip', path: folder.path, itemType: 'folder' });
+      skipped++;
+    } else {
+      fs.mkdirSync(fullPath, { recursive: true });
+      results.push({ type: 'download', path: folder.path, itemType: 'folder' });
+      foldersCreated++;
+      downloaded++;
+      if (serverMeta && serverMeta.source === 'web') {
+        filesToMarkAsPulled.push(folder.path);
+      }
+    }
+  }
 
-    // 서버 메타 정보 조회
-    const serverMeta = serverPathToMeta[file.path];
+  // ========================================
+  // 파일 처리 - Batch API 사용 (200개씩)
+  // ========================================
+  const BATCH_SIZE = 200;
 
-    // 서버 경로에 따라 로컬 저장 위치 결정
+  // 먼저 로컬 해시 기반으로 스킵할 파일 필터링
+  const filesToDownload = [];
+  for (const file of files_only) {
     let fullPath;
     if (file.path.startsWith('xx_') ||
         file.path.startsWith('yy_All_Docu/') ||
@@ -1246,119 +1280,159 @@ export async function handlePullInternal(args) {
       fullPath = path.join(mainFolderPath, file.path);
     }
 
-    try {
-      // ========================================
-      // 폴더 처리 (type === 'folder')
-      // ========================================
-      if (file.type === 'folder') {
-        if (fs.existsSync(fullPath)) {
-          // 이미 있음 → 스킵
-          results.push({ type: 'skip', path: file.path, itemType: 'folder' });
-          skipped++;
-        } else {
-          // 폴더 생성
-          fs.mkdirSync(fullPath, { recursive: true });
-          results.push({ type: 'download', path: file.path, itemType: 'folder' });
-          foldersCreated++;
-          downloaded++;
+    const serverHash = serverPathToHash[file.path];
 
-          // source: 'web' 폴더면 mark-as-pulled 대상
-          if (serverMeta && serverMeta.source === 'web') {
-            filesToMarkAsPulled.push(file.path);
-          }
+    // 로컬 파일 해시 비교
+    if (fs.existsSync(fullPath)) {
+      try {
+        const localStat = fs.statSync(fullPath);
+        if (localStat.isDirectory()) {
+          results.push({ type: 'skip', path: file.path, error: '같은 이름의 폴더 존재' });
+          skipped++;
+          current++;
+          continue;
         }
-        continue;
+        const localBuffer = fs.readFileSync(fullPath);
+        const localHash = crypto.createHash('sha256').update(localBuffer).digest('hex');
+        if (localHash === serverHash) {
+          results.push({ type: 'skip', path: file.path });
+          skipped++;
+          current++;
+          continue;
+        }
+      } catch (e) {
+        // 해시 계산 실패 시 다운로드 대상
       }
+    }
 
-      // ========================================
-      // 파일 처리 (type === 'file')
-      // ========================================
-      const response = await fetch(
-        `${API_ENDPOINT}/files?projectId=${projectId}&path=${encodeURIComponent(file.path)}`,
-        {
-          headers: {
-            'Authorization': `Bearer ${apiKey}`,
-          },
-        }
-      );
+    filesToDownload.push({ ...file, fullPath });
+  }
 
-      if (!response.ok) {
-        results.push({ type: 'fail', path: file.path, error: await response.text() });
-        failed++;
-        continue;
-      }
+  console.error(`[DocuKing] Batch Pull: ${filesToDownload.length}개 파일 다운로드 필요 (${Math.ceil(filesToDownload.length / BATCH_SIZE)}개 배치)`);
 
-      const data = await response.json();
+  // Batch API로 파일 다운로드
+  for (let i = 0; i < filesToDownload.length; i += BATCH_SIZE) {
+    const batch = filesToDownload.slice(i, i + BATCH_SIZE);
+    const batchPaths = batch.map(f => f.path);
+    const batchNum = Math.floor(i / BATCH_SIZE) + 1;
+    const totalBatches = Math.ceil(filesToDownload.length / BATCH_SIZE);
 
-      // 인코딩에 따라 저장
-      const content = data.file?.content || data.content || '';
-      const encoding = data.file?.encoding || data.encoding || 'utf-8';
+    console.error(`[DocuKing] Batch ${batchNum}/${totalBatches}: ${batch.length}개 파일 다운로드 중...`);
 
-      // 서버 파일 해시 계산
-      let serverBuffer;
-      if (encoding === 'base64') {
-        serverBuffer = Buffer.from(content, 'base64');
-      } else {
-        serverBuffer = Buffer.from(content, 'utf-8');
+    try {
+      const batchResponse = await fetch(`${API_ENDPOINT}/files/batch-pull`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${apiKey}`,
+        },
+        body: JSON.stringify({
+          projectId,
+          paths: batchPaths,
+        }),
+      });
+
+      if (!batchResponse.ok) {
+        const errorText = await batchResponse.text();
+        console.error(`[DocuKing] Batch Pull 실패: ${errorText}`);
+        // 배치 실패 시 모든 파일 실패 처리
+        for (const file of batch) {
+          results.push({ type: 'fail', path: file.path, error: 'Batch request failed' });
+          failed++;
+          current++;
+        }
+        continue;
       }
-      const serverHash = crypto.createHash('sha256').update(serverBuffer).digest('hex');
 
-      // 로컬 파일 존재 여부 및 해시 비교
-      let localExists = fs.existsSync(fullPath);
-      let localHash = null;
+      const batchData = await batchResponse.json();
+      const filesData = batchData.files || [];
+
+      // 파일별 처리
+      for (const file of batch) {
+        current++;
+        const fileData = filesData.find(f => f.path === file.path);
+        const serverMeta = serverPathToMeta[file.path];
+
+        if (!fileData || fileData.error) {
+          results.push({ type: 'fail', path: file.path, error: fileData?.error || 'Not found' });
+          failed++;
+          continue;
+        }
 
-      if (localExists) {
         try {
-          const localStat = fs.statSync(fullPath);
-          if (localStat.isDirectory()) {
-            // 로컬에 같은 이름의 폴더가 있으면 건너뜀
-            results.push({ type: 'skip', path: file.path, error: '같은 이름의 폴더 존재' });
-            skipped++;
-            continue;
+          const content = fileData.content || '';
+          const encoding = fileData.encoding || 'utf-8';
+
+          // 서버 파일 버퍼
+          let serverBuffer;
+          if (encoding === 'base64') {
+            serverBuffer = Buffer.from(content, 'base64');
+          } else {
+            serverBuffer = Buffer.from(content, 'utf-8');
           }
-          const localBuffer = fs.readFileSync(fullPath);
-          localHash = crypto.createHash('sha256').update(localBuffer).digest('hex');
-        } catch (e) {
-          localExists = false;
-        }
-      }
+          const serverHash = fileData.hash || crypto.createHash('sha256').update(serverBuffer).digest('hex');
 
-      // 변경 감지
-      if (localExists && localHash === serverHash) {
-        // 변경 없음 - 스킵
-        results.push({ type: 'skip', path: file.path });
-        skipped++;
-        continue;
-      }
+          // 로컬 존재 여부
+          const localExists = fs.existsSync(file.fullPath);
 
-      // 디렉토리 생성
-      fs.mkdirSync(path.dirname(fullPath), { recursive: true });
+          // 디렉토리 생성
+          fs.mkdirSync(path.dirname(file.fullPath), { recursive: true });
 
-      // 파일 저장
-      if (encoding === 'base64') {
-        fs.writeFileSync(fullPath, serverBuffer);
-      } else {
-        fs.writeFileSync(fullPath, content, 'utf-8');
-      }
+          // 파일 저장
+          if (encoding === 'base64') {
+            fs.writeFileSync(file.fullPath, serverBuffer);
+          } else {
+            fs.writeFileSync(file.fullPath, content, 'utf-8');
+          }
 
-      if (localExists) {
-        results.push({ type: 'update', path: file.path });
-        updated++;
-      } else {
-        results.push({ type: 'download', path: file.path });
-        downloaded++;
-      }
+          if (localExists) {
+            results.push({ type: 'update', path: file.path });
+            updated++;
+          } else {
+            results.push({ type: 'download', path: file.path });
+            downloaded++;
+          }
 
-      // 인덱스에 다운로드된 파일 기록
-      syncedFiles.push({ serverPath: file.path, fullPath, hash: serverHash });
+          // 인덱스에 기록
+          syncedFiles.push({ serverPath: file.path, fullPath: file.fullPath, hash: serverHash });
 
-      // source: 'web' 파일이면 mark-as-pulled 대상
-      if (serverMeta && serverMeta.source === 'web') {
-        filesToMarkAsPulled.push(file.path);
+          // source: 'web' 파일이면 mark-as-pulled 대상
+          if (serverMeta && serverMeta.source === 'web') {
+            filesToMarkAsPulled.push(file.path);
+          }
+        } catch (e) {
+          results.push({ type: 'fail', path: file.path, error: e.message });
+          failed++;
+        }
       }
     } catch (e) {
-      results.push({ type: 'fail', path: file.path, error: e.message });
-      failed++;
+      console.error(`[DocuKing] Batch ${batchNum} 네트워크 오류: ${e.message}`);
+      // 배치 실패 시 개별 요청으로 폴백
+      for (const file of batch) {
+        current++;
+        try {
+          const response = await fetch(
+            `${API_ENDPOINT}/files?projectId=${projectId}&path=${encodeURIComponent(file.path)}`,
+            { headers: { 'Authorization': `Bearer ${apiKey}` } }
+          );
+          if (!response.ok) {
+            results.push({ type: 'fail', path: file.path, error: await response.text() });
+            failed++;
+            continue;
+          }
+          const data = await response.json();
+          const content = data.file?.content || data.content || '';
+          const encoding = data.file?.encoding || data.encoding || 'utf-8';
+          let serverBuffer = encoding === 'base64' ? Buffer.from(content, 'base64') : Buffer.from(content, 'utf-8');
+          fs.mkdirSync(path.dirname(file.fullPath), { recursive: true });
+          fs.writeFileSync(file.fullPath, encoding === 'base64' ? serverBuffer : content, encoding === 'base64' ? undefined : 'utf-8');
+          results.push({ type: 'download', path: file.path });
+          downloaded++;
+        } catch (fallbackErr) {
+          results.push({ type: 'fail', path: file.path, error: fallbackErr.message });
+          failed++;
+        }
+      }
     }
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "docuking-mcp",
-  "version": "2.12.0",
+  "version": "2.13.0",
   "description": "DocuKing MCP Server - AI 시대의 문서 협업 플랫폼",
   "type": "module",
   "main": "index.js",