auto-cr-cmd 2.0.92 → 2.0.96

package/dist/index.js

@@ -63,14 +63,17 @@ var _a;
 Object.defineProperty(exports, "__esModule", { value: true });
 var consola_1 = require("consola");
 var fs_1 = __importDefault(require("fs"));
+var os_1 = __importDefault(require("os"));
 var path_1 = __importDefault(require("path"));
 var commander_1 = require("commander");
-var wasm_1 = require("@swc/wasm");
 var config_1 = require("./config");
 var report_1 = require("./report");
 var i18n_1 = require("./i18n");
 var file_1 = require("./utils/file");
 var stdin_1 = require("./utils/stdin");
+var analyzeFile_1 = require("./scan/analyzeFile");
+var runtime_1 = require("./scan/runtime");
+var workerPool_1 = require("./scan/workerPool");
 var loader_1 = require("./rules/loader");
 var autocrrc_1 = require("./config/autocrrc");
 var ignore_1 = require("./config/ignore");
@@ -83,7 +86,7 @@ var consolaLoggers = {
 // 仅扫描 JS/TS 源码扩展名，避免把配置文件/JSON/图片等送进 SWC 解析导致报错。
 var SCANNABLE_EXTENSIONS = ['.ts', '.tsx', '.js', '.jsx'];
 // 运行 TS 源码时优先加载本地 rules，避免开发时拿到旧的 workspace 依赖。
-var rulesRuntime = loadRulesRuntime();
+var rulesRuntime = (0, runtime_1.loadRulesRuntime)();
 var builtinRules = rulesRuntime.builtinRules;
 var createRuleContext = rulesRuntime.createRuleContext;
 var RuleSeverity = rulesRuntime.RuleSeverity;
@@ -91,33 +94,28 @@ var RuleSeverity = rulesRuntime.RuleSeverity;
 var isScannableFile = function (filePath) {
     return SCANNABLE_EXTENSIONS.some(function (extension) { return filePath.endsWith(extension); });
 };
-function loadRulesRuntime() {
-    var fallback = require('auto-cr-rules');
-    var localRuntime = resolveLocalRulesRuntime();
-    if (localRuntime) {
-        return localRuntime;
-    }
-    return fallback;
-}
-function resolveLocalRulesRuntime() {
-    if (!__filename.endsWith('.ts')) {
-        return null;
-    }
-    var localEntry = path_1.default.resolve(__dirname, '../../auto-cr-rules/src/index.ts');
-    if (!fs_1.default.existsSync(localEntry)) {
-        return null;
-    }
-    try {
-        var localRuntime = require(localEntry);
-        if (Array.isArray(localRuntime.builtinRules)) {
-            return localRuntime;
+// 小文件集无需并行，避免 worker 初始化开销大于收益。
+var MIN_FILES_FOR_WORKERS = 20;
+var resolveWorkerCount = function (totalFiles) {
+    // 支持环境变量覆盖并发数：
+    // - AUTO_CR_WORKERS=0 表示强制单线程；
+    // - AUTO_CR_WORKERS=1 表示不启用并行；
+    // - >1 则按指定数量启用 worker。
+    var override = process.env.AUTO_CR_WORKERS;
+    if (override) {
+        var parsed = Number(override);
+        if (!Number.isNaN(parsed) && parsed >= 0) {
+            return Math.min(totalFiles, Math.floor(parsed));
         }
     }
-    catch (_a) {
-        return null;
+    if (totalFiles < MIN_FILES_FOR_WORKERS) {
+        return 1;
     }
-    return null;
-}
+    // 默认按 CPU 数量 - 1 取值，给主线程留出空间。
+    var cpuCount = os_1.default.cpus().length || 1;
+    var maxWorkers = Math.max(1, cpuCount - 1);
+    return Math.min(maxWorkers, totalFiles);
+};
 /**
  * CLI 主流程：
  * 1. 校验输入路径并应用 ignore；
@@ -127,7 +125,7 @@ function resolveLocalRulesRuntime() {
  */
 function run() {
     return __awaiter(this, arguments, void 0, function (filePaths, ruleDir, format, configPath, ignorePath, progressOption) {
-        var t, notifications, progressMode, progressStream, progressStreamHasTty, progressEnabled, progressPinned, progressStyle, progressTotal, progressCurrent, progressLastPercent, clearProgressLine, renderProgress, startProgress, advanceProgress, finishProgress, reporterHooks, log, validPaths, ignoreConfig, isIgnored_1, allFiles, _i, validPaths_1, targetPath, stat, directoryFiles, scannableFiles, customRules, rcConfig, rules, filesWithErrors, filesWithWarnings, filesWithOptimizing, totalViolations, totalErrorViolations, totalWarningViolations, totalOptimizingViolations, fileSummaries, _a, scannableFiles_1, file, summary, error_1;
+        var t, notifications, progressMode, progressStream, progressStreamHasTty, progressEnabled, progressPinned, progressStyle, progressTotal, progressCurrent, progressLastPercent, clearProgressLine, renderProgress, startProgress, advanceProgress, finishProgress, reporterHooks, createNotification, logRecord, log, validPaths, ignoreConfig, isIgnored_1, allFiles, _i, validPaths_1, targetPath, stat, directoryFiles, scannableFiles_2, customRules, rcConfig, rules, filesWithErrors_1, filesWithWarnings_1, filesWithOptimizing_1, totalViolations_1, totalErrorViolations_1, totalWarningViolations_1, totalOptimizingViolations_1, fileSummaries_1, analysisFormat, workerCount, applyFileSummary_1, fileOccurrences_1, uniqueFiles, pendingResults_1, nextOutputIndex_1, flushReadyResults_1, initData, fileSummaryCache, _loop_1, _a, scannableFiles_1, file, error_1;
         var _b;
         if (filePaths === void 0) { filePaths = []; }
         return __generator(this, function (_c) {
@@ -230,7 +228,7 @@ function run() {
                     reporterHooks = {
                         onAfterReport: function () { return renderProgress(true); },
                     };
-                    log = function (level, message, detail) {
+                    createNotification = function (level, message, detail) {
                         var detailText;
                         if (detail !== undefined) {
                             if (detail instanceof Error) {
@@ -248,21 +246,27 @@ function run() {
                                 }
                             }
                         }
-                        notifications.push({ level: level, message: message, detail: detailText });
+                        return { level: level, message: message, detail: detailText };
+                    };
+                    logRecord = function (record) {
+                        notifications.push(record);
                         if (format === 'text') {
-                            var logger = consolaLoggers[level];
-                            if (detail === undefined) {
-                                logger(message);
+                            var logger = consolaLoggers[record.level];
+                            if (record.detail === undefined) {
+                                logger(record.message);
                             }
                             else {
-                                logger(message, detail);
+                                logger(record.message, record.detail);
                             }
                             renderProgress(true);
                         }
                     };
+                    log = function (level, message, detail) {
+                        logRecord(createNotification(level, message, detail));
+                    };
                     _c.label = 1;
                 case 1:
-                    _c.trys.push([1, 6, , 7]);
+                    _c.trys.push([1, 8, , 9]);
                     if (filePaths.length === 0) {
                         log('info', t.noPathsProvided());
                         return [2 /*return*/, {
@@ -322,8 +326,10 @@ function run() {
                                 notifications: notifications,
                             }];
                     }
-                    scannableFiles = allFiles.filter(function (candidate) { return !candidate.endsWith('.d.ts') && !isIgnored_1(candidate); });
-                    customRules = (0, loader_1.loadCustomRules)(ruleDir);
+                    scannableFiles_2 = allFiles.filter(function (candidate) { return !candidate.endsWith('.d.ts') && !isIgnored_1(candidate); });
+                    customRules = (0, loader_1.loadCustomRules)(ruleDir, {
+                        onWarning: function (message, detail) { return log('warn', message, detail); },
+                    });
                     rcConfig = (0, autocrrc_1.loadAutoCrRc)(configPath);
                     rcConfig.warnings.forEach(function (warning) { return log('warn', warning); });
                     rules = (0, autocrrc_1.applyRuleConfig)(__spreadArray(__spreadArray([], builtinRules, true), customRules, true), rcConfig.rules, function (warning) {
@@ -341,234 +347,167 @@ function run() {
                                 notifications: notifications,
                             }];
                     }
-                    filesWithErrors = 0;
-                    filesWithWarnings = 0;
-                    filesWithOptimizing = 0;
-                    totalViolations = 0;
-                    totalErrorViolations = 0;
-                    totalWarningViolations = 0;
-                    totalOptimizingViolations = 0;
-                    fileSummaries = [];
-                    startProgress(scannableFiles.length);
-                    _a = 0, scannableFiles_1 = scannableFiles;
-                    _c.label = 2;
+                    filesWithErrors_1 = 0;
+                    filesWithWarnings_1 = 0;
+                    filesWithOptimizing_1 = 0;
+                    totalViolations_1 = 0;
+                    totalErrorViolations_1 = 0;
+                    totalWarningViolations_1 = 0;
+                    totalOptimizingViolations_1 = 0;
+                    fileSummaries_1 = [];
+                    analysisFormat = 'json';
+                    workerCount = resolveWorkerCount(scannableFiles_2.length);
+                    applyFileSummary_1 = function (filePath, summary) {
+                        if (summary.severityCounts.error > 0) {
+                            filesWithErrors_1 += 1;
+                        }
+                        if (summary.severityCounts.warning > 0) {
+                            filesWithWarnings_1 += 1;
+                        }
+                        if (summary.severityCounts.optimizing > 0) {
+                            filesWithOptimizing_1 += 1;
+                        }
+                        totalViolations_1 += summary.totalViolations;
+                        totalErrorViolations_1 += summary.errorViolations;
+                        totalWarningViolations_1 += summary.severityCounts.warning;
+                        totalOptimizingViolations_1 += summary.severityCounts.optimizing;
+                        fileSummaries_1.push({
+                            filePath: filePath,
+                            severityCounts: summary.severityCounts,
+                            totalViolations: summary.totalViolations,
+                            errorViolations: summary.errorViolations,
+                            violations: summary.violations,
+                        });
+                    };
+                    startProgress(scannableFiles_2.length);
+                    if (!(workerCount > 1)) return [3 /*break*/, 3];
+                    fileOccurrences_1 = new Map();
+                    scannableFiles_2.forEach(function (filePath, index) {
+                        var _a;
+                        var bucket = (_a = fileOccurrences_1.get(filePath)) !== null && _a !== void 0 ? _a : [];
+                        bucket.push(index);
+                        fileOccurrences_1.set(filePath, bucket);
+                    });
+                    uniqueFiles = Array.from(fileOccurrences_1.keys());
+                    pendingResults_1 = new Map();
+                    nextOutputIndex_1 = 0;
+                    flushReadyResults_1 = function () {
+                        while (pendingResults_1.has(nextOutputIndex_1)) {
+                            var entry = pendingResults_1.get(nextOutputIndex_1);
+                            if (!entry) {
+                                break;
+                            }
+                            pendingResults_1.delete(nextOutputIndex_1);
+                            var filePath = scannableFiles_2[nextOutputIndex_1];
+                            // 先回放日志，再回放违规输出，确保日志顺序与单线程一致。
+                            entry.logs.forEach(function (record) { return logRecord(record); });
+                            if (format === 'text') {
+                                (0, report_1.renderViolations)(filePath, entry.summary.violations, __assign({ format: format }, reporterHooks));
+                            }
+                            applyFileSummary_1(filePath, entry.summary);
+                            nextOutputIndex_1 += 1;
+                        }
+                    };
+                    initData = {
+                        ruleDir: ruleDir,
+                        ruleSettings: rcConfig.rules,
+                        language: (0, i18n_1.getLanguage)(),
+                        tsconfigPath: resolvedTsconfigPath,
+                    };
+                    return [4 /*yield*/, (0, workerPool_1.runWorkerPool)({
+                            files: uniqueFiles,
+                            workerCount: workerCount,
+                            initData: initData,
+                            onResult: function (_a) {
+                                var _b;
+                                var filePath = _a.filePath, summary = _a.summary, logs = _a.logs;
+                                var occurrences = (_b = fileOccurrences_1.get(filePath)) !== null && _b !== void 0 ? _b : [];
+                                for (var _i = 0, occurrences_1 = occurrences; _i < occurrences_1.length; _i++) {
+                                    var index = occurrences_1[_i];
+                                    pendingResults_1.set(index, { summary: summary, logs: logs });
+                                    // 进度按“原始文件列表”推进，保持百分比与实际扫描一致。
+                                    advanceProgress();
+                                }
+                                flushReadyResults_1();
+                            },
+                        })];
                 case 2:
-                    if (!(_a < scannableFiles_1.length)) return [3 /*break*/, 5];
-                    file = scannableFiles_1[_a];
-                    return [4 /*yield*/, analyzeFile(file, rules, format, log, reporterHooks)];
+                    _c.sent();
+                    flushReadyResults_1();
+                    return [3 /*break*/, 7];
                 case 3:
-                    summary = _c.sent();
-                    if (summary.severityCounts.error > 0) {
-                        filesWithErrors += 1;
-                    }
-                    if (summary.severityCounts.warning > 0) {
-                        filesWithWarnings += 1;
-                    }
-                    if (summary.severityCounts.optimizing > 0) {
-                        filesWithOptimizing += 1;
-                    }
-                    totalViolations += summary.totalViolations;
-                    totalErrorViolations += summary.errorViolations;
-                    totalWarningViolations += summary.severityCounts.warning;
-                    totalOptimizingViolations += summary.severityCounts.optimizing;
-                    fileSummaries.push({
-                        filePath: file,
-                        severityCounts: summary.severityCounts,
-                        totalViolations: summary.totalViolations,
-                        errorViolations: summary.errorViolations,
-                        violations: summary.violations,
-                    });
-                    advanceProgress();
+                    fileSummaryCache = new Map();
+                    _loop_1 = function (file) {
+                        var cached, summary, logs, capturedLogs_1, logForFile;
+                        return __generator(this, function (_d) {
+                            switch (_d.label) {
+                                case 0:
+                                    cached = fileSummaryCache.get(file);
+                                    summary = void 0;
+                                    logs = [];
+                                    if (!cached) return [3 /*break*/, 1];
+                                    summary = cached.summary;
+                                    logs = cached.logs;
+                                    logs.forEach(function (record) { return logRecord(record); });
+                                    return [3 /*break*/, 3];
+                                case 1:
+                                    capturedLogs_1 = [];
+                                    logForFile = function (level, message, detail) {
+                                        var record = createNotification(level, message, detail);
+                                        capturedLogs_1.push(record);
+                                        logRecord(record);
+                                    };
+                                    return [4 /*yield*/, (0, analyzeFile_1.analyzeFile)(file, rules, analysisFormat, logForFile, createRuleContext)];
+                                case 2:
+                                    summary = _d.sent();
+                                    fileSummaryCache.set(file, { summary: summary, logs: capturedLogs_1 });
+                                    _d.label = 3;
+                                case 3:
+                                    if (format === 'text') {
+                                        (0, report_1.renderViolations)(file, summary.violations, __assign({ format: format }, reporterHooks));
+                                    }
+                                    applyFileSummary_1(file, summary);
+                                    advanceProgress();
+                                    return [2 /*return*/];
+                            }
+                        });
+                    };
+                    _a = 0, scannableFiles_1 = scannableFiles_2;
                     _c.label = 4;
                 case 4:
-                    _a++;
-                    return [3 /*break*/, 2];
+                    if (!(_a < scannableFiles_1.length)) return [3 /*break*/, 7];
+                    file = scannableFiles_1[_a];
+                    return [5 /*yield**/, _loop_1(file)];
                 case 5:
+                    _c.sent();
+                    _c.label = 6;
+                case 6:
+                    _a++;
+                    return [3 /*break*/, 4];
+                case 7:
                     finishProgress();
                     return [2 /*return*/, {
-                            scannedFiles: scannableFiles.length,
-                            filesWithErrors: filesWithErrors,
-                            filesWithWarnings: filesWithWarnings,
-                            filesWithOptimizing: filesWithOptimizing,
+                            scannedFiles: scannableFiles_2.length,
+                            filesWithErrors: filesWithErrors_1,
+                            filesWithWarnings: filesWithWarnings_1,
+                            filesWithOptimizing: filesWithOptimizing_1,
                             violationTotals: {
-                                total: totalViolations,
-                                error: totalErrorViolations,
-                                warning: totalWarningViolations,
-                                optimizing: totalOptimizingViolations,
+                                total: totalViolations_1,
+                                error: totalErrorViolations_1,
+                                warning: totalWarningViolations_1,
+                                optimizing: totalOptimizingViolations_1,
                             },
-                            files: fileSummaries,
+                            files: fileSummaries_1,
                             notifications: notifications,
                         }];
-                case 6:
+                case 8:
                     error_1 = _c.sent();
                     throw error_1 instanceof Error ? error_1 : new Error(String(error_1));
-                case 7: return [2 /*return*/];
+                case 9: return [2 /*return*/];
             }
         });
     });
 }
-/**
- * 单文件扫描流程：
- * - 读取源码并解析 AST；
- * - 基于语言/源码构建规则上下文；
- * - 逐条执行规则，收集 reporter 输出；
- * - 汇总为文件级统计。
- */
-function analyzeFile(file, rules, format, log, reporterHooks) {
-    return __awaiter(this, void 0, void 0, function () {
-        var source, reporter, t, ast, parseOptions, language, baseContext, sharedHelpers, _loop_1, _i, rules_1, rule, summary;
-        return __generator(this, function (_a) {
-            switch (_a.label) {
-                case 0:
-                    source = (0, file_1.readFile)(file);
-                    reporter = (0, report_1.createReporter)(file, source, __assign({ format: format }, reporterHooks));
-                    t = (0, i18n_1.getTranslator)();
-                    try {
-                        parseOptions = (0, config_1.loadParseOptions)(file);
-                        ast = (0, wasm_1.parseSync)(source, parseOptions);
-                    }
-                    catch (error) {
-                        log('error', t.parseFileFailed({ file: file }), error);
-                        return [2 /*return*/, {
-                                severityCounts: {
-                                    error: 1,
-                                    warning: 0,
-                                    optimizing: 0,
-                                },
-                                totalViolations: 1,
-                                errorViolations: 1,
-                                violations: [],
-                            }];
-                    }
-                    language = (0, i18n_1.getLanguage)();
-                    baseContext = createRuleContext({
-                        ast: ast,
-                        filePath: file,
-                        source: source,
-                        reporter: reporter,
-                        language: language,
-                    });
-                    sharedHelpers = baseContext.helpers;
-                    _loop_1 = function (rule) {
-                        var scopedReporter_1, reporterWithRecord_1, helpers, context, error_2;
-                        return __generator(this, function (_b) {
-                            switch (_b.label) {
-                                case 0:
-                                    _b.trys.push([0, 2, , 3]);
-                                    scopedReporter_1 = reporter.forRule(rule);
-                                    reporterWithRecord_1 = scopedReporter_1;
-                                    helpers = __assign(__assign({}, sharedHelpers), { reportViolation: (function (input, span) {
-                                            var normalized = normalizeViolationInput(input, span);
-                                            if (typeof reporterWithRecord_1.record === 'function') {
-                                                reporterWithRecord_1.record({
-                                                    description: normalized.message,
-                                                    code: normalized.code,
-                                                    suggestions: normalized.suggestions,
-                                                    span: normalized.span,
-                                                    line: normalized.line,
-                                                });
-                                                return;
-                                            }
-                                            if (normalized.span) {
-                                                scopedReporter_1.errorAtSpan(normalized.span, normalized.message);
-                                                return;
-                                            }
-                                            if (typeof normalized.line === 'number') {
-                                                scopedReporter_1.errorAtLine(normalized.line, normalized.message);
-                                                return;
-                                            }
-                                            scopedReporter_1.error(normalized.message);
-                                        }) });
-                                    context = __assign(__assign({}, baseContext), { reporter: scopedReporter_1, helpers: helpers });
-                                    return [4 /*yield*/, rule.run(context)];
-                                case 1:
-                                    _b.sent();
-                                    return [3 /*break*/, 3];
-                                case 2:
-                                    error_2 = _b.sent();
-                                    log('error', t.ruleExecutionFailed({ ruleName: rule.name, file: file }), error_2);
-                                    return [3 /*break*/, 3];
-                                case 3: return [2 /*return*/];
-                            }
-                        });
-                    };
-                    _i = 0, rules_1 = rules;
-                    _a.label = 1;
-                case 1:
-                    if (!(_i < rules_1.length)) return [3 /*break*/, 4];
-                    rule = rules_1[_i];
-                    return [5 /*yield**/, _loop_1(rule)];
-                case 2:
-                    _a.sent();
-                    _a.label = 3;
-                case 3:
-                    _i++;
-                    return [3 /*break*/, 1];
-                case 4:
-                    summary = reporter.flush();
-                    return [2 /*return*/, {
-                            severityCounts: summary.severityCounts,
-                            totalViolations: summary.totalViolations,
-                            errorViolations: summary.errorViolations,
-                            violations: summary.violations,
-                        }];
-            }
-        });
-    });
-}
-function normalizeViolationInput(input, spanArg) {
-    var _a;
-    // 规则既可以直接输出字符串，也可以输出结构化对象；这里统一为标准格式。
-    if (typeof input === 'string') {
-        return {
-            message: input,
-            span: spanArg,
-        };
-    }
-    if (input && typeof input === 'object') {
-        var candidate = input;
-        var description = typeof candidate.description === 'string'
-            ? candidate.description
-            : typeof candidate.message === 'string'
-                ? candidate.message
-                : undefined;
-        var code = typeof candidate.code === 'string' ? candidate.code : undefined;
-        var suggestions = void 0;
-        if (Array.isArray(candidate.suggestions)) {
-            var normalizedSuggestions = [];
-            for (var _i = 0, _b = candidate.suggestions; _i < _b.length; _i++) {
-                var entry = _b[_i];
-                if (typeof entry === 'string') {
-                    normalizedSuggestions.push({ text: entry });
-                    continue;
-                }
-                if (entry && typeof entry === 'object') {
-                    var suggestion = entry;
-                    if (typeof suggestion.text === 'string') {
-                        normalizedSuggestions.push({
-                            text: suggestion.text,
-                            link: typeof suggestion.link === 'string' ? suggestion.link : undefined,
-                        });
-                    }
-                }
-            }
-            if (normalizedSuggestions.length > 0) {
-                suggestions = normalizedSuggestions;
-            }
-        }
-        return {
-            message: description !== null && description !== void 0 ? description : 'Rule violation detected.',
-            span: (_a = candidate.span) !== null && _a !== void 0 ? _a : spanArg,
-            line: typeof candidate.line === 'number' ? candidate.line : undefined,
-            code: code,
-            suggestions: suggestions,
-        };
-    }
-    return {
-        message: 'Rule violation detected.',
-        span: spanArg,
-    };
-}
 // CLI 输出格式解析：仅允许 text/json。
 function parseOutputFormat(value) {
     if (!value) {
@@ -660,7 +599,8 @@ commander_1.program
 var options = commander_1.program.opts();
 var cliArguments = commander_1.program.args;
 (0, i18n_1.setLanguage)((_a = options.language) !== null && _a !== void 0 ? _a : process.env.LANG);
-(0, config_1.setTsConfigPath)(options.tsconfig ? path_1.default.resolve(process.cwd(), options.tsconfig) : undefined);
+var resolvedTsconfigPath = options.tsconfig ? path_1.default.resolve(process.cwd(), options.tsconfig) : undefined;
+(0, config_1.setTsConfigPath)(resolvedTsconfigPath);
 var outputFormat;
 var progressOption;
 try {
@@ -681,7 +621,7 @@ catch (error) {
 }
 ;
 (function () { return __awaiter(void 0, void 0, void 0, function () {
-    var stdinTargets, combinedTargets, filePaths, result, t, payload, exitCode, language, resultMessage, exitCode, error_3, t, detail, payload;
+    var stdinTargets, combinedTargets, filePaths, result, t, payload, exitCode, language, resultMessage, exitCode, error_2, t, detail, payload;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
@@ -716,9 +656,9 @@ catch (error) {
                 }
                 return [3 /*break*/, 4];
             case 3:
-                error_3 = _a.sent();
+                error_2 = _a.sent();
                 t = (0, i18n_1.getTranslator)();
-                detail = error_3 instanceof Error ? error_3.message : String(error_3);
+                detail = error_2 instanceof Error ? error_2.message : String(error_2);
                 if (outputFormat === 'json') {
                     payload = {
                         error: {