From 29c40a06aaaa9b8556c6b640d3c659c893aee4a1 Mon Sep 17 00:00:00 2001
From: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
Date: Mon, 19 Feb 2024 18:17:06 +0200
Subject: [PATCH] fix: azure credential condition (fixes #375)

---
 .../codegpt/credentials/AbstractCredentialsManager.java     | 6 ++++--
 .../codegpt/credentials/AzureCredentialsManager.java        | 4 ++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/main/java/ee/carlrobert/codegpt/credentials/AbstractCredentialsManager.java b/src/main/java/ee/carlrobert/codegpt/credentials/AbstractCredentialsManager.java
index 1e597497..8b90e995 100644
--- a/src/main/java/ee/carlrobert/codegpt/credentials/AbstractCredentialsManager.java
+++ b/src/main/java/ee/carlrobert/codegpt/credentials/AbstractCredentialsManager.java
@@ -47,9 +47,11 @@ abstract class AbstractCredentialsManager {
 
   protected void setCredential(String key, String credential) {
     if (credential == null) {
-      return;
+      credentialCache.remove(key);
+    } else {
+      credentialCache.put(key, credential);
     }
+
     passwordSafe.setPassword(credentialMapping.get(key), credential);
-    credentialCache.put(key, credential);
   }
 }
diff --git a/src/main/java/ee/carlrobert/codegpt/credentials/AzureCredentialsManager.java b/src/main/java/ee/carlrobert/codegpt/credentials/AzureCredentialsManager.java
index f91a4455..c8e9842a 100644
--- a/src/main/java/ee/carlrobert/codegpt/credentials/AzureCredentialsManager.java
+++ b/src/main/java/ee/carlrobert/codegpt/credentials/AzureCredentialsManager.java
@@ -21,9 +21,9 @@ public final class AzureCredentialsManager extends AbstractCredentialsManager {
   @Override
   public boolean isCredentialSet() {
     if (AzureSettings.getCurrentState().isUseAzureApiKeyAuthentication()) {
-      return isCredentialSet(ACTIVE_DIRECTORY_TOKEN);
+      return isCredentialSet(API_KEY);
     }
-    return isCredentialSet(API_KEY);
+    return isCredentialSet(ACTIVE_DIRECTORY_TOKEN);
   }
 
   @Override