DEV2-4340 - inline chat interaction added

package.json

@@ -481,6 +481,10 @@
       {
         "key": "ctrl+shift+q",
         "command": "tabnine.chat.focus-input"
+      },
+      {
+        "key": "ctrl+i",
+        "command": "tabnine.chat.commands.inline.action"
       }
     ]
   },

src/tabnineChatWidget/extensionCommands/ChatActionProvider.ts

@@ -0,0 +1,66 @@
+/* eslint-disable class-methods-use-this */
+import {
+  CodeAction,
+  CodeActionContext,
+  CodeActionKind,
+  ExtensionContext,
+  Range,
+  TextDocument,
+  languages,
+  CodeActionProvider,
+} from "vscode";
+import { languagesFilter } from "./const";
+
+export function registerChatActionProvider(context: ExtensionContext) {
+  context.subscriptions.push(
+    languages.registerCodeActionsProvider(
+      languagesFilter,
+      new ChatActionProvider(),
+      {
+        providedCodeActionKinds: [
+          CodeActionKind.RefactorRewrite,
+          CodeActionKind.QuickFix,
+        ],
+      }
+    )
+  );
+}
+
+class ChatActionProvider implements CodeActionProvider {
+  provideCodeActions(
+    document: TextDocument,
+    range: Range,
+    codeActionContext: CodeActionContext & {
+      triggerKind: number;
+    }
+  ): CodeAction[] {
+    if (codeActionContext.triggerKind !== 1) {
+      return [];
+    }
+    const resultActions: CodeAction[] = [];
+
+    if (codeActionContext.diagnostics[0]?.range) {
+      const fixAction = new CodeAction(
+        "Fix with Tabnine",
+        CodeActionKind.QuickFix
+      );
+
+      fixAction.command = {
+        title: fixAction.title,
+        command: "tabnine.chat.commands.fix-code",
+      };
+      resultActions.push(fixAction);
+    }
+    const refactor = new CodeAction(
+      "Ask Tabnine",
+      CodeActionKind.RefactorRewrite
+    );
+
+    refactor.command = {
+      title: refactor.title,
+      command: "tabnine.chat.commands.inline.action",
+    };
+    resultActions.push(refactor);
+    return resultActions;
+  }
+}

src/tabnineChatWidget/extensionCommands/codeLens/ChatCodeLensProvider.ts → src/tabnineChatWidget/extensionCommands/ChatCodeLensProvider.ts

@@ -3,40 +3,37 @@ import {
   CancellationToken,
   CodeLens,
   CodeLensProvider,
-  commands,
   Diagnostic,
   DiagnosticSeverity,
-  DocumentSymbol,
   Event,
   EventEmitter,
+  ExtensionContext,
   languages,
   Location,
-  SymbolInformation,
-  SymbolKind,
   TextDocument,
   Uri,
 } from "vscode";
-import { fireEvent } from "../../../binary/requests/requests";
-
-export type Intents = "test" | "fix" | "explain" | "document";
-export type Actions =
-  | "generate-test-for-code"
-  | "fix-code"
-  | "explain-code"
-  | "document-code";
-
-const CODE_LENS_ACTIONS: [Intents, Actions][] = [
-  ["test", "generate-test-for-code"],
-  ["fix", "fix-code"],
-  ["explain", "explain-code"],
-  ["document", "document-code"],
-];
-
-const VALID_SYMBOLS = [SymbolKind.Function, SymbolKind.Method];
+import { fireEvent } from "../../binary/requests/requests";
+import { getFuctionsSymbols } from "./getFuctionsSymbols";
+import { Action, CODE_LENS_COMMANDS } from "./commands";
+import tabnineExtensionProperties from "../../globals/tabnineExtensionProperties";
+import { languagesFilter } from "./const";
 
 const MAX_LINES = 2500;
 
-export class ChatCodeLensProvider implements CodeLensProvider {
+export default function registerChatCodeLens(context: ExtensionContext) {
+  if (!tabnineExtensionProperties.codeLensEnabled) {
+    return;
+  }
+  context.subscriptions.push(
+    languages.registerCodeLensProvider(
+      languagesFilter,
+      new ChatCodeLensProvider()
+    )
+  );
+}
+
+class ChatCodeLensProvider implements CodeLensProvider {
   private visitedFiles: Set<Uri> = new Set();
 
   private didChangeCodeLenses = new EventEmitter<void>();
@@ -54,9 +51,8 @@ export class ChatCodeLensProvider implements CodeLensProvider {
       return [];
     }
 
-    const documnetSymbols = await commands.executeCommand<
-      (SymbolInformation & DocumentSymbol)[]
-    >("vscode.executeDocumentSymbolProvider", document.uri);
+    const documnetSymbols = await getFuctionsSymbols(document);
+
     if (!documnetSymbols?.length) {
       return [];
     }
@@ -72,25 +68,9 @@ export class ChatCodeLensProvider implements CodeLensProvider {
 
     const lenses: CodeLens[] = [];
 
-    documnetSymbols
-      ?.filter((fn) => VALID_SYMBOLS.includes(fn.kind))
-      .forEach(({ location }) =>
-        lenses.push(...toIntentLens(location, diagnostic), toAskLens(location))
-      );
-
-    documnetSymbols
-      ?.filter((symbol) => symbol.kind === SymbolKind.Class)
-      .forEach((classSymbol) =>
-        classSymbol.children
-          .filter((child) => VALID_SYMBOLS.includes(child.kind))
-          .forEach((method) => {
-            const { location } = (method as unknown) as SymbolInformation;
-            lenses.push(
-              ...toIntentLens(location, diagnostic),
-              toAskLens(location)
-            );
-          })
-      );
+    documnetSymbols.forEach(({ location }) => {
+      lenses.push(...toIntentLens(location, diagnostic), toAskLens(location));
+    });
 
     if (!this.visitedFiles.has(document.uri)) {
       this.visitedFiles.add(document.uri);
@@ -128,20 +108,22 @@ function toIntentLens(
   location: Location,
   diagnostics: Diagnostic[]
 ): CodeLens[] {
-  return CODE_LENS_ACTIONS.filter(([text]) =>
+  return CODE_LENS_COMMANDS.filter(({ text }) =>
     filterRelevantActions(text, location, diagnostics)
-  ).map(
-    ([text, action], index) =>
-      new CodeLens(location.range, {
-        title: `${index === 0 ? "tabnine: " : ""}${text}`,
-        tooltip: `tabnine ${text}`,
-        command: "tabnine.chat.commands.any",
-        arguments: [location.range, action],
-      })
-  );
+  )
+    .sort((a, b) => a.lensOrder - b.lensOrder)
+    .map(
+      ({ text, intent }, index) =>
+        new CodeLens(location.range, {
+          title: `${index === 0 ? "tabnine: " : ""}${text}`,
+          tooltip: `tabnine ${text}`,
+          command: "tabnine.chat.commands.any",
+          arguments: [location.range, intent],
+        })
+    );
 }
 function filterRelevantActions(
-  text: Intents,
+  text: Action,
   location: Location,
   diagnostics: Diagnostic[]
 ): boolean {

src/tabnineChatWidget/extensionCommands/commands.ts

@@ -0,0 +1,77 @@
+export type Scope = "block" | "selection" | "none";
+export type Action = "test" | "fix" | "explain" | "document" | "workspace";
+export type Intent =
+  | "/generate-test-for-code"
+  | "/fix-code"
+  | "/explain-code"
+  | "/document-code"
+  | "/workspace";
+
+enum LensOrder {
+  test = 1,
+  fix = 2,
+  explain = 3,
+  document = 4,
+}
+
+export type Command = {
+  label: string;
+  text: Action;
+  intent: Intent;
+  description: string;
+  scope: Scope[];
+  multistep: boolean;
+};
+export type LensCommand = Command & { lensOrder: number };
+
+export const CODE_LENS_COMMANDS: LensCommand[] = [
+  {
+    label: "$(feedback) explain",
+    text: "explain",
+    intent: "/explain-code",
+    description: "Explain the selected code",
+    scope: ["selection", "block"],
+    multistep: false,
+    lensOrder: LensOrder.explain,
+  },
+  {
+    label: "$(beaker) test",
+    text: "test",
+    intent: "/generate-test-for-code",
+    description: "Write tests for the selected code",
+    scope: ["block"],
+    multistep: false,
+    lensOrder: LensOrder.test,
+  },
+  {
+    label: "$(checklist) document",
+    text: "document",
+    intent: "/document-code",
+    description: "Add documentation for the selected code",
+    scope: ["block"],
+    multistep: false,
+    lensOrder: LensOrder.document,
+  },
+  {
+    label: "$(symbol-property) fix",
+    text: "fix",
+    intent: "/fix-code",
+    description: "Find errors in the selected code and fix them",
+    scope: ["block"],
+    multistep: false,
+    lensOrder: LensOrder.fix,
+  },
+];
+
+export const COMANDS: Command[] = [
+  ...CODE_LENS_COMMANDS,
+  {
+    label: "$(search) workspace",
+    text: "workspace",
+    intent: "/workspace",
+    description:
+      "Ask a question related to any code within your current workspace",
+    scope: ["none"],
+    multistep: true,
+  },
+];

src/tabnineChatWidget/extensionCommands/const.ts

@@ -0,0 +1,16 @@
+export const languagesFilter = [
+  { language: "javascript" },
+  { pattern: "**/*[!d].ts", scheme: "file" },
+  { language: "javascriptreact" },
+  { language: "typescriptreact" },
+  { language: "python" },
+  { language: "ruby" },
+  { language: "go" },
+  { language: "rust" },
+  { language: "swift" },
+  { language: "java" },
+  { language: "c" },
+  { language: "cpp" },
+  { language: "csharp" },
+  { language: "php" },
+];