Merge pull request #13 from VectorCamp/feature/autocomplete

Feature/autocomplete

Author: markos

README.md

@@ -112,6 +112,9 @@ We are actively working to address these issues in an upcoming release.
 
 ## 1.0.7
 - Fix ai chat issues on session switching
+
+## 1.0.8
+- Add intrinsics autocomplete on typing!
 ---
 
 

code-simd-ai-1.0.7.vsix code-simd-ai-1.0.8.vsixcode-simd-ai-1.0.7.vsix renamed to code-simd-ai-1.0.8.vsix

@@ -2,7 +2,7 @@
   "name": "code-simd-ai",
   "displayName": "code.simd.ai",
   "description": "VScode plugin for porting of code between different cpu architectures",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "publisher": "VectorCamp",
   "engines": {
     "vscode": "^1.100.0"

package.json

@@ -0,0 +1,76 @@
+import * as vscode from 'vscode';
+import { getIntrinsics } from './intrinsicsCache';
+import { fetchTooltip } from './api/tooltipFetcher';
+
+let intrinsics: string[] = [];
+
+export async function activate(context: vscode.ExtensionContext) {
+  try {
+    intrinsics = await getIntrinsics();
+    intrinsics.sort(); // required for binary search
+  } catch (err) {
+    console.error('Failed to fetch intrinsics:', err);
+    intrinsics = [];
+  }
+
+  const languages = ['rust', 'c', 'cpp'];
+
+  const provider = vscode.languages.registerCompletionItemProvider(
+    languages.map(lang => ({ scheme: 'file', language: lang })),
+    {
+        provideCompletionItems(document, position) {
+          const linePrefix = document.lineAt(position).text.substring(0, position.character);
+          
+          const wordMatch = linePrefix.match(/[a-zA-Z_][a-zA-Z0-9_]*$/);
+          const prefix = wordMatch ? wordMatch[0] : '';
+          
+          if (!prefix) {return undefined;}
+          
+          const start = lowerBound(intrinsics, prefix);
+          const maxResults = 50;
+          const items: vscode.CompletionItem[] = [];
+          
+          for (let i = start; i < intrinsics.length && items.length < maxResults; i++) {
+              if (!intrinsics[i].startsWith(prefix)) {break;}
+              const name = intrinsics[i];
+              const item = new vscode.CompletionItem(name, vscode.CompletionItemKind.Function);
+              item.detail = 'SIMD.info intrinsic';
+              item.filterText = name;
+              item.sortText = name;
+              
+              items.push(item);
+          }
+          return items;
+        },
+        async resolveCompletionItem(item: vscode.CompletionItem) {
+          try {
+              const doc = await fetchTooltip(item.label.toString());
+              if (doc) {
+                const markdown = new vscode.MarkdownString(doc);
+                markdown.isTrusted = true;
+                markdown.supportHtml = true; 
+                item.documentation = markdown;
+              }
+          } catch (err) {
+            console.error('Error fetching tooltip:', err);
+          }
+          return item;
+        }
+    },
+    '_','m','v' // trigger characters
+    );
+
+
+  context.subscriptions.push(provider);
+}
+
+// Returns the index of the first string >= prefix
+function lowerBound(arr: string[], prefix: string) {
+  let lo = 0, hi = arr.length;
+  while (lo < hi) {
+    const mid = Math.floor((lo + hi) / 2);
+    if (arr[mid] < prefix) {lo = mid + 1;}
+    else {hi = mid;}
+  }
+  return lo;
+}

src/completionProvider.ts

@@ -24,6 +24,7 @@ import { registerTranslateCommand } from './translation/translator';
 import { registerAcceptRejectCommands, registerEditSIMDCommand } from './translation/commands';
 import { TranslationCodeLensProvider } from './translation/codelens';
 import { highlightIntrinsics, initIntrinsicHighlighting, deactivateHighlighting } from './syntaxHighlighting';
+import { activate as activateCompletion } from './completionProvider'; // Update to correct filename
 
 export function activate(context: vscode.ExtensionContext) {
 
@@ -51,6 +52,8 @@ export function activate(context: vscode.ExtensionContext) {
 	vscode.window.visibleTextEditors.forEach(editor => {
         highlightIntrinsics(editor);
     });
+
+	activateCompletion(context);
 }
 
 export function deactivate() {

src/extension.ts

@@ -0,0 +1,15 @@
+import { fetchIntrinsicNames } from './api/simdAi';
+
+let intrinsics: string[] | null = null;
+
+export async function getIntrinsics(): Promise<string[]> {
+  if (!intrinsics) {
+    try {
+      intrinsics = await fetchIntrinsicNames();
+    } catch (err) {
+      console.error("Failed to fetch intrinsics:", err);
+      intrinsics = [];
+    }
+  }
+  return intrinsics;
+}

src/intrinsicsCache.ts

@@ -16,15 +16,14 @@
 import * as vscode from 'vscode';
 import * as fs from 'fs';
 import { fetchTooltip } from './api/tooltipFetcher';
-import { fetchIntrinsicNames } from './api/simdAi';
-
+import { getIntrinsics } from './intrinsicsCache';
 
 let intrinsics: string[] = [];
 let decorationType: vscode.TextEditorDecorationType | null = null;
 
 export async function initIntrinsicHighlighting(context: vscode.ExtensionContext) {
   try {
-    intrinsics = await fetchIntrinsicNames();
+    intrinsics = await getIntrinsics();
   } catch (error) {
     console.error('Failed to fetch intrinsics:', error);
   }