utils: module loader register

This commit is contained in:
undefined committed 2022-08-07 05:05:45 +08:00
1 parent 20bbda794e
commit 7dafab8344
50 files changed
+83 -275

No files matched your search

+1 -81
View File
@@ -1,87 +1,7 @@
#!/usr/bin/env node
require('@hydrooj/utils/lib/register');
/* eslint-disable consistent-return */
const map = {};
require('source-map-support').install({
handleUncaughtExceptions: false,
environment: 'node',
retrieveSourceMap(file) {
if (map[file]) {
return {
url: file,
map: map[file],
};
}
return null;
},
});
const path = require('path');
const vm = require('vm');
const fs = require('fs-extra');
const esbuild = require('esbuild');
const { default: hook } = require('require-resolve-hook');
const { bypass } = hook(/^(hydrooj|@hydrooj\/utils)/, (id) => bypass(() => require.resolve(id)));
if (!process.env.NODE_APP_INSTANCE) process.env.NODE_APP_INSTANCE = '0';
const major = +process.version.split('.')[0].split('v')[1];
const minor = +process.version.split('.')[1];
function transform(filename) {
const code = fs.readFileSync(filename, 'utf-8');
const result = esbuild.transformSync(code, {
sourcefile: filename,
sourcemap: 'both',
format: 'cjs',
loader: 'tsx',
target: `node${major}.${minor}`,
jsx: 'transform',
});
if (result.warnings.length) console.warn(result.warnings);
map[filename] = result.map;
return result.code;
}
const ESM = ['p-queue', 'p-timeout'];
const _script = new vm.Script('"Hydro"', { produceCachedData: true });
const bytecode = (_script.createCachedData && _script.createCachedData.call)
? _script.createCachedData()
: _script.cachedData;
require.extensions['.js'] = function loader(module, filename) {
if (ESM.filter((i) => filename.includes(i)).length || major < 14) {
return module._compile(transform(filename), filename);
}
const content = fs.readFileSync(filename, 'utf-8');
return module._compile(content, filename);
};
require.extensions['.ts'] = require.extensions['.tsx'] = function loader(module, filename) {
return module._compile(transform(filename), filename);
};
require.extensions['.jsc'] = function loader(module, filename) {
const buf = fs.readFileSync(filename);
bytecode.slice(12, 16).copy(buf, 12);
if (![12, 13, 14, 15, 16, 17].filter((i) => process.version.startsWith(`v${i}`)).length) {
bytecode.slice(16, 20).copy(buf, 16);
}
// eslint-disable-next-line no-return-assign
const length = buf.slice(8, 12).reduce((sum, number, power) => sum += number * (256 ** power), 0);
let dummyCode = '';
if (length > 1) dummyCode = `"${'\u200b'.repeat(length - 2)}"`;
const script = new vm.Script(dummyCode, {
filename,
lineOffset: 0,
displayErrors: true,
cachedData: buf,
});
if (script.cachedDataRejected) throw new Error(`cacheDataRejected on ${filename}`);
const compiledWrapper = script.runInThisContext({
filename,
lineOffset: 0,
columnOffset: 0,
displayErrors: true,
});
const dirname = path.dirname(filename);
const args = [module.exports, require, module, filename, dirname, process, global];
return compiledWrapper.apply(module.exports, args);
};
require('./commands');
+9 -12
View File
@@ -1,6 +1,6 @@
{
"name": "hydrooj",
"version": "3.15.5",
"version": "3.15.6",
"bin": "bin/hydrooj.js",
"main": "src/loader",
"module": "src/loader",
@@ -12,7 +12,7 @@
},
"preferUnplugged": true,
"dependencies": {
"@graphql-tools/schema": "^8.5.0",
"@graphql-tools/schema": "^8.5.1",
"@hydrooj/utils": "workspace:*",
"adm-zip": "0.5.5",
"ajv": "^8.11.0",
@@ -21,7 +21,6 @@
"detect-browser": "^5.3.0",
"emoji-regex": "^10.1.0",
"emojis-list": "2.1.0",
"esbuild": "0.14.3",
"fs-extra": "^10.1.0",
"graphql": "^16.5.0",
"graphql-scalars": "1.17.0",
@@ -31,7 +30,7 @@
"koa-body": "^5.0.0",
"koa-compress": "^5.1.0",
"koa-proxies": "^0.12.2",
"koa-router": "^10.1.1",
"koa-router": "^12.0.0",
"koa-static-cache": "^5.1.4",
"lodash": "^4.17.21",
"lru-cache": "7.12.0",
@@ -39,17 +38,16 @@
"minio": "7.0.25",
"moment-timezone": "^0.5.34",
"mongodb": "^3.7.3",
"nanoid": "^3.3.4",
"nanoid": "^4.0.0",
"nodemailer": "^6.7.7",
"notp": "^2.0.3",
"p-queue": "^7.2.0",
"p-queue": "^7.3.0",
"path-to-regexp": "^6.2.1",
"require-resolve-hook": "^1.1.0",
"schemastery": "^3.5.0",
"schemastery": "^3.5.1",
"semver": "^7.3.7",
"serialize-javascript": "^6.0.0",
"source-map-support": "^0.5.21",
"superagent": "^7.1.5",
"superagent": "^8.0.0",
"thirty-two": "^1.0.2",
"tx2": "^1.0.5"
},
@@ -57,7 +55,7 @@
"@types/adm-zip": "^0.4.34",
"@types/fs-extra": "^9.0.13",
"@types/js-yaml": "^4.0.5",
"@types/koa": "^2.13.4",
"@types/koa": "^2.13.5",
"@types/koa-compress": "^4.0.3",
"@types/koa-router": "^7.4.4",
"@types/koa-static-cache": "^5.1.1",
@@ -66,11 +64,10 @@
"@types/mime-types": "^2.1.1",
"@types/minio": "^7.0.13",
"@types/mongodb": "^3.6.20",
"@types/nodemailer": "^6.4.4",
"@types/nodemailer": "^6.4.5",
"@types/notp": "^2.0.2",
"@types/semver": "^7.3.10",
"@types/serialize-javascript": "^5.0.2",
"@types/source-map-support": "^0.5.4",
"@types/superagent": "^4.1.15",
"moment": "^2.29.4",
"pm2": "^5.2.0"
-5
View File
@@ -16,11 +16,6 @@ export * from './interface';
export { Schema, Logger };
const argv = cac().parse();
if (argv.options.debug) {
console.log('Debug mode enabled');
process.env.NODE_ENV = 'development';
process.env.DEV = 'on';
} else process.env.NODE_ENV = process.env.NODE_ENV || 'production';
const logger = new Logger('loader');
logger.debug('%o', argv);