More reliable user input detect + undo

package.json

@@ -2,7 +2,7 @@
 	"name": "ollama-autocoder",
 	"displayName": "Ollama Autocoder",
 	"description": "A simple to use Ollama autocompletion engine with options exposed and streaming functionality",
-	"version": "0.0.6",
+	"version": "0.0.7",
 	"icon": "icon.png",
 	"publisher": "10nates",
 	"license": "MIT",

src/extension.ts

@@ -38,6 +38,18 @@ updateVSConfig();
 // No need for restart for any of these settings
 vscode.workspace.onDidChangeConfiguration(updateVSConfig);
 
+// internal function to handle undo functionality
+async function handleUndo(document: vscode.TextDocument, startingPosition: vscode.Position, endingPosition: vscode.Position, optons?: vscode.MessageOptions) {
+	vscode.window.showInformationMessage("Ollama autocompletion was cancelled. Would you like it to undo its changes?", "Keep", "Undo").then(r => {
+		if (r === "Undo") {	
+			const edit = new vscode.WorkspaceEdit();
+			const range = new vscode.Range(startingPosition, endingPosition);
+			edit.delete(document.uri, range);
+			vscode.workspace.applyEdit(edit);
+		}
+	});
+}
+
 // internal function for autocomplete, not directly exposed
 async function autocompleteCommand(textEditor: vscode.TextEditor, cancellationToken?: vscode.CancellationToken) {
 	const document = textEditor.document;
@@ -58,10 +70,16 @@ async function autocompleteCommand(textEditor: vscode.TextEditor, cancellationTo
 			try {
 				progress.report({ message: "Starting model..." });
 
+				//tracker
+				// let oldPosition = position;
+				let currentPosition = position;
+				// let lastToken = "";
+
 				let axiosCancelPost: () => void;
 				const axiosCancelToken = new axios.CancelToken((c) => {
 					const cancelPost = function () {
 						c("Autocompletion request terminated by user cancel");
+						handleUndo(document, position, currentPosition);
 					};
 					axiosCancelPost = cancelPost;
 					if (cancellationToken) cancellationToken.onCancellationRequested(cancelPost);
@@ -85,27 +103,31 @@ async function autocompleteCommand(textEditor: vscode.TextEditor, cancellationTo
 				}
 				);
 
-				//tracker
-				let oldPosition = position;
-				let currentPosition = position;
-				let lastToken = "";
-
-
 				response.data.on('data', async (d: Uint8Array) => {
 					progress.report({ message: "Generating..." });
 
 					// Check for user input (cancel)
-					if (lastToken != "") {
+					if (currentPosition != textEditor.selection.start) {
-						const lastInput = document.getText(new vscode.Range(oldPosition, textEditor.selection.active));
+						console.log(currentPosition, textEditor.selection.start);
-						if (lastInput !== lastToken) {
 						axiosCancelPost(); // cancel axios => cancel finished promise => close notification
 						return;
 					}
-					}
+					// if (lastToken != "") {
+					// 	const lastInput = document.getText(new vscode.Range(oldPosition, textEditor.selection.active));
+					// 	if (lastInput !== lastToken) {
+					// 		console.log(lastInput, lastToken);
+					// 		axiosCancelPost(); // cancel axios => cancel finished promise => close notification
+					// 		return;
+					// 	}
+					// }
 
 					// Get a completion from the response
 					const completion: string = JSON.parse(d.toString()).response;
-					lastToken = completion;
+					// lastToken = completion;
+					
+					if (completion === "") {
+						return;
+					}
 
 					//complete edit for token
 					const edit = new vscode.WorkspaceEdit();
@@ -126,7 +148,7 @@ async function autocompleteCommand(textEditor: vscode.TextEditor, cancellationTo
 						newPosition,
 						newPosition
 					);
-					oldPosition = currentPosition;
+					// oldPosition = currentPosition;
 					currentPosition = newPosition;
 
 					// completion bar
@@ -161,11 +183,8 @@ async function autocompleteCommand(textEditor: vscode.TextEditor, cancellationTo
 	);
 }
 
-// This method is called when extension is activated
+// Completion item provider callback for activate
-function activate(context: vscode.ExtensionContext) {
+async function provideCompletionItems(document: vscode.TextDocument, position: vscode.Position, cancellationToken: vscode.CancellationToken) {
-	// Register a completion provider for JavaScript files
-	const completionProvider = vscode.languages.registerCompletionItemProvider("*", {
-		async provideCompletionItems(document, position, cancellationToken) {
 
 	// Create a completion item
 	const item = new vscode.CompletionItem("Autocomplete with Ollama");
@@ -218,7 +237,13 @@ function activate(context: vscode.ExtensionContext) {
 	};
 	// Return the completion item
 	return [item];
-		},
+}
+
+// This method is called when extension is activated
+function activate(context: vscode.ExtensionContext) {
+	// Register a completion provider for JavaScript files
+	const completionProvider = vscode.languages.registerCompletionItemProvider("*", {
+		provideCompletionItems
 	},
 		...completionKeys.split("")
 	);