From e206b502b5577485eff0e4b9f3e2f484f767192a Mon Sep 17 00:00:00 2001
From: Kirigaya <1193466151@qq.com>
Date: Fri, 2 May 2025 20:00:06 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=97=A0=E6=B3=95=E8=BF=9B?=
 =?UTF-8?q?=E8=A1=8C=E7=A6=BB=E7=BA=BF=20OCR=20=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore                                    |  3 +-
 .vscodeignore                                 |  1 +
 README.md                                     |  2 +-
 package-lock.json                             |  4 +--
 package.json                                  |  6 ++++
 .../main-panel/chat/message/streaming-box.vue |  5 +++-
 .../main-panel/chat/options/system-prompt.vue |  7 +++--
 renderer/src/i18n/ar.json                     |  4 ++-
 renderer/src/i18n/de.json                     |  4 ++-
 renderer/src/i18n/en.json                     |  4 ++-
 renderer/src/i18n/fr.json                     |  4 ++-
 renderer/src/i18n/ja.json                     |  4 ++-
 renderer/src/i18n/ko.json                     |  4 ++-
 renderer/src/i18n/ru.json                     |  4 ++-
 renderer/src/i18n/zh-cn.json                  |  4 ++-
 renderer/src/i18n/zh-tw.json                  |  4 ++-
 service/src/hook/setting.ts                   |  5 ++++
 service/src/index.ts                          |  2 +-
 service/src/mcp/ocr.service.ts                | 12 +++++++-
 src/common/entry.ts                           |  4 ++-
 src/extension.ts                              |  2 ++
 src/hook/hook.controller.ts                   | 28 +++++++++++++++++++
 22 files changed, 97 insertions(+), 20 deletions(-)
 create mode 100644 src/hook/hook.controller.ts

diff --git a/.gitignore b/.gitignore
index fb3d5fc..6e8f07e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -8,4 +8,5 @@ resources
 .DS_Store
 .exe
 
-.idea
\ No newline at end of file
+.idea
+*.traineddata
\ No newline at end of file
diff --git a/.vscodeignore b/.vscodeignore
index b40231d..64fc789 100644
--- a/.vscodeignore
+++ b/.vscodeignore
@@ -23,3 +23,4 @@ software/**
 
 .editorconfig
 .gitattributes
+*.vsix
\ No newline at end of file
diff --git a/README.md b/README.md
index 590d433..d70b6f8 100644
--- a/README.md
+++ b/README.md
@@ -12,7 +12,7 @@
 
 ## OpenMCP
 
-一款用于 MCP 服务端调试的一体化 vscode/trae 插件。
+一款用于 MCP 服务端调试的一体化 vscode/trae/cursor 插件。
 
 集成 Inspector + MCP 客户端基础功能，开发测试一体化。
 
diff --git a/package-lock.json b/package-lock.json
index 0858cd0..f25e4af 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
     "name": "openmcp",
-    "version": "0.0.5",
+    "version": "0.0.6",
     "lockfileVersion": 3,
     "requires": true,
     "packages": {
         "": {
             "name": "openmcp",
-            "version": "0.0.5",
+            "version": "0.0.6",
             "dependencies": {
                 "@modelcontextprotocol/sdk": "^1.10.2",
                 "@seald-io/nedb": "^4.1.1",
diff --git a/package.json b/package.json
index aa44c58..7b710f6 100644
--- a/package.json
+++ b/package.json
@@ -97,6 +97,12 @@
                 "title": "打开配置",
                 "category": "openmcp",
                 "icon": "$(gear)"
+            },
+            {
+                "command": "openmcp.hook.test-ocr",
+                "title": "测试 OCR",
+                "category": "openmcp",
+                "icon": "$(test)"
             }
         ],
         "menus": {
diff --git a/renderer/src/components/main-panel/chat/message/streaming-box.vue b/renderer/src/components/main-panel/chat/message/streaming-box.vue
index c519dcf..d48244c 100644
--- a/renderer/src/components/main-panel/chat/message/streaming-box.vue
+++ b/renderer/src/components/main-panel/chat/message/streaming-box.vue
@@ -6,7 +6,7 @@
         <div class="message-role">
             Agent
             <span class="message-reminder">
-                正在生成答案
+                {{ t('generate-answer') }}
                 <span class="tool-loading iconfont icon-double-loading">
                 </span>
             </span>
@@ -19,8 +19,11 @@
 
 <script setup lang="ts">
 import { defineProps } from 'vue';
+import { useI18n } from 'vue-i18n';
 import { markdownToHtml } from '../markdown/markdown';
 
+const { t } = useI18n();
+
 const props = defineProps({
     streamingContent: {
         type: String,
diff --git a/renderer/src/components/main-panel/chat/options/system-prompt.vue b/renderer/src/components/main-panel/chat/options/system-prompt.vue
index e128a25..b06869e 100644
--- a/renderer/src/components/main-panel/chat/options/system-prompt.vue
+++ b/renderer/src/components/main-panel/chat/options/system-prompt.vue
@@ -9,7 +9,8 @@
     <el-dialog v-model="showSystemPromptDialog" :title="t('system-prompt')" width="600px">
 
         <div v-if="!showAdd">
-            <el-select v-model="tabStorage.settings.systemPrompt" placeholder="选择预设"
+            <el-select v-model="tabStorage.settings.systemPrompt"
+                :placeholder="t('choose-presetting')"
                 style="width: 100%; margin-bottom: 20px;">
 
                 <el-option v-for="prompt in systemPrompts"
@@ -59,10 +60,10 @@
 </template>
 
 <script setup lang="ts">
-import { ref, computed, inject, watch, onMounted } from 'vue';
+import { ref, computed, inject, onMounted } from 'vue';
 import { useI18n } from 'vue-i18n';
 import { ChatStorage } from '../chat';
-import { systemPrompts, saveSystemPrompts, setSystemPrompt, loadSystemPrompts, deleteSystemPrompt } from './system-prompt';
+import { systemPrompts, setSystemPrompt, loadSystemPrompts, deleteSystemPrompt } from './system-prompt';
 import { ElMessage } from 'element-plus';
 import { debounce } from 'lodash';
 
diff --git a/renderer/src/i18n/ar.json b/renderer/src/i18n/ar.json
index 12a7445..a991985 100644
--- a/renderer/src/i18n/ar.json
+++ b/renderer/src/i18n/ar.json
@@ -147,5 +147,7 @@
     "connect-sigature": "توقيع الاتصال",
     "finish-refresh": "تم التحديث",
     "add-system-prompt.name-placeholder": "عنوان prompt المخصص",
-    "enter-message-dot": "أدخل الرسالة..."
+    "enter-message-dot": "أدخل الرسالة...",
+    "generate-answer": "جارٍ إنشاء الإجابة",
+    "choose-presetting": "اختر الإعداد المسبق"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/de.json b/renderer/src/i18n/de.json
index 15a1e3c..200934b 100644
--- a/renderer/src/i18n/de.json
+++ b/renderer/src/i18n/de.json
@@ -147,5 +147,7 @@
     "connect-sigature": "Verbindungssignatur",
     "finish-refresh": "Aktualisierung abgeschlossen",
     "add-system-prompt.name-placeholder": "Titel für benutzerdefinierte Eingabeaufforderung",
-    "enter-message-dot": "Nachricht eingeben..."
+    "enter-message-dot": "Nachricht eingeben...",
+    "generate-answer": "Antwort wird generiert",
+    "choose-presetting": "Voreinstellung auswählen"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/en.json b/renderer/src/i18n/en.json
index 5252960..106f6a9 100644
--- a/renderer/src/i18n/en.json
+++ b/renderer/src/i18n/en.json
@@ -147,5 +147,7 @@
     "connect-sigature": "Connection signature",
     "finish-refresh": "Refresh completed",
     "add-system-prompt.name-placeholder": "Title for custom prompt",
-    "enter-message-dot": "Enter message..."
+    "enter-message-dot": "Enter message...",
+    "generate-answer": "Generating answer",
+    "choose-presetting": "Select preset"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/fr.json b/renderer/src/i18n/fr.json
index 25e4a04..cb7b28d 100644
--- a/renderer/src/i18n/fr.json
+++ b/renderer/src/i18n/fr.json
@@ -147,5 +147,7 @@
     "connect-sigature": "Signature de connexion",
     "finish-refresh": "Actualisation terminée",
     "add-system-prompt.name-placeholder": "Titre de l'invite personnalisée",
-    "enter-message-dot": "Entrez un message..."
+    "enter-message-dot": "Entrez un message...",
+    "generate-answer": "Génération de la réponse",
+    "choose-presetting": "Sélectionner un préréglage"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/ja.json b/renderer/src/i18n/ja.json
index 02db8fc..a3a2f27 100644
--- a/renderer/src/i18n/ja.json
+++ b/renderer/src/i18n/ja.json
@@ -147,5 +147,7 @@
     "connect-sigature": "接続署名",
     "finish-refresh": "更新が完了しました",
     "add-system-prompt.name-placeholder": "カスタムプロンプトのタイトル",
-    "enter-message-dot": "メッセージを入力..."
+    "enter-message-dot": "メッセージを入力...",
+    "generate-answer": "回答を生成中",
+    "choose-presetting": "プリセットを選択"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/ko.json b/renderer/src/i18n/ko.json
index 601e09b..3b25bce 100644
--- a/renderer/src/i18n/ko.json
+++ b/renderer/src/i18n/ko.json
@@ -147,5 +147,7 @@
     "connect-sigature": "연결 서명",
     "finish-refresh": "새로 고침 완료",
     "add-system-prompt.name-placeholder": "사용자 지정 프롬프트 제목",
-    "enter-message-dot": "메시지를 입력하세요..."
+    "enter-message-dot": "메시지를 입력하세요...",
+    "generate-answer": "답변 생성 중",
+    "choose-presetting": "프리셋 선택"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/ru.json b/renderer/src/i18n/ru.json
index 1234304..6d5a8b1 100644
--- a/renderer/src/i18n/ru.json
+++ b/renderer/src/i18n/ru.json
@@ -147,5 +147,7 @@
     "connect-sigature": "Подпись соединения",
     "finish-refresh": "Обновление завершено",
     "add-system-prompt.name-placeholder": "Заголовок пользовательского prompt",
-    "enter-message-dot": "Введите сообщение..."
+    "enter-message-dot": "Введите сообщение...",
+    "generate-answer": "Генерация ответа",
+    "choose-presetting": "Выбрать预设"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/zh-cn.json b/renderer/src/i18n/zh-cn.json
index 980d56c..10cba07 100644
--- a/renderer/src/i18n/zh-cn.json
+++ b/renderer/src/i18n/zh-cn.json
@@ -147,5 +147,7 @@
     "connect-sigature": "连接签名",
     "finish-refresh": "完成刷新",
     "add-system-prompt.name-placeholder": "输入自定义 prompt 的标题",
-    "enter-message-dot": "输入消息..."
+    "enter-message-dot": "输入消息...",
+    "generate-answer": "正在生成答案",
+    "choose-presetting": "选择预设"
 }
\ No newline at end of file
diff --git a/renderer/src/i18n/zh-tw.json b/renderer/src/i18n/zh-tw.json
index ad43897..74bd24e 100644
--- a/renderer/src/i18n/zh-tw.json
+++ b/renderer/src/i18n/zh-tw.json
@@ -147,5 +147,7 @@
     "connect-sigature": "連接簽名",
     "finish-refresh": "刷新完成",
     "add-system-prompt.name-placeholder": "自定義提示的標題",
-    "enter-message-dot": "輸入訊息..."
+    "enter-message-dot": "輸入訊息...",
+    "generate-answer": "正在生成答案",
+    "choose-presetting": "選擇預設"
 }
\ No newline at end of file
diff --git a/service/src/hook/setting.ts b/service/src/hook/setting.ts
index a991252..32d190e 100644
--- a/service/src/hook/setting.ts
+++ b/service/src/hook/setting.ts
@@ -1,5 +1,10 @@
 export let VSCODE_WORKSPACE = '';
+export let RUNNING_CWD = '';
 
 export function setVscodeWorkspace(workspace: string) {
     VSCODE_WORKSPACE = workspace;
+}
+
+export function setRunningCWD(path: string) {
+    RUNNING_CWD = path;
 }
\ No newline at end of file
diff --git a/service/src/index.ts b/service/src/index.ts
index 5cc80bd..53374a1 100644
--- a/service/src/index.ts
+++ b/service/src/index.ts
@@ -1,5 +1,5 @@
 export { routeMessage } from './common/router';
 export { VSCodeWebViewLike } from './hook/adapter';
-export { setVscodeWorkspace } from './hook/setting';
+export { setVscodeWorkspace, setRunningCWD } from './hook/setting';
 // TODO: 更加规范
 export { client } from './mcp/connect.service';
\ No newline at end of file
diff --git a/service/src/mcp/ocr.service.ts b/service/src/mcp/ocr.service.ts
index 818204f..74058f5 100644
--- a/service/src/mcp/ocr.service.ts
+++ b/service/src/mcp/ocr.service.ts
@@ -6,6 +6,7 @@ import { diskStorage, ocrDB } from '../hook/db';
 import * as fs from 'fs';
 import * as os from 'os';
 import * as path from 'path';
+import { RUNNING_CWD } from '../hook/setting';
 
 export const ocrWorkerStorage = new Map<string, OcrWorker>();
 
@@ -46,12 +47,17 @@ export async function tesseractOCR(
     logger: (message: Tesseract.LoggerMessage) => void,
     lang: string = 'eng+chi_sim'
 ) {
+    
     try {
         const { data: { text } } = await Tesseract.recognize(
             imagePath,
             lang,
             {
-                logger
+                logger,
+                langPath: './',
+                gzip: false,
+                cacheMethod: 'cache',
+                cachePath: RUNNING_CWD
             }
         );
 
@@ -67,6 +73,10 @@ export function createOcrWorker(filename: string, webview: PostMessageble): OcrW
     const workerId = uuidv4();
 
     const logger = (message: Tesseract.LoggerMessage) => {
+        console.log('report ocr status');
+        console.log(message);
+        
+        
         webview.postMessage({
             command: 'ocr/worker/log',
             data: {
diff --git a/src/common/entry.ts b/src/common/entry.ts
index 4758fbb..5b7570a 100644
--- a/src/common/entry.ts
+++ b/src/common/entry.ts
@@ -4,12 +4,14 @@ import { HelpProvider } from '../sidebar/help.controller';
 import { McpWorkspaceConnectProvider } from '../sidebar/workspace.controller';
 import { McpInstalledConnectProvider } from '../sidebar/installed.controller';
 import { WebviewController } from '../webview/webview.controller';
+import { HookController } from '../hook/hook.controller';
 
 export const InstallModules = [
     McpWorkspaceConnectProvider,
     McpInstalledConnectProvider,
     HelpProvider,
-    WebviewController
+    WebviewController,
+    HookController
 ];
 
 const registerSingles = new Map<string, any>();
diff --git a/src/extension.ts b/src/extension.ts
index 7c20070..e48d0ce 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -9,7 +9,9 @@ export function activate(context: vscode.ExtensionContext) {
     // 获取当前打开的项目的路径
     const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
     const workspace = workspaceFolder?.uri.fsPath || '';
+
     OpenMCPService.setVscodeWorkspace(workspace);
+    OpenMCPService.setRunningCWD(context.extensionPath);
 
     launch(context);
 }
diff --git a/src/hook/hook.controller.ts b/src/hook/hook.controller.ts
new file mode 100644
index 0000000..0d95d13
--- /dev/null
+++ b/src/hook/hook.controller.ts
@@ -0,0 +1,28 @@
+import { RegisterCommand } from "../common";
+import * as vscode from 'vscode';
+import * as path from 'path';
+import Tesseract from 'tesseract.js';
+
+
+export class HookController {
+
+    @RegisterCommand('openmcp.hook.test-ocr')
+    async testOcr(context: vscode.ExtensionContext) {
+        const testImage = path.join(context.extensionPath, 'icons/openmcp.resource.png');
+        
+        const { data: { text } } = await Tesseract.recognize(
+            testImage,
+            'eng+chi_sim',
+            {
+                logger: (m) => console.log(m),
+                langPath: './',
+                gzip: false,
+                cacheMethod: 'cache',
+                cachePath: context.extensionPath
+            }
+        );
+
+        vscode.window.showInformationMessage('ocr result: ' + text);
+    }
+
+}
\ No newline at end of file