@diffmind/cli 0.1.0
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/LICENSE +21 -0
- package/dist/formatters.test.js +71 -0
- package/dist/formatters.test.js.map +1 -0
- package/dist/index.js +334 -0
- package/dist/index.js.map +1 -0
- package/dist/worker.js +68 -0
- package/dist/worker.js.map +1 -0
- package/package.json +45 -0
package/LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2026 Diffmind Contributors
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const index_1 = require("./index");
|
|
4
|
+
// Mock chalk to return plain strings for easier testing
|
|
5
|
+
jest.mock("chalk", () => {
|
|
6
|
+
const createMock = () => {
|
|
7
|
+
const fn = (str) => str;
|
|
8
|
+
const proxy = new Proxy(fn, {
|
|
9
|
+
get: (_target, prop) => {
|
|
10
|
+
if (prop === "default")
|
|
11
|
+
return proxy;
|
|
12
|
+
return proxy;
|
|
13
|
+
},
|
|
14
|
+
});
|
|
15
|
+
return proxy;
|
|
16
|
+
};
|
|
17
|
+
return createMock();
|
|
18
|
+
});
|
|
19
|
+
jest.mock("ora", () => {
|
|
20
|
+
return jest.fn(() => ({
|
|
21
|
+
start: jest.fn().mockReturnThis(),
|
|
22
|
+
succeed: jest.fn().mockReturnThis(),
|
|
23
|
+
fail: jest.fn().mockReturnThis(),
|
|
24
|
+
}));
|
|
25
|
+
});
|
|
26
|
+
describe("cli formatters", () => {
|
|
27
|
+
const mockReport = [
|
|
28
|
+
{
|
|
29
|
+
file: "auth.ts",
|
|
30
|
+
line: 10,
|
|
31
|
+
severity: "high",
|
|
32
|
+
category: "security",
|
|
33
|
+
issue: "Hardcoded secret",
|
|
34
|
+
suggested_fix: "Use env",
|
|
35
|
+
},
|
|
36
|
+
];
|
|
37
|
+
describe("formatJson", () => {
|
|
38
|
+
it("should return formatted JSON string", () => {
|
|
39
|
+
const output = (0, index_1.formatJson)(mockReport);
|
|
40
|
+
expect(JSON.parse(output)).toEqual(mockReport);
|
|
41
|
+
});
|
|
42
|
+
});
|
|
43
|
+
describe("formatMarkdown", () => {
|
|
44
|
+
it("should return a summary message when no issues are found", () => {
|
|
45
|
+
const output = (0, index_1.formatMarkdown)([], "main");
|
|
46
|
+
expect(output).toContain("No issues found");
|
|
47
|
+
});
|
|
48
|
+
it("should return a formatted report when issues are found", () => {
|
|
49
|
+
const output = (0, index_1.formatMarkdown)(mockReport, "feat-branch");
|
|
50
|
+
expect(output).toContain("diffmind Code Review Report");
|
|
51
|
+
expect(output).toContain("Branch: feat-branch");
|
|
52
|
+
expect(output).toContain("auth.ts:10");
|
|
53
|
+
expect(output).toContain("Hardcoded secret");
|
|
54
|
+
expect(output).toContain("Fix: Use env");
|
|
55
|
+
});
|
|
56
|
+
});
|
|
57
|
+
describe("badges", () => {
|
|
58
|
+
it("should return correct severity badges", () => {
|
|
59
|
+
expect((0, index_1.severityBadge)("high")).toContain("HIGH");
|
|
60
|
+
expect((0, index_1.severityBadge)("medium")).toContain("MED");
|
|
61
|
+
expect((0, index_1.severityBadge)("low")).toContain("LOW");
|
|
62
|
+
});
|
|
63
|
+
it("should return correct category badges", () => {
|
|
64
|
+
expect((0, index_1.categoryBadge)("security")).toContain("SECURITY");
|
|
65
|
+
expect((0, index_1.categoryBadge)("quality")).toContain("QUALITY");
|
|
66
|
+
expect((0, index_1.categoryBadge)("performance")).toContain("PERF");
|
|
67
|
+
expect((0, index_1.categoryBadge)("maintainability")).toContain("MAINT");
|
|
68
|
+
});
|
|
69
|
+
});
|
|
70
|
+
});
|
|
71
|
+
//# sourceMappingURL=formatters.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"formatters.test.js","sourceRoot":"","sources":["../src/formatters.test.ts"],"names":[],"mappings":";;AAAA,mCAKiB;AAGjB,wDAAwD;AACxD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;IACtB,MAAM,UAAU,GAAG,GAAG,EAAE;QACtB,MAAM,EAAE,GAAG,CAAC,GAAW,EAAE,EAAE,CAAC,GAAG,CAAC;QAChC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,EAAE;YAC1B,GAAG,EAAE,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE;gBACrB,IAAI,IAAI,KAAK,SAAS;oBAAE,OAAO,KAAK,CAAC;gBACrC,OAAO,KAAK,CAAC;YACf,CAAC;SACF,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;IACF,OAAO,UAAU,EAAE,CAAC;AACtB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE;IACpB,OAAO,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;QACpB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;QACjC,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;QACnC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;KACjC,CAAC,CAAC,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,UAAU,GAAiB;QAC/B;YACE,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,MAAM;YAChB,QAAQ,EAAE,UAAU;YACpB,KAAK,EAAE,kBAAkB;YACzB,aAAa,EAAE,SAAS;SACzB;KACF,CAAC;IAEF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,MAAM,GAAG,IAAA,kBAAU,EAAC,UAAU,CAAC,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,MAAM,GAAG,IAAA,sBAAc,EAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,MAAM,GAAG,IAAA,sBAAc,EAAC,UAAU,EAAE,aAAa,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,IAAA,qBAAa,EAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,IAAA,qBAAa,EAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACjD,MAAM,CAAC,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,IAAA,qBAAa,EAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACxD,MAAM,CAAC,IAAA,qBAAa,EAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACtD,MAAM,CAAC,IAAA,qBAAa,EAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACvD,MAAM,CAAC,IAAA,qBAAa,EAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,334 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
"use strict";
|
|
3
|
+
/**
|
|
4
|
+
* diffmind CLI
|
|
5
|
+
*
|
|
6
|
+
* Local-first AI code review for your git diffs.
|
|
7
|
+
* Powered by Qwen2.5-Coder-3B running entirely on-device via WebAssembly.
|
|
8
|
+
*
|
|
9
|
+
* Usage:
|
|
10
|
+
* diffmind --branch main
|
|
11
|
+
* diffmind --branch develop --format json
|
|
12
|
+
* git diff main...HEAD | diffmind --stdin
|
|
13
|
+
*/
|
|
14
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
15
|
+
if (k2 === undefined) k2 = k;
|
|
16
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
17
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
18
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
19
|
+
}
|
|
20
|
+
Object.defineProperty(o, k2, desc);
|
|
21
|
+
}) : (function(o, m, k, k2) {
|
|
22
|
+
if (k2 === undefined) k2 = k;
|
|
23
|
+
o[k2] = m[k];
|
|
24
|
+
}));
|
|
25
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
26
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
27
|
+
}) : function(o, v) {
|
|
28
|
+
o["default"] = v;
|
|
29
|
+
});
|
|
30
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
31
|
+
var ownKeys = function(o) {
|
|
32
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
33
|
+
var ar = [];
|
|
34
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
35
|
+
return ar;
|
|
36
|
+
};
|
|
37
|
+
return ownKeys(o);
|
|
38
|
+
};
|
|
39
|
+
return function (mod) {
|
|
40
|
+
if (mod && mod.__esModule) return mod;
|
|
41
|
+
var result = {};
|
|
42
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
43
|
+
__setModuleDefault(result, mod);
|
|
44
|
+
return result;
|
|
45
|
+
};
|
|
46
|
+
})();
|
|
47
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
48
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
49
|
+
};
|
|
50
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
51
|
+
exports.formatJson = formatJson;
|
|
52
|
+
exports.formatMarkdown = formatMarkdown;
|
|
53
|
+
exports.severityBadge = severityBadge;
|
|
54
|
+
exports.categoryBadge = categoryBadge;
|
|
55
|
+
const commander_1 = require("commander");
|
|
56
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
57
|
+
const ora_1 = __importDefault(require("ora"));
|
|
58
|
+
const fs = __importStar(require("fs"));
|
|
59
|
+
const path = __importStar(require("path"));
|
|
60
|
+
const os = __importStar(require("os"));
|
|
61
|
+
const https = __importStar(require("https"));
|
|
62
|
+
const http = __importStar(require("http"));
|
|
63
|
+
const child_process = __importStar(require("child_process"));
|
|
64
|
+
const cli_progress_1 = require("cli-progress");
|
|
65
|
+
const shared_types_1 = require("@diffmind/shared-types");
|
|
66
|
+
// ─── Constants ────────────────────────────────────────────────────────────────
|
|
67
|
+
const MODEL_DIR = path.join(os.homedir(), ".diffmind", "models");
|
|
68
|
+
const MODEL_FILENAME = "Qwen2.5-Coder-3B-Instruct-Q4_K_M.gguf";
|
|
69
|
+
const TOKENIZER_FILENAME = "tokenizer.json";
|
|
70
|
+
// HuggingFace Hub URLs
|
|
71
|
+
const MODEL_URL = "https://huggingface.co/Qwen/Qwen2.5-Coder-3B-Instruct-GGUF/resolve/main/Qwen2.5-Coder-3B-Instruct-Q4_K_M.gguf";
|
|
72
|
+
const TOKENIZER_URL = "https://huggingface.co/Qwen/Qwen2.5-Coder-3B-Instruct/resolve/main/tokenizer.json";
|
|
73
|
+
// ─── CLI Definition ───────────────────────────────────────────────────────────
|
|
74
|
+
const program = new commander_1.Command();
|
|
75
|
+
program
|
|
76
|
+
.name("diffmind")
|
|
77
|
+
.description("Local-first AI code review for your git diffs")
|
|
78
|
+
.version("0.1.0")
|
|
79
|
+
.option("-b, --branch <name>", "Target branch to diff against", "main")
|
|
80
|
+
.option("-f, --format <type>", 'Output format: "markdown" or "json"', "markdown")
|
|
81
|
+
.option("-o, --output <file>", "Write output to a file instead of stdout")
|
|
82
|
+
.option("--min-severity <level>", 'Minimum severity to report: "high", "medium", or "low"', "low")
|
|
83
|
+
.option("--stdin", "Read git diff from stdin instead of running git diff")
|
|
84
|
+
.option("--no-color", "Disable colored output")
|
|
85
|
+
.parse(process.argv);
|
|
86
|
+
const opts = program.opts();
|
|
87
|
+
// ─── Main ─────────────────────────────────────────────────────────────────────
|
|
88
|
+
const worker_threads_1 = require("worker_threads");
|
|
89
|
+
// ─── Main ─────────────────────────────────────────────────────────────────────
|
|
90
|
+
async function main() {
|
|
91
|
+
printBanner();
|
|
92
|
+
// 1. Ensure model files are downloaded
|
|
93
|
+
await ensureModelFiles();
|
|
94
|
+
// 2. Get the git diff
|
|
95
|
+
const diff = await getDiff();
|
|
96
|
+
if (!diff.trim()) {
|
|
97
|
+
console.log(chalk_1.default.green("✓ No changes detected. Nothing to analyze."));
|
|
98
|
+
process.exit(0);
|
|
99
|
+
}
|
|
100
|
+
// 3. Run analysis in a background worker
|
|
101
|
+
// This keeps the process responsive and the spinner animated.
|
|
102
|
+
const analyzeSpinner = (0, ora_1.default)("Initializing engine & analyzing diff...").start();
|
|
103
|
+
try {
|
|
104
|
+
const reportRaw = await runAnalysisInWorker({
|
|
105
|
+
modelPath: path.join(MODEL_DIR, MODEL_FILENAME),
|
|
106
|
+
tokenizerPath: path.join(MODEL_DIR, TOKENIZER_FILENAME),
|
|
107
|
+
diff,
|
|
108
|
+
maxTokens: 2048,
|
|
109
|
+
});
|
|
110
|
+
report = (0, shared_types_1.sortFindings)((0, shared_types_1.filterBySeverity)((0, shared_types_1.parseReport)(reportRaw), opts.minSeverity));
|
|
111
|
+
analyzeSpinner.succeed(`Analysis complete — ${report.length} finding(s)`);
|
|
112
|
+
}
|
|
113
|
+
catch (err) {
|
|
114
|
+
analyzeSpinner.fail("Analysis failed");
|
|
115
|
+
console.error(chalk_1.default.red(String(err)));
|
|
116
|
+
process.exit(1);
|
|
117
|
+
}
|
|
118
|
+
// 4. Format and output results
|
|
119
|
+
const output = opts.format === "json"
|
|
120
|
+
? formatJson(report)
|
|
121
|
+
: formatMarkdown(report, opts.branch);
|
|
122
|
+
if (opts.output) {
|
|
123
|
+
fs.writeFileSync(opts.output, output, "utf-8");
|
|
124
|
+
console.log(chalk_1.default.green(`✓ Report saved to ${opts.output}`));
|
|
125
|
+
}
|
|
126
|
+
else {
|
|
127
|
+
console.log("\n" + output);
|
|
128
|
+
}
|
|
129
|
+
// Exit code 1 if any HIGH severity findings (useful for CI)
|
|
130
|
+
const hasHigh = report.some((f) => f.severity === "high");
|
|
131
|
+
process.exit(hasHigh ? 1 : 0);
|
|
132
|
+
}
|
|
133
|
+
let report;
|
|
134
|
+
function runAnalysisInWorker(workerData) {
|
|
135
|
+
return new Promise((resolve, reject) => {
|
|
136
|
+
// Determine the worker path. During development with ts-node, we point to the .ts file
|
|
137
|
+
// In production, we point to the transpiled .js file.
|
|
138
|
+
const isTsNode = process.argv.some(arg => arg.includes('ts-node'));
|
|
139
|
+
const workerPath = isTsNode
|
|
140
|
+
? path.join(__dirname, "worker.ts")
|
|
141
|
+
: path.join(__dirname, "worker.js");
|
|
142
|
+
const worker = new worker_threads_1.Worker(workerPath, {
|
|
143
|
+
workerData,
|
|
144
|
+
// If running via ts-node, we need to register it for the worker thread too
|
|
145
|
+
execArgv: isTsNode ? ["-r", "ts-node/register"] : [],
|
|
146
|
+
});
|
|
147
|
+
worker.on("message", (message) => {
|
|
148
|
+
if (message.success) {
|
|
149
|
+
resolve(message.data);
|
|
150
|
+
}
|
|
151
|
+
else {
|
|
152
|
+
reject(new Error(message.error));
|
|
153
|
+
}
|
|
154
|
+
});
|
|
155
|
+
worker.on("error", reject);
|
|
156
|
+
worker.on("exit", (code) => {
|
|
157
|
+
if (code !== 0) {
|
|
158
|
+
reject(new Error(`Worker stopped with exit code ${code}`));
|
|
159
|
+
}
|
|
160
|
+
});
|
|
161
|
+
});
|
|
162
|
+
}
|
|
163
|
+
// ─── Diff Acquisition ─────────────────────────────────────────────────────────
|
|
164
|
+
async function getDiff() {
|
|
165
|
+
if (opts.stdin) {
|
|
166
|
+
return readStdin();
|
|
167
|
+
}
|
|
168
|
+
const spinner = (0, ora_1.default)(`Running git diff ${opts.branch}...HEAD`).start();
|
|
169
|
+
try {
|
|
170
|
+
const result = child_process.spawnSync("git", ["diff", `${opts.branch}...HEAD`], {
|
|
171
|
+
maxBuffer: 10 * 1024 * 1024, // 10MB max diff
|
|
172
|
+
encoding: "utf-8",
|
|
173
|
+
shell: false, // Explicitly disable shell for security
|
|
174
|
+
});
|
|
175
|
+
if (result.status !== 0) {
|
|
176
|
+
throw new Error(result.stderr?.toString() || "Unknown git error");
|
|
177
|
+
}
|
|
178
|
+
const diff = result.stdout.toString();
|
|
179
|
+
spinner.succeed(`Diff captured (${Math.round(diff.length / 1024)}KB)`);
|
|
180
|
+
return diff;
|
|
181
|
+
}
|
|
182
|
+
catch (err) {
|
|
183
|
+
spinner.fail("Failed to get git diff");
|
|
184
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
185
|
+
if (msg.includes("not a git repository")) {
|
|
186
|
+
console.error(chalk_1.default.red("Error: not a git repository. Run diffmind from within a git project."));
|
|
187
|
+
}
|
|
188
|
+
else if (msg.includes("unknown revision")) {
|
|
189
|
+
console.error(chalk_1.default.red(`Error: branch "${opts.branch}" not found. Try a different --branch value.`));
|
|
190
|
+
}
|
|
191
|
+
else {
|
|
192
|
+
console.error(chalk_1.default.red(msg));
|
|
193
|
+
}
|
|
194
|
+
process.exit(1);
|
|
195
|
+
}
|
|
196
|
+
}
|
|
197
|
+
function readStdin() {
|
|
198
|
+
return new Promise((resolve) => {
|
|
199
|
+
let data = "";
|
|
200
|
+
process.stdin.setEncoding("utf-8");
|
|
201
|
+
process.stdin.on("data", (chunk) => (data += chunk));
|
|
202
|
+
process.stdin.on("end", () => resolve(data));
|
|
203
|
+
});
|
|
204
|
+
}
|
|
205
|
+
// ─── Model Management ─────────────────────────────────────────────────────────
|
|
206
|
+
async function ensureModelFiles() {
|
|
207
|
+
fs.mkdirSync(MODEL_DIR, { recursive: true });
|
|
208
|
+
const modelPath = path.join(MODEL_DIR, MODEL_FILENAME);
|
|
209
|
+
const tokenizerPath = path.join(MODEL_DIR, TOKENIZER_FILENAME);
|
|
210
|
+
if (!fs.existsSync(tokenizerPath)) {
|
|
211
|
+
console.log(chalk_1.default.cyan("Downloading tokenizer.json..."));
|
|
212
|
+
await downloadFile(TOKENIZER_URL, tokenizerPath);
|
|
213
|
+
}
|
|
214
|
+
if (!fs.existsSync(modelPath)) {
|
|
215
|
+
console.log(chalk_1.default.cyan(`\nDownloading ${MODEL_FILENAME} (~2.2GB). This only happens once.\n`));
|
|
216
|
+
await downloadFileWithProgress(MODEL_URL, modelPath);
|
|
217
|
+
}
|
|
218
|
+
}
|
|
219
|
+
function downloadFile(url, dest) {
|
|
220
|
+
return new Promise((resolve, reject) => {
|
|
221
|
+
const file = fs.createWriteStream(dest);
|
|
222
|
+
const get = url.startsWith("https") ? https.get : http.get;
|
|
223
|
+
get(url, { headers: { "User-Agent": "diffmind/0.1.0" } }, (res) => {
|
|
224
|
+
if (res.statusCode === 302 || res.statusCode === 301) {
|
|
225
|
+
file.close();
|
|
226
|
+
fs.unlinkSync(dest);
|
|
227
|
+
downloadFile(res.headers.location, dest).then(resolve).catch(reject);
|
|
228
|
+
return;
|
|
229
|
+
}
|
|
230
|
+
res.pipe(file);
|
|
231
|
+
file.on("finish", () => file.close(() => resolve()));
|
|
232
|
+
}).on("error", (err) => {
|
|
233
|
+
fs.unlinkSync(dest);
|
|
234
|
+
reject(err);
|
|
235
|
+
});
|
|
236
|
+
});
|
|
237
|
+
}
|
|
238
|
+
function downloadFileWithProgress(url, dest) {
|
|
239
|
+
return new Promise((resolve, reject) => {
|
|
240
|
+
const get = url.startsWith("https") ? https.get : http.get;
|
|
241
|
+
get(url, { headers: { "User-Agent": "diffmind/0.1.0" } }, (res) => {
|
|
242
|
+
if (res.statusCode === 302 || res.statusCode === 301) {
|
|
243
|
+
downloadFileWithProgress(res.headers.location, dest)
|
|
244
|
+
.then(resolve)
|
|
245
|
+
.catch(reject);
|
|
246
|
+
return;
|
|
247
|
+
}
|
|
248
|
+
const total = parseInt(res.headers["content-length"] ?? "0", 10);
|
|
249
|
+
const bar = new cli_progress_1.SingleBar({
|
|
250
|
+
format: `{bar} {percentage}% | {value}/{total} MB | ETA: {eta}s`,
|
|
251
|
+
formatValue: (v, _, type) => {
|
|
252
|
+
if (type === "value" || type === "total")
|
|
253
|
+
return (v / 1024 / 1024).toFixed(1);
|
|
254
|
+
return String(v);
|
|
255
|
+
},
|
|
256
|
+
}, cli_progress_1.Presets.shades_classic);
|
|
257
|
+
bar.start(total, 0);
|
|
258
|
+
let downloaded = 0;
|
|
259
|
+
const file = fs.createWriteStream(dest);
|
|
260
|
+
res.on("data", (chunk) => {
|
|
261
|
+
downloaded += chunk.length;
|
|
262
|
+
bar.update(downloaded);
|
|
263
|
+
file.write(chunk);
|
|
264
|
+
});
|
|
265
|
+
res.on("end", () => {
|
|
266
|
+
bar.stop();
|
|
267
|
+
file.close(() => resolve());
|
|
268
|
+
});
|
|
269
|
+
res.on("error", (err) => {
|
|
270
|
+
bar.stop();
|
|
271
|
+
fs.unlinkSync(dest);
|
|
272
|
+
reject(err);
|
|
273
|
+
});
|
|
274
|
+
}).on("error", reject);
|
|
275
|
+
});
|
|
276
|
+
}
|
|
277
|
+
// ─── Formatters ───────────────────────────────────────────────────────────────
|
|
278
|
+
function formatJson(report) {
|
|
279
|
+
return JSON.stringify(report, null, 2);
|
|
280
|
+
}
|
|
281
|
+
function formatMarkdown(report, branch) {
|
|
282
|
+
const lines = [];
|
|
283
|
+
if (report.length === 0) {
|
|
284
|
+
lines.push(chalk_1.default.green("✓ No issues found in this diff.\n"));
|
|
285
|
+
return lines.join("\n");
|
|
286
|
+
}
|
|
287
|
+
const high = report.filter((f) => f.severity === "high");
|
|
288
|
+
const medium = report.filter((f) => f.severity === "medium");
|
|
289
|
+
const low = report.filter((f) => f.severity === "low");
|
|
290
|
+
lines.push(chalk_1.default.bold.white("╔══════════════════════════════════════╗"));
|
|
291
|
+
lines.push(chalk_1.default.bold.white("║ diffmind Code Review Report ║"));
|
|
292
|
+
lines.push(chalk_1.default.bold.white("╚══════════════════════════════════════╝"));
|
|
293
|
+
lines.push("");
|
|
294
|
+
lines.push(`Branch: ${chalk_1.default.cyan(branch)} | Findings: ${chalk_1.default.white(report.length)} | ${chalk_1.default.red(`High: ${high.length}`)} ${chalk_1.default.yellow(`Medium: ${medium.length}`)} ${chalk_1.default.blue(`Low: ${low.length}`)}`);
|
|
295
|
+
lines.push("");
|
|
296
|
+
for (const finding of report) {
|
|
297
|
+
const sBadge = severityBadge(finding.severity);
|
|
298
|
+
const cBadge = categoryBadge(finding.category);
|
|
299
|
+
const conf = finding.confidence != null
|
|
300
|
+
? chalk_1.default.dim(` (confidence: ${Math.round(finding.confidence * 100)}%)`)
|
|
301
|
+
: "";
|
|
302
|
+
lines.push(`${sBadge} ${cBadge} ${chalk_1.default.bold(finding.file)}:${chalk_1.default.cyan(String(finding.line))}${conf}`);
|
|
303
|
+
lines.push(` ${chalk_1.default.white(finding.issue)}`);
|
|
304
|
+
lines.push(` ${chalk_1.default.dim("Fix:")} ${chalk_1.default.green(finding.suggested_fix)}`);
|
|
305
|
+
lines.push("");
|
|
306
|
+
}
|
|
307
|
+
return lines.join("\n");
|
|
308
|
+
}
|
|
309
|
+
function severityBadge(severity) {
|
|
310
|
+
switch (severity) {
|
|
311
|
+
case "high": return chalk_1.default.bgRed.white.bold(" HIGH ");
|
|
312
|
+
case "medium": return chalk_1.default.bgYellow.black.bold(" MED ");
|
|
313
|
+
case "low": return chalk_1.default.bgBlue.white.bold(" LOW ");
|
|
314
|
+
}
|
|
315
|
+
}
|
|
316
|
+
function categoryBadge(category) {
|
|
317
|
+
switch (category) {
|
|
318
|
+
case "security": return chalk_1.default.bgMagenta.white.bold(" SECURITY ");
|
|
319
|
+
case "quality": return chalk_1.default.bgCyan.black.bold(" QUALITY ");
|
|
320
|
+
case "performance": return chalk_1.default.bgBlackBright.white.bold(" PERF ");
|
|
321
|
+
case "maintainability": return chalk_1.default.bgBlueBright.white.bold(" MAINT ");
|
|
322
|
+
}
|
|
323
|
+
}
|
|
324
|
+
function printBanner() {
|
|
325
|
+
console.log(chalk_1.default.cyan.bold("\n diffmind") + chalk_1.default.dim(" v0.1.0 — local-first AI code review"));
|
|
326
|
+
console.log(chalk_1.default.dim(" Model: Qwen2.5-Coder-3B-Instruct Q4_K_M | Inference: on-device Wasm\n"));
|
|
327
|
+
}
|
|
328
|
+
if (require.main === module) {
|
|
329
|
+
main().catch((err) => {
|
|
330
|
+
console.error(chalk_1.default.red("\nUnexpected error:"), err);
|
|
331
|
+
process.exit(1);
|
|
332
|
+
});
|
|
333
|
+
}
|
|
334
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;AACA;;;;;;;;;;GAUG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyTH,gCAEC;AAED,wCAgCC;AAED,sCAMC;AAED,sCAOC;AA5WD,yCAAoC;AACpC,kDAA0B;AAC1B,8CAAsB;AACtB,uCAAyB;AACzB,2CAA6B;AAC7B,uCAAyB;AACzB,6CAA+B;AAC/B,2CAA6B;AAC7B,6DAA+C;AAC/C,+CAAkD;AAClD,yDAOgC;AAEhC,iFAAiF;AAEjF,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;AACjE,MAAM,cAAc,GAAG,uCAAuC,CAAC;AAC/D,MAAM,kBAAkB,GAAG,gBAAgB,CAAC;AAE5C,uBAAuB;AACvB,MAAM,SAAS,GACb,+GAA+G,CAAC;AAClH,MAAM,aAAa,GACjB,mFAAmF,CAAC;AAEtF,iFAAiF;AAEjF,MAAM,OAAO,GAAG,IAAI,mBAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,UAAU,CAAC;KAChB,WAAW,CAAC,+CAA+C,CAAC;KAC5D,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,CAAC,qBAAqB,EAAE,+BAA+B,EAAE,MAAM,CAAC;KACtE,MAAM,CACL,qBAAqB,EACrB,qCAAqC,EACrC,UAAU,CACX;KACA,MAAM,CAAC,qBAAqB,EAAE,0CAA0C,CAAC;KACzE,MAAM,CACL,wBAAwB,EACxB,wDAAwD,EACxD,KAAK,CACN;KACA,MAAM,CAAC,SAAS,EAAE,sDAAsD,CAAC;KACzE,MAAM,CAAC,YAAY,EAAE,wBAAwB,CAAC;KAC9C,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;AAEvB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAOrB,CAAC;AAEL,iFAAiF;AAEjF,mDAAwC;AAExC,iFAAiF;AAEjF,KAAK,UAAU,IAAI;IACjB,WAAW,EAAE,CAAC;IAEd,uCAAuC;IACvC,MAAM,gBAAgB,EAAE,CAAC;IAEzB,sBAAsB;IACtB,MAAM,IAAI,GAAG,MAAM,OAAO,EAAE,CAAC;IAC7B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QACjB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC,CAAC;QACvE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,yCAAyC;IACzC,8DAA8D;IAC9D,MAAM,cAAc,GAAG,IAAA,aAAG,EAAC,yCAAyC,CAAC,CAAC,KAAK,EAAE,CAAC;IAE9E,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,mBAAmB,CAAC;YAC1C,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC;YAC/C,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,kBAAkB,CAAC;YACvD,IAAI;YACJ,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAEH,MAAM,GAAG,IAAA,2BAAY,EACnB,IAAA,+BAAgB,EAAC,IAAA,0BAAW,EAAC,SAAS,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,CAC3D,CAAC;QACF,cAAc,CAAC,OAAO,CAAC,uBAAuB,MAAM,CAAC,MAAM,aAAa,CAAC,CAAC;IAC5E,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,cAAc,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACvC,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACtC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,+BAA+B;IAC/B,MAAM,MAAM,GACV,IAAI,CAAC,MAAM,KAAK,MAAM;QACpB,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC;QACpB,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAE1C,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAC/C,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC;IAC7B,CAAC;IAED,4DAA4D;IAC5D,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC;IAC1D,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAChC,CAAC;AAED,IAAI,MAAoB,CAAC;AAEzB,SAAS,mBAAmB,CAAC,UAK5B;IACC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,uFAAuF;QACvF,sDAAsD;QACtD,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,QAAQ;YACzB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC;YACnC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QAEtC,MAAM,MAAM,GAAG,IAAI,uBAAM,CAAC,UAAU,EAAE;YACpC,UAAU;YACV,2EAA2E;YAC3E,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC,EAAE;SACrD,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;YAC/B,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACpB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;YACnC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACzB,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,KAAK,CAAC,iCAAiC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iFAAiF;AAEjF,KAAK,UAAU,OAAO;IACpB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,OAAO,SAAS,EAAE,CAAC;IACrB,CAAC;IACD,MAAM,OAAO,GAAG,IAAA,aAAG,EAAC,oBAAoB,IAAI,CAAC,MAAM,SAAS,CAAC,CAAC,KAAK,EAAE,CAAC;IACtE,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CACpC,KAAK,EACL,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,SAAS,CAAC,EACjC;YACE,SAAS,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE,gBAAgB;YAC7C,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,KAAK,EAAE,wCAAwC;SACvD,CACF,CAAC;QAEF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,mBAAmB,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACtC,OAAO,CAAC,OAAO,CAAC,kBAAkB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACvC,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7D,IAAI,GAAG,CAAC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;YACzC,OAAO,CAAC,KAAK,CACX,eAAK,CAAC,GAAG,CACP,sEAAsE,CACvE,CACF,CAAC;QACJ,CAAC;aAAM,IAAI,GAAG,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;YAC5C,OAAO,CAAC,KAAK,CACX,eAAK,CAAC,GAAG,CACP,kBAAkB,IAAI,CAAC,MAAM,8CAA8C,CAC5E,CACF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAChC,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,SAAS,SAAS;IAChB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACnC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC;QACrD,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iFAAiF;AAEjF,KAAK,UAAU,gBAAgB;IAC7B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAE7C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;IACvD,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;IAE/D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC,CAAC;QACzD,MAAM,YAAY,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,OAAO,CAAC,GAAG,CACT,eAAK,CAAC,IAAI,CACR,iBAAiB,cAAc,sCAAsC,CACtE,CACF,CAAC;QACF,MAAM,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IACvD,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,GAAW,EAAE,IAAY;IAC7C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,IAAI,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,GAAG,GAAG,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;QAC3D,GAAG,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,gBAAgB,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE;YAChE,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;gBACrD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACb,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACpB,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,QAAS,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACtE,OAAO;YACT,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACf,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACrB,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACpB,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,wBAAwB,CAAC,GAAW,EAAE,IAAY;IACzD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,GAAG,GAAG,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;QAC3D,GAAG,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,gBAAgB,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE;YAChE,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;gBACrD,wBAAwB,CAAC,GAAG,CAAC,OAAO,CAAC,QAAS,EAAE,IAAI,CAAC;qBAClD,IAAI,CAAC,OAAO,CAAC;qBACb,KAAK,CAAC,MAAM,CAAC,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;YACjE,MAAM,GAAG,GAAG,IAAI,wBAAS,CACvB;gBACE,MAAM,EAAE,wDAAwD;gBAChE,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE;oBAC1B,IAAI,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,OAAO;wBACtC,OAAO,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;oBACtC,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;gBACnB,CAAC;aACF,EACD,sBAAO,CAAC,cAAc,CACvB,CAAC;YACF,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAEpB,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,MAAM,IAAI,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAExC,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBAC/B,UAAU,IAAI,KAAK,CAAC,MAAM,CAAC;gBAC3B,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBACvB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACpB,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACjB,GAAG,CAAC,IAAI,EAAE,CAAC;gBACX,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACtB,GAAG,CAAC,IAAI,EAAE,CAAC;gBACX,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACpB,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACzB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iFAAiF;AAEjF,SAAgB,UAAU,CAAC,MAAoB;IAC7C,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AACzC,CAAC;AAED,SAAgB,cAAc,CAAC,MAAoB,EAAE,MAAc;IACjE,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,KAAK,CAAC,IAAI,CAAC,eAAK,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC,CAAC;QAC7D,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC;IAC7D,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAC;IAEvD,KAAK,CAAC,IAAI,CAAC,eAAK,CAAC,IAAI,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC,CAAC;IACzE,KAAK,CAAC,IAAI,CAAC,eAAK,CAAC,IAAI,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC,CAAC;IACzE,KAAK,CAAC,IAAI,CAAC,eAAK,CAAC,IAAI,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC,CAAC;IACzE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,WAAW,eAAK,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,eAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,eAAK,CAAC,GAAG,CAAC,SAAS,IAAI,CAAC,MAAM,EAAE,CAAC,KAAK,eAAK,CAAC,MAAM,CAAC,WAAW,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,eAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;IACnN,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,OAAO,CAAC,UAAU,IAAI,IAAI;YACrC,CAAC,CAAC,eAAK,CAAC,GAAG,CAAC,iBAAiB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,GAAG,GAAG,CAAC,IAAI,CAAC;YACtE,CAAC,CAAC,EAAE,CAAC;QACP,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,MAAM,IAAI,eAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,eAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC;QACzG,KAAK,CAAC,IAAI,CAAC,KAAK,eAAK,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9C,KAAK,CAAC,IAAI,CAAC,KAAK,eAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,eAAK,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAC3E,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,SAAgB,aAAa,CAAC,QAAkB;IAC9C,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,MAAM,CAAC,CAAG,OAAO,eAAK,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,KAAK,QAAQ,CAAC,CAAC,OAAO,eAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1D,KAAK,KAAK,CAAC,CAAI,OAAO,eAAK,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED,SAAgB,aAAa,CAAC,QAAkB;IAC9C,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,UAAU,CAAC,CAAQ,OAAO,eAAK,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACxE,KAAK,SAAS,CAAC,CAAS,OAAO,eAAK,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACrE,KAAK,aAAa,CAAC,CAAK,OAAO,eAAK,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC5E,KAAK,iBAAiB,CAAC,CAAC,OAAO,eAAK,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC7E,CAAC;AACH,CAAC;AAED,SAAS,WAAW;IAClB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,eAAK,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC,CAAC;IACjG,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,GAAG,CAAC,yEAAyE,CAAC,CAAC,CAAC;AACpG,CAAC;AAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;IAC5B,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;QACnB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,qBAAqB,CAAC,EAAE,GAAG,CAAC,CAAC;QACrD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC"}
|
package/dist/worker.js
ADDED
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
19
|
+
var ownKeys = function(o) {
|
|
20
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
21
|
+
var ar = [];
|
|
22
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
23
|
+
return ar;
|
|
24
|
+
};
|
|
25
|
+
return ownKeys(o);
|
|
26
|
+
};
|
|
27
|
+
return function (mod) {
|
|
28
|
+
if (mod && mod.__esModule) return mod;
|
|
29
|
+
var result = {};
|
|
30
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
31
|
+
__setModuleDefault(result, mod);
|
|
32
|
+
return result;
|
|
33
|
+
};
|
|
34
|
+
})();
|
|
35
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
36
|
+
const worker_threads_1 = require("worker_threads");
|
|
37
|
+
const fs = __importStar(require("fs"));
|
|
38
|
+
/**
|
|
39
|
+
* Diffmind Background Worker
|
|
40
|
+
*
|
|
41
|
+
* Handles heavy Wasm inference off the main thread to keep the CLI responsive.
|
|
42
|
+
*/
|
|
43
|
+
async function runWorker() {
|
|
44
|
+
if (!worker_threads_1.parentPort)
|
|
45
|
+
return;
|
|
46
|
+
try {
|
|
47
|
+
const { modelPath, tokenizerPath, diff, maxTokens } = worker_threads_1.workerData;
|
|
48
|
+
// 1. Load Wasm bindings
|
|
49
|
+
// Note: In Node.js, we require the generated CommonJS bindings from pkg/
|
|
50
|
+
const { ReviewAnalyzer } = require("@diffmind/core-wasm");
|
|
51
|
+
// 2. Read model files
|
|
52
|
+
const modelBytes = fs.readFileSync(modelPath);
|
|
53
|
+
const tokenizerBytes = fs.readFileSync(tokenizerPath);
|
|
54
|
+
// 3. Initialize and run
|
|
55
|
+
const analyzer = new ReviewAnalyzer(modelBytes, tokenizerBytes);
|
|
56
|
+
const resultJson = analyzer.analyze_diff_chunked(diff, maxTokens);
|
|
57
|
+
// 4. Return results
|
|
58
|
+
worker_threads_1.parentPort.postMessage({ success: true, data: resultJson });
|
|
59
|
+
}
|
|
60
|
+
catch (err) {
|
|
61
|
+
worker_threads_1.parentPort.postMessage({
|
|
62
|
+
success: false,
|
|
63
|
+
error: err instanceof Error ? err.message : String(err)
|
|
64
|
+
});
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
runWorker();
|
|
68
|
+
//# sourceMappingURL=worker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"worker.js","sourceRoot":"","sources":["../src/worker.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,mDAAwD;AACxD,uCAAyB;AAEzB;;;;GAIG;AAEH,KAAK,UAAU,SAAS;IACtB,IAAI,CAAC,2BAAU;QAAE,OAAO;IAExB,IAAI,CAAC;QACH,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,2BAAU,CAAC;QAEjE,wBAAwB;QACxB,yEAAyE;QACzE,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAE1D,sBAAsB;QACtB,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;QAEtD,wBAAwB;QACxB,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QAChE,MAAM,UAAU,GAAG,QAAQ,CAAC,oBAAoB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAElE,oBAAoB;QACpB,2BAAU,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IAC9D,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,2BAAU,CAAC,WAAW,CAAC;YACrB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,SAAS,EAAE,CAAC"}
|
package/package.json
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@diffmind/cli",
|
|
3
|
+
"version": "0.1.0",
|
|
4
|
+
"description": "diffmind CLI — local-first AI code review for your git diffs",
|
|
5
|
+
"author": "Thinkgrid Labs <hello@thinkgrid.com>",
|
|
6
|
+
"license": "MIT",
|
|
7
|
+
"repository": {
|
|
8
|
+
"type": "git",
|
|
9
|
+
"url": "git+https://github.com/thinkgrid-labs/diffmind.git",
|
|
10
|
+
"directory": "apps/local-cli"
|
|
11
|
+
},
|
|
12
|
+
"publishConfig": {
|
|
13
|
+
"access": "public"
|
|
14
|
+
},
|
|
15
|
+
"files": [
|
|
16
|
+
"dist/"
|
|
17
|
+
],
|
|
18
|
+
"bin": {
|
|
19
|
+
"diffmind": "./dist/index.js"
|
|
20
|
+
},
|
|
21
|
+
"main": "./dist/index.js",
|
|
22
|
+
"dependencies": {
|
|
23
|
+
"chalk": "^5.3.0",
|
|
24
|
+
"cli-progress": "^3.12.0",
|
|
25
|
+
"commander": "^12.0.0",
|
|
26
|
+
"ora": "^8.0.1",
|
|
27
|
+
"@diffmind/core-wasm": "0.1.0",
|
|
28
|
+
"@diffmind/shared-types": "0.1.0"
|
|
29
|
+
},
|
|
30
|
+
"devDependencies": {
|
|
31
|
+
"@types/cli-progress": "^3.11.5",
|
|
32
|
+
"@types/node": "^20.0.0",
|
|
33
|
+
"ts-node": "^10.9.2",
|
|
34
|
+
"typescript": "^5.4.0"
|
|
35
|
+
},
|
|
36
|
+
"engines": {
|
|
37
|
+
"node": ">=18.0.0"
|
|
38
|
+
},
|
|
39
|
+
"scripts": {
|
|
40
|
+
"build": "tsc",
|
|
41
|
+
"dev": "ts-node src/index.ts",
|
|
42
|
+
"watch": "tsc --watch",
|
|
43
|
+
"start": "node dist/index.js"
|
|
44
|
+
}
|
|
45
|
+
}
|