Change to just listen to binary state

codota · Dec 4, 2023 · a256003 · a256003
1 parent 13347cc
commit a256003
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/src/enterprise/tabnineChatWidget/SelfHostedChatEnabledState.ts b/src/enterprise/tabnineChatWidget/SelfHostedChatEnabledState.ts
@@ -1,10 +1,11 @@
-import { ExtensionContext, authentication } from "vscode";
+import { ExtensionContext } from "vscode";
 import ChatEnabledState, {
   ChatEnabledStateData,
   ChatStates,
 } from "../../tabnineChatWidget/ChatEnabledState";
 import EventEmitterBasedNonNullState from "../../state/EventEmitterBasedNonNullState";
 import getUserInfo from "../requests/UserInfo";
+import { useDerviedState } from "../../state/deriveState";
 import BINARY_STATE from "../../binary/binaryStateSingleton";
 
 export default class SelfHostedChatEnabledState
@@ -13,18 +14,14 @@ export default class SelfHostedChatEnabledState
   constructor(context: ExtensionContext) {
     super(ChatStates.loading);
 
-    const stateListenDisposable = BINARY_STATE.onChange(() => {
-      void this.updateState();
-
-      if (stateListenDisposable) {
-        stateListenDisposable.dispose();
-      }
-    });
-
     context.subscriptions.push(
-      authentication.onDidChangeSessions(() => {
-        void this.updateState();
-      })
+      useDerviedState(
+        BINARY_STATE,
+        (state) => state.is_logged_in,
+        () => {
+          void this.updateState();
+        }
+      )
     );
   }