|
|
"use strict";
|
|
|
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
|
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
|
};
|
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
exports.initUniXCompilerRootWorkers = exports.uniJavaScriptWorkersPlugin = exports.resolveWorkersDir = exports.uniWorkersPlugin = exports.initWorkers = exports.getWorkersRootDirs = exports.resolveWorkersRootDir = exports.getWorkers = void 0;
|
|
|
const path_1 = __importDefault(require("path"));
|
|
|
const fs_extra_1 = __importDefault(require("fs-extra"));
|
|
|
const debug_1 = __importDefault(require("debug"));
|
|
|
const fast_glob_1 = require("fast-glob");
|
|
|
const utils_1 = require("./utils");
|
|
|
const json_1 = require("./json");
|
|
|
const uts_1 = require("./uts");
|
|
|
const uni_modules_1 = require("./vite/plugins/uts/uni_modules");
|
|
|
const resolve_1 = require("./resolve");
|
|
|
const debugWorkers = (0, debug_1.default)('uni:workers');
|
|
|
let workersRootDir = null;
|
|
|
let workersRootDirs = [];
|
|
|
let workers = {};
|
|
|
function getWorkers() {
|
|
|
return workers;
|
|
|
}
|
|
|
exports.getWorkers = getWorkers;
|
|
|
function resolveWorkersRootDir() {
|
|
|
// 默认是 workers
|
|
|
return workersRootDir || 'workers';
|
|
|
}
|
|
|
exports.resolveWorkersRootDir = resolveWorkersRootDir;
|
|
|
function getWorkersRootDirs() {
|
|
|
return workersRootDirs;
|
|
|
}
|
|
|
exports.getWorkersRootDirs = getWorkersRootDirs;
|
|
|
/**
|
|
|
* 遍历目录下的所有uts文件，读取文件内容，正则匹配出定义的worker，返回文件名和类名的映射关系
|
|
|
* export class MyWorkerTask extends WorkerTaskImpl {}
|
|
|
* @param dir
|
|
|
*/
|
|
|
function initWorkers(workersDirs, rootDir) {
|
|
|
workers = {};
|
|
|
for (const workersDir of workersDirs) {
|
|
|
const dir = path_1.default.join(rootDir, workersDir);
|
|
|
if (!fs_extra_1.default.existsSync(dir)) {
|
|
|
continue;
|
|
|
}
|
|
|
(0, fast_glob_1.sync)('**/*.uts', { cwd: dir }).forEach((file) => {
|
|
|
const content = fs_extra_1.default.readFileSync(path_1.default.join(dir, file), 'utf-8');
|
|
|
const match = content.match(/class\s+(.*)\s+extends\s+WorkerTaskImpl/);
|
|
|
if (match && match[1]) {
|
|
|
const key = (0, utils_1.normalizePath)(path_1.default.join(workersDir, file));
|
|
|
workers[key] = match[1];
|
|
|
}
|
|
|
});
|
|
|
}
|
|
|
debugWorkers('workers', workers);
|
|
|
return workers;
|
|
|
}
|
|
|
exports.initWorkers = initWorkers;
|
|
|
function uniWorkersPlugin() {
|
|
|
const inputDir = process.env.UNI_INPUT_DIR;
|
|
|
const platform = process.env.UNI_UTS_PLATFORM;
|
|
|
const resolveWorkers = () => getWorkers();
|
|
|
function refreshWorkers() {
|
|
|
workersRootDirs = resolveWorkersDir(inputDir);
|
|
|
initWorkers(workersRootDirs, inputDir);
|
|
|
return Object.keys(getWorkers()).length > 0;
|
|
|
}
|
|
|
refreshWorkers();
|
|
|
const preprocessor = platform === 'app-android'
|
|
|
? (0, uni_modules_1.createAppAndroidUniModulesSyncFilePreprocessorOnce)(false)
|
|
|
: platform === 'app-ios'
|
|
|
? (0, uni_modules_1.createAppIosUniModulesSyncFilePreprocessorOnce)(false)
|
|
|
: platform === 'app-harmony'
|
|
|
? (0, uni_modules_1.createAppHarmonyUniModulesSyncFilePreprocessorOnce)(false)
|
|
|
: null;
|
|
|
const cache = {};
|
|
|
const uniXKotlinCompiler = platform === 'app-android'
|
|
|
? (0, uts_1.resolveUTSCompiler)().createUniXKotlinCompilerOnce({
|
|
|
resolveWorkers,
|
|
|
})
|
|
|
: null;
|
|
|
const uniXSwiftCompiler = platform === 'app-ios'
|
|
|
? (0, uts_1.resolveUTSCompiler)().createUniXSwiftCompilerOnce({
|
|
|
resolveWorkers,
|
|
|
})
|
|
|
: null;
|
|
|
const uniXArkTSCompiler = platform === 'app-harmony'
|
|
|
? (0, uts_1.resolveUTSCompiler)().createUniXArkTSCompilerOnce({
|
|
|
resolveWorkers,
|
|
|
})
|
|
|
: null;
|
|
|
return {
|
|
|
name: 'uni-workers',
|
|
|
enforce: 'pre',
|
|
|
async buildStart() {
|
|
|
if (refreshWorkers()) {
|
|
|
if (preprocessor) {
|
|
|
await syncWorkersFiles(platform, inputDir, preprocessor, cache);
|
|
|
}
|
|
|
}
|
|
|
// 需要等待 workers 文件同步完之后，添加到 rootFiles 中，触发 tsc 的编译
|
|
|
if (uniXKotlinCompiler) {
|
|
|
await initUniXCompilerRootWorkers((0, uts_1.tscOutDir)('app-android'), uniXKotlinCompiler);
|
|
|
}
|
|
|
if (uniXSwiftCompiler) {
|
|
|
await initUniXCompilerRootWorkers((0, uts_1.tscOutDir)('app-ios'), uniXSwiftCompiler);
|
|
|
}
|
|
|
if (uniXArkTSCompiler) {
|
|
|
await initUniXCompilerRootWorkers((0, uts_1.tscOutDir)('app-harmony'), uniXArkTSCompiler);
|
|
|
}
|
|
|
},
|
|
|
};
|
|
|
}
|
|
|
exports.uniWorkersPlugin = uniWorkersPlugin;
|
|
|
async function syncWorkersFiles(platform, inputDir, preprocessor, cache) {
|
|
|
if (platform !== 'app-harmony' &&
|
|
|
platform !== 'app-android' &&
|
|
|
platform !== 'app-ios') {
|
|
|
return;
|
|
|
}
|
|
|
const workersDirs = resolveWorkersDir(inputDir);
|
|
|
if (workersDirs.length) {
|
|
|
const { syncUTSFiles } = (0, uts_1.resolveUTSCompiler)();
|
|
|
for (const workersDir of workersDirs) {
|
|
|
await syncUTSFiles((0, utils_1.normalizePath)(path_1.default.join(workersDir, '**/*.uts')), inputDir, (0, uts_1.tscOutDir)(platform), true, preprocessor, cache);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
function resolveWorkersDir(inputDir) {
|
|
|
const workersDirs = [];
|
|
|
const manifestJson = (0, json_1.parseManifestJsonOnce)(inputDir);
|
|
|
if (manifestJson.workers) {
|
|
|
let workersDir = typeof manifestJson.workers === 'string'
|
|
|
? manifestJson.workers
|
|
|
: manifestJson.workers.path;
|
|
|
if (workersDir) {
|
|
|
workersDir = (0, utils_1.normalizePath)(workersDir);
|
|
|
const dir = path_1.default.join(inputDir, workersDir);
|
|
|
if (fs_extra_1.default.existsSync(dir)) {
|
|
|
workersRootDir = workersDir;
|
|
|
workersDirs.push(workersDir);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
// 遍历uni_modules插件目录是否有workers目录
|
|
|
const uniModulesDir = path_1.default.join(inputDir, 'uni_modules');
|
|
|
if (fs_extra_1.default.existsSync(uniModulesDir)) {
|
|
|
fs_extra_1.default.readdirSync(uniModulesDir).forEach((dir) => {
|
|
|
if (fs_extra_1.default.existsSync(path_1.default.join(uniModulesDir, dir, 'workers'))) {
|
|
|
workersDirs.push('uni_modules/' + dir + '/workers');
|
|
|
}
|
|
|
});
|
|
|
}
|
|
|
debugWorkers('workersDirs', workersDirs);
|
|
|
return workersDirs;
|
|
|
}
|
|
|
exports.resolveWorkersDir = resolveWorkersDir;
|
|
|
function uniJavaScriptWorkersPlugin() {
|
|
|
// 仅小程序平台外置uni-worker.mp.js
|
|
|
const external = (process.env.UNI_UTS_PLATFORM || '').startsWith('mp-');
|
|
|
let workerPolyfillCode = '';
|
|
|
let isWrite = false;
|
|
|
const UniAppWorkerJSName = external ? 'uni-worker.mp.js' : 'uni-worker.web.js';
|
|
|
let viteServer = null;
|
|
|
const workersRootPaths = [];
|
|
|
const workerPolyfillPath = `@dcloudio/uni-app/dist-x/${UniAppWorkerJSName}`;
|
|
|
const workerPolyfillAbsPath = (0, utils_1.normalizePath)((0, resolve_1.resolveBuiltIn)(workerPolyfillPath));
|
|
|
function isWorkerFile(id) {
|
|
|
if (workersRootPaths.length) {
|
|
|
return workersRootPaths.some((dir) => id.startsWith(dir));
|
|
|
}
|
|
|
return false;
|
|
|
}
|
|
|
function parseWorkerEntryFile(workerJsPath) {
|
|
|
const workerPath = workerJsPath.slice(1).replace('.js', '.uts');
|
|
|
if (workerPath in workers) {
|
|
|
return (0, utils_1.normalizePath)(path_1.default.resolve(process.env.UNI_INPUT_DIR, workerPath));
|
|
|
}
|
|
|
}
|
|
|
function parseWorkerClass(id) {
|
|
|
const filename = id.split('?')[0];
|
|
|
if (isWorkerFile(filename)) {
|
|
|
const workerPath = (0, utils_1.normalizePath)(path_1.default.relative(process.env.UNI_INPUT_DIR, filename));
|
|
|
return workers[workerPath] || '';
|
|
|
}
|
|
|
return false;
|
|
|
}
|
|
|
return {
|
|
|
name: 'uni:javascript-workers',
|
|
|
configureServer(server) {
|
|
|
viteServer = server;
|
|
|
},
|
|
|
buildStart() {
|
|
|
if (!workerPolyfillCode && Object.keys(getWorkers()).length) {
|
|
|
workerPolyfillCode = fs_extra_1.default.readFileSync(workerPolyfillAbsPath, 'utf-8');
|
|
|
}
|
|
|
workersRootPaths.length = 0;
|
|
|
for (const workersRootDir of getWorkersRootDirs()) {
|
|
|
workersRootPaths.push((0, utils_1.normalizePath)(path_1.default.resolve(process.env.UNI_INPUT_DIR, workersRootDir)));
|
|
|
}
|
|
|
},
|
|
|
resolveId(id) {
|
|
|
// uni.createWorker('workers/request/index.uts')
|
|
|
// 编译阶段调整为 uni.createWorker('workers/request/index.js')，确保开发和运行时都是用.js后缀加载
|
|
|
// 不调整成js后缀或.uts?import这些格式， vite 是不会走transform逻辑的，而是直接读取文件内容
|
|
|
if (viteServer) {
|
|
|
const workerEntryFile = parseWorkerEntryFile(id);
|
|
|
if (workerEntryFile) {
|
|
|
return workerEntryFile;
|
|
|
}
|
|
|
if (id === workerPolyfillPath) {
|
|
|
return workerPolyfillAbsPath;
|
|
|
}
|
|
|
}
|
|
|
},
|
|
|
load(id) {
|
|
|
const filename = id.split('?')[0];
|
|
|
const workerClass = parseWorkerClass(filename);
|
|
|
if (workerClass === false) {
|
|
|
return;
|
|
|
}
|
|
|
if (fs_extra_1.default.existsSync(filename)) {
|
|
|
let code = (viteServer ? `import '${workerPolyfillPath}';` : '') +
|
|
|
fs_extra_1.default
|
|
|
.readFileSync(filename, 'utf-8')
|
|
|
// 移除 export
|
|
|
.replace(/export\s+class\s+(.*)\s+extends\s+WorkerTaskImpl\s*{/, 'class $1 extends WorkerTaskImpl {');
|
|
|
// 如果是入口文件，需要追加初始化代码
|
|
|
if (workerClass) {
|
|
|
code += `\n;new ${workerClass}().entry()`;
|
|
|
}
|
|
|
return code;
|
|
|
}
|
|
|
},
|
|
|
generateBundle(_, bundle) {
|
|
|
const workers = getWorkers();
|
|
|
const workerRootDir = resolveWorkersRootDir();
|
|
|
const workerPaths = Object.keys(workers).map((key) => {
|
|
|
if (key.startsWith('uni_modules')) {
|
|
|
key = workerRootDir + '/' + key;
|
|
|
}
|
|
|
return key.replace('.uts', '.js');
|
|
|
});
|
|
|
if (workerPaths.length) {
|
|
|
Object.keys(bundle).forEach((file) => {
|
|
|
if (workerPaths.includes(file)) {
|
|
|
const chunk = bundle[file];
|
|
|
if (chunk.type === 'chunk') {
|
|
|
const workerCode = external
|
|
|
? `require('${(0, utils_1.normalizePath)(path_1.default.relative(path_1.default.dirname(file), path_1.default.join(resolveWorkersRootDir(), 'uni-worker.js')))}')`
|
|
|
: workerPolyfillCode;
|
|
|
chunk.code = `${workerCode}\n${chunk.code}`;
|
|
|
}
|
|
|
}
|
|
|
});
|
|
|
}
|
|
|
},
|
|
|
writeBundle() {
|
|
|
if (external && Object.keys(getWorkers()).length && !isWrite) {
|
|
|
isWrite = true;
|
|
|
// 写入uni-worker.js
|
|
|
fs_extra_1.default.outputFileSync(path_1.default.resolve(process.env.UNI_OUTPUT_DIR, resolveWorkersRootDir(), 'uni-worker.js'), workerPolyfillCode);
|
|
|
}
|
|
|
},
|
|
|
};
|
|
|
}
|
|
|
exports.uniJavaScriptWorkersPlugin = uniJavaScriptWorkersPlugin;
|
|
|
async function initUniXCompilerRootWorkers(rootDir, compiler) {
|
|
|
const workers = getWorkers();
|
|
|
if (Object.keys(workers).length) {
|
|
|
for (const key in workers) {
|
|
|
const file = path_1.default.join(rootDir, key + '.ts');
|
|
|
if (fs_extra_1.default.existsSync(file)) {
|
|
|
if (!compiler.hasRootFile(file)) {
|
|
|
await compiler.addRootFile(file);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
exports.initUniXCompilerRootWorkers = initUniXCompilerRootWorkers;
|
