Compare commits
12
Commits
cecec8bf23
...
1.4.0
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
88bc27be40 | ||
|
|
019dfa96a0 | ||
|
|
645421e59c | ||
|
|
d51ae3a574 | ||
|
|
0c2c35ad72 | ||
|
|
558b6a9953 | ||
|
|
a7f07dbeef | ||
|
|
4fc21100a9 | ||
|
|
18061a9156 | ||
|
|
c887635a0c | ||
|
|
55df24e0fd | ||
|
|
fd4fd87ea8 |
@@ -0,0 +1,46 @@
|
||||
# AGENTS.md
|
||||
|
||||
## Build & Run
|
||||
|
||||
- **Compile**: `npm run compile` (or `tsc -p ./`)
|
||||
- **Watch**: `npm run watch`
|
||||
- **Package**: `npm run build` → produces `.vsix` in project root
|
||||
- **Test**: `npm run test` — currently a no-op (`echo "No tests yet"`); press F5 in VS Code to debug the extension
|
||||
- **Verify**: `npx tsc --noEmit` — strict mode, TS 6.0 (uses `ignoreDeprecations: "6.0"`)
|
||||
- No linter or formatter is configured
|
||||
|
||||
## Architecture
|
||||
|
||||
| File | Role |
|
||||
|---|---|
|
||||
| `src/extension.ts` | Entrypoint; registers `aiCommitExt.generate` command; orchestrates flow |
|
||||
| `src/opencodeService.ts` | Spawns `opencode run --format default --variant minimal` with the prompt on stdin; parses stdout for the commit message |
|
||||
| `src/gitService.ts` | Uses VS Code's built-in Git extension API (`vscode.git`) for diffs and repo root |
|
||||
|
||||
## Key Behaviors
|
||||
|
||||
- **Activation**: `onCommand` only — not on startup
|
||||
- **Diff logic**: Prefers staged changes. Falls back to unstaged **only if** `aiCommitExt.includeUnstaged` is `true` (default `false`).
|
||||
- **Prompt**: Hardcoded in `opencodeService.ts` — enforces Conventional Commit format (`<type>(<scope>): <description>`, max 72-char subject). Types: feat, fix, refactor, docs, style, test, chore, perf, ci, build, revert.
|
||||
- **Parsing**: Uses `\b(type)(\(scope\))?:\s` regex that matches anywhere in a line (not just line-start), so backtick-wrapped or prose-embedded commit messages are found. ANSI escape sequences are stripped before matching. Falls back to first content line, then `"chore: generated commit message"`.
|
||||
- **Stdout is logged** to the output channel ("ai-commit-ext") for debugging.
|
||||
- **`--pure` flag**: Not used — opencode loads its user config (`~/.config/opencode/opencode.json`) which needs to configure the model. The extension does NOT pass a hardcoded `--model` flag; the user's config determines the model.
|
||||
- **`shell: true`**: Spawn uses a shell so opencode can find `git` and other tools via the shell's PATH (VS Code's GUI-launched process has a minimal PATH).
|
||||
- **Timeout**: 120 seconds for OpenCode response.
|
||||
- **Model**: Default is whatever opencode's user config sets. Override via `aiCommitExt.model` setting or `GenerateOptions.model`.
|
||||
- **OpenCode CLI**: Required on `$PATH`; resolved to an absolute path via `which opencode` and cached — used in `spawn()` to avoid PATH lookup mismatches between the shell and VS Code's process environment.
|
||||
|
||||
## Configuration (`aiCommitExt.*`)
|
||||
|
||||
| Setting | Default | Purpose |
|
||||
|---|---|---|
|
||||
| `model` | `""` | OpenCode model override |
|
||||
| `includeUnstaged` | `false` | Use unstaged changes when nothing staged |
|
||||
| `showNotification` | `true` | Show VS Code notifications on success/error |
|
||||
|
||||
## Build Artifacts & Cleanup
|
||||
|
||||
- Compiled output → `out/` (gitignored)
|
||||
- Packaged extension → `*.vsix` (gitignored)
|
||||
- `node_modules/` (gitignored)
|
||||
- `npm run compile` before packaging (`vscode:prepublish` hook)
|
||||
Generated
+570
-570
File diff suppressed because it is too large
Load Diff
+1
-1
@@ -2,7 +2,7 @@
|
||||
"name": "ai-commit-ext",
|
||||
"displayName": "AI Commit Ext",
|
||||
"description": "Generate commit messages using OpenCode AI",
|
||||
"version": "1.0.0",
|
||||
"version": "1.4.0",
|
||||
"publisher": "local",
|
||||
"engines": {
|
||||
"vscode": "^1.110.0"
|
||||
|
||||
+95
-45
@@ -1,47 +1,33 @@
|
||||
import * as vscode from "vscode";
|
||||
import { generateCommitMessage, isOpenCodeAvailable } from "./opencodeService";
|
||||
|
||||
interface GitExtension {
|
||||
getAPI(version: number): GitAPI;
|
||||
}
|
||||
|
||||
interface GitAPI {
|
||||
repositories: Repository[];
|
||||
}
|
||||
|
||||
interface Repository {
|
||||
root: string;
|
||||
rootUri: vscode.Uri;
|
||||
state: RepositoryState;
|
||||
inputBox: { value: string };
|
||||
diffIndexWithHEAD(path?: string): Promise<string>;
|
||||
diffWithHEAD(path?: string): Promise<string>;
|
||||
status(): Promise<void>;
|
||||
}
|
||||
|
||||
interface RepositoryState {
|
||||
indexChanges: any[];
|
||||
workingTreeChanges: any[];
|
||||
}
|
||||
import {
|
||||
Repository,
|
||||
getAllRepositories,
|
||||
getRepositoryLabel,
|
||||
getRepositoryRoot,
|
||||
} from "./gitService";
|
||||
|
||||
export async function activate(
|
||||
context: vscode.ExtensionContext,
|
||||
): Promise<void> {
|
||||
const commandHandler = vscode.commands.registerCommand(
|
||||
"aiCommitExt.generate",
|
||||
async () => {
|
||||
await handleGenerateCommitMessage();
|
||||
async (sc?: vscode.SourceControl) => {
|
||||
await handleGenerateCommitMessage(sc);
|
||||
},
|
||||
);
|
||||
context.subscriptions.push(commandHandler);
|
||||
}
|
||||
|
||||
async function handleGenerateCommitMessage(): Promise<void> {
|
||||
async function handleGenerateCommitMessage(
|
||||
sc?: vscode.SourceControl,
|
||||
): Promise<void> {
|
||||
const config = vscode.workspace.getConfiguration("aiCommitExt");
|
||||
const showNotification = config.get<boolean>("showNotification", true);
|
||||
|
||||
try {
|
||||
const opencodeAvailable = await isOpenCodeAvailable();
|
||||
|
||||
if (!opencodeAvailable) {
|
||||
if (showNotification) {
|
||||
vscode.window.showErrorMessage(
|
||||
@@ -51,27 +37,15 @@ async function handleGenerateCommitMessage(): Promise<void> {
|
||||
return;
|
||||
}
|
||||
|
||||
const gitExtension =
|
||||
vscode.extensions.getExtension<GitExtension>("vscode.git");
|
||||
if (!gitExtension) {
|
||||
if (showNotification) {
|
||||
vscode.window.showErrorMessage("Git extension not found");
|
||||
}
|
||||
return;
|
||||
}
|
||||
const repository = await resolveRepository(sc);
|
||||
|
||||
const gitApi = gitExtension.exports.getAPI(1);
|
||||
const repositories = gitApi.repositories;
|
||||
|
||||
if (repositories.length === 0) {
|
||||
if (!repository) {
|
||||
if (showNotification) {
|
||||
vscode.window.showErrorMessage("No Git repository found");
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
const repository = repositories[0];
|
||||
|
||||
await repository.status();
|
||||
|
||||
const state = repository.state;
|
||||
@@ -86,21 +60,41 @@ async function handleGenerateCommitMessage(): Promise<void> {
|
||||
return;
|
||||
}
|
||||
|
||||
const userSuggestion = await vscode.window.showInputBox({
|
||||
prompt: "Suggest a commit message (optional)",
|
||||
placeHolder: "e.g., added login button",
|
||||
ignoreFocusOut: true,
|
||||
});
|
||||
|
||||
if (userSuggestion === undefined) {
|
||||
return;
|
||||
}
|
||||
|
||||
const repoLabel = getRepositoryLabel(repository);
|
||||
|
||||
const commitMessage = await vscode.window.withProgress(
|
||||
{
|
||||
location: vscode.ProgressLocation.Notification,
|
||||
cancellable: false,
|
||||
title: "Generating commit message...",
|
||||
cancellable: true,
|
||||
title: userSuggestion
|
||||
? `Improving commit message (${repoLabel})...`
|
||||
: `Generating commit message (${repoLabel})...`,
|
||||
},
|
||||
async () => {
|
||||
return await generateCommitMessage();
|
||||
async (_progress, token) => {
|
||||
return await generateCommitMessage({
|
||||
repository,
|
||||
userSuggestion: userSuggestion || undefined,
|
||||
cancellationToken: token,
|
||||
});
|
||||
},
|
||||
);
|
||||
|
||||
repository.inputBox.value = commitMessage;
|
||||
|
||||
if (showNotification) {
|
||||
vscode.window.showInformationMessage("Commit message generated");
|
||||
vscode.window.showInformationMessage(
|
||||
`Commit message generated for ${repoLabel}`,
|
||||
);
|
||||
}
|
||||
} catch (error) {
|
||||
const message =
|
||||
@@ -113,4 +107,60 @@ async function handleGenerateCommitMessage(): Promise<void> {
|
||||
}
|
||||
}
|
||||
|
||||
async function resolveRepository(
|
||||
sc?: vscode.SourceControl,
|
||||
): Promise<Repository | undefined> {
|
||||
const repositories = getAllRepositories();
|
||||
|
||||
if (repositories.length === 0) {
|
||||
return undefined;
|
||||
}
|
||||
|
||||
if (sc) {
|
||||
output.appendLine(`[resolveRepository] sc provided, sc.rootUri=${sc.rootUri?.fsPath}`);
|
||||
for (const r of repositories) {
|
||||
const rRoot = getRepositoryRoot(r);
|
||||
const match = r.inputBox === sc.inputBox;
|
||||
output.appendLine(`[resolveRepository] repo root=${rRoot}, inputBox match=${match}, sc.inputBox===r.inputBox=${sc.inputBox === r.inputBox}`);
|
||||
}
|
||||
|
||||
const byInputBox = repositories.find((r) => r.inputBox === sc.inputBox);
|
||||
if (byInputBox) {
|
||||
output.appendLine(`[resolveRepository] matched by inputBox`);
|
||||
return byInputBox;
|
||||
}
|
||||
|
||||
if (sc.rootUri) {
|
||||
const byRootUri = repositories.find((r) => {
|
||||
const rRoot = getRepositoryRoot(r);
|
||||
return rRoot === sc.rootUri!.fsPath;
|
||||
});
|
||||
if (byRootUri) {
|
||||
output.appendLine(`[resolveRepository] matched by rootUri`);
|
||||
return byRootUri;
|
||||
}
|
||||
}
|
||||
|
||||
output.appendLine(`[resolveRepository] no match found for sc`);
|
||||
}
|
||||
|
||||
if (repositories.length === 1) {
|
||||
return repositories[0];
|
||||
}
|
||||
|
||||
const items = repositories.map((repo) => ({
|
||||
label: getRepositoryLabel(repo),
|
||||
description: getRepositoryRoot(repo),
|
||||
repository: repo,
|
||||
}));
|
||||
|
||||
const selected = await vscode.window.showQuickPick(items, {
|
||||
placeHolder: "Select a repository to generate a commit message for",
|
||||
});
|
||||
|
||||
return selected?.repository;
|
||||
}
|
||||
|
||||
export function deactivate(): void {}
|
||||
|
||||
export const output = vscode.window.createOutputChannel("ai-commit-ext");
|
||||
|
||||
+158
-143
@@ -1,170 +1,185 @@
|
||||
import * as vscode from 'vscode';
|
||||
import * as vscode from "vscode";
|
||||
|
||||
interface GitExtension {
|
||||
getAPI(version: number): GitAPI;
|
||||
export interface GitExtension {
|
||||
getAPI(version: number): GitAPI;
|
||||
}
|
||||
|
||||
interface GitAPI {
|
||||
repositories: Repository[];
|
||||
export interface GitAPI {
|
||||
repositories: Repository[];
|
||||
}
|
||||
|
||||
interface Repository {
|
||||
root: string;
|
||||
rootUri: vscode.Uri;
|
||||
state: RepositoryState;
|
||||
inputBox: { value: string };
|
||||
diffWithHEAD(path?: string): Promise<string>;
|
||||
diffIndexWithHEAD(path?: string): Promise<string>;
|
||||
status(): Promise<void>;
|
||||
export interface Repository {
|
||||
root: string;
|
||||
rootUri: vscode.Uri;
|
||||
state: RepositoryState;
|
||||
inputBox: { value: string };
|
||||
diffIndexWithHEAD(path?: string): Promise<string>;
|
||||
diffWithHEAD(path?: string): Promise<string>;
|
||||
status(): Promise<void>;
|
||||
}
|
||||
|
||||
interface RepositoryState {
|
||||
indexChanges: Change[];
|
||||
workingTreeChanges: Change[];
|
||||
export interface RepositoryState {
|
||||
indexChanges: Change[];
|
||||
workingTreeChanges: Change[];
|
||||
}
|
||||
|
||||
interface Change {
|
||||
uri: vscode.Uri;
|
||||
status: number;
|
||||
uri: vscode.Uri;
|
||||
status: number;
|
||||
}
|
||||
|
||||
export interface GitChange {
|
||||
path: string;
|
||||
status: 'added' | 'modified' | 'deleted' | 'renamed' | 'untracked';
|
||||
diff?: string;
|
||||
path: string;
|
||||
status: "added" | "modified" | "deleted" | "renamed" | "untracked";
|
||||
diff?: string;
|
||||
}
|
||||
|
||||
export async function getGitChanges(): Promise<GitChange[]> {
|
||||
const repository = await getActiveGitRepository();
|
||||
if (!repository) {
|
||||
throw new Error('No Git repository found');
|
||||
}
|
||||
export async function getGitChanges(
|
||||
repository: Repository,
|
||||
): Promise<GitChange[]> {
|
||||
const state = repository.state;
|
||||
const changes: GitChange[] = [];
|
||||
|
||||
const state = repository.state;
|
||||
const changes: GitChange[] = [];
|
||||
const stagedChanges = state.indexChanges || [];
|
||||
const unstagedChanges = state.workingTreeChanges || [];
|
||||
|
||||
const stagedChanges = state.indexChanges || [];
|
||||
const unstagedChanges = state.workingTreeChanges || [];
|
||||
|
||||
if (stagedChanges.length > 0) {
|
||||
for (const change of stagedChanges) {
|
||||
changes.push({
|
||||
path: change.uri.fsPath,
|
||||
status: getChangeStatus(change.status),
|
||||
});
|
||||
}
|
||||
} else if (unstagedChanges.length > 0) {
|
||||
const config = vscode.workspace.getConfiguration('aiCommitExt');
|
||||
const includeUnstaged = config.get<boolean>('includeUnstaged', false);
|
||||
|
||||
if (includeUnstaged) {
|
||||
for (const change of unstagedChanges) {
|
||||
changes.push({
|
||||
path: change.uri.fsPath,
|
||||
status: getChangeStatus(change.status),
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return changes;
|
||||
}
|
||||
|
||||
export async function getGitDiff(): Promise<string> {
|
||||
const repository = await getActiveGitRepository();
|
||||
if (!repository) {
|
||||
throw new Error('No Git repository found');
|
||||
}
|
||||
|
||||
const state = repository.state;
|
||||
let diffOutput = '';
|
||||
|
||||
const stagedChanges = state.indexChanges || [];
|
||||
const unstagedChanges = state.workingTreeChanges || [];
|
||||
const config = vscode.workspace.getConfiguration('aiCommitExt');
|
||||
const includeUnstaged = config.get<boolean>('includeUnstaged', false);
|
||||
|
||||
if (stagedChanges.length > 0) {
|
||||
for (const change of stagedChanges) {
|
||||
const fileName = change.uri.fsPath.split('/').pop() || '';
|
||||
diffOutput += `## ${fileName} (staged)\n`;
|
||||
|
||||
try {
|
||||
const diff = await repository.diffIndexWithHEAD(change.uri.fsPath);
|
||||
if (diff) {
|
||||
diffOutput += diff + '\n';
|
||||
if (stagedChanges.length > 0) {
|
||||
for (const change of stagedChanges) {
|
||||
changes.push({
|
||||
path: change.uri.fsPath,
|
||||
status: getChangeStatus(change.status),
|
||||
});
|
||||
}
|
||||
} catch {
|
||||
diffOutput += '(Unable to get diff)\n';
|
||||
}
|
||||
}
|
||||
} else if (unstagedChanges.length > 0 && includeUnstaged) {
|
||||
for (const change of unstagedChanges) {
|
||||
const fileName = change.uri.fsPath.split('/').pop() || '';
|
||||
diffOutput += `## ${fileName} (unstaged)\n`;
|
||||
|
||||
try {
|
||||
const diff = await repository.diffWithHEAD(change.uri.fsPath);
|
||||
if (diff) {
|
||||
diffOutput += diff + '\n';
|
||||
} else if (unstagedChanges.length > 0) {
|
||||
const config = vscode.workspace.getConfiguration("aiCommitExt");
|
||||
const includeUnstaged = config.get<boolean>("includeUnstaged", false);
|
||||
|
||||
if (includeUnstaged) {
|
||||
for (const change of unstagedChanges) {
|
||||
changes.push({
|
||||
path: change.uri.fsPath,
|
||||
status: getChangeStatus(change.status),
|
||||
});
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
diffOutput += '(Unable to get diff)\n';
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!diffOutput) {
|
||||
throw new Error('No changes to commit');
|
||||
}
|
||||
|
||||
return diffOutput;
|
||||
return changes;
|
||||
}
|
||||
|
||||
async function getActiveGitRepository(): Promise<Repository | null> {
|
||||
const gitExtension = vscode.extensions.getExtension<GitExtension>('vscode.git');
|
||||
if (!gitExtension) {
|
||||
throw new Error('Git extension not found');
|
||||
}
|
||||
|
||||
const api = gitExtension.exports.getAPI(1);
|
||||
if (api.repositories.length === 0) {
|
||||
return null;
|
||||
}
|
||||
return api.repositories[0];
|
||||
export async function getGitDiff(
|
||||
repository: Repository,
|
||||
): Promise<string> {
|
||||
const state = repository.state;
|
||||
let diffOutput = "";
|
||||
|
||||
const stagedChanges = state.indexChanges || [];
|
||||
const unstagedChanges = state.workingTreeChanges || [];
|
||||
const config = vscode.workspace.getConfiguration("aiCommitExt");
|
||||
const includeUnstaged = config.get<boolean>("includeUnstaged", false);
|
||||
|
||||
if (stagedChanges.length > 0) {
|
||||
for (const change of stagedChanges) {
|
||||
const fileName = change.uri.fsPath.split("/").pop() || "";
|
||||
diffOutput += `## ${fileName} (staged)\n`;
|
||||
|
||||
try {
|
||||
const diff = await repository.diffIndexWithHEAD(
|
||||
change.uri.fsPath,
|
||||
);
|
||||
if (diff) {
|
||||
diffOutput += diff + "\n";
|
||||
}
|
||||
} catch {
|
||||
diffOutput += "(Unable to get diff)\n";
|
||||
}
|
||||
}
|
||||
} else if (unstagedChanges.length > 0 && includeUnstaged) {
|
||||
for (const change of unstagedChanges) {
|
||||
const fileName = change.uri.fsPath.split("/").pop() || "";
|
||||
diffOutput += `## ${fileName} (unstaged)\n`;
|
||||
|
||||
try {
|
||||
const diff = await repository.diffWithHEAD(change.uri.fsPath);
|
||||
if (diff) {
|
||||
diffOutput += diff + "\n";
|
||||
}
|
||||
} catch {
|
||||
diffOutput += "(Unable to get diff)\n";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!diffOutput) {
|
||||
throw new Error("No changes to commit");
|
||||
}
|
||||
|
||||
return diffOutput;
|
||||
}
|
||||
|
||||
function getChangeStatus(status: number): GitChange['status'] {
|
||||
const Status = {
|
||||
INDEX_MODIFIED: 0,
|
||||
INDEX_ADDED: 1,
|
||||
INDEX_DELETED: 2,
|
||||
INDEX_RENAMED: 3,
|
||||
MODIFIED: 4,
|
||||
DELETED: 5,
|
||||
UNTRACKED: 6,
|
||||
};
|
||||
|
||||
if (status === Status.INDEX_ADDED || status === Status.UNTRACKED) {
|
||||
return 'added';
|
||||
}
|
||||
if (status === Status.INDEX_DELETED || status === Status.DELETED) {
|
||||
return 'deleted';
|
||||
}
|
||||
if (status === Status.INDEX_RENAMED) {
|
||||
return 'renamed';
|
||||
}
|
||||
|
||||
return 'modified';
|
||||
export function getRepositoryRoot(repository: Repository): string {
|
||||
if (typeof repository.root === "string") {
|
||||
return repository.root;
|
||||
}
|
||||
return repository.rootUri.fsPath;
|
||||
}
|
||||
|
||||
export async function getRepositoryRoot(): Promise<string | null> {
|
||||
const repository = await getActiveGitRepository();
|
||||
if (!repository) {
|
||||
return null;
|
||||
}
|
||||
if (typeof repository.root === 'string') {
|
||||
return repository.root;
|
||||
}
|
||||
return repository.rootUri.fsPath;
|
||||
}
|
||||
export function getRepositoryLabel(repository: Repository): string {
|
||||
const root = getRepositoryRoot(repository);
|
||||
const folderName = root.split("/").pop() || root;
|
||||
return folderName;
|
||||
}
|
||||
|
||||
export function getAllRepositories(): Repository[] {
|
||||
const gitExtension =
|
||||
vscode.extensions.getExtension<GitExtension>("vscode.git");
|
||||
if (!gitExtension) {
|
||||
throw new Error("Git extension not found");
|
||||
}
|
||||
|
||||
const api = gitExtension.exports.getAPI(1);
|
||||
return api.repositories;
|
||||
}
|
||||
|
||||
export function findRepositoryForFile(
|
||||
repositories: Repository[],
|
||||
filePath: string,
|
||||
): Repository | undefined {
|
||||
const normalizedPath = filePath.toLowerCase();
|
||||
let bestMatch: Repository | undefined;
|
||||
let bestLength = 0;
|
||||
|
||||
for (const repo of repositories) {
|
||||
const root = getRepositoryRoot(repo).toLowerCase();
|
||||
if (normalizedPath.startsWith(root) && root.length > bestLength) {
|
||||
bestMatch = repo;
|
||||
bestLength = root.length;
|
||||
}
|
||||
}
|
||||
|
||||
return bestMatch;
|
||||
}
|
||||
|
||||
function getChangeStatus(status: number): GitChange["status"] {
|
||||
const Status = {
|
||||
INDEX_MODIFIED: 0,
|
||||
INDEX_ADDED: 1,
|
||||
INDEX_DELETED: 2,
|
||||
INDEX_RENAMED: 3,
|
||||
MODIFIED: 4,
|
||||
DELETED: 5,
|
||||
UNTRACKED: 6,
|
||||
};
|
||||
|
||||
if (status === Status.INDEX_ADDED || status === Status.UNTRACKED) {
|
||||
return "added";
|
||||
}
|
||||
if (status === Status.INDEX_DELETED || status === Status.DELETED) {
|
||||
return "deleted";
|
||||
}
|
||||
if (status === Status.INDEX_RENAMED) {
|
||||
return "renamed";
|
||||
}
|
||||
|
||||
return "modified";
|
||||
}
|
||||
|
||||
+129
-46
@@ -1,11 +1,17 @@
|
||||
import { exec, spawn, ExecException } from "child_process";
|
||||
import * as vscode from "vscode";
|
||||
import { getGitDiff, getRepositoryRoot } from "./gitService";
|
||||
import { getGitDiff, getRepositoryRoot, Repository } from "./gitService";
|
||||
import { output } from "./extension";
|
||||
|
||||
export interface GenerateOptions {
|
||||
repository: Repository;
|
||||
model?: string;
|
||||
userSuggestion?: string;
|
||||
cancellationToken?: vscode.CancellationToken;
|
||||
}
|
||||
|
||||
let opencodePathCache: string | null = null;
|
||||
|
||||
const DEFAULT_PROMPT = `You are a helpful assistant that generates git commit messages.
|
||||
Generate a concise Conventional Commit message (max 72 characters for the subject line).
|
||||
Format: <type>(<scope>): <description>
|
||||
@@ -15,47 +21,117 @@ Types: feat, fix, refactor, docs, style, test, chore, perf, ci, build, revert
|
||||
Only output the commit message, nothing else.`;
|
||||
|
||||
export async function isOpenCodeAvailable(): Promise<boolean> {
|
||||
return new Promise((resolve) => {
|
||||
exec("which opencode", (error: ExecException | null) => {
|
||||
resolve(!error);
|
||||
});
|
||||
try {
|
||||
await getOpenCodePath();
|
||||
return true;
|
||||
} catch {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
async function getOpenCodePath(): Promise<string> {
|
||||
if (opencodePathCache !== null) {
|
||||
return opencodePathCache;
|
||||
}
|
||||
return new Promise((resolve, reject) => {
|
||||
exec(
|
||||
"which opencode",
|
||||
(error: ExecException | null, stdout: string) => {
|
||||
if (error) {
|
||||
reject(
|
||||
new Error(
|
||||
"OpenCode not found. Please install from https://opencode.ai",
|
||||
),
|
||||
);
|
||||
} else {
|
||||
opencodePathCache = stdout.trim();
|
||||
resolve(opencodePathCache);
|
||||
}
|
||||
},
|
||||
);
|
||||
});
|
||||
}
|
||||
|
||||
export async function generateCommitMessage(
|
||||
options: GenerateOptions = {},
|
||||
options: GenerateOptions,
|
||||
): Promise<string> {
|
||||
const opencodeAvailable = await isOpenCodeAvailable();
|
||||
if (!opencodeAvailable) {
|
||||
throw new Error(
|
||||
"OpenCode is not installed. Please install it from https://opencode.ai",
|
||||
);
|
||||
}
|
||||
const { repository } = options;
|
||||
|
||||
const diff = await getGitDiff();
|
||||
const repoRoot = await getRepositoryRoot();
|
||||
const [opencodePath, diff] = await Promise.all([
|
||||
getOpenCodePath(),
|
||||
getGitDiff(repository),
|
||||
]);
|
||||
|
||||
const repoRoot = getRepositoryRoot(repository);
|
||||
const config = vscode.workspace.getConfiguration("aiCommitExt");
|
||||
const model = options.model || config.get<string>("model", "");
|
||||
|
||||
const prompt = `${DEFAULT_PROMPT}
|
||||
let prompt: string;
|
||||
if (options.userSuggestion) {
|
||||
prompt = `The user suggested: "${options.userSuggestion}"
|
||||
|
||||
Improve this commit message to be concise and follow Conventional Commit format.
|
||||
Format: <type>(<scope>): <description>
|
||||
Max 72 characters for the subject line.
|
||||
Types: feat, fix, refactor, docs, style, test, chore, perf, ci, build, revert
|
||||
|
||||
Here are the git changes:
|
||||
${diff}
|
||||
|
||||
Only output the improved commit message, nothing else.`;
|
||||
} else {
|
||||
prompt = `${DEFAULT_PROMPT}
|
||||
|
||||
Here are the git changes:
|
||||
${diff}
|
||||
|
||||
Generate a concise Conventional Commit message for these changes:`;
|
||||
}
|
||||
|
||||
output.appendLine(`[${Date.now()}]\r\n${prompt}`);
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const args: string[] = ["run", "--format", "default"];
|
||||
const args: string[] = [
|
||||
"run",
|
||||
"--format",
|
||||
"default",
|
||||
"--variant",
|
||||
"minimal",
|
||||
];
|
||||
|
||||
if (model) {
|
||||
args.push("--model", model);
|
||||
}
|
||||
|
||||
const proc = spawn("opencode", args, {
|
||||
const proc = spawn(opencodePath, args, {
|
||||
stdio: ["pipe", "pipe", "pipe"],
|
||||
shell: false,
|
||||
cwd: repoRoot || undefined,
|
||||
shell: true,
|
||||
cwd: repoRoot,
|
||||
env: {
|
||||
...process.env,
|
||||
OPENCODE_SERVER_PASSWORD: undefined,
|
||||
OPENCODE_SERVER_USERNAME: undefined,
|
||||
OPENCODE_CLIENT: undefined,
|
||||
OPENCODE_HOST: undefined,
|
||||
OPENCODE_PORT: undefined,
|
||||
OPENCODE_SKIP_START: undefined,
|
||||
OPENCODE_BINARY: undefined,
|
||||
},
|
||||
});
|
||||
|
||||
let settled = false;
|
||||
const timeout = setTimeout(() => {
|
||||
proc.kill();
|
||||
reject(new Error("OpenCode timed out"));
|
||||
}, 120000);
|
||||
|
||||
const cancelDisposable = options.cancellationToken?.onCancellationRequested(() => {
|
||||
if (!settled) {
|
||||
settled = true;
|
||||
clearTimeout(timeout);
|
||||
proc.kill();
|
||||
reject(new Error("Cancelled"));
|
||||
}
|
||||
});
|
||||
|
||||
let stdout = "";
|
||||
@@ -69,12 +145,18 @@ Generate a concise Conventional Commit message for these changes:`;
|
||||
});
|
||||
proc.stdin?.write(prompt + "\n", (err) => {
|
||||
if (err) {
|
||||
new Error(`OpenCode write error with ${err}`);
|
||||
reject(new Error(`OpenCode write error with ${err}`));
|
||||
return;
|
||||
}
|
||||
proc.stdin?.end();
|
||||
});
|
||||
|
||||
proc.on("close", (code: number | null) => {
|
||||
if (settled) return;
|
||||
settled = true;
|
||||
clearTimeout(timeout);
|
||||
cancelDisposable?.dispose();
|
||||
|
||||
if (code !== 0) {
|
||||
reject(
|
||||
new Error(`OpenCode exited with code ${code}: ${stderr}`),
|
||||
@@ -82,47 +164,48 @@ Generate a concise Conventional Commit message for these changes:`;
|
||||
return;
|
||||
}
|
||||
|
||||
output.appendLine(stdout);
|
||||
const message = parseCommitMessage(stdout);
|
||||
resolve(message);
|
||||
});
|
||||
|
||||
proc.on("error", (err: Error) => {
|
||||
if (settled) return;
|
||||
settled = true;
|
||||
clearTimeout(timeout);
|
||||
cancelDisposable?.dispose();
|
||||
reject(new Error(`Failed to run opencode: ${err.message}`));
|
||||
});
|
||||
|
||||
setTimeout(() => {
|
||||
proc.kill();
|
||||
reject(new Error("OpenCode timed out"));
|
||||
}, 120000);
|
||||
});
|
||||
}
|
||||
|
||||
function parseCommitMessage(output: string): string {
|
||||
const lines = output.split("\n").filter((line) => line.trim());
|
||||
const clean = output.replace(/\x1b\[[0-9;]*m/g, "");
|
||||
const lines = clean.split("\n").filter((line) => line.trim());
|
||||
|
||||
const ccTypes = [
|
||||
"feat",
|
||||
"fix",
|
||||
"refactor",
|
||||
"docs",
|
||||
"style",
|
||||
"test",
|
||||
"chore",
|
||||
"perf",
|
||||
"ci",
|
||||
"build",
|
||||
"revert",
|
||||
];
|
||||
|
||||
const pat = new RegExp(`\\b(${ccTypes.join("|")})(\\([^)]*\\))?:\\s`);
|
||||
|
||||
for (const line of lines) {
|
||||
const trimmed = line.trim();
|
||||
|
||||
if (
|
||||
trimmed.includes(":") &&
|
||||
(trimmed.startsWith("feat") ||
|
||||
trimmed.startsWith("fix") ||
|
||||
trimmed.startsWith("refactor") ||
|
||||
trimmed.startsWith("docs") ||
|
||||
trimmed.startsWith("style") ||
|
||||
trimmed.startsWith("test") ||
|
||||
trimmed.startsWith("chore") ||
|
||||
trimmed.startsWith("perf") ||
|
||||
trimmed.startsWith("ci") ||
|
||||
trimmed.startsWith("build") ||
|
||||
trimmed.startsWith("revert"))
|
||||
) {
|
||||
return trimmed.substring(0, 200);
|
||||
const m = trimmed.match(pat);
|
||||
if (m) {
|
||||
return trimmed.substring(m.index!).substring(0, 200);
|
||||
}
|
||||
}
|
||||
|
||||
const cleaned =
|
||||
output.trim().split("\n")[0]?.substring(0, 200) ||
|
||||
"chore: generated commit message";
|
||||
return cleaned;
|
||||
return lines[0]?.substring(0, 200) || "chore: generated commit message";
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user