Merge pull request #16 from andaoai/dev

Dev

Author: andaoai

package-lock.json

@@ -2,7 +2,7 @@
   "name": "yolo-labeling-vs",
   "displayName": "YOLO Labeling",
   "description": "A VS Code extension for YOLO dataset labeling",
-  "version": "0.0.51",
+  "version": "0.0.54",
   "publisher": "andaoai",
   "license": "MIT",
   "icon": "docs/images/icon.png",
@@ -131,6 +131,7 @@
     "@types/react-dom": "^18.2.0",
     "js-yaml": "^4.1.0",
     "react": "^18.2.0",
-    "react-dom": "^18.2.0"
+    "react-dom": "^18.2.0",
+    "sharp": "^0.34.1"
   }
 }

package.json

@@ -37,7 +37,7 @@ export interface ErrorMessage extends ExtensionToWebviewMessage {
 
 // 从Webview到扩展的消息
 export interface WebviewToExtensionMessage extends WebviewMessage {
-  command: 'save' | 'next' | 'previous' | 'loadImage' | 'getImageList' | 'reload' | 'openImageInNewTab' | 'openTxtInNewTab';
+  command: 'save' | 'next' | 'previous' | 'loadImage' | 'getImageList' | 'reload' | 'openImageInNewTab' | 'openTxtInNewTab' | 'getImagePreviews';
 }
 
 // 保存标签消息

src/model/types.ts

@@ -2,6 +2,7 @@ import * as fs from 'fs';
 import * as path from 'path';
 import { CacheManager } from '../model/CacheManager';
 import { ErrorHandler, ErrorType } from '../ErrorHandler';
+import sharp from 'sharp';
 
 /**
  * 图像服务类
@@ -92,4 +93,18 @@ export class ImageService {
     public clearCache(): void {
         this._imageCache.clear();
     }
+
+    /**
+     * 生成图片缩略图，返回base64字符串（不带data:image/png;base64,前缀）
+     */
+    public async generateThumbnail(imagePath: string, width: number, height: number): Promise<string> {
+        // 检查文件是否存在
+        await fs.promises.access(imagePath, fs.constants.R_OK);
+        // 用 sharp 生成缩略图
+        const buffer = await sharp(imagePath)
+            .resize(width, height, { fit: 'cover' })
+            .toFormat('png')
+            .toBuffer();
+        return buffer.toString('base64');
+    }
 } 

src/services/ImageService.ts

@@ -74,6 +74,10 @@ export class WebviewMessageHandler {
                 await this.handleOpenTxtInNewTabCommand(openTxtMessage);
                 break;
 
+            case 'getImagePreviews':
+                await this.handleGetImagePreviewsCommand(message);
+                break;
+                
             default:
                 console.warn(`Unknown command: ${extMessage.command}`);
         }
@@ -325,4 +329,21 @@ export class WebviewMessageHandler {
             );
         }
     }
+    
+    /**
+     * 处理批量获取图片缩略图命令
+     */
+    private async handleGetImagePreviewsCommand(message: any): Promise<void> {
+        const paths: string[] = message.paths || [];
+        const previews: string[] = [];
+        for (const p of paths) {
+            try {
+                const thumb = await this._imageService.generateThumbnail(p, 120, 80); // 120x80
+                previews.push(thumb);
+            } catch {
+                previews.push(''); // 失败用空字符串
+            }
+        }
+        this._webview.postMessage({ command: 'imagePreviews', previews });
+    }
 }