DEV2-4342 fix tabnine.enterprise key (#1390)

codota · Dec 3, 2023 · 7c1f1ea · 7c1f1ea
1 parent 03d842e
commit 7c1f1ea
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 8 deletions.
diff --git a/src/enterprise/extension.ts b/src/enterprise/extension.ts
@@ -34,7 +34,7 @@ import TabnineAuthenticationProvider from "../authentication/TabnineAuthenticati
 import {
   BRAND_NAME,
   ENTERPRISE_BRAND_NAME,
-  TABNINE_ENTERPRISE_KEY,
+  IS_SELF_HOSTED_CONTEXT_KEY,
 } from "../globals/consts";
 import { StatusBar } from "./statusBar";
 import { isHealthyServer } from "./update/isHealthyServer";
@@ -46,6 +46,8 @@ import SignInUsingCustomTokenCommand from "../authentication/loginWithCustomToke
 import { SIGN_IN_AUTH_TOKEN_COMMAND } from "../commandsHandler";
 import { WorkspaceUpdater } from "../WorkspaceUpdater";
 
+const TABNINE_ENTERPISE_CONTEXT_KEY = "tabnine.enterprise";
+
 export async function activate(
   context: vscode.ExtensionContext
 ): Promise<void> {
@@ -119,16 +121,16 @@ async function setEnterpriseContext(
 ): Promise<vscode.Disposable> {
   await vscode.commands.executeCommand(
     "setContext",
-    TABNINE_ENTERPRISE_KEY,
+    TABNINE_ENTERPISE_CONTEXT_KEY,
     true
   );
-  await context.workspaceState.update(TABNINE_ENTERPRISE_KEY, true);
+  await context.workspaceState.update(IS_SELF_HOSTED_CONTEXT_KEY, true);
 
   return new vscode.Disposable(() => {
-    void context.workspaceState.update(TABNINE_ENTERPRISE_KEY, undefined);
+    void context.workspaceState.update(IS_SELF_HOSTED_CONTEXT_KEY, undefined);
     void vscode.commands.executeCommand(
       "setContext",
-      TABNINE_ENTERPRISE_KEY,
+      TABNINE_ENTERPISE_CONTEXT_KEY,
       undefined
     );
   });

diff --git a/src/globals/consts.ts b/src/globals/consts.ts
@@ -178,4 +178,4 @@ export enum SuggestionTrigger {
 }
 
 export const TLS_CONFIG_MIN_SUPPORTED_VERSION = "4.22.0";
-export const TABNINE_ENTERPRISE_KEY = "tabnine.enterprise";
+export const IS_SELF_HOSTED_CONTEXT_KEY = "tabnine.isSelfHosted";
diff --git a/src/tabnineChatWidget/ChatApi.ts b/src/tabnineChatWidget/ChatApi.ts
@@ -15,7 +15,7 @@ import { peekDefinition } from "./handlers/peekDefinition";
 import { ServiceLevel } from "../binary/state";
 import {
   GET_CHAT_STATE_COMMAND,
-  TABNINE_ENTERPRISE_KEY,
+  IS_SELF_HOSTED_CONTEXT_KEY,
 } from "../globals/consts";
 import {
   BasicContext,
@@ -122,7 +122,10 @@ export function initChatApi(
         ].includes(vscode.window.activeColorTheme.kind),
         isTelemetryEnabled: isCapabilityEnabled(Capability.ALPHA_CAPABILITY),
         serverUrl,
-        isSelfHosted: context.workspaceState.get(TABNINE_ENTERPRISE_KEY, false),
+        isSelfHosted: context.workspaceState.get(
+          IS_SELF_HOSTED_CONTEXT_KEY,
+          false
+        ),
       });
     })
     .registerEvent<void, GetUserResponse>("get_user", async () => {