Merge pull request #14 from VectorCamp/patch/1.0.9

Patch/1.0.9

Author: markos

README.md

@@ -115,6 +115,13 @@ We are actively working to address these issues in an upcoming release.
 
 ## 1.0.8
 - Add intrinsics autocomplete on typing!
+
+## 1.0.9
+- Datatype highlighting
+- Refactoring backend api
+- Select highlight style
+- Fix bug on first session
+
 ---
 
 

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

@@ -616,9 +616,15 @@
             const sessionTabsContainer = document.getElementById('sessionTabs');
             sessionTabsContainer.innerHTML = ''; 
 
-            message.sessions.forEach(sessionId => {
-                const tab = createSessionTab(sessionId, sessionId === message.currentSession);
-            });
+            if (message.sessions && message.sessions.length > 0) {
+                message.sessions.forEach(sessionId => {
+                    createSessionTab(sessionId, sessionId === message.currentSession);
+                });
+            } else {
+                // If extension returned no sessions, auto-create Chat 1
+                createDefaultTabIfMissing();
+                vscode.postMessage({ type: 'switchSession', sessionId: 'Chat 1' });
+            }
         } else if (message.type === 'error') {
             removeSpinner(message.sessionId);
             showError(`⚠️ ${message.error}`, message.sessionId);
@@ -678,7 +684,7 @@
             } else {
                 createDefaultTabIfMissing();
                 vscode.postMessage({ type: 'switchSession', sessionId: 'Chat 1' });
-                vscode.postMessage({ type: 'requestHistory', sessionId: 'Chat 1' });
+                // vscode.postMessage({ type: 'requestHistory', sessionId: 'Chat 1' });
             }
         };
 
@@ -785,6 +791,7 @@
         const createSessionBtn = document.getElementById('createSessionBtn');
         const lastSessionId = localStorage.getItem('lastActiveSession') || defaultSessionId;
 
+        
         sessionNameInput.addEventListener('keydown', (e) => {
             if (e.key === 'Enter') {
                 e.preventDefault();

media/chat.html

@@ -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.8",
+  "version": "1.0.9",
   "publisher": "VectorCamp",
   "engines": {
     "vscode": "^1.100.0"
@@ -117,6 +117,10 @@
       {
         "command": "code.simd.ai.editSIMD",
         "title": "Edit SIMD"
+      },
+      {
+        "command": "code.simd.ai.cycleHighlightMode",
+        "title": "Cycle SIMD Syntax Highlighting Mode"
       }
     ],
     "submenus": [

package.json

@@ -0,0 +1,69 @@
+/*
+* Copyright (c) 2025, VectorCamp PC
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+*    http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+
+import fetch from 'node-fetch';
+
+/**
+ * Fetch datatypes grouped by architecture.
+ *
+ * Example response:
+ * {
+ *   "NEON": ["uint16x8_t", "float32x8_t"],
+ *   "INTEL": ["__m128h", "__m256h"],
+ *   "POWER": ["vector signed char", "vector bool char"]
+ * }
+ */
+
+import { API_BASE, getApiToken } from '../config';
+// process.env.NODE_TLS_REJECT_UNAUTHORIZED = '0';
+
+
+export async function fetchDatatypesByArch(): Promise<Record<string, string[]>> {
+  const apiKey = getApiToken();
+  if (!apiKey) {
+    console.warn('⚠️ API key missing');
+    return {};
+  }
+
+  const endpoint = `${API_BASE}/v1/plugin-datatypes/get-datatypes-list`;
+
+  try {
+    const res = await fetch(endpoint, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Accept': 'application/json'
+      },
+      body: JSON.stringify({ api_key: apiKey })
+    });
+
+    if (!res.ok) {
+      throw new Error(`HTTP ${res.status} ${res.statusText}`);
+    }
+
+    const data = await res.json();
+    if (!data || typeof data !== 'object' || !data.datatypes_list) {
+      throw new Error('Invalid JSON structure');
+    }
+
+    // console.log('✅ Datatypes fetched successfully');
+    return data.datatypes_list as Record<string, string[]>; 
+
+  } catch (err) {
+    console.error('❌ Failed to fetch datatypes:', err);
+    return {};
+  }
+}

src/api/datatypeFetcher.ts

@@ -16,22 +16,21 @@
 
 import { get } from 'http';
 import { getApiToken } from '../config';
-import { PLUGIN_DEFAULT_TOKEN } from '../config';
-
+import { API_BASE, MODEL_NAME , PLUGIN_DEFAULT_TOKEN } from '../config';
 
 export async function callSimdAiWithHistory(messages: { role: string; content: string }[]): Promise<string> {
   const apiToken = getApiToken();
   if (!apiToken) {return '⚠️ API token missing';}
 
   try {
-    const res = await fetch('https://simd.ai/api/chat/completions', {
+    const res = await fetch(`${API_BASE}/chat/completions`, {
       method: 'POST',
       headers: {
         'Content-Type': 'application/json',
         'Authorization': `Bearer ${apiToken}`
       },
       body: JSON.stringify({
-        model: 'SIMD-ai-2506.1.ai:24b',
+        model: MODEL_NAME,
         messages
       })
     });
@@ -56,7 +55,7 @@ let cachedIntrinsics: string[] | null = null;
 export async function fetchIntrinsicNames(): Promise<string[]> {
   let apiToken = getApiToken();
 
-  // if user has not specified api token, use predifined to only see Intel intrinsics
+  // if user has not specified api token, use predifined to only see Intel intrinsics and some Preview
   if (!apiToken) {
     apiToken = PLUGIN_DEFAULT_TOKEN;
   }
@@ -66,7 +65,7 @@ export async function fetchIntrinsicNames(): Promise<string[]> {
   }
 
   try {
-    const response = await fetch('https://simd.ai/api/v1/plugin-intrinsics-list/get-intrinsics-list', {
+    const response = await fetch(`${API_BASE}/v1/plugin-intrinsics-list/get-intrinsics-list`, {
       method: 'POST',
       headers: {
         'Content-Type': 'application/json'
@@ -111,6 +110,32 @@ export async function fetchIntrinsicNames(): Promise<string[]> {
   }
 }
 
+export async function sendToSimdAI(userPrompt: string) {
+  const apiToken = getApiToken();
+  if (!apiToken) {throw new Error('API token missing');}
+
+  const endpoint = `${API_BASE}/chat/completions`;
+
+  const response = await fetch(endpoint, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${apiToken}`
+    },
+    body: JSON.stringify({
+      model: MODEL_NAME,
+      messages: [{ role: 'user', content: userPrompt }]
+    })
+  });
+
+  if (!response.ok) {
+    throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+  }
+
+  const data = await response.json();
+  return data;
+}
+
 interface Prototype {
   key: string;
   output?: string;
@@ -149,7 +174,7 @@ export async function fetchIntrinsicInfo(word: string): Promise<TooltipData | nu
   }
 
   try {
-    const response = await fetch(`https://simd.ai/api/v1/plugin-intrinsic-info/get-intrinsics-info`, {
+    const response = await fetch(`${API_BASE}/v1/plugin-intrinsic-info/get-intrinsics-info`, {
       method: 'POST',
       headers: {
         'Content-Type': 'application/json'

src/api/simdAi.ts

@@ -19,15 +19,10 @@ import fetch from 'node-fetch';
 
 import * as vscode from 'vscode';
 import { fetchIntrinsicInfo } from './simdAi';
-
-
-
 // process.env.NODE_TLS_REJECT_UNAUTHORIZED = '0'; // Dev only — allow self-signed certs
 
 const tooltipCache: Record<string, string> = {};
 
-
-
 export async function fetchTooltip(word: string): Promise<string> {
   if (tooltipCache[word]) {return tooltipCache[word];}
 
@@ -102,3 +97,46 @@ export async function fetchTooltip(word: string): Promise<string> {
     return '';
   }
 }
+
+
+import { fetchDatatypesByArch } from './datatypeFetcher';
+
+const datatypeTooltipCache: Record<string, string> = {};
+
+export async function fetchDatatypeTooltip(name: string): Promise<string> {
+  return ''; // Coming soon datatype tooltips
+  if (datatypeTooltipCache[name]) {return datatypeTooltipCache[name];}
+
+  try {
+    const datatypesByArch = await fetchDatatypesByArch();
+    if (!datatypesByArch || !Object.keys(datatypesByArch).length) {
+      console.warn('No datatypes available');
+      return '';
+    }
+
+    const md = new vscode.MarkdownString();
+    md.isTrusted = true;
+    md.supportHtml = true;
+
+    let found = false;
+    for (const [arch, names] of Object.entries(datatypesByArch)) {
+      if (names.includes(name)) {
+        found = true;
+        const link = `https://simd.info/datatype/${encodeURIComponent(name)}`;
+        md.appendMarkdown(`### [${name}](${link}) (${arch})\n`);
+        md.appendMarkdown(`Description for ${name} on ${arch}.\n`); 
+      }
+    }
+
+    if (!found) {
+      md.appendMarkdown(`No datatype info found for **${name}**`);
+    }
+
+    const markdown = md.value;
+    datatypeTooltipCache[name] = markdown;
+    return markdown;
+  } catch (error) {
+    console.error('Failed to fetch datatype tooltip for', name, error);
+    return '';
+  }
+}

src/api/tooltipFetcher.ts

@@ -24,9 +24,11 @@ export function getApiToken(): string | undefined {
   return token;
 }
 
-import * as dotenv from 'dotenv';
-import path from 'path';
+// import * as dotenv from 'dotenv';
+// import path from 'path';
 
-dotenv.config({ path: path.resolve(__dirname, '../.env') });
+// dotenv.config({ path: path.resolve(__dirname, '../.env') });
 
-export const PLUGIN_DEFAULT_TOKEN = "sk-b5204816b07e49ec8096c9e8b34f28bb";
+export const PLUGIN_DEFAULT_TOKEN = "sk-b5204816b07e49ec8096c9e8b34f28bb";
+export const API_BASE = 'https://simd.ai/api';
+export const MODEL_NAME = 'SIMD-ai-2506.1.ai:24b';

src/config.ts

@@ -23,8 +23,8 @@ import { translationState, clearState } from './translation/state';
 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
+import { highlightIntrinsicsAndDatatypes, initIntrinsicHighlighting, deactivateHighlighting, cycleHighlightMode } from './syntaxHighlighting';
+import { activate as activateCompletion } from './completionProvider';
 
 export function activate(context: vscode.ExtensionContext) {
 
@@ -47,10 +47,14 @@ export function activate(context: vscode.ExtensionContext) {
 	context.subscriptions.push(
 		vscode.languages.registerCodeLensProvider({ scheme: 'file', language: '*' }, new TranslationCodeLensProvider())
 	);
-
+	context.subscriptions.push(
+		vscode.commands.registerCommand('code.simd.ai.cycleHighlightMode', () => {
+			cycleHighlightMode(context);
+		})
+	);
 	initIntrinsicHighlighting(context);
 	vscode.window.visibleTextEditors.forEach(editor => {
-        highlightIntrinsics(editor);
+        highlightIntrinsicsAndDatatypes(editor);
     });
 
 	activateCompletion(context);

src/extension.ts

@@ -0,0 +1,37 @@
+/*
+* Copyright (c) 2025, VectorCamp PC
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+*    http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+
+export const HighlightConfig: Record<
+  string,
+  { gradient: string; flat: string }
+> = {
+  NEON: {
+    gradient: 'linear-gradient(90deg, #00FF7F, #32CD32, #228B22)',
+    flat: '#32CD32',
+  },
+  INTEL: {
+    gradient: 'linear-gradient(90deg, #00BFFF, #1E90FF, #4169E1)',
+    flat: '#1E90FF',
+  },
+  POWER: {
+    gradient: 'linear-gradient(90deg, #7B68EE, #6A5ACD, #483D8B)',
+    flat: '#6A5ACD',
+  },
+  DEFAULT: {
+    gradient: 'linear-gradient(90deg, #ADFF2F, #7FFF00, #32CD32)',
+    flat: '#7FFF00',
+  },
+};