@dinyangetoh/codeplug-cli 0.1.2 → 0.1.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +286 -45
- package/dist/cli/commands/convention.d.ts +1 -1
- package/dist/cli/commands/convention.d.ts.map +1 -1
- package/dist/cli/commands/convention.js +174 -31
- package/dist/cli/commands/convention.js.map +1 -1
- package/dist/cli/commands/docs.d.ts +3 -1
- package/dist/cli/commands/docs.d.ts.map +1 -1
- package/dist/cli/commands/docs.js +13 -2
- package/dist/cli/commands/docs.js.map +1 -1
- package/dist/cli/commands/export.d.ts.map +1 -1
- package/dist/cli/commands/export.js +6 -1
- package/dist/cli/commands/export.js.map +1 -1
- package/dist/cli/commands/start.d.ts +2 -0
- package/dist/cli/commands/start.d.ts.map +1 -0
- package/dist/cli/commands/start.js +172 -0
- package/dist/cli/commands/start.js.map +1 -0
- package/dist/cli/index.js +74 -66
- package/dist/cli/index.js.map +1 -1
- package/dist/config/ConfigManager.d.ts +18 -0
- package/dist/config/ConfigManager.d.ts.map +1 -1
- package/dist/config/ConfigManager.js +55 -2
- package/dist/config/ConfigManager.js.map +1 -1
- package/dist/config/ConventionSchema.d.ts +59 -0
- package/dist/config/ConventionSchema.d.ts.map +1 -1
- package/dist/config/ConventionSchema.js +11 -0
- package/dist/config/ConventionSchema.js.map +1 -1
- package/dist/config/defaults.d.ts +60 -2
- package/dist/config/defaults.d.ts.map +1 -1
- package/dist/config/defaults.js +136 -20
- package/dist/config/defaults.js.map +1 -1
- package/dist/config/types.d.ts +71 -0
- package/dist/config/types.d.ts.map +1 -1
- package/dist/core/analyzer/AstAnalyzer.d.ts +11 -2
- package/dist/core/analyzer/AstAnalyzer.d.ts.map +1 -1
- package/dist/core/analyzer/AstAnalyzer.js +21 -5
- package/dist/core/analyzer/AstAnalyzer.js.map +1 -1
- package/dist/core/analyzer/ConventionDetector.d.ts +12 -1
- package/dist/core/analyzer/ConventionDetector.d.ts.map +1 -1
- package/dist/core/analyzer/ConventionDetector.js +120 -20
- package/dist/core/analyzer/ConventionDetector.js.map +1 -1
- package/dist/core/analyzer/ConventionMlResolver.d.ts +4 -0
- package/dist/core/analyzer/ConventionMlResolver.d.ts.map +1 -0
- package/dist/core/analyzer/ConventionMlResolver.js +17 -0
- package/dist/core/analyzer/ConventionMlResolver.js.map +1 -0
- package/dist/core/analyzer/PatternAggregator.d.ts +11 -2
- package/dist/core/analyzer/PatternAggregator.d.ts.map +1 -1
- package/dist/core/analyzer/PatternAggregator.js +87 -15
- package/dist/core/analyzer/PatternAggregator.js.map +1 -1
- package/dist/core/analyzer/SemanticCoherencePhase.d.ts +10 -0
- package/dist/core/analyzer/SemanticCoherencePhase.d.ts.map +1 -0
- package/dist/core/analyzer/SemanticCoherencePhase.js +191 -0
- package/dist/core/analyzer/SemanticCoherencePhase.js.map +1 -0
- package/dist/core/analyzer/SemanticCoherenceService.d.ts +11 -0
- package/dist/core/analyzer/SemanticCoherenceService.d.ts.map +1 -0
- package/dist/core/analyzer/SemanticCoherenceService.js +56 -0
- package/dist/core/analyzer/SemanticCoherenceService.js.map +1 -0
- package/dist/core/analyzer/visitors/ComponentVisitor.d.ts.map +1 -1
- package/dist/core/analyzer/visitors/ComponentVisitor.js +2 -2
- package/dist/core/analyzer/visitors/ComponentVisitor.js.map +1 -1
- package/dist/core/analyzer/visitors/ErrorHandlingVisitor.d.ts.map +1 -1
- package/dist/core/analyzer/visitors/ErrorHandlingVisitor.js +2 -2
- package/dist/core/analyzer/visitors/ErrorHandlingVisitor.js.map +1 -1
- package/dist/core/analyzer/visitors/ImportVisitor.d.ts +5 -0
- package/dist/core/analyzer/visitors/ImportVisitor.d.ts.map +1 -1
- package/dist/core/analyzer/visitors/ImportVisitor.js +8 -3
- package/dist/core/analyzer/visitors/ImportVisitor.js.map +1 -1
- package/dist/core/analyzer/visitors/NamingVisitor.d.ts +20 -2
- package/dist/core/analyzer/visitors/NamingVisitor.d.ts.map +1 -1
- package/dist/core/analyzer/visitors/NamingVisitor.js +464 -25
- package/dist/core/analyzer/visitors/NamingVisitor.js.map +1 -1
- package/dist/core/analyzer/visitors/SchemaVisitor.d.ts +7 -0
- package/dist/core/analyzer/visitors/SchemaVisitor.d.ts.map +1 -0
- package/dist/core/analyzer/visitors/SchemaVisitor.js +50 -0
- package/dist/core/analyzer/visitors/SchemaVisitor.js.map +1 -0
- package/dist/core/analyzer/visitors/StructureVisitor.d.ts +4 -1
- package/dist/core/analyzer/visitors/StructureVisitor.d.ts.map +1 -1
- package/dist/core/analyzer/visitors/StructureVisitor.js +38 -4
- package/dist/core/analyzer/visitors/StructureVisitor.js.map +1 -1
- package/dist/core/analyzer/visitors/types.d.ts +4 -0
- package/dist/core/analyzer/visitors/types.d.ts.map +1 -1
- package/dist/core/classifier/CodeBertEmbedder.d.ts +9 -0
- package/dist/core/classifier/CodeBertEmbedder.d.ts.map +1 -0
- package/dist/core/classifier/CodeBertEmbedder.js +33 -0
- package/dist/core/classifier/CodeBertEmbedder.js.map +1 -0
- package/dist/core/classifier/ConfidenceGate.d.ts +2 -0
- package/dist/core/classifier/ConfidenceGate.d.ts.map +1 -1
- package/dist/core/classifier/ConfidenceGate.js +6 -2
- package/dist/core/classifier/ConfidenceGate.js.map +1 -1
- package/dist/core/exporter/FreshnessChecker.d.ts +5 -1
- package/dist/core/exporter/FreshnessChecker.d.ts.map +1 -1
- package/dist/core/exporter/FreshnessChecker.js +5 -10
- package/dist/core/exporter/FreshnessChecker.js.map +1 -1
- package/dist/core/generator/DocGenerator.d.ts +6 -1
- package/dist/core/generator/DocGenerator.d.ts.map +1 -1
- package/dist/core/generator/DocGenerator.js +62 -7
- package/dist/core/generator/DocGenerator.js.map +1 -1
- package/dist/core/generator/StalenessTracker.d.ts +7 -1
- package/dist/core/generator/StalenessTracker.d.ts.map +1 -1
- package/dist/core/generator/StalenessTracker.js +12 -12
- package/dist/core/generator/StalenessTracker.js.map +1 -1
- package/dist/core/generator/documents/ArchitectureGenerator.d.ts +1 -0
- package/dist/core/generator/documents/ArchitectureGenerator.d.ts.map +1 -1
- package/dist/core/generator/documents/ArchitectureGenerator.js +37 -12
- package/dist/core/generator/documents/ArchitectureGenerator.js.map +1 -1
- package/dist/core/generator/documents/ContributingGenerator.d.ts.map +1 -1
- package/dist/core/generator/documents/ContributingGenerator.js +20 -13
- package/dist/core/generator/documents/ContributingGenerator.js.map +1 -1
- package/dist/core/generator/documents/ConventionsGenerator.d.ts.map +1 -1
- package/dist/core/generator/documents/ConventionsGenerator.js +16 -2
- package/dist/core/generator/documents/ConventionsGenerator.js.map +1 -1
- package/dist/core/generator/documents/OnboardingGenerator.d.ts.map +1 -1
- package/dist/core/generator/documents/OnboardingGenerator.js +26 -18
- package/dist/core/generator/documents/OnboardingGenerator.js.map +1 -1
- package/dist/core/generator/documents/ReadmeGenerator.d.ts.map +1 -1
- package/dist/core/generator/documents/ReadmeGenerator.js +26 -14
- package/dist/core/generator/documents/ReadmeGenerator.js.map +1 -1
- package/dist/core/generator/documents/types.d.ts +8 -1
- package/dist/core/generator/documents/types.d.ts.map +1 -1
- package/dist/core/generator/mlPipelineBuilder.d.ts +8 -0
- package/dist/core/generator/mlPipelineBuilder.d.ts.map +1 -0
- package/dist/core/generator/mlPipelineBuilder.js +117 -0
- package/dist/core/generator/mlPipelineBuilder.js.map +1 -0
- package/dist/core/scorer/ComplianceScorer.d.ts +5 -1
- package/dist/core/scorer/ComplianceScorer.d.ts.map +1 -1
- package/dist/core/scorer/ComplianceScorer.js +47 -23
- package/dist/core/scorer/ComplianceScorer.js.map +1 -1
- package/dist/core/scorer/ViolationDetector.d.ts +19 -2
- package/dist/core/scorer/ViolationDetector.d.ts.map +1 -1
- package/dist/core/scorer/ViolationDetector.js +169 -53
- package/dist/core/scorer/ViolationDetector.js.map +1 -1
- package/dist/models/ModelManager.d.ts +5 -1
- package/dist/models/ModelManager.d.ts.map +1 -1
- package/dist/models/ModelManager.js +19 -5
- package/dist/models/ModelManager.js.map +1 -1
- package/dist/models/ModelRegistry.d.ts +1 -1
- package/dist/models/ModelRegistry.d.ts.map +1 -1
- package/dist/models/ModelRegistry.js +28 -0
- package/dist/models/ModelRegistry.js.map +1 -1
- package/dist/storage/CustomRuleStore.d.ts +9 -0
- package/dist/storage/CustomRuleStore.d.ts.map +1 -0
- package/dist/storage/CustomRuleStore.js +22 -0
- package/dist/storage/CustomRuleStore.js.map +1 -0
- package/dist/storage/ScoreStore.d.ts +2 -5
- package/dist/storage/ScoreStore.d.ts.map +1 -1
- package/dist/storage/ScoreStore.js +23 -71
- package/dist/storage/ScoreStore.js.map +1 -1
- package/package.json +6 -4
- package/dist/core/classifier/DriftClassifier.d.ts +0 -17
- package/dist/core/classifier/DriftClassifier.d.ts.map +0 -1
- package/dist/core/classifier/DriftClassifier.js +0 -293
- package/dist/core/classifier/DriftClassifier.js.map +0 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ComplianceScorer.d.ts","sourceRoot":"","sources":["../../../src/core/scorer/ComplianceScorer.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"ComplianceScorer.d.ts","sourceRoot":"","sources":["../../../src/core/scorer/ComplianceScorer.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,eAAe,EAEf,aAAa,EAEb,SAAS,EACV,MAAM,uBAAuB,CAAC;AAE/B,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,aAAa,CAAgB;gBAEzB,OAAO,CAAC,EAAE;QAAE,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE;IAIvD,SAAS,CAAC,UAAU,EAAE,SAAS,EAAE,GAAG,eAAe;IA+B7C,eAAe,CACnB,UAAU,EAAE,SAAS,EAAE,EACvB,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,eAAe,CAAC;IAmCrB,WAAW,CACf,KAAK,EAAE,eAAe,EACtB,UAAU,EAAE,SAAS,EAAE,GACtB,OAAO,CAAC,IAAI,CAAC;CA+DjB"}
|
|
@@ -1,41 +1,53 @@
|
|
|
1
|
-
|
|
2
|
-
critical: 15,
|
|
3
|
-
high: 8,
|
|
4
|
-
medium: 3,
|
|
5
|
-
low: 1,
|
|
6
|
-
};
|
|
1
|
+
import { DEFAULT_SCORING } from "../../config/defaults.js";
|
|
7
2
|
export class ComplianceScorer {
|
|
3
|
+
scoringConfig;
|
|
4
|
+
constructor(options) {
|
|
5
|
+
this.scoringConfig = options?.scoringConfig ?? DEFAULT_SCORING;
|
|
6
|
+
}
|
|
8
7
|
calculate(violations) {
|
|
9
|
-
const
|
|
8
|
+
const scoringWeights = this.scoringConfig.weights ??
|
|
9
|
+
DEFAULT_SCORING.weights ??
|
|
10
|
+
{ critical: 15, high: 8, medium: 3, low: 1 };
|
|
11
|
+
const threshold = this.scoringConfig.threshold ?? DEFAULT_SCORING.threshold ?? 70;
|
|
12
|
+
const breakdown = {
|
|
13
|
+
critical: 0,
|
|
14
|
+
high: 0,
|
|
15
|
+
medium: 0,
|
|
16
|
+
low: 0,
|
|
17
|
+
};
|
|
10
18
|
let totalDeduction = 0;
|
|
11
19
|
for (const v of violations) {
|
|
12
20
|
breakdown[v.severity]++;
|
|
13
|
-
totalDeduction +=
|
|
21
|
+
totalDeduction += scoringWeights[v.severity];
|
|
14
22
|
}
|
|
15
23
|
return {
|
|
16
24
|
total: Math.max(0, 100 - totalDeduction),
|
|
17
25
|
breakdown,
|
|
18
26
|
violationCount: violations.length,
|
|
19
|
-
threshold
|
|
27
|
+
threshold,
|
|
20
28
|
};
|
|
21
29
|
}
|
|
22
30
|
async scoreAndPersist(violations, projectRoot) {
|
|
23
31
|
const score = this.calculate(violations);
|
|
24
|
-
const { ScoreStore } = await import(
|
|
32
|
+
const { ScoreStore } = await import("../../storage/ScoreStore.js");
|
|
25
33
|
const store = new ScoreStore(projectRoot);
|
|
26
34
|
try {
|
|
27
|
-
const { randomUUID, createHash } = await import(
|
|
35
|
+
const { randomUUID, createHash } = await import("node:crypto");
|
|
28
36
|
const record = {
|
|
29
37
|
id: randomUUID(),
|
|
30
|
-
projectHash: createHash(
|
|
38
|
+
projectHash: createHash("sha256")
|
|
39
|
+
.update(projectRoot)
|
|
40
|
+
.digest("hex")
|
|
41
|
+
.slice(0, 12),
|
|
31
42
|
score: score.total,
|
|
32
43
|
breakdown: score.breakdown,
|
|
33
44
|
createdAt: new Date().toISOString(),
|
|
34
45
|
};
|
|
35
46
|
await store.save(record);
|
|
36
|
-
const { TrendTracker } = await import(
|
|
47
|
+
const { TrendTracker } = await import("./TrendTracker.js");
|
|
37
48
|
const tracker = new TrendTracker();
|
|
38
|
-
const
|
|
49
|
+
const trendWindow = this.scoringConfig.trendWindow ?? DEFAULT_SCORING.trendWindow ?? 8;
|
|
50
|
+
const history = await store.getHistory(trendWindow);
|
|
39
51
|
score.trend = tracker.computeTrend(history);
|
|
40
52
|
}
|
|
41
53
|
finally {
|
|
@@ -44,33 +56,45 @@ export class ComplianceScorer {
|
|
|
44
56
|
return score;
|
|
45
57
|
}
|
|
46
58
|
async printReport(score, violations) {
|
|
47
|
-
const chalk = (await import(
|
|
48
|
-
const trendIcon = score.trend ===
|
|
59
|
+
const chalk = (await import("chalk")).default;
|
|
60
|
+
const trendIcon = score.trend === "improving"
|
|
61
|
+
? "\ud83d\udcc8"
|
|
62
|
+
: score.trend === "declining"
|
|
63
|
+
? "\ud83d\udcc9"
|
|
64
|
+
: "\u27a1\ufe0f";
|
|
49
65
|
console.log(chalk.bold(`\n\ud83d\udcca Compliance Score: ${score.total}/100 ${trendIcon}\n`));
|
|
50
66
|
if (violations.length === 0) {
|
|
51
|
-
console.log(chalk.green(
|
|
67
|
+
console.log(chalk.green("\u2705 No violations found.\n"));
|
|
52
68
|
return;
|
|
53
69
|
}
|
|
54
|
-
const severityOrder = [
|
|
55
|
-
const icons = {
|
|
70
|
+
const severityOrder = ["critical", "high", "medium", "low"];
|
|
71
|
+
const icons = {
|
|
72
|
+
critical: "\ud83d\udd34",
|
|
73
|
+
high: "\ud83d\udfe0",
|
|
74
|
+
medium: "\ud83d\udfe1",
|
|
75
|
+
low: "\ud83d\udd35",
|
|
76
|
+
};
|
|
56
77
|
for (const sev of severityOrder) {
|
|
57
78
|
const group = violations.filter((v) => v.severity === sev);
|
|
58
79
|
if (group.length === 0)
|
|
59
80
|
continue;
|
|
60
81
|
console.log(`${icons[sev]} ${chalk.bold(sev.toUpperCase())} (${group.length}):`);
|
|
61
82
|
for (const v of group) {
|
|
62
|
-
|
|
83
|
+
const scoreSuffix = v.fileScorePercent !== undefined
|
|
84
|
+
? chalk.dim(` (${v.fileScorePercent}%)`)
|
|
85
|
+
: "";
|
|
86
|
+
console.log(chalk.red(` ${v.file}${v.line ? `:${v.line}` : ""}${scoreSuffix}`));
|
|
63
87
|
console.log(chalk.dim(` ${v.message}`));
|
|
64
88
|
console.log(chalk.dim(` Expected: ${v.expected}`));
|
|
65
89
|
console.log(chalk.dim(` Found: ${v.found}`));
|
|
66
90
|
if (v.autoFixable) {
|
|
67
91
|
console.log(chalk.cyan(` \u2192 Auto-fixable: codeplug convention fix --id ${v.id}`));
|
|
68
92
|
}
|
|
69
|
-
console.log(
|
|
93
|
+
console.log("");
|
|
70
94
|
}
|
|
71
95
|
}
|
|
72
|
-
const passIcon = score.total >= (score.threshold ?? 70) ?
|
|
73
|
-
console.log(`${passIcon} Threshold: ${score.threshold ?? 70} | Score: ${score.total}\n`);
|
|
96
|
+
const passIcon = score.total >= (score.threshold ?? 70) ? "\u2705" : "\u274c";
|
|
97
|
+
console.log(`${passIcon} Threshold: ${score.threshold ?? 70} | Score: ${score.total} | Issues: ${violations.length}\n`);
|
|
74
98
|
}
|
|
75
99
|
}
|
|
76
100
|
//# sourceMappingURL=ComplianceScorer.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ComplianceScorer.js","sourceRoot":"","sources":["../../../src/core/scorer/ComplianceScorer.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"ComplianceScorer.js","sourceRoot":"","sources":["../../../src/core/scorer/ComplianceScorer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAS3D,MAAM,OAAO,gBAAgB;IACnB,aAAa,CAAgB;IAErC,YAAY,OAA2C;QACrD,IAAI,CAAC,aAAa,GAAG,OAAO,EAAE,aAAa,IAAI,eAAe,CAAC;IACjE,CAAC;IAED,SAAS,CAAC,UAAuB;QAC/B,MAAM,cAAc,GAClB,IAAI,CAAC,aAAa,CAAC,OAAO;YAC1B,eAAe,CAAC,OAAO;YACtB,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAGzC,CAAC;QACL,MAAM,SAAS,GACb,IAAI,CAAC,aAAa,CAAC,SAAS,IAAI,eAAe,CAAC,SAAS,IAAI,EAAE,CAAC;QAClE,MAAM,SAAS,GAA6B;YAC1C,QAAQ,EAAE,CAAC;YACX,IAAI,EAAE,CAAC;YACP,MAAM,EAAE,CAAC;YACT,GAAG,EAAE,CAAC;SACP,CAAC;QAEF,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;YAC3B,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxB,cAAc,IAAI,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAED,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,GAAG,cAAc,CAAC;YACxC,SAAS;YACT,cAAc,EAAE,UAAU,CAAC,MAAM;YACjC,SAAS;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,UAAuB,EACvB,WAAmB;QAEnB,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAEzC,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,6BAA6B,CAAC,CAAC;QACnE,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAgB;gBAC1B,EAAE,EAAE,UAAU,EAAE;gBAChB,WAAW,EAAE,UAAU,CAAC,QAAQ,CAAC;qBAC9B,MAAM,CAAC,WAAW,CAAC;qBACnB,MAAM,CAAC,KAAK,CAAC;qBACb,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;gBACf,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YAEF,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEzB,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;YAC3D,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;YACnC,MAAM,WAAW,GACf,IAAI,CAAC,aAAa,CAAC,WAAW,IAAI,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC;YACrE,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YACpD,KAAK,CAAC,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;gBAAS,CAAC;YACT,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,WAAW,CACf,KAAsB,EACtB,UAAuB;QAEvB,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAE9C,MAAM,SAAS,GACb,KAAK,CAAC,KAAK,KAAK,WAAW;YACzB,CAAC,CAAC,cAAc;YAChB,CAAC,CAAC,KAAK,CAAC,KAAK,KAAK,WAAW;gBAC3B,CAAC,CAAC,cAAc;gBAChB,CAAC,CAAC,cAAc,CAAC;QACvB,OAAO,CAAC,GAAG,CACT,KAAK,CAAC,IAAI,CACR,oCAAoC,KAAK,CAAC,KAAK,SAAS,SAAS,IAAI,CACtE,CACF,CAAC;QAEF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,MAAM,aAAa,GAAe,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QACxE,MAAM,KAAK,GAA6B;YACtC,QAAQ,EAAE,cAAc;YACxB,IAAI,EAAE,cAAc;YACpB,MAAM,EAAE,cAAc;YACtB,GAAG,EAAE,cAAc;SACpB,CAAC;QAEF,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,GAAG,CAAC,CAAC;YAC3D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAEjC,OAAO,CAAC,GAAG,CACT,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,KAAK,KAAK,CAAC,MAAM,IAAI,CACpE,CAAC;YACF,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;gBACtB,MAAM,WAAW,GACf,CAAC,CAAC,gBAAgB,KAAK,SAAS;oBAC9B,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,gBAAgB,IAAI,CAAC;oBACxC,CAAC,CAAC,EAAE,CAAC;gBACT,OAAO,CAAC,GAAG,CACT,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,WAAW,EAAE,CAAC,CACpE,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC3C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;gBACtD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;gBACnD,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;oBAClB,OAAO,CAAC,GAAG,CACT,KAAK,CAAC,IAAI,CACR,yDAAyD,CAAC,CAAC,EAAE,EAAE,CAChE,CACF,CAAC;gBACJ,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GACZ,KAAK,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC/D,OAAO,CAAC,GAAG,CACT,GAAG,QAAQ,eAAe,KAAK,CAAC,SAAS,IAAI,EAAE,aAAa,KAAK,CAAC,KAAK,cAAc,UAAU,CAAC,MAAM,IAAI,CAC3G,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -1,7 +1,24 @@
|
|
|
1
|
-
import type {
|
|
1
|
+
import type { AnalysisConfig, Convention, ConventionAuditOptions, ConventionConfig, ModelTier, NamingConfig, StructureConfig, Violation } from "../../config/types.js";
|
|
2
|
+
export interface SpinnerLike {
|
|
3
|
+
text: (msg: string) => void;
|
|
4
|
+
succeed?: (msg?: string) => void;
|
|
5
|
+
}
|
|
2
6
|
export declare class ViolationDetector {
|
|
3
7
|
private projectRoot;
|
|
4
|
-
|
|
8
|
+
private analysisConfig;
|
|
9
|
+
private structureConfig?;
|
|
10
|
+
private namingConfig?;
|
|
11
|
+
private modelTier?;
|
|
12
|
+
private conventionConfig?;
|
|
13
|
+
private spinner?;
|
|
14
|
+
constructor(projectRoot: string, options?: {
|
|
15
|
+
analysisConfig?: AnalysisConfig;
|
|
16
|
+
structureConfig?: StructureConfig;
|
|
17
|
+
namingConfig?: NamingConfig;
|
|
18
|
+
modelTier?: ModelTier;
|
|
19
|
+
conventionConfig?: ConventionConfig;
|
|
20
|
+
spinner?: SpinnerLike;
|
|
21
|
+
});
|
|
5
22
|
detect(conventions: Convention[], options: ConventionAuditOptions): Promise<Violation[]>;
|
|
6
23
|
private persistViolations;
|
|
7
24
|
private getTargetFiles;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ViolationDetector.d.ts","sourceRoot":"","sources":["../../../src/core/scorer/ViolationDetector.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,
|
|
1
|
+
{"version":3,"file":"ViolationDetector.d.ts","sourceRoot":"","sources":["../../../src/core/scorer/ViolationDetector.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,cAAc,EACd,UAAU,EACV,sBAAsB,EACtB,gBAAgB,EAChB,SAAS,EACT,YAAY,EACZ,eAAe,EACf,SAAS,EACV,MAAM,uBAAuB,CAAC;AAK/B,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5B,OAAO,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;CAClC;AAED,qBAAa,iBAAiB;IAS1B,OAAO,CAAC,WAAW;IARrB,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,eAAe,CAAC,CAAkB;IAC1C,OAAO,CAAC,YAAY,CAAC,CAAe;IACpC,OAAO,CAAC,SAAS,CAAC,CAAY;IAC9B,OAAO,CAAC,gBAAgB,CAAC,CAAmB;IAC5C,OAAO,CAAC,OAAO,CAAC,CAAc;gBAGpB,WAAW,EAAE,MAAM,EAC3B,OAAO,CAAC,EAAE;QACR,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC,eAAe,CAAC,EAAE,eAAe,CAAC;QAClC,YAAY,CAAC,EAAE,YAAY,CAAC;QAC5B,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;QACpC,OAAO,CAAC,EAAE,WAAW,CAAC;KACvB;IAUG,MAAM,CACV,WAAW,EAAE,UAAU,EAAE,EACzB,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,SAAS,EAAE,CAAC;YAqJT,iBAAiB;YAMjB,cAAc;YA2Bd,oBAAoB;YA2BpB,SAAS;YA0BT,cAAc;IA8B5B,OAAO,CAAC,eAAe;IAOvB,OAAO,CAAC,uBAAuB;IAiB/B,OAAO,CAAC,UAAU;IAQlB,OAAO,CAAC,WAAW;IAYnB,OAAO,CAAC,YAAY;CAOrB"}
|
|
@@ -1,20 +1,37 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
1
|
+
import { readFile } from "node:fs/promises";
|
|
2
|
+
import { basename, extname, join } from "node:path";
|
|
3
|
+
import { DEFAULT_ANALYSIS } from "../../config/defaults.js";
|
|
3
4
|
export class ViolationDetector {
|
|
4
5
|
projectRoot;
|
|
5
|
-
|
|
6
|
+
analysisConfig;
|
|
7
|
+
structureConfig;
|
|
8
|
+
namingConfig;
|
|
9
|
+
modelTier;
|
|
10
|
+
conventionConfig;
|
|
11
|
+
spinner;
|
|
12
|
+
constructor(projectRoot, options) {
|
|
6
13
|
this.projectRoot = projectRoot;
|
|
14
|
+
this.analysisConfig = options?.analysisConfig ?? DEFAULT_ANALYSIS;
|
|
15
|
+
this.structureConfig = options?.structureConfig;
|
|
16
|
+
this.namingConfig = options?.namingConfig;
|
|
17
|
+
this.modelTier = options?.modelTier;
|
|
18
|
+
this.conventionConfig = options?.conventionConfig;
|
|
19
|
+
this.spinner = options?.spinner;
|
|
7
20
|
}
|
|
8
21
|
async detect(conventions, options) {
|
|
9
22
|
const confirmed = conventions.filter((c) => c.confirmed);
|
|
10
|
-
|
|
11
|
-
return [];
|
|
23
|
+
this.spinner?.text("Finding files to audit...");
|
|
12
24
|
const files = await this.getTargetFiles(options);
|
|
13
25
|
if (files.length === 0)
|
|
14
26
|
return [];
|
|
27
|
+
this.spinner?.text("Parsing files and checking conventions...");
|
|
28
|
+
const customRuleStore = new (await import("../../storage/CustomRuleStore.js")).CustomRuleStore(this.projectRoot);
|
|
29
|
+
const customRules = (await customRuleStore.exists()) ? await customRuleStore.load() : [];
|
|
30
|
+
if (confirmed.length === 0 && customRules.length === 0)
|
|
31
|
+
return [];
|
|
15
32
|
const [visitors, { randomUUID }] = await Promise.all([
|
|
16
|
-
this.createVisitors(),
|
|
17
|
-
import(
|
|
33
|
+
this.createVisitors(files),
|
|
34
|
+
import("node:crypto"),
|
|
18
35
|
]);
|
|
19
36
|
const conventionMap = new Map();
|
|
20
37
|
for (const c of confirmed) {
|
|
@@ -23,6 +40,38 @@ export class ViolationDetector {
|
|
|
23
40
|
const violations = [];
|
|
24
41
|
for (const filePath of files) {
|
|
25
42
|
const parsed = await this.parseFile(filePath);
|
|
43
|
+
for (const rule of customRules) {
|
|
44
|
+
try {
|
|
45
|
+
let target;
|
|
46
|
+
if (rule.scope === "filename") {
|
|
47
|
+
target = basename(filePath);
|
|
48
|
+
}
|
|
49
|
+
else if (rule.scope === "path") {
|
|
50
|
+
target = filePath;
|
|
51
|
+
}
|
|
52
|
+
else {
|
|
53
|
+
if (!parsed)
|
|
54
|
+
continue;
|
|
55
|
+
target = parsed.code;
|
|
56
|
+
}
|
|
57
|
+
const re = new RegExp(rule.pattern);
|
|
58
|
+
if (re.test(target)) {
|
|
59
|
+
violations.push({
|
|
60
|
+
id: randomUUID(),
|
|
61
|
+
conventionId: `custom:${rule.id}`,
|
|
62
|
+
severity: rule.severity ?? "medium",
|
|
63
|
+
file: filePath,
|
|
64
|
+
message: rule.message,
|
|
65
|
+
expected: rule.message,
|
|
66
|
+
found: target.slice(0, 80),
|
|
67
|
+
autoFixable: false,
|
|
68
|
+
});
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
catch {
|
|
72
|
+
// skip invalid rule or file
|
|
73
|
+
}
|
|
74
|
+
}
|
|
26
75
|
if (!parsed)
|
|
27
76
|
continue;
|
|
28
77
|
for (const visitor of visitors) {
|
|
@@ -35,11 +84,27 @@ export class ViolationDetector {
|
|
|
35
84
|
const convention = conventionMap.get(key);
|
|
36
85
|
if (!convention)
|
|
37
86
|
continue;
|
|
38
|
-
const isNamingFile = finding.dimension ===
|
|
87
|
+
const isNamingFile = finding.dimension === "naming" && finding.total === 1;
|
|
39
88
|
const ext = extname(filePath);
|
|
40
|
-
const
|
|
41
|
-
|
|
42
|
-
|
|
89
|
+
const isClassNamesPattern = finding.pattern === "Class names use PascalCase";
|
|
90
|
+
let expectedName = null;
|
|
91
|
+
let foundName;
|
|
92
|
+
let autoFixable = false;
|
|
93
|
+
if (isClassNamesPattern && finding.exportedName && finding.foundValue) {
|
|
94
|
+
expectedName = finding.exportedName;
|
|
95
|
+
foundName = finding.foundValue;
|
|
96
|
+
}
|
|
97
|
+
else if (finding.exportedName && isNamingFile) {
|
|
98
|
+
expectedName = finding.exportedName + ext;
|
|
99
|
+
foundName = basename(filePath);
|
|
100
|
+
autoFixable = !isClassNamesPattern;
|
|
101
|
+
}
|
|
102
|
+
else if (isNamingFile) {
|
|
103
|
+
expectedName = this.computeExpectedFilename(basename(filePath, ext), ext, finding.pattern);
|
|
104
|
+
foundName = basename(filePath);
|
|
105
|
+
autoFixable = expectedName !== null;
|
|
106
|
+
}
|
|
107
|
+
const fileScorePercent = finding.total > 0 ? Math.round((finding.count / finding.total) * 100) : undefined;
|
|
43
108
|
violations.push({
|
|
44
109
|
id: randomUUID(),
|
|
45
110
|
conventionId: convention.id,
|
|
@@ -47,8 +112,9 @@ export class ViolationDetector {
|
|
|
47
112
|
file: filePath,
|
|
48
113
|
message: `Violates: ${finding.pattern}`,
|
|
49
114
|
expected: expectedName ?? finding.pattern,
|
|
50
|
-
found: isNamingFile ? basename(filePath) : this.describeFinding(finding, filePath),
|
|
51
|
-
autoFixable
|
|
115
|
+
found: foundName ?? (isNamingFile ? basename(filePath) : this.describeFinding(finding, filePath)),
|
|
116
|
+
autoFixable,
|
|
117
|
+
fileScorePercent,
|
|
52
118
|
});
|
|
53
119
|
}
|
|
54
120
|
}
|
|
@@ -57,39 +123,81 @@ export class ViolationDetector {
|
|
|
57
123
|
}
|
|
58
124
|
}
|
|
59
125
|
}
|
|
126
|
+
const runSemantic = this.modelTier && files.length >= 3;
|
|
127
|
+
const semanticConvention = confirmed.find((c) => c.dimension === 'naming' && c.rule === 'Export semantically fits file context');
|
|
128
|
+
const semanticEnabled = semanticConvention || this.conventionConfig?.enableSemanticCoherence;
|
|
129
|
+
if (runSemantic && semanticEnabled) {
|
|
130
|
+
try {
|
|
131
|
+
this.spinner?.text("Running semantic coherence (HF)...");
|
|
132
|
+
const { detectSemanticViolations } = await import("../analyzer/SemanticCoherencePhase.js");
|
|
133
|
+
const convList = semanticConvention ? confirmed : [...confirmed, { id: 'naming-export-semantically-fits-file-context', dimension: 'naming', rule: 'Export semantically fits file context', confidence: 100, confirmed: true, examples: [], severity: 'medium' }];
|
|
134
|
+
const spinner = this.spinner;
|
|
135
|
+
const onProgress = spinner ? (msg) => spinner.text(msg) : undefined;
|
|
136
|
+
const semanticViolations = await detectSemanticViolations(this.projectRoot, files, convList, this.modelTier, {
|
|
137
|
+
semanticFitThreshold: this.conventionConfig?.semanticFitThreshold,
|
|
138
|
+
useSentenceSimilarityFallback: this.conventionConfig?.useSentenceSimilarityFallback,
|
|
139
|
+
onProgress,
|
|
140
|
+
});
|
|
141
|
+
violations.push(...semanticViolations);
|
|
142
|
+
this.spinner?.text(`Semantic coherence: ${semanticViolations.length} violation(s)`);
|
|
143
|
+
}
|
|
144
|
+
catch (err) {
|
|
145
|
+
const chalk = (await import('chalk')).default;
|
|
146
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
147
|
+
this.spinner?.text(chalk.yellow(`Semantic coherence skipped (${msg})`));
|
|
148
|
+
}
|
|
149
|
+
}
|
|
60
150
|
await this.persistViolations(violations);
|
|
61
151
|
return violations;
|
|
62
152
|
}
|
|
63
153
|
async persistViolations(violations) {
|
|
64
|
-
const { ViolationStore } = await import(
|
|
154
|
+
const { ViolationStore } = await import("../../storage/ViolationStore.js");
|
|
65
155
|
const store = new ViolationStore(this.projectRoot);
|
|
66
156
|
await store.save(violations);
|
|
67
157
|
}
|
|
68
158
|
async getTargetFiles(options) {
|
|
69
|
-
const { globby } = await import(
|
|
70
|
-
const
|
|
159
|
+
const { globby } = await import("globby");
|
|
160
|
+
const include = this.analysisConfig.include ?? DEFAULT_ANALYSIS.include;
|
|
161
|
+
const ignore = this.analysisConfig.ignore ?? DEFAULT_ANALYSIS.ignore;
|
|
162
|
+
let allFiles = await globby(include ?? ['**/*.{ts,tsx,js,jsx,mjs,cjs}'], {
|
|
71
163
|
cwd: this.projectRoot,
|
|
72
164
|
gitignore: true,
|
|
73
|
-
ignore
|
|
165
|
+
ignore,
|
|
74
166
|
absolute: false,
|
|
75
167
|
});
|
|
76
|
-
if (
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
168
|
+
if (options.filePaths?.length) {
|
|
169
|
+
const pathSet = new Set(options.filePaths);
|
|
170
|
+
allFiles = allFiles.filter((f) => pathSet.has(f));
|
|
171
|
+
}
|
|
172
|
+
else if (options.since) {
|
|
173
|
+
const changedFiles = await this.getChangedFilesSince(options.since);
|
|
174
|
+
if (changedFiles.length === 0)
|
|
175
|
+
return [];
|
|
176
|
+
const changedSet = new Set(changedFiles);
|
|
177
|
+
allFiles = allFiles.filter((f) => changedSet.has(f));
|
|
178
|
+
}
|
|
179
|
+
return allFiles;
|
|
83
180
|
}
|
|
84
181
|
async getChangedFilesSince(since) {
|
|
85
182
|
try {
|
|
86
|
-
const { simpleGit } = await import(
|
|
183
|
+
const { simpleGit } = await import("simple-git");
|
|
87
184
|
const git = simpleGit(this.projectRoot);
|
|
88
185
|
const isRepo = await git.checkIsRepo();
|
|
89
186
|
if (!isRepo)
|
|
90
187
|
return [];
|
|
91
|
-
const result = await git.raw([
|
|
92
|
-
|
|
188
|
+
const result = await git.raw([
|
|
189
|
+
"log",
|
|
190
|
+
"--since",
|
|
191
|
+
since,
|
|
192
|
+
"--name-only",
|
|
193
|
+
"--format=",
|
|
194
|
+
]);
|
|
195
|
+
return [
|
|
196
|
+
...new Set(result
|
|
197
|
+
.split("\n")
|
|
198
|
+
.map((l) => l.trim())
|
|
199
|
+
.filter(Boolean)),
|
|
200
|
+
];
|
|
93
201
|
}
|
|
94
202
|
catch {
|
|
95
203
|
return [];
|
|
@@ -98,18 +206,18 @@ export class ViolationDetector {
|
|
|
98
206
|
async parseFile(filePath) {
|
|
99
207
|
try {
|
|
100
208
|
const absolutePath = join(this.projectRoot, filePath);
|
|
101
|
-
const code = await readFile(absolutePath,
|
|
102
|
-
const { parse } = await import(
|
|
209
|
+
const code = await readFile(absolutePath, "utf-8");
|
|
210
|
+
const { parse } = await import("@babel/parser");
|
|
103
211
|
const ast = parse(code, {
|
|
104
|
-
sourceType:
|
|
212
|
+
sourceType: "module",
|
|
105
213
|
plugins: [
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
214
|
+
"typescript",
|
|
215
|
+
"jsx",
|
|
216
|
+
"decorators-legacy",
|
|
217
|
+
"classProperties",
|
|
218
|
+
"optionalChaining",
|
|
219
|
+
"nullishCoalescingOperator",
|
|
220
|
+
"dynamicImport",
|
|
113
221
|
],
|
|
114
222
|
errorRecovery: true,
|
|
115
223
|
});
|
|
@@ -119,20 +227,24 @@ export class ViolationDetector {
|
|
|
119
227
|
return null;
|
|
120
228
|
}
|
|
121
229
|
}
|
|
122
|
-
async createVisitors() {
|
|
123
|
-
const [{ NamingVisitor }, { ComponentVisitor }, { TestVisitor }, { ErrorHandlingVisitor }, { ImportVisitor },] = await Promise.all([
|
|
124
|
-
import(
|
|
125
|
-
import(
|
|
126
|
-
import(
|
|
127
|
-
import(
|
|
128
|
-
import(
|
|
230
|
+
async createVisitors(filePaths) {
|
|
231
|
+
const [{ NamingVisitor }, { ComponentVisitor }, { TestVisitor }, { ErrorHandlingVisitor }, { ImportVisitor }, { SchemaVisitor }, { StructureVisitor },] = await Promise.all([
|
|
232
|
+
import("../analyzer/visitors/NamingVisitor.js"),
|
|
233
|
+
import("../analyzer/visitors/ComponentVisitor.js"),
|
|
234
|
+
import("../analyzer/visitors/TestVisitor.js"),
|
|
235
|
+
import("../analyzer/visitors/ErrorHandlingVisitor.js"),
|
|
236
|
+
import("../analyzer/visitors/ImportVisitor.js"),
|
|
237
|
+
import("../analyzer/visitors/SchemaVisitor.js"),
|
|
238
|
+
import("../analyzer/visitors/StructureVisitor.js"),
|
|
129
239
|
]);
|
|
130
240
|
return [
|
|
131
|
-
new NamingVisitor(),
|
|
241
|
+
new NamingVisitor({ namingConfig: this.namingConfig }),
|
|
132
242
|
new ComponentVisitor(),
|
|
133
243
|
new TestVisitor(),
|
|
134
244
|
new ErrorHandlingVisitor(),
|
|
135
|
-
new ImportVisitor(),
|
|
245
|
+
new ImportVisitor({ namingConfig: this.namingConfig }),
|
|
246
|
+
new SchemaVisitor(),
|
|
247
|
+
new StructureVisitor(filePaths, this.structureConfig),
|
|
136
248
|
];
|
|
137
249
|
}
|
|
138
250
|
describeFinding(finding, filePath) {
|
|
@@ -143,11 +255,11 @@ export class ViolationDetector {
|
|
|
143
255
|
}
|
|
144
256
|
computeExpectedFilename(stem, ext, pattern) {
|
|
145
257
|
const lower = pattern.toLowerCase();
|
|
146
|
-
if (lower.includes(
|
|
258
|
+
if (lower.includes("pascalcase")) {
|
|
147
259
|
const converted = this.toPascalCase(stem);
|
|
148
260
|
return converted !== stem ? converted + ext : null;
|
|
149
261
|
}
|
|
150
|
-
if (lower.includes(
|
|
262
|
+
if (lower.includes("camelcase")) {
|
|
151
263
|
const converted = this.toCamelCase(stem);
|
|
152
264
|
return converted !== stem ? converted + ext : null;
|
|
153
265
|
}
|
|
@@ -155,8 +267,8 @@ export class ViolationDetector {
|
|
|
155
267
|
}
|
|
156
268
|
splitWords(str) {
|
|
157
269
|
return str
|
|
158
|
-
.replace(/([a-z0-9])([A-Z])/g,
|
|
159
|
-
.replace(/([A-Z]+)([A-Z][a-z])/g,
|
|
270
|
+
.replace(/([a-z0-9])([A-Z])/g, "$1_$2")
|
|
271
|
+
.replace(/([A-Z]+)([A-Z][a-z])/g, "$1_$2")
|
|
160
272
|
.split(/[-_.\s]+/)
|
|
161
273
|
.filter(Boolean);
|
|
162
274
|
}
|
|
@@ -165,14 +277,18 @@ export class ViolationDetector {
|
|
|
165
277
|
if (words.length === 0)
|
|
166
278
|
return str;
|
|
167
279
|
return words
|
|
168
|
-
.map((w, i) => i === 0
|
|
169
|
-
.
|
|
280
|
+
.map((w, i) => i === 0
|
|
281
|
+
? w.toLowerCase()
|
|
282
|
+
: w.charAt(0).toUpperCase() + w.slice(1).toLowerCase())
|
|
283
|
+
.join("");
|
|
170
284
|
}
|
|
171
285
|
toPascalCase(str) {
|
|
172
286
|
const words = this.splitWords(str);
|
|
173
287
|
if (words.length === 0)
|
|
174
288
|
return str;
|
|
175
|
-
return words
|
|
289
|
+
return words
|
|
290
|
+
.map((w) => w.charAt(0).toUpperCase() + w.slice(1).toLowerCase())
|
|
291
|
+
.join("");
|
|
176
292
|
}
|
|
177
293
|
}
|
|
178
294
|
//# sourceMappingURL=ViolationDetector.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ViolationDetector.js","sourceRoot":"","sources":["../../../src/core/scorer/ViolationDetector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAK5C,MAAM,OAAO,iBAAiB;IACR;IAApB,YAAoB,WAAmB;QAAnB,gBAAW,GAAX,WAAW,CAAQ;IAAG,CAAC;IAE3C,KAAK,CAAC,MAAM,CAAC,WAAyB,EAAE,OAA+B;QACrE,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACzD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,MAAM,CAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACnD,IAAI,CAAC,cAAc,EAAE;YACrB,MAAM,CAAC,aAAa,CAAC;SACtB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,GAAG,EAAsB,CAAC;QACpD,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1B,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,UAAU,GAAgB,EAAE,CAAC;QAEnC,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9C,IAAI,CAAC,MAAM;gBAAE,SAAS;YAEtB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBACvC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;wBAC/B,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK;4BAAE,SAAS;wBAE7C,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;wBACtD,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wBAC1C,IAAI,CAAC,UAAU;4BAAE,SAAS;wBAE1B,MAAM,YAAY,GAAG,OAAO,CAAC,SAAS,KAAK,QAAQ,IAAI,OAAO,CAAC,KAAK,KAAK,CAAC,CAAC;wBAC3E,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;wBAC9B,MAAM,YAAY,GAAG,YAAY;4BAC/B,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,OAAO,CAAC;4BAC7E,CAAC,CAAC,IAAI,CAAC;wBAET,UAAU,CAAC,IAAI,CAAC;4BACd,EAAE,EAAE,UAAU,EAAE;4BAChB,YAAY,EAAE,UAAU,CAAC,EAAE;4BAC3B,QAAQ,EAAE,UAAU,CAAC,QAAQ;4BAC7B,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,aAAa,OAAO,CAAC,OAAO,EAAE;4BACvC,QAAQ,EAAE,YAAY,IAAI,OAAO,CAAC,OAAO;4BACzC,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,QAAQ,CAAC;4BAClF,WAAW,EAAE,YAAY,KAAK,IAAI;yBACnC,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,qDAAqD;gBACvD,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACzC,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,UAAuB;QACrD,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,iCAAiC,CAAC,CAAC;QAC3E,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnD,MAAM,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/B,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,OAA+B;QAC1D,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,CAAC,8BAA8B,CAAC,EAAE;YAC9D,GAAG,EAAE,IAAI,CAAC,WAAW;YACrB,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,CAAC,oBAAoB,EAAE,YAAY,EAAE,aAAa,EAAE,iBAAiB,CAAC;YAC9E,QAAQ,EAAE,KAAK;SAChB,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,KAAK;YAAE,OAAO,QAAQ,CAAC;QAEpC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACpE,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEzC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC;QACzC,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACnD,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,KAAa;QAC9C,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,CAAC;YACjD,MAAM,GAAG,GAAG,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM;gBAAE,OAAO,EAAE,CAAC;YAEvB,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC,CAAC;YACpF,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC/E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,QAAgB;QACtC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;YAEhD,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,EAAE;gBACtB,UAAU,EAAE,QAAQ;gBACpB,OAAO,EAAE;oBACP,YAAY;oBACZ,KAAK;oBACL,mBAAmB;oBACnB,iBAAiB;oBACjB,kBAAkB;oBAClB,2BAA2B;oBAC3B,eAAe;iBAChB;gBACD,aAAa,EAAE,IAAI;aACpB,CAAC,CAAC;YAEH,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,MAAM,CACJ,EAAE,aAAa,EAAE,EACjB,EAAE,gBAAgB,EAAE,EACpB,EAAE,WAAW,EAAE,EACf,EAAE,oBAAoB,EAAE,EACxB,EAAE,aAAa,EAAE,EAClB,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpB,MAAM,CAAC,uCAAuC,CAAC;YAC/C,MAAM,CAAC,0CAA0C,CAAC;YAClD,MAAM,CAAC,qCAAqC,CAAC;YAC7C,MAAM,CAAC,8CAA8C,CAAC;YACtD,MAAM,CAAC,uCAAuC,CAAC;SAChD,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,aAAa,EAAE;YACnB,IAAI,gBAAgB,EAAE;YACtB,IAAI,WAAW,EAAE;YACjB,IAAI,oBAAoB,EAAE;YAC1B,IAAI,aAAa,EAAE;SACpB,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,OAAuB,EAAE,QAAgB;QAC/D,IAAI,OAAO,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,GAAG,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,aAAa,CAAC;IACxD,CAAC;IAEO,uBAAuB,CAAC,IAAY,EAAE,GAAW,EAAE,OAAe;QACxE,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC1C,OAAO,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QACrD,CAAC;QACD,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAChC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACzC,OAAO,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,UAAU,CAAC,GAAW;QAC5B,OAAO,GAAG;aACP,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC;aACtC,OAAO,CAAC,uBAAuB,EAAE,OAAO,CAAC;aACzC,KAAK,CAAC,UAAU,CAAC;aACjB,MAAM,CAAC,OAAO,CAAC,CAAC;IACrB,CAAC;IAEO,WAAW,CAAC,GAAW;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,GAAG,CAAC;QACnC,OAAO,KAAK;aACT,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACZ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CACjF;aACA,IAAI,CAAC,EAAE,CAAC,CAAC;IACd,CAAC;IAEO,YAAY,CAAC,GAAW;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,GAAG,CAAC;QACnC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzF,CAAC;CACF"}
|
|
1
|
+
{"version":3,"file":"ViolationDetector.js","sourceRoot":"","sources":["../../../src/core/scorer/ViolationDetector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAWpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAS5D,MAAM,OAAO,iBAAiB;IASlB;IARF,cAAc,CAAiB;IAC/B,eAAe,CAAmB;IAClC,YAAY,CAAgB;IAC5B,SAAS,CAAa;IACtB,gBAAgB,CAAoB;IACpC,OAAO,CAAe;IAE9B,YACU,WAAmB,EAC3B,OAOC;QARO,gBAAW,GAAX,WAAW,CAAQ;QAU3B,IAAI,CAAC,cAAc,GAAG,OAAO,EAAE,cAAc,IAAI,gBAAgB,CAAC;QAClE,IAAI,CAAC,eAAe,GAAG,OAAO,EAAE,eAAe,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,OAAO,EAAE,YAAY,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;QACpC,IAAI,CAAC,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,CAAC;QAClD,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,MAAM,CACV,WAAyB,EACzB,OAA+B;QAE/B,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACzD,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,2CAA2C,CAAC,CAAC;QAEhE,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,MAAM,CAAC,kCAAkC,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACjH,MAAM,WAAW,GAAG,CAAC,MAAM,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAEzF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElE,MAAM,CAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACnD,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC;YAC1B,MAAM,CAAC,aAAa,CAAC;SACtB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,GAAG,EAAsB,CAAC;QACpD,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1B,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,UAAU,GAAgB,EAAE,CAAC;QAEnC,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAE9C,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,IAAI,MAAc,CAAC;oBACnB,IAAI,IAAI,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;wBAC9B,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBAC9B,CAAC;yBAAM,IAAI,IAAI,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;wBACjC,MAAM,GAAG,QAAQ,CAAC;oBACpB,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,MAAM;4BAAE,SAAS;wBACtB,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC;oBACvB,CAAC;oBACD,MAAM,EAAE,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACpC,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;wBACpB,UAAU,CAAC,IAAI,CAAC;4BACd,EAAE,EAAE,UAAU,EAAE;4BAChB,YAAY,EAAE,UAAU,IAAI,CAAC,EAAE,EAAE;4BACjC,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,QAAQ;4BACnC,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,IAAI,CAAC,OAAO;4BACrB,QAAQ,EAAE,IAAI,CAAC,OAAO;4BACtB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;4BAC1B,WAAW,EAAE,KAAK;yBACnB,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,4BAA4B;gBAC9B,CAAC;YACH,CAAC;YAED,IAAI,CAAC,MAAM;gBAAE,SAAS;YAEtB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBACvC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;wBAC/B,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK;4BAAE,SAAS;wBAE7C,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;wBACtD,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wBAC1C,IAAI,CAAC,UAAU;4BAAE,SAAS;wBAE1B,MAAM,YAAY,GAChB,OAAO,CAAC,SAAS,KAAK,QAAQ,IAAI,OAAO,CAAC,KAAK,KAAK,CAAC,CAAC;wBACxD,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;wBAC9B,MAAM,mBAAmB,GAAG,OAAO,CAAC,OAAO,KAAK,4BAA4B,CAAC;wBAC7E,IAAI,YAAY,GAAkB,IAAI,CAAC;wBACvC,IAAI,SAA6B,CAAC;wBAClC,IAAI,WAAW,GAAG,KAAK,CAAC;wBAExB,IAAI,mBAAmB,IAAI,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;4BACtE,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;4BACpC,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC;wBACjC,CAAC;6BAAM,IAAI,OAAO,CAAC,YAAY,IAAI,YAAY,EAAE,CAAC;4BAChD,YAAY,GAAG,OAAO,CAAC,YAAY,GAAG,GAAG,CAAC;4BAC1C,SAAS,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;4BAC/B,WAAW,GAAG,CAAC,mBAAmB,CAAC;wBACrC,CAAC;6BAAM,IAAI,YAAY,EAAE,CAAC;4BACxB,YAAY,GAAG,IAAI,CAAC,uBAAuB,CACzC,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,EACvB,GAAG,EACH,OAAO,CAAC,OAAO,CAChB,CAAC;4BACF,SAAS,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;4BAC/B,WAAW,GAAG,YAAY,KAAK,IAAI,CAAC;wBACtC,CAAC;wBAED,MAAM,gBAAgB,GACpB,OAAO,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAEpF,UAAU,CAAC,IAAI,CAAC;4BACd,EAAE,EAAE,UAAU,EAAE;4BAChB,YAAY,EAAE,UAAU,CAAC,EAAE;4BAC3B,QAAQ,EAAE,UAAU,CAAC,QAAQ;4BAC7B,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,aAAa,OAAO,CAAC,OAAO,EAAE;4BACvC,QAAQ,EAAE,YAAY,IAAI,OAAO,CAAC,OAAO;4BACzC,KAAK,EAAE,SAAS,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;4BACjG,WAAW;4BACX,gBAAgB;yBACjB,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,qDAAqD;gBACvD,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC;QACxD,MAAM,kBAAkB,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,KAAK,uCAAuC,CAAC,CAAC;QACjI,MAAM,eAAe,GAAG,kBAAkB,IAAI,IAAI,CAAC,gBAAgB,EAAE,uBAAuB,CAAC;QAC7F,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,IAAI,CAAC;gBACH,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,oCAAoC,CAAC,CAAC;gBACzD,MAAM,EAAE,wBAAwB,EAAE,GAAG,MAAM,MAAM,CAAC,uCAAuC,CAAC,CAAC;gBAC3F,MAAM,QAAQ,GAAG,kBAAkB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,EAAE,EAAE,EAAE,EAAE,8CAA8C,EAAE,SAAS,EAAE,QAAiB,EAAE,IAAI,EAAE,uCAAuC,EAAE,UAAU,EAAE,GAAG,EAAE,SAAS,EAAE,IAAa,EAAE,QAAQ,EAAE,EAAc,EAAE,QAAQ,EAAE,QAAiB,EAAE,CAAC,CAAC;gBACxS,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;gBAC7B,MAAM,UAAU,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAAW,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC5E,MAAM,kBAAkB,GAAG,MAAM,wBAAwB,CACvD,IAAI,CAAC,WAAW,EAChB,KAAK,EACL,QAAQ,EACR,IAAI,CAAC,SAAsB,EAC3B;oBACE,oBAAoB,EAAE,IAAI,CAAC,gBAAgB,EAAE,oBAAoB;oBACjE,6BAA6B,EAAE,IAAI,CAAC,gBAAgB,EAAE,6BAA6B;oBACnF,UAAU;iBACX,CACF,CAAC;gBACF,UAAU,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,CAAC;gBACvC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,uBAAuB,kBAAkB,CAAC,MAAM,eAAe,CAAC,CAAC;YACtF,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC9C,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC7D,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,+BAA+B,GAAG,GAAG,CAAC,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACzC,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,UAAuB;QACrD,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,iCAAiC,CAAC,CAAC;QAC3E,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnD,MAAM,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/B,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,OAA+B;QAE/B,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,IAAI,gBAAgB,CAAC,OAAO,CAAC;QACxE,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,CAAC;QAErE,IAAI,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,IAAI,CAAC,8BAA8B,CAAC,EAAE;YACvE,GAAG,EAAE,IAAI,CAAC,WAAW;YACrB,SAAS,EAAE,IAAI;YACf,MAAM;YACN,QAAQ,EAAE,KAAK;SAChB,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC3C,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC;aAAM,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACzB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACpE,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YACzC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC;YACzC,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,KAAa;QAC9C,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,CAAC;YACjD,MAAM,GAAG,GAAG,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM;gBAAE,OAAO,EAAE,CAAC;YAEvB,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC;gBAC3B,KAAK;gBACL,SAAS;gBACT,KAAK;gBACL,aAAa;gBACb,WAAW;aACZ,CAAC,CAAC;YACH,OAAO;gBACL,GAAG,IAAI,GAAG,CACR,MAAM;qBACH,KAAK,CAAC,IAAI,CAAC;qBACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;qBACpB,MAAM,CAAC,OAAO,CAAC,CACnB;aACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,QAAgB;QACtC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;YAEhD,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,EAAE;gBACtB,UAAU,EAAE,QAAQ;gBACpB,OAAO,EAAE;oBACP,YAAY;oBACZ,KAAK;oBACL,mBAAmB;oBACnB,iBAAiB;oBACjB,kBAAkB;oBAClB,2BAA2B;oBAC3B,eAAe;iBAChB;gBACD,aAAa,EAAE,IAAI;aACpB,CAAC,CAAC;YAEH,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,SAAmB;QAC9C,MAAM,CACJ,EAAE,aAAa,EAAE,EACjB,EAAE,gBAAgB,EAAE,EACpB,EAAE,WAAW,EAAE,EACf,EAAE,oBAAoB,EAAE,EACxB,EAAE,aAAa,EAAE,EACjB,EAAE,aAAa,EAAE,EACjB,EAAE,gBAAgB,EAAE,EACrB,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpB,MAAM,CAAC,uCAAuC,CAAC;YAC/C,MAAM,CAAC,0CAA0C,CAAC;YAClD,MAAM,CAAC,qCAAqC,CAAC;YAC7C,MAAM,CAAC,8CAA8C,CAAC;YACtD,MAAM,CAAC,uCAAuC,CAAC;YAC/C,MAAM,CAAC,uCAAuC,CAAC;YAC/C,MAAM,CAAC,0CAA0C,CAAC;SACnD,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,aAAa,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC;YACtD,IAAI,gBAAgB,EAAE;YACtB,IAAI,WAAW,EAAE;YACjB,IAAI,oBAAoB,EAAE;YAC1B,IAAI,aAAa,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC;YACtD,IAAI,aAAa,EAAE;YACnB,IAAI,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,eAAe,CAAC;SACtD,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,OAAuB,EAAE,QAAgB;QAC/D,IAAI,OAAO,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,GAAG,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,aAAa,CAAC;IACxD,CAAC;IAEO,uBAAuB,CAC7B,IAAY,EACZ,GAAW,EACX,OAAe;QAEf,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC1C,OAAO,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QACrD,CAAC;QACD,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAChC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACzC,OAAO,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,UAAU,CAAC,GAAW;QAC5B,OAAO,GAAG;aACP,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC;aACtC,OAAO,CAAC,uBAAuB,EAAE,OAAO,CAAC;aACzC,KAAK,CAAC,UAAU,CAAC;aACjB,MAAM,CAAC,OAAO,CAAC,CAAC;IACrB,CAAC;IAEO,WAAW,CAAC,GAAW;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,GAAG,CAAC;QACnC,OAAO,KAAK;aACT,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACZ,CAAC,KAAK,CAAC;YACL,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE;YACjB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CACzD;aACA,IAAI,CAAC,EAAE,CAAC,CAAC;IACd,CAAC;IAEO,YAAY,CAAC,GAAW;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,GAAG,CAAC;QACnC,OAAO,KAAK;aACT,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;aAChE,IAAI,CAAC,EAAE,CAAC,CAAC;IACd,CAAC;CACF"}
|
|
@@ -1,10 +1,14 @@
|
|
|
1
1
|
import type { ModelTier } from '../config/types.js';
|
|
2
2
|
import { type ModelRole, type ModelSpec } from './ModelRegistry.js';
|
|
3
|
+
export interface ModelManagerOptions {
|
|
4
|
+
onLoaded?: (description: string) => void;
|
|
5
|
+
}
|
|
3
6
|
export declare class ModelManager {
|
|
4
7
|
private loadedModel;
|
|
5
8
|
private cacheDir;
|
|
6
9
|
private tier;
|
|
7
|
-
|
|
10
|
+
private onLoaded?;
|
|
11
|
+
constructor(tier: ModelTier, cacheDir?: string, opts?: ModelManagerOptions);
|
|
8
12
|
ensureCacheDir(): Promise<void>;
|
|
9
13
|
loadModel(role: ModelRole): Promise<unknown>;
|
|
10
14
|
disposeModel(): Promise<void>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ModelManager.d.ts","sourceRoot":"","sources":["../../src/models/ModelManager.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAgB,KAAK,SAAS,EAAE,KAAK,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAIlF,qBAAa,YAAY;IACvB,OAAO,CAAC,WAAW,CAAuD;IAC1E,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,IAAI,CAAY;
|
|
1
|
+
{"version":3,"file":"ModelManager.d.ts","sourceRoot":"","sources":["../../src/models/ModelManager.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAgB,KAAK,SAAS,EAAE,KAAK,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAIlF,MAAM,WAAW,mBAAmB;IAClC,QAAQ,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,KAAK,IAAI,CAAC;CAC1C;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,WAAW,CAAuD;IAC1E,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,IAAI,CAAY;IACxB,OAAO,CAAC,QAAQ,CAAC,CAAgC;gBAErC,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,mBAAmB;IAMpE,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAM/B,SAAS,CAAC,IAAI,EAAE,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC;IAmD5C,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAUnC,OAAO,CAAC,IAAI,EAAE,SAAS,GAAG,SAAS;IAInC,aAAa,IAAI,OAAO;IAIxB,cAAc,IAAI,SAAS,GAAG,IAAI;CAGnC"}
|
|
@@ -8,9 +8,11 @@ export class ModelManager {
|
|
|
8
8
|
loadedModel = null;
|
|
9
9
|
cacheDir;
|
|
10
10
|
tier;
|
|
11
|
-
|
|
11
|
+
onLoaded;
|
|
12
|
+
constructor(tier, cacheDir, opts) {
|
|
12
13
|
this.tier = tier;
|
|
13
14
|
this.cacheDir = cacheDir ?? process.env['CODEPLUG_MODEL_CACHE'] ?? DEFAULT_CACHE_DIR;
|
|
15
|
+
this.onLoaded = opts?.onLoaded;
|
|
14
16
|
}
|
|
15
17
|
async ensureCacheDir() {
|
|
16
18
|
if (!existsSync(this.cacheDir)) {
|
|
@@ -26,8 +28,12 @@ export class ModelManager {
|
|
|
26
28
|
}
|
|
27
29
|
const spec = getModelSpec(role, this.tier);
|
|
28
30
|
await this.ensureCacheDir();
|
|
29
|
-
const
|
|
30
|
-
|
|
31
|
+
const useCallback = !!this.onLoaded;
|
|
32
|
+
let spinner = null;
|
|
33
|
+
if (!useCallback) {
|
|
34
|
+
const ora = (await import('ora')).default;
|
|
35
|
+
spinner = ora(`Loading ${spec.description} (~${spec.sizeEstimateMb}MB)...`).start();
|
|
36
|
+
}
|
|
31
37
|
try {
|
|
32
38
|
const { pipeline, env } = await import('@huggingface/transformers');
|
|
33
39
|
env.cacheDir = this.cacheDir;
|
|
@@ -37,16 +43,24 @@ export class ModelManager {
|
|
|
37
43
|
summarizer: 'summarization',
|
|
38
44
|
extractor: 'question-answering',
|
|
39
45
|
ner: 'token-classification',
|
|
46
|
+
zeroShot: 'zero-shot-classification',
|
|
47
|
+
sentenceSimilarity: 'feature-extraction',
|
|
40
48
|
};
|
|
41
49
|
const pipe = await pipeline(taskMap[role], spec.huggingFaceId, {
|
|
42
50
|
dtype: spec.dtype,
|
|
43
51
|
});
|
|
44
52
|
this.loadedModel = { role, pipeline: pipe };
|
|
45
|
-
|
|
53
|
+
if (useCallback && this.onLoaded) {
|
|
54
|
+
this.onLoaded(spec.description);
|
|
55
|
+
}
|
|
56
|
+
else if (spinner) {
|
|
57
|
+
spinner.succeed(`Loaded ${spec.description}`);
|
|
58
|
+
}
|
|
46
59
|
return pipe;
|
|
47
60
|
}
|
|
48
61
|
catch (err) {
|
|
49
|
-
spinner
|
|
62
|
+
if (spinner)
|
|
63
|
+
spinner.fail(`Failed to load ${spec.description}`);
|
|
50
64
|
throw err;
|
|
51
65
|
}
|
|
52
66
|
}
|