Merge branch 'main' into feature/support-deepseek

Author: olasunkanmi-SE

README.md

@@ -9,58 +9,97 @@ https://marketplace.visualstudio.com/items?itemName=fiatinnovations.ola-code-bud
 
 
 ## Architecture
-
 ```mermaid
-graph TB
-    subgraph VSCode["VS Code Extension"]
-        Editor["Editor Interface"]
-        Commands["Commands Layer"]
-        Webview["Webview UI (React)"]
+flowchart TD
+
+    %% VS Code Extension Frontend
+    subgraph "VS Code Extension Frontend"
+        CE["Editor Interface"]:::frontend
+        CL["Commands Layer"]:::frontend
+        WV1["Webview (VS Code)"]:::frontend
+        WV2["Webview (React UI)"]:::frontend
     end
 
-    subgraph Core["Core Application"]
-        direction TB
-        Agents["AI Agents Layer"]
-        AppServices["Application Services"]
-        Infrastructure["Infrastructure Layer"]
-        Memory["Memory System"]
+    %% Core Application Layer
+    subgraph "Core Application"
+        AA["AI Agents"]:::core
+        MS["Memory System"]:::core
+        subgraph "Application Services"
+            BL["Business Logic"]:::core
+            AI["Application Interfaces"]:::core
+        end
+        subgraph "Infrastructure Layer"
+            HTTP["HTTP Services"]:::infra
+            LOG["Logging"]:::infra
+            REP["Repository"]:::infra
+            LS["Local Storage"]:::infra
+        end
     end
 
-    subgraph AIProviders["AI Providers"]
-        Gemini["Gemini API"]
-        Groq["Groq API"]
-        Anthropic["Anthropic API"]
-        XGrok["XGrok API"]
-        Deepseek["Deepseel API"]
+    %% AI Providers
+    subgraph "AI Providers"
+        LLM["Language Model Integrations"]:::provider
+        ESP["External Service Providers"]:::provider
     end
 
-    subgraph Storage["Storage Layer"]
-        SQLite["SQLite Database"]
-        VectorDB["Vector Embeddings"]
-        FileSystem["File System Service"]
+    %% Storage Layer
+    subgraph "Storage Layer"
+        DB["Database (SQLite)"]:::storage
+        FS["File System"]:::storage
+        VD["Vector Database"]:::storage
     end
 
-    %% Main Flow Connections
-    Editor -->|"User Input"| Commands
-    Commands -->|"Process Request"| Agents
-    Agents -->|"Context Management"| Memory
-    Agents -->|"API Requests"| Infrastructure
-    Infrastructure -->|"Model Selection"| AIProviders
-    AppServices -->|"Data Access"| Storage
-    Memory -->|"Store Context"| Storage
-    Infrastructure -->|"Response"| Webview
-
-    %% Additional Connections
-    Agents -->|"Business Logic"| AppServices
-    Commands -->|"UI Updates"| Webview
-    FileSystem -->|"Code Analysis"| Agents
-    
-
-    class VSCode vscode
-    class Core core
-    class AIProviders ai
-    class Storage storage
+    %% Connections between VS Code Extension Frontend
+    CE -->|"UserInput"| CL
+    CL -->|"ProcessRequest"| AA
+    CL -->|"UIUpdate"| WV1
+
+    %% Connections within Core Application
+    AA -->|"ContextManagement"| MS
+    AA -->|"Orchestration"| BL
+    AA -->|"ContractCall"| AI
+
+    %% Connections to Infrastructure
+    AA -->|"APIRequest"| HTTP
+    HTTP -->|"APICall"| LLM
+    HTTP -->|"APICall"| ESP
+    HTTP -->|"Feedback"| WV1
+
+    %% Connections from Application Services to Storage
+    BL -->|"DataAccess"| DB
+    BL -->|"FileAccess"| FS
+    BL -->|"EmbedData"| VD
+    AI -->|"RepositoryAccess"| REP
+
+    %% Memory System stores context to Storage
+    MS -->|"StoreContext"| DB
+
+    %% Styling Classes
+    classDef frontend fill:#cce5ff,stroke:#2a6592,stroke-width:2px;
+    classDef core fill:#d4edda,stroke:#155724,stroke-width:2px;
+    classDef infra fill:#f8d7da,stroke:#a71d2a,stroke-width:2px;
+    classDef provider fill:#fff3cd,stroke:#856404,stroke-width:2px;
+    classDef storage fill:#d1ecf1,stroke:#0c5460,stroke-width:2px;
+
+    %% Click Events
+    click CE "https://github.com/olasunkanmi-se/codebuddy/blob/main/src/extension.ts"
+    click CL "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/commands"
+    click WV1 "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/webview"
+    click WV2 "https://github.com/olasunkanmi-se/codebuddy/tree/main/webviewUi"
+    click AA "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/agents"
+    click MS "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/memory"
+    click BL "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/services"
+    click AI "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/application"
+    click HTTP "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/infrastructure/http"
+    click LOG "https://github.com/olasunkanmi-se/codebuddy/blob/main/src/infrastructure/logger/logger.ts"
+    click REP "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/infrastructure/repository"
+    click LS "https://github.com/olasunkanmi-se/codebuddy/blob/main/src/infrastructure/storage/local-storage.ts"
+    click LLM "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/llms"
+    click ESP "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/providers"
+    click DB "https://github.com/olasunkanmi-se/codebuddy/tree/main/src/infrastructure/repository"
+    click FS "https://github.com/olasunkanmi-se/codebuddy/blob/main/src/services/file-system.ts"
 ```
+
 ### Database
 - SQLite database for code pattern storage
 - Vector embeddings for semantic code search

package.json

@@ -78,15 +78,6 @@
           "command": "CodeBuddy.explain",
           "group": "CodeBuddy"
         },
-        {
-          "when": "editorHasSelection",
-          "command": "CodeBuddy.sendChatMessage",
-          "group": "CodeBuddy"
-        },
-        {
-          "command": "CodeBuddy.savePattern",
-          "group": "CodeBuddy"
-        },
         {
           "when": "editorHasSelection",
           "command": "CodeBuddy.readFromKnowledgeBase",
@@ -134,18 +125,10 @@
         "command": "CodeBuddy.codeOptimize",
         "title": "CodeBuddy. Optimize these selected code."
       },
-      {
-        "command": "CodeBuddy.sendChatMessage",
-        "title": "Send to CodeBuddy"
-      },
       {
         "command": "CodeBuddy.explain",
         "title": "CodeBuddy. Explain this code"
       },
-      {
-        "command": "CodeBuddy.savePattern",
-        "title": "CodeBuddy. Save this code pattern"
-      },
       {
         "command": "CodeBuddy.readFromKnowledgeBase",
         "title": "CodeBuddy. Generate based on knowledgeBase"

src/application/interfaces/workspace.interface.ts

@@ -0,0 +1,31 @@
+import { UUID } from "crypto";
+
+export interface IWorkspaceService {
+  getWorkspaceFiles(rootPath: string): Promise<Record<string, string>>;
+  getActiveFileContent(): string | undefined;
+  getContextInfo(useWorkspaceContext: boolean): Promise<IContextInfo>;
+}
+
+export interface IContextInfo {
+  activeFileContent?: string;
+  workspaceFiles?: Map<string, FolderEntry[]>;
+  openFiles: IOpenFiles[];
+}
+
+export interface IOpenFiles {
+  path: string;
+  language: string;
+}
+
+export interface FolderEntry {
+  id: UUID;
+  type: "folder";
+  name: string;
+  children: (FolderEntry | FileEntry)[];
+}
+
+export interface FileEntry {
+  id: UUID;
+  type: "file";
+  name: string;
+}

src/extension.ts

@@ -1,3 +1,4 @@
+import * as path from "path";
 import * as vscode from "vscode";
 import {
   APP_CONFIG,
@@ -18,6 +19,7 @@ import { OptimizeCode } from "./commands/optimize";
 import { RefactorCode } from "./commands/refactor";
 import { ReviewCode } from "./commands/review";
 import { EventEmitter } from "./emitter/agent-emitter";
+import { Logger } from "./infrastructure/logger/logger";
 import { dbManager } from "./infrastructure/repository/data-base-manager";
 import { Memory } from "./memory/base";
 import { AnthropicWebViewProvider } from "./providers/anthropic";
@@ -43,40 +45,48 @@ const {
   deepseekModel,
 } = APP_CONFIG;
 
-const connectDB = async () => {
-  await dbManager.connect(
-    "file:/Users/olasunkanmi/Documents/Github/codebuddy/patterns/dev.db",
-  );
-};
+const logger = new Logger("extension");
 
 let quickFixCodeAction: vscode.Disposable;
 let agentEventEmmitter: EventEmitter;
 
+async function connectToDatabase() {
+  try {
+    const dbPath = path.join(__dirname, "..", "patterns", "dev.db");
+    const urlPath = dbPath.replace(/\\/g, "/");
+    const isWindows = dbPath.includes("\\");
+    const filePath = isWindows ? `file:/${urlPath}` : `file:${urlPath}`;
+    await dbManager.connect(filePath);
+  } catch (error: any) {
+    logger.error("Unable to connect to DB", error);
+    throw new Error(error);
+  }
+}
+
+async function createFileDB(context: vscode.ExtensionContext) {
+  try {
+    const fileUploader = new FileUploader(context);
+    const files = await fileUploader.getFiles();
+    if (!files?.find((file) => file.includes("dev.db"))) {
+      await fileUploader.createFile("dev.db");
+    }
+  } catch (error: any) {
+    logger.error("Unable to ", error);
+    throw new Error(error);
+  }
+}
+
 export async function activate(context: vscode.ExtensionContext) {
   try {
+    await createFileDB(context);
+    await connectToDatabase();
     const credentials = new Credentials();
     await credentials.initialize(context);
     const session: vscode.AuthenticationSession | undefined =
       await credentials.getSession();
-    vscode.window.showInformationMessage(
-      `Logged into GitHub as ${session?.account.label}`,
-    );
+    logger.info(`Logged into GitHub as ${session?.account.label}`);
     Memory.getInstance();
-    await connectDB();
-    // const web = WebSearchService.getInstance();
-    // const x = CodeRepository.getInstance();
-    // const apiKey = getGeminiAPIKey();
-    // const embeddingService = new EmbeddingService(apiKey);
-    // const embedding = await embeddingService.generateEmbedding("is jwt web token used withnin this app ?");
-    // const y = await x.searchSimilarFunctions(embedding, 2);
-    // console.log(y);
     const fileUpload = new FileUploader(context);
-    // await fileUpload.createFile("aiix.db");
-
-    // const files = await fileUpload.getFiles();
-    // const names = await fileUpload.getFileNames();
-    // console.log(files, names);
-
     // const index = CodeIndexingService.createInstance();
     // Get each of the folders and call the next line for each
     // const result = await index.buildFunctionStructureMap();