add getWorkspaceRootPaths util (#1389)

src/WorkspaceUpdater.ts

@@ -2,8 +2,8 @@ import * as vscode from "vscode";
 import { BINARY_UPDATE_WORKSPACE_INTERVAL } from "./globals/consts";
 import sendUpdateWorkspaceRequest from "./binary/requests/workspace";
 import { Logger } from "./utils/logger";
-import { fireEvent, tabNineProcess } from "./binary/requests/requests";
-import { Capability, isCapabilityEnabled } from "./capabilities/capabilities";
+import { tabNineProcess } from "./binary/requests/requests";
+import { getWorkspaceRootPaths } from "./utils/workspaceFolders";
 
 const INITIAL_DELAY = 3000;
 
@@ -29,40 +29,15 @@ export class WorkspaceUpdater implements vscode.Disposable {
   }
 }
 
-function sendEvent(event: string, payload: Record<string, unknown> = {}) {
-  if (isCapabilityEnabled(Capability.WORKSPACE_TRACE)) {
-    void fireEvent({
-      name: event,
-      ...payload,
-    });
-  }
-}
-
 function updateWorkspace() {
-  sendEvent("workspace_starting");
-  const rootPaths =
-    vscode.workspace.workspaceFolders
-      ?.filter((wf) => {
-        const isFileUrl = wf.uri.scheme === "file";
-        if (!isFileUrl) {
-          sendEvent("workspace_protocol_not_file", {
-            path: wf.uri.toString(),
-          });
-        }
-        return isFileUrl;
-      })
-      .map((wf) => wf.uri.fsPath) || [];
+  const rootPaths = getWorkspaceRootPaths() || [];
 
   Logger.debug(
     `Updating root paths for project ${
       vscode.workspace.name || "unknown"
     }: ${JSON.stringify(rootPaths)}`
   );
 
-  sendEvent("workspace_sending_request", {
-    root_paths_len: rootPaths.length,
-  });
-
   void sendUpdateWorkspaceRequest({
     root_paths: rootPaths,
   });

src/capabilities/capabilities.ts

@@ -44,7 +44,6 @@ export enum Capability {
   TEST_GEN = "vscode_test_gen",
   FORCE_REGISTRATION = "plugin.feature.force_registration",
   TABNINE_CHAT = "plugin.feature.tabnine_chat",
-  WORKSPACE_TRACE = "plugin.feature.workspace_trace",
 }
 
 let enabledCapabilities: Record<string, boolean> | null = null;

src/tabnineChatWidget/ChatApi.ts

@@ -31,6 +31,7 @@ import {
   NavigateToLocationPayload,
   navigateToLocation,
 } from "./handlers/navigateToLocation";
+import { getWorkspaceRootPaths } from "../utils/workspaceFolders";
 
 type GetUserResponse = {
   token: string;
@@ -235,9 +236,7 @@ export function initChatApi(
     .registerEvent<void, WorkspaceFolders | undefined>(
       "workspace_folders",
       () => {
-        const rootPaths = vscode.workspace.workspaceFolders
-          ?.filter((wf) => wf.uri.scheme === "file")
-          .map((wf) => wf.uri.fsPath);
+        const rootPaths = getWorkspaceRootPaths();
         if (!rootPaths) return undefined;
 
         return {

src/utils/workspaceFolders.ts

@@ -0,0 +1,7 @@
+import * as vscode from "vscode";
+
+export function getWorkspaceRootPaths() {
+  return vscode.workspace.workspaceFolders
+    ?.filter((wf) => wf.uri.scheme === "file")
+    .map((wf) => wf.uri.fsPath);
+}