Replaced ChatUserCredentialPolicy with BearerTokenAuthenticationPolicy

Azure · Nov 10, 2020 · 7ff99e8 · 7ff99e8
1 parent 3540be5
commit 7ff99e8
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 3 deletions.
diff --git a/...zure-communication-chat/src/main/java/com/azure/communication/chat/ChatClientBuilder.java b/...zure-communication-chat/src/main/java/com/azure/communication/chat/ChatClientBuilder.java
@@ -4,6 +4,7 @@
 package com.azure.communication.chat;
 
 import com.azure.communication.chat.implementation.AzureCommunicationChatServiceImplBuilder;
+import com.azure.communication.common.CommunicationTokenCredential;
 import com.azure.communication.common.CommunicationUserCredential;
 
 import java.util.ArrayList;
@@ -14,6 +15,7 @@
 import com.azure.core.http.HttpClient;
 import com.azure.core.http.HttpPipeline;
 import com.azure.core.http.HttpPipelineBuilder;
+import com.azure.core.http.policy.BearerTokenAuthenticationPolicy;
 import com.azure.core.http.policy.CookiePolicy;
 import com.azure.core.http.policy.HttpLogOptions;
 import com.azure.core.http.policy.HttpLoggingPolicy;
@@ -139,7 +141,7 @@ public ChatClientBuilder configuration(Configuration configuration) {
     }
 
     /**
-     * Create synchronous client applying ChatUserCredentialPolicy, UserAgentPolicy,
+     * Create synchronous client applying CommunicationTokenCredential, UserAgentPolicy,
      * RetryPolicy, and CookiePolicy.
      * Additional HttpPolicies specified by additionalPolicies will be applied after them
      *
@@ -151,7 +153,7 @@ public ChatClient buildClient() {
     }
 
     /**
-     * Create asynchronous client applying ChatUserCredentialPolicy, UserAgentPolicy,
+     * Create asynchronous client applying CommunicationTokenCredential, UserAgentPolicy,
      * RetryPolicy, and CookiePolicy.
      * Additional HttpPolicies specified by additionalPolicies will be applied after them
      *
@@ -166,9 +168,10 @@ public ChatAsyncClient buildAsyncClient() {
         } else {
             Objects.requireNonNull(communicationUserCredential);
             Objects.requireNonNull(httpClient);
+            CommunicationTokenCredential tokenCredential = new CommunicationTokenCredential(communicationUserCredential);
 
             pipeline = createHttpPipeline(httpClient,
-                new ChatUserCredentialPolicy(communicationUserCredential),
+                new BearerTokenAuthenticationPolicy(tokenCredential, ""),
                 customPolicies);
         }
 

diff --git a/...ion-common/src/main/java/com/azure/communication/common/CommunicationTokenCredential.java b/...ion-common/src/main/java/com/azure/communication/common/CommunicationTokenCredential.java
@@ -0,0 +1,32 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.communication.common;
+
+import java.util.concurrent.ExecutionException;
+
+import com.azure.core.credential.AccessToken;
+import com.azure.core.credential.TokenCredential;
+import com.azure.core.credential.TokenRequestContext;
+
+import reactor.core.publisher.Mono;
+
+public class CommunicationTokenCredential implements TokenCredential {
+    private CommunicationUserCredential credential;
+
+    public CommunicationTokenCredential(CommunicationUserCredential communicationUserCrendential){
+        credential = communicationUserCrendential;
+    }
+
+    @Override
+    public Mono<AccessToken> getToken(TokenRequestContext request){
+        try{
+            return Mono.just(credential.getToken().get());
+        }
+        catch (InterruptedException ex) {
+            return Mono.error(ex);
+        } catch (ExecutionException ex) {
+            return Mono.error(ex);
+        }
+    }
+}