90 lines
3.5 KiB
JavaScript
90 lines
3.5 KiB
JavaScript
"use strict";
|
||
var __importDefault = (this && this.__importDefault) || function (mod) {
|
||
return (mod && mod.__esModule) ? mod : { "default": mod };
|
||
};
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
exports.appendConsoleExpr = exports.restoreConsoleExpr = exports.rewriteConsoleExpr = void 0;
|
||
const magic_string_1 = __importDefault(require("magic-string"));
|
||
const utils_1 = require("../utils");
|
||
function rewriteConsoleExpr(method, id, filename, code, sourceMap = false) {
|
||
filename = (0, utils_1.normalizePath)(filename);
|
||
const re = /(console\.(log|info|debug|warn|error))\s*\(([^)]+)\)/g;
|
||
const locate = getLocator(code);
|
||
const s = new magic_string_1.default(code);
|
||
let match;
|
||
while ((match = re.exec(code))) {
|
||
const [, expr, type] = match;
|
||
s.overwrite(match.index, match.index + expr.length + 1, method + `('${type}','at ${filename}:${locate(match.index).line + 1}',`);
|
||
}
|
||
if (s.hasChanged()) {
|
||
return {
|
||
code: s.toString(),
|
||
map: sourceMap ? s.generateMap({ hires: true }) : { mappings: '' },
|
||
};
|
||
}
|
||
return { code, map: { mappings: '' } };
|
||
}
|
||
exports.rewriteConsoleExpr = rewriteConsoleExpr;
|
||
function restoreConsoleExpr(code) {
|
||
return code.replace(/(?:uni\.)?__f__\('([^']+)','at ([^:]+):(\d+)',/g, 'console.$1(');
|
||
}
|
||
exports.restoreConsoleExpr = restoreConsoleExpr;
|
||
function getLocator(source) {
|
||
const originalLines = source.split('\n');
|
||
const lineOffsets = [];
|
||
for (let i = 0, pos = 0; i < originalLines.length; i++) {
|
||
lineOffsets.push(pos);
|
||
pos += originalLines[i].length + 1;
|
||
}
|
||
return function locate(index) {
|
||
let i = 0;
|
||
let j = lineOffsets.length;
|
||
while (i < j) {
|
||
const m = (i + j) >> 1;
|
||
if (index < lineOffsets[m]) {
|
||
j = m;
|
||
}
|
||
else {
|
||
i = m + 1;
|
||
}
|
||
}
|
||
const line = i - 1;
|
||
const column = index - lineOffsets[line];
|
||
return { line, column };
|
||
};
|
||
}
|
||
const methods = ['log', 'info', 'debug', 'warn', 'error'];
|
||
function appendConsoleExpr(filename, code, ts) {
|
||
if (!ts) {
|
||
return code;
|
||
}
|
||
const s = new magic_string_1.default(code);
|
||
const sourceFile = ts.createSourceFile(filename, code, ts.ScriptTarget.Latest);
|
||
// 遍历sourceFile，查找console的方法调用
|
||
const traverse = (node) => {
|
||
ts.forEachChild(node, (node) => traverse(node));
|
||
if (ts.isCallExpression(node) &&
|
||
node.arguments.length > 0 &&
|
||
ts.isPropertyAccessExpression(node.expression)) {
|
||
const propertyAccess = node.expression;
|
||
if (ts.isIdentifier(propertyAccess.expression) &&
|
||
propertyAccess.expression.text === 'console' &&
|
||
ts.isIdentifier(propertyAccess.name) &&
|
||
methods.includes(propertyAccess.name.text)) {
|
||
const lastArg = node.arguments[node.arguments.length - 1];
|
||
if (lastArg) {
|
||
const { line } = sourceFile.getLineAndCharacterOfPosition(propertyAccess.name.end);
|
||
// 重要，需要用双引号，因为混编的kt，swift，java不能用单引号（char类型）
|
||
s.prependRight(lastArg.getEnd(), `, " at ${filename}:${line + 1}"`);
|
||
}
|
||
}
|
||
}
|
||
};
|
||
traverse(sourceFile);
|
||
if (s.hasChanged()) {
|
||
return s.toString();
|
||
}
|
||
return code;
|
||
}
|
||
exports.appendConsoleExpr = appendConsoleExpr;
|