fix: 优化 OCR 加载逻辑

ocsjs · Apr 27, 2022 · 9dd69dc · 9dd69dc
1 parent db1219b
commit 9dd69dc
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/packages/core/src/script/cx/index.ts b/packages/core/src/script/cx/index.ts
@@ -179,18 +179,24 @@ export const CXScript = defineScript({
     },
     {
       name: '文字识别脚本',
-      url: '**/work/doHomeWorkNew**',
+      url: ['**/mycourse/studentstudy**', '**/work/doHomeWorkNew**'],
       async onload() {
         /** 文字识别 */
-        const fonts = CXAnalyses.getSecretFont();
-        if (fonts.length) {
-          const ocr = new OCR({
-            langPath: 'https://cdn.ocs.enncy.cn/resources/tessdata'
+        const ocr = new OCR({
+          langPath: 'https://cdn.ocs.enncy.cn/resources/tessdata'
+        });
 
-          });
+        // 顶层初始化
+        if (window === top) {
           logger('debug', '加载文字识别功能, 如果是初始化请耐心等待...');
           await ocr.load();
           logger('info', '文字识别功能加载成功');
+        }
+
+        const fonts = CXAnalyses.getSecretFont();
+        if (fonts.length) {
+          logger('info', '文字识别功能启动');
+          await ocr.load();
           for (const font of fonts) {
             const text = await ocr.recognize(OCR.suit(font));
             font.innerHTML = text;