[extension] 增加对自定义多线程扩展的支持，并更新 meta 文件

extension/main.js

@@ -6,6 +6,7 @@ const path = require('path');
  * 环境信息
  */
 let engineEditorPath = path.dirname(path.dirname(Editor.frameworkPath));
+let engineVersion = null;
 
 const engineMinigameAdapterPath = path.join(engineEditorPath, "builtin", "adapters");
 const engineWechatMinigameWorkerMainMacroPath = path.join(engineEditorPath, "builtin", "adapters", "platforms/wechat/worker/macro.js");
@@ -13,6 +14,7 @@ const engineWechatMinigameWorkerSubMacroPath = path.join(engineEditorPath, "buil
 
 const WECHAT_MINIGAME_WORKER_SUB_PATH = "platforms/wechat/res/workers";
 const WECHAT_MINIGAME_CONFIG_PATH = "platforms/wechat/res/game.json";
+const WECHAT_MINIGAME_WORKER_CUSTOM_PATH = "platforms/wechat/res/workers/custom";
 
 function t(str) {
     return Editor.T('enhance-kit.' + str);
@@ -27,6 +29,68 @@ function getMinigameAdapterVersion() {
     }
 }
 
+function getProjectWorkerDtsPath() {
+    return path.join(getProjectCustomWorkerPath(), "creator-worker.d.ts");
+}
+
+function getProjectCustomWorkerPath() {
+    return path.join(path.dirname(Editor.url("db://assets/")), "worker");
+}
+
+function getProjectCustomWorkerSrcPath() {
+    return path.join(getProjectCustomWorkerPath(), "src");
+}
+
+function getTemplatePath(version, name) {
+    return path.join(Editor.url("packages://enhance-kit"), "templates", String(engineVersion), version, name);
+}
+
+function copyFolder(src, dest) {
+    function createDir(path) {
+        fs.mkdirSync(path)
+    }
+
+    if (fs.existsSync(src)) {
+        createDir(dest)
+        /**
+         * @des 方式一：利用子进程操作命令行方式
+         */
+        // child_process.spawn('cp', ['-r', copiedPath, resultPath])
+
+        /**
+         * @des 方式二：
+         */
+        const files = fs.readdirSync(src, { withFileTypes: true });
+        for (let i = 0; i < files.length; i++) {
+            const cf = files[i]
+            const ccp = path.join(src, cf.name)
+            const crp = path.join(dest, cf.name)
+            if (cf.isFile()) {
+                fs.copyFileSync(ccp, crp)
+            } else if (cf.isDirectory()) {
+                try {
+                    /**
+                     * @des 判断读(R_OK | W_OK)写权限
+                     */
+                    fs.accessSync(path.join(crp, '..'), fs.constants.W_OK)
+                    copyFolder(ccp, crp, true);
+                } catch (error) {
+                    Editor.error('folder write error:', error);
+                }
+
+            } else if (cf.isSymbolicLink()) {
+                fs.symlinkSync(fs.readlinkSync(ccp), crp)
+            } else if (cf.isBlockDevice() || cf.isCharacterDevice()) {
+                fs.copyFileSync(ccp, crp)
+            } else {
+                Editor.error('not file or directory: ', ccp);
+            }
+        }
+    } else {
+        Editor.error('do not exist path: ', src);
+    }
+}
+
 function getSettings() {
     const minigameVersion = getMinigameAdapterVersion();
     const isUninstalled = minigameVersion === "";
@@ -45,6 +109,7 @@ function getSettings() {
             CC_WORKER_AUDIO_SYSTEM: getMacroBooleanValue(content, "CC_WORKER_AUDIO_SYSTEM"),
             CC_WORKER_SCHEDULER: getMacroBooleanValue(content, "CC_WORKER_SCHEDULER"),
             CC_WORKER_AUDIO_SYSTEM_SYNC_INTERVAL: getMacroIntegerValue(content, "CC_WORKER_AUDIO_SYSTEM_SYNC_INTERVAL"),
+            CC_CUSTOM_WORKER: getMacroBooleanValue(content, "CC_CUSTOM_WORKER"),
         };
     }
 }
@@ -71,6 +136,7 @@ function setSettings(macro, value) {
     }
 
     checkAndModifyWorkerFiles();
+    refreshCustomThreadCode();
     syncSettingsToSubWorker();
 }
 
@@ -108,7 +174,7 @@ function checkAndModifyWorkerFiles() {
         const gameJson = JSON.parse(fs.readFileSync(gameJsonPath, { encoding: "utf-8" }));
 
         // 是否启用 Worker
-        if (result.CC_WORKER_ASSET_PIPELINE || result.CC_WORKER_AUDIO_SYSTEM) {
+        if (result.CC_WORKER_ASSET_PIPELINE || result.CC_WORKER_AUDIO_SYSTEM || result.CC_CUSTOM_WORKER) {
             // 没有 Worker 目录与配置的话提醒用户重新安装
             if (!(gameJson.workers && fs.existsSync(workerDir))) {
                 Editor.error(t('thread_not_right_workers_dir'));
@@ -119,6 +185,45 @@ function checkAndModifyWorkerFiles() {
     }
 }
 
+function refreshCustomThreadCode() {
+    const result = getSettings();
+    if (result.code === 0) {
+        const customDir = path.join(engineMinigameAdapterPath, WECHAT_MINIGAME_WORKER_CUSTOM_PATH);
+        const src = getProjectCustomWorkerSrcPath();
+        // 是否启用 Worker
+        if (result.CC_CUSTOM_WORKER) {
+            if (fs.existsSync(src)) {
+                fs.rmSync(customDir, { force: true, recursive: true });
+                copyFolder(src, customDir);
+                fs.writeFileSync(path.join(customDir, "ipc-worker.js"), "module.exports = require('../ipc-worker.js');");
+                Editor.success(t('refresh_thread_custom_success'));
+            } else {
+                fs.rmSync(customDir, { force: true, recursive: true });
+                Editor.error(t('thread_custom_not_exists_1') + Editor.T('i18n:MAIN_MENU.package.title') + "-" + t('COCOS_CREATE_EXTENSION') + "-" + t('thread_create_custom_thread_menu') + t('thread_custom_not_exists_2'));
+            }
+        } else {
+            fs.rmSync(customDir, { force: true, recursive: true });
+        }
+    }
+}
+
+function createThreadTemplate() {
+    const version = getMinigameAdapterVersion();
+    const src = getTemplatePath(version, "worker");
+    const dest = getProjectCustomWorkerPath();
+    if (fs.existsSync(dest)) {
+        // 写入 creator-worker.d.ts
+        const workerDtsPath = path.join(getTemplatePath(version, "worker"), "creator-worker.d.ts");
+        const projectWorkerDtsPath = getProjectWorkerDtsPath();
+        if (exist(workerDtsPath) && exist(projectWorkerDtsPath)) {
+            fs.writeFileSync(projectWorkerDtsPath, fs.readFileSync(workerDtsPath));
+        }
+        Editor.success(t('create_thread_custom_already_exists'));
+    } else {
+        copyFolder(src, dest);
+        Editor.success(t('create_thread_custom_success'));
+    }
+}
 
 module.exports = {
 
@@ -142,10 +247,24 @@ module.exports = {
             event.reply(null);
         },
 
-        "scene:ready"(event) {
-            checkAndModifyWorkerFiles();
+        createThreadTemplate() {
+            createThreadTemplate();
+            refreshCustomThreadCode();
         },
 
+        refreshCustomThreadCode() {
+            refreshCustomThreadCode();
+        },
+
+        "scene:ready"(event) {
+            Editor.Scene.callSceneScript('enhance-kit', 'scene-get-engine-version', function (err, version) {
+                if (version) {
+                    engineVersion = version;
+                    checkAndModifyWorkerFiles();
+                    refreshCustomThreadCode();
+                }
+            });
+        },
     },
 
 };