Some improvements for VS code extension Library index

Author: wladimiiir

packages/common/src/dto/function/function.dto.ts

@@ -19,6 +19,7 @@ export interface FunctionDefinitionDto {
   id: string;
   context: string;
   name: string;
+  description: string;
   arguments: FunctionArgument[];
   returnType: string;
   customCode?: string;

packages/vscode-extension/CHANGELOG.md

@@ -1,3 +1,6 @@
+### 0.1.8
+* Some library index tree related improvements
+
 ### 0.1.7
 * Added library index tree
 

packages/vscode-extension/package.json

@@ -3,7 +3,7 @@
   "displayName": "Poly Assistant",
   "description": "Poly AI assistant for VS Code",
   "icon": "resources/poly-logo.png",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "publisher": "Poly API Corp",
   "engines": {
     "vscode": "^1.75.0"
@@ -76,12 +76,14 @@
         {
           "type": "tree",
           "id": "poly.library-index-view",
-          "name": "Library",
+          "icon": "resources/poly-library-index.png",
+          "name": "poly",
           "when": "polyLibraryInstalled == true"
         },
         {
           "id": "poly.library-info",
-          "name": "Library",
+          "name": "poly",
+          "icon": "resources/poly-library-index.png",
           "when": "polyLibraryInstalled == false"
         }
       ]

packages/vscode-extension/resources/poly-library-index.png

@@ -11,13 +11,22 @@ class LibraryTreeItem extends vscode.TreeItem {
     super(label, collapsibleState);
     switch (data.type) {
       case 'function':
-        this.tooltip = new vscode.MarkdownString(`**Function**  \n${data.name}(${data.arguments.map(arg => `${arg.name}: ${arg.type}`).join(', ')})`);
+        const title = data.customCode ? 'Custom function' : 'Function';
+        this.tooltip = new vscode.MarkdownString(
+          `**${title}**\n\n---\n\n${data.description
+            ? `${data.description}\n\n---\n\n`
+            : ''}${data.name}(${data.arguments.map(arg => `${arg.name}: ${arg.type}`).join(', ')})`,
+        );
         break;
       case 'webhookHandle':
-        this.tooltip = new vscode.MarkdownString(`**Webhook listener**  \n${data.name}()`);
+        this.tooltip = new vscode.MarkdownString(
+          `**Webhook listener**\n\n---\n\n${data.name}()`,
+        );
         break;
       default:
-        this.tooltip = new vscode.MarkdownString(`**Context**  \n${label}`);
+        this.tooltip = new vscode.MarkdownString(
+          `**Context**\n\n---\n\n${label}`,
+        );
         break;
     }
   }
@@ -55,13 +64,11 @@ export default class LibraryIndexViewProvider implements vscode.TreeDataProvider
   }
 
   getTreeItem(element: LibraryTreeItem): TreeItem | Thenable<TreeItem> {
-    if (element.data.type === 'function' || element.data.type === 'webhookHandle') {
-      element.command = {
-        title: 'Copy to clipboard',
-        command: 'poly.copyLibraryItem',
-        arguments: [element],
-      };
-    }
+    element.command = {
+      title: 'Copy to clipboard',
+      command: 'poly.copyLibraryItem',
+      arguments: [element],
+    };
     return element;
   }
 
@@ -72,7 +79,7 @@ export default class LibraryIndexViewProvider implements vscode.TreeDataProvider
   }
 
   static copyLibraryItem(item: LibraryTreeItem) {
-    const { parentPath, data } = item;
+    const { parentPath, data, label } = item;
     switch (data.type) {
       case 'function':
         vscode.env.clipboard.writeText(`await ${parentPath}.${data.name}(${data.arguments.map(arg => `${arg.name}`).join(', ')});`);
@@ -81,7 +88,10 @@ export default class LibraryIndexViewProvider implements vscode.TreeDataProvider
         vscode.env.clipboard.writeText(`${parentPath}.${data.name}(data => {\n\n});`);
         break;
       default:
+        console.log('%c ', 'background: yellow; color: black', item);
+        vscode.env.clipboard.writeText(`await ${parentPath}.${label}.`);
         break;
     }
+    vscode.window.showInformationMessage('Copied');
   }
 }

packages/vscode-extension/src/library-index-view-provider.ts

@@ -292,6 +292,7 @@ export class FunctionService {
     return {
       id: urlFunction.publicId,
       name: urlFunction.name,
+      description: urlFunction.description,
       context: urlFunction.context,
       arguments: this.getArguments(urlFunction),
       returnType: urlFunction.responseType,
@@ -313,6 +314,7 @@ export class FunctionService {
     return {
       id: customFunction.publicId,
       name: customFunction.name,
+      description: customFunction.description,
       context: customFunction.context,
       arguments: JSON.parse(customFunction.arguments),
       returnType: customFunction.returnType,