Compare commits
No commits in common. "12f411fbac5d1d8fa3bca09e07508c9f06bf4425" and "e234969638214e02083dd03702f667cc6b0b77a9" have entirely different histories.
12f411fbac
...
e234969638
28
package.json
28
package.json
|
@ -4,7 +4,7 @@
|
||||||
"displayName": "Fabelous Autocoder",
|
"displayName": "Fabelous Autocoder",
|
||||||
"description": "A simple to use Ollama autocompletion Plugin",
|
"description": "A simple to use Ollama autocompletion Plugin",
|
||||||
"icon": "icon.png",
|
"icon": "icon.png",
|
||||||
"publisher": "Falko Habel",
|
"publisher": "fabel",
|
||||||
"license": "CC BY-ND 4.0",
|
"license": "CC BY-ND 4.0",
|
||||||
"bugs": {
|
"bugs": {
|
||||||
"url": "https://gitea.fabelous.app/fabel/Fabelous-Autocoder/issues"
|
"url": "https://gitea.fabelous.app/fabel/Fabelous-Autocoder/issues"
|
||||||
|
@ -90,6 +90,7 @@
|
||||||
"type": "boolean",
|
"type": "boolean",
|
||||||
"default": true,
|
"default": true,
|
||||||
"description": "Ollama continues autocompletion after what is previewed inline. Disabling disables that feature as some may find it irritating. Multiline completion is still accessible through the shortcut even after disabling."
|
"description": "Ollama continues autocompletion after what is previewed inline. Disabling disables that feature as some may find it irritating. Multiline completion is still accessible through the shortcut even after disabling."
|
||||||
|
|
||||||
},
|
},
|
||||||
"fabelous-autocoder.temperature": {
|
"fabelous-autocoder.temperature": {
|
||||||
"type": "number",
|
"type": "number",
|
||||||
|
@ -103,38 +104,15 @@
|
||||||
},
|
},
|
||||||
"fabelous-autocoder.top p": {
|
"fabelous-autocoder.top p": {
|
||||||
"type": "number",
|
"type": "number",
|
||||||
"default": 1,
|
|
||||||
"description": "Top p sampling for the model."
|
"description": "Top p sampling for the model."
|
||||||
},
|
|
||||||
"fabelous-autocoder.enableLineByLineAcceptance": {
|
|
||||||
"type": "boolean",
|
|
||||||
"default": false,
|
|
||||||
"description": "Enable line-by-line acceptance of the generated code."
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"keybindings": [
|
|
||||||
{
|
|
||||||
"command": "fabelous-autocoder.handleTab",
|
|
||||||
"key": "tab",
|
|
||||||
"when": "editorTextFocus && !editorTabMovesFocus"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"command": "fabelous-autocoder.handleBackspace",
|
|
||||||
"key": "backspace",
|
|
||||||
"when": "editorTextFocus"
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"commands": [
|
"commands": [
|
||||||
{
|
{
|
||||||
"command": "fabelous-autocoder.autocomplete",
|
"command": "fabelous-autocoder.autocomplete",
|
||||||
"title": "Fabelous Autocompletion"
|
"title": "Fabelous Autocompletion"
|
||||||
},
|
}]
|
||||||
{
|
|
||||||
"command": "fabelous-autocoder.handleTab",
|
|
||||||
"title": "Handle Tab"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
},
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"vscode:prepublish": "npm run compile",
|
"vscode:prepublish": "npm run compile",
|
||||||
|
|
184
src/extension.ts
184
src/extension.ts
|
@ -91,7 +91,6 @@ class CompletionManager {
|
||||||
private document: vscode.TextDocument;
|
private document: vscode.TextDocument;
|
||||||
private startPosition: vscode.Position;
|
private startPosition: vscode.Position;
|
||||||
private completionText: string;
|
private completionText: string;
|
||||||
private insertedLineCount: number = 0; // Track the number of inserted lines
|
|
||||||
|
|
||||||
constructor(textEditor: vscode.TextEditor, startPosition: vscode.Position, completionText: string) {
|
constructor(textEditor: vscode.TextEditor, startPosition: vscode.Position, completionText: string) {
|
||||||
this.textEditor = textEditor;
|
this.textEditor = textEditor;
|
||||||
|
@ -101,36 +100,24 @@ class CompletionManager {
|
||||||
}
|
}
|
||||||
|
|
||||||
public async showPreview() {
|
public async showPreview() {
|
||||||
if (!previewDecorationType) {
|
this.completionText = '\n' + this.completionText;
|
||||||
createPreviewDecorationType();
|
const completionLines = this.completionText.split('\n');
|
||||||
}
|
|
||||||
|
|
||||||
const completionLines = this.completionText.split('\n').length;
|
const emptyLine = ''; // Empty line for spacing
|
||||||
|
const previewLines = [emptyLine, ...completionLines, emptyLine];
|
||||||
|
|
||||||
// Adjust the start position to line after the original start position
|
const previewRanges: vscode.DecorationOptions[] = previewLines.map((line, index) => {
|
||||||
const adjustedStartPosition = this.startPosition.translate(0, 0);
|
const actualLineNumber = this.startPosition.line + index;
|
||||||
|
const totalLines = this.textEditor.document.lineCount;
|
||||||
// Step 1: Insert blank lines to make space for the preview
|
const lineNumber = Math.min(totalLines - 1, actualLineNumber);
|
||||||
const edit = new vscode.WorkspaceEdit();
|
|
||||||
const linePadding = '\n'.repeat(completionLines + 1); // Include extra line break for visual separation
|
|
||||||
edit.insert(this.document.uri, adjustedStartPosition, linePadding);
|
|
||||||
await vscode.workspace.applyEdit(edit);
|
|
||||||
|
|
||||||
this.insertedLineCount = completionLines + 1;
|
|
||||||
|
|
||||||
// Step 2: Apply decorations
|
|
||||||
const previewRanges: vscode.DecorationOptions[] = this.completionText.split('\n').map((line, index) => {
|
|
||||||
const lineNumber = adjustedStartPosition.line + index + 1; // Start preview one line later
|
|
||||||
return {
|
return {
|
||||||
range: new vscode.Range(
|
range: new vscode.Range(
|
||||||
new vscode.Position(lineNumber, 0),
|
new vscode.Position(lineNumber, 0),
|
||||||
new vscode.Position(lineNumber, 0)
|
new vscode.Position(lineNumber, Number.MAX_VALUE)
|
||||||
),
|
),
|
||||||
renderOptions: {
|
renderOptions: {
|
||||||
after: {
|
after: {
|
||||||
contentText: line,
|
contentText: line.length > 0 ? ` ${line}` : '',
|
||||||
color: '#888888',
|
|
||||||
fontStyle: 'italic',
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
@ -139,91 +126,27 @@ class CompletionManager {
|
||||||
this.textEditor.setDecorations(previewDecorationType, previewRanges);
|
this.textEditor.setDecorations(previewDecorationType, previewRanges);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
public async acceptCompletion() {
|
public async acceptCompletion() {
|
||||||
const edit = new vscode.WorkspaceEdit();
|
const edit = new vscode.WorkspaceEdit();
|
||||||
const completionLines = this.completionText.split('\n');
|
const startLine = Math.max(0, this.startPosition.line - 1);
|
||||||
const numberOfLines = completionLines.length;
|
|
||||||
|
|
||||||
// Ensure the start position is never negative
|
|
||||||
const safeStartPosition = new vscode.Position(Math.max(0, this.startPosition.line - 1), 0);
|
|
||||||
|
|
||||||
// Prepare the range to replace
|
|
||||||
const rangeToReplace = new vscode.Range(
|
|
||||||
safeStartPosition,
|
|
||||||
this.startPosition.translate(numberOfLines, 0)
|
|
||||||
);
|
|
||||||
|
|
||||||
// Construct the content to insert
|
|
||||||
const contentToInsert = (safeStartPosition.line === 0 ? '' : '\n') + this.completionText + '\n';
|
|
||||||
edit.replace(this.document.uri, rangeToReplace, contentToInsert);
|
|
||||||
|
|
||||||
await vscode.workspace.applyEdit(edit);
|
|
||||||
|
|
||||||
// Clear the preview decorations
|
|
||||||
this.clearPreview();
|
|
||||||
|
|
||||||
// Set activeCompletionManager to null
|
|
||||||
activeCompletionManager = null;
|
|
||||||
|
|
||||||
// Calculate the new cursor position from the inserted content
|
|
||||||
const lastCompletionLine = completionLines[completionLines.length - 1];
|
|
||||||
const newPosition = new vscode.Position(
|
|
||||||
this.startPosition.line + numberOfLines - 1,
|
|
||||||
lastCompletionLine.length
|
|
||||||
);
|
|
||||||
|
|
||||||
// Set the new cursor position
|
|
||||||
this.textEditor.selection = new vscode.Selection(newPosition, newPosition);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
public clearPreview() {
|
|
||||||
this.textEditor.setDecorations(previewDecorationType, []); // Remove all preview decorations
|
|
||||||
}
|
|
||||||
|
|
||||||
public async declineCompletion() {
|
|
||||||
this.clearPreview(); // Clear the preview decorations
|
|
||||||
|
|
||||||
try {
|
|
||||||
const document = this.textEditor.document;
|
|
||||||
const currentPosition = this.textEditor.selection.active;
|
|
||||||
|
|
||||||
// Calculate the range of lines to remove
|
|
||||||
const startLine = this.startPosition.line + 1;
|
|
||||||
const endLine = currentPosition.line;
|
|
||||||
if (endLine > startLine) {
|
|
||||||
const workspaceEdit = new vscode.WorkspaceEdit();
|
|
||||||
|
|
||||||
// Create a range from start of startLine to end of endLine
|
|
||||||
const range = new vscode.Range(
|
const range = new vscode.Range(
|
||||||
new vscode.Position(startLine, 0),
|
new vscode.Position(startLine, 0),
|
||||||
new vscode.Position(endLine, document.lineAt(endLine).text.length)
|
this.startPosition.translate(0, Number.MAX_VALUE)
|
||||||
);
|
);
|
||||||
|
edit.replace(this.document.uri, range, this.completionText);
|
||||||
// Delete the range
|
await vscode.workspace.applyEdit(edit);
|
||||||
workspaceEdit.delete(document.uri, range);
|
this.clearPreview();
|
||||||
|
|
||||||
// Apply the edit
|
|
||||||
await vscode.workspace.applyEdit(workspaceEdit);
|
|
||||||
|
|
||||||
// Move the cursor back to the original position
|
|
||||||
this.textEditor.selection = new vscode.Selection(this.startPosition, this.startPosition);
|
|
||||||
|
|
||||||
console.log(`Lines ${startLine + 1} to ${endLine + 1} removed successfully`);
|
|
||||||
activeCompletionManager = null;
|
|
||||||
} else {
|
|
||||||
console.log('No lines to remove');
|
|
||||||
}
|
}
|
||||||
} catch (error) {
|
|
||||||
console.error('Error declining completion:', error);
|
public clearPreview() {
|
||||||
vscode.window.showErrorMessage(`Error removing lines: ${error}`);
|
this.textEditor.setDecorations(previewDecorationType, []);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public declineCompletion() {
|
||||||
|
this.clearPreview();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
async function autocompleteCommand(textEditor: vscode.TextEditor, edit: vscode.TextEditorEdit, ...args: any[]) {
|
async function autocompleteCommand(textEditor: vscode.TextEditor, edit: vscode.TextEditorEdit, ...args: any[]) {
|
||||||
const cancellationTokenSource = new vscode.CancellationTokenSource();
|
const cancellationTokenSource = new vscode.CancellationTokenSource();
|
||||||
const cancellationToken = cancellationTokenSource.token;
|
const cancellationToken = cancellationTokenSource.token;
|
||||||
|
@ -250,6 +173,49 @@ async function autocompleteCommand(textEditor: vscode.TextEditor, edit: vscode.T
|
||||||
await completionManager.showPreview();
|
await completionManager.showPreview();
|
||||||
activeCompletionManager = completionManager;
|
activeCompletionManager = completionManager;
|
||||||
|
|
||||||
|
let isDisposed = false;
|
||||||
|
|
||||||
|
const dispose = () => {
|
||||||
|
if (!isDisposed) {
|
||||||
|
console.log('Disposing listeners');
|
||||||
|
disposable.dispose();
|
||||||
|
typeDisposable.dispose();
|
||||||
|
activeCompletionManager = null;
|
||||||
|
isDisposed = true;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const disposable = vscode.Disposable.from(
|
||||||
|
vscode.window.onDidChangeTextEditorSelection(async (event) => {
|
||||||
|
if (event.textEditor !== textEditor) return;
|
||||||
|
|
||||||
|
if (event.kind === vscode.TextEditorSelectionChangeKind.Keyboard) {
|
||||||
|
console.log('Accepting completion');
|
||||||
|
await completionManager.acceptCompletion();
|
||||||
|
dispose();
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
vscode.window.onDidChangeActiveTextEditor(() => {
|
||||||
|
console.log('Active editor changed, clearing preview');
|
||||||
|
completionManager.clearPreview();
|
||||||
|
dispose();
|
||||||
|
}),
|
||||||
|
vscode.workspace.onDidChangeTextDocument((event) => {
|
||||||
|
if (event.document === document) {
|
||||||
|
console.log('Document changed, clearing preview');
|
||||||
|
completionManager.clearPreview();
|
||||||
|
dispose();
|
||||||
|
}
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
const typeDisposable = vscode.commands.registerCommand('type', async (args) => {
|
||||||
|
if (args.text === '\b') { // Backspace key
|
||||||
|
console.log('Declining completion');
|
||||||
|
completionManager.declineCompletion();
|
||||||
|
dispose();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
} catch (err: any) {
|
} catch (err: any) {
|
||||||
console.error('Error in autocompleteCommand:', err);
|
console.error('Error in autocompleteCommand:', err);
|
||||||
|
@ -259,19 +225,24 @@ async function autocompleteCommand(textEditor: vscode.TextEditor, edit: vscode.T
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async function handleTab() {
|
async function acceptCompletion() {
|
||||||
if (activeCompletionManager) {
|
if (activeCompletionManager) {
|
||||||
await activeCompletionManager.acceptCompletion();
|
await activeCompletionManager.acceptCompletion();
|
||||||
} else {
|
const editor = vscode.window.activeTextEditor;
|
||||||
await vscode.commands.executeCommand('tab');
|
if (editor) {
|
||||||
|
const lastLine = editor.document.lineAt(editor.document.lineCount - 1);
|
||||||
|
const newPosition = new vscode.Position(lastLine.lineNumber, lastLine.text.length);
|
||||||
|
editor.selection = new vscode.Selection(newPosition, newPosition);
|
||||||
|
}
|
||||||
|
activeCompletionManager = null;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async function handleBackspace() {
|
async function handleTab() {
|
||||||
if (activeCompletionManager) {
|
if (activeCompletionManager) {
|
||||||
await activeCompletionManager.declineCompletion();
|
await acceptCompletion();
|
||||||
} else {
|
} else {
|
||||||
await vscode.commands.executeCommand('deleteLeft');
|
await vscode.commands.executeCommand('tab');
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -318,11 +289,10 @@ export function activate(context: vscode.ExtensionContext) {
|
||||||
vscode.workspace.onDidChangeConfiguration(updateConfig),
|
vscode.workspace.onDidChangeConfiguration(updateConfig),
|
||||||
vscode.languages.registerCompletionItemProvider('*', { provideCompletionItems }, ...config.completionKeys),
|
vscode.languages.registerCompletionItemProvider('*', { provideCompletionItems }, ...config.completionKeys),
|
||||||
vscode.commands.registerTextEditorCommand('fabelous-autocoder.autocomplete', autocompleteCommand),
|
vscode.commands.registerTextEditorCommand('fabelous-autocoder.autocomplete', autocompleteCommand),
|
||||||
vscode.commands.registerCommand('fabelous-autocoder.handleTab', handleTab),
|
vscode.commands.registerCommand('fabelous-autocoder.acceptCompletion', acceptCompletion),
|
||||||
vscode.commands.registerCommand('fabelous-autocoder.handleBackspace', handleBackspace) // Add this line
|
vscode.commands.registerCommand('fabelous-autocoder.handleTab', handleTab)
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
export function deactivate() {}
|
export function deactivate() {}
|
||||||
|
|
Loading…
Reference in New Issue