Add logs and fix notification function name

codota · Dec 6, 2023 · d6c304c · d6c304c
1 parent 8e2b5d8
commit d6c304c
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 5 deletions.
diff --git a/src/enterprise/statusBar/StatusBar.ts b/src/enterprise/statusBar/StatusBar.ts
@@ -1,6 +1,6 @@
 import { Disposable, ExtensionContext, window } from "vscode";
 import { StatusItem } from "./StatusItem";
-import { showLoginNotification } from "./statusAction";
+import { showPleaseLoginNotification } from "./statusAction";
 import { CONGRATS_MESSAGE_SHOWN_KEY } from "../../globals/consts";
 import StatusBarState from "./StatusBarState";
 import { useDerviedState } from "../../state/deriveState";
@@ -30,8 +30,8 @@ export class StatusBar implements Disposable {
         USER_INFO_STATE,
         (s) => s.isLoggedIn,
         (isLoggedIn) => {
-          if (isLoggedIn) {
-            showLoginNotification();
+          if (!isLoggedIn) {
+            showPleaseLoginNotification();
           }
         }
       )

diff --git a/src/enterprise/statusBar/calculateStatusBarState.ts b/src/enterprise/statusBar/calculateStatusBarState.ts
@@ -22,6 +22,8 @@ export default function calculateStatusBarState(
   userInfo: UserInfo | null
 ): StatusBarStateData {
   if (!processStartedState.resolved) {
+    Logger.info("Waiting for process to start...");
+
     return INITIAL_STATE;
   }
 
@@ -36,20 +38,36 @@ export default function calculateStatusBarState(
   }
 
   if (!serverHealthOnPluginStart.resolved) {
+    Logger.info("Waiting for Tabnine server to be ready...");
+
     return {
       type: "loading",
     };
   }
 
   if (serverHealthOnPluginStart.isError || !serverHealthOnPluginStart.data) {
+    Logger.error("Tabnine server is unhealthy.");
+
     return {
       type: "error",
       message: "Please set your Tabnine server URL",
       command: StatusState.SetServer,
     };
   }
 
+  if (cloudConnection === undefined || cloudConnection === null) {
+    Logger.info("Waiting for Tabnine cloud connection...");
+
+    return {
+      type: "loading",
+    };
+  }
+
   if (cloudConnection !== "Ok") {
+    Logger.warn(
+      `Tabnine is not connected to your cloud. Connection status: ${cloudConnection}`
+    );
+
     return {
       type: "warning",
       message: "Connectivity issue - Tabnine is unable to reach the server",
@@ -58,19 +76,25 @@ export default function calculateStatusBarState(
   }
 
   if (!isCompletionsEnabled) {
+    Logger.debug("Showing completions disabled status.");
+
     return {
       type: "warning",
       command: StatusState.Ready,
     };
   }
 
   if (!userInfo) {
+    Logger.info("Waiting for user info...");
+
     return {
       type: "loading",
     };
   }
 
   if (!userInfo.isLoggedIn) {
+    Logger.debug("User is logged out. Showing logged out status.");
+
     return {
       type: "warning",
       message: "Please sign in to access Tabnine",
@@ -79,6 +103,8 @@ export default function calculateStatusBarState(
   }
 
   if (!userInfo.team) {
+    Logger.debug("User is not part of a team. Showing logged out status.");
+
     return {
       type: "warning",
       message: "You are not part of a team",

diff --git a/src/enterprise/statusBar/statusAction.ts b/src/enterprise/statusBar/statusAction.ts
@@ -27,7 +27,7 @@ export function action(state: StatusState): void {
       );
       break;
     case StatusState.LogIn:
-      showLoginNotification();
+      showPleaseLoginNotification();
       break;
     case StatusState.ErrorWaitingForProcess:
       void window
@@ -102,7 +102,7 @@ export function action(state: StatusState): void {
   }
 }
 
-export function showLoginNotification() {
+export function showPleaseLoginNotification() {
   void window
     .showInformationMessage("Please sign in to access Tabnine.", "Sign in")
     .then((selection) => {