feat: Added OAuth Support for Public APIs with TokenManager Integrati…

…on (#813) Added OAuth functionality for public APIs
twilio · Oct 3, 2024 · 30e91a6 · 30e91a6
1 parent fab34c1
commit 30e91a6
Show file tree

Hide file tree

Showing 21 changed files with 501 additions and 41 deletions.
diff --git a/.github/workflows/test-and-deploy.yml b/.github/workflows/test-and-deploy.yml
@@ -53,6 +53,9 @@ jobs:
           TWILIO_ORGS_CLIENT_ID: ${{ secrets.TWILIO_ORGS_CLIENT_ID }}
           TWILIO_ORGS_CLIENT_SECRET: ${{ secrets.TWILIO_ORGS_CLIENT_SECRET }}
           TWILIO_ORG_SID: ${{ secrets.TWILIO_ORG_SID }}
+          TWILIO_CLIENT_ID: ${{ secrets.TWILIO_CLIENT_ID }}
+          TWILIO_CLIENT_SECRET: ${{ secrets.TWILIO_CLIENT_SECRET }}
+          TWILIO_MESSAGE_SID: ${{ secrets.TWILIO_MESSAGE_SID }}
         run:  mvn test -DTest="ClusterTest" -B
 
       - uses: actions/setup-java@v4

diff --git a/README.md b/README.md
@@ -214,6 +214,14 @@ public class Example {
 }
 ```
 
+
+
+### OAuth Feature for Twilio APIs
+We are introducing Client Credentials Flow-based OAuth 2.0 authentication. 
+This feature is currently in `beta` and its implementation is subject to change.
+
+Detailed examples [here](https://github.com/twilio/twilio-java/blob/main/examples/FetchMessageUsingOAuth.md)
+
 ### Iterate through records
 
 The library automatically handles paging for you. With the `read` method, you can specify the number of records you want to receive (`limit`) and the maximum size you want each page fetch to be (`pageSize`). The library will then handle the task for you, fetching new pages under the hood as you iterate over the records.

diff --git a/examples/FetchMessageUsingOAuth.md b/examples/FetchMessageUsingOAuth.md
@@ -0,0 +1,21 @@
+```
+import com.twilio.Twilio;
+import com.twilio.credential.ClientCredentialProvider;
+import com.twilio.rest.api.v2010.account.Message;
+
+public class FetchMessageUsingOAuth {
+    public static void main(String[] args) {
+        String clientId = "YOUR_CLIENT_ID";
+        String clientSecret = "YOUR_CLIENT_SECRET";
+        String accountSid = "YOUR_ACCOUNT_SID";
+        Twilio.init(new ClientCredentialProvider(clientId, clientSecret), accountSid);
+        /*
+          Or use the following if accountSid is not required as a path parameter for an API or when setting accountSid in the API.
+          Twilio.init(new ClientCredentialProvider(clientId, clientSecret));
+        */
+        String messageSid = "YOUR_MESSAGE_SID";
+        Message message = Message.fetcher(messageSid).fetch();
+    }
+}
+```
+
diff --git a/src/main/java/com/twilio/Twilio.java b/src/main/java/com/twilio/Twilio.java
@@ -1,8 +1,11 @@
 package com.twilio;
 
+import com.twilio.annotations.Beta;
+import com.twilio.auth_strategy.AuthStrategy;
 import com.twilio.exception.ApiException;
 import com.twilio.exception.AuthenticationException;
 import com.twilio.exception.CertificateValidationException;
+import com.twilio.credential.CredentialProvider;
 import com.twilio.http.HttpMethod;
 import com.twilio.http.NetworkHttpClient;
 import com.twilio.http.Request;
@@ -34,6 +37,8 @@ public class Twilio {
     private static String edge = System.getenv("TWILIO_EDGE");
     private static volatile TwilioRestClient restClient;
     private static volatile ExecutorService executorService;
+
+    private static CredentialProvider credentialProvider;
 
     private Twilio() {
     }
@@ -64,6 +69,31 @@ public static synchronized void init(final String username, final String passwor
         Twilio.setAccountSid(null);
     }
 
+    @Beta
+    public static synchronized void init(final CredentialProvider credentialProvider) {
+        Twilio.setCredentialProvider(credentialProvider);
+        Twilio.setAccountSid(null);
+    }
+
+    @Beta
+    public static synchronized void init(final CredentialProvider credentialProvider, String accountSid) {
+        Twilio.setCredentialProvider(credentialProvider);
+        Twilio.setAccountSid(accountSid);
+    }
+
+    private static void setCredentialProvider(final CredentialProvider credentialProvider) {
+        if (credentialProvider == null) {
+            throw new AuthenticationException("Credential Provider can not be null");
+        }
+
+        if (!credentialProvider.equals(Twilio.credentialProvider)) {
+            Twilio.invalidate();
+        }
+        // Invalidate Basic Creds as they might be initialized via environment variables.
+        invalidateBasicCreds();
+        Twilio.credentialProvider = credentialProvider;
+    }
+
     /**
      * Initialize the Twilio environment.
      *
@@ -91,6 +121,7 @@ public static synchronized void setUsername(final String username) {
         if (!username.equals(Twilio.username)) {
             Twilio.invalidate();
         }
+        Twilio.invalidateOAuthCreds();
 
         Twilio.username = username;
     }
@@ -109,6 +140,7 @@ public static synchronized void setPassword(final String password) {
         if (!password.equals(Twilio.password)) {
             Twilio.invalidate();
         }
+        Twilio.invalidateOAuthCreds();
 
         Twilio.password = password;
     }
@@ -181,12 +213,19 @@ public static TwilioRestClient getRestClient() {
 
     private static TwilioRestClient buildRestClient() {
         if (Twilio.username == null || Twilio.password == null) {
-            throw new AuthenticationException(
-                "TwilioRestClient was used before AccountSid and AuthToken were set, please call Twilio.init()"
-            );
+            if (credentialProvider == null) {
+                throw new AuthenticationException(
+                        "Credentials have not been initialized or changed, please call Twilio.init()"
+                );
+            }
+        }
+        TwilioRestClient.Builder builder;
+        if (credentialProvider != null) {
+            AuthStrategy authStrategy = credentialProvider.toAuthStrategy();
+            builder = new TwilioRestClient.Builder(authStrategy);
+        } else {
+            builder = new TwilioRestClient.Builder(Twilio.username, Twilio.password);
         }
-
-        TwilioRestClient.Builder builder = new TwilioRestClient.Builder(Twilio.username, Twilio.password);
 
         if (Twilio.accountSid != null) {
             builder.accountSid(Twilio.accountSid);
@@ -273,6 +312,15 @@ private static void invalidate() {
         Twilio.restClient = null;
     }
 
+    private static void invalidateOAuthCreds() {
+        Twilio.credentialProvider = null;
+    }
+
+    private static void invalidateBasicCreds() {
+        Twilio.username = null;
+        Twilio.password = null;
+    }
+
     /**
      * Attempts to gracefully shutdown the ExecutorService if it is present.
      */

diff --git a/src/main/java/com/twilio/TwilioNoAuth.java b/src/main/java/com/twilio/TwilioNoAuth.java
@@ -1,13 +1,13 @@
 package com.twilio;
 
-import com.twilio.annotations.Preview;
+import com.twilio.annotations.Beta;
 import com.twilio.http.noauth.NoAuthTwilioRestClient;
 import lombok.Getter;
 
 import java.util.List;
 import com.twilio.exception.AuthenticationException;
 
-@Preview
+@Beta
 public class TwilioNoAuth {
     @Getter
     private static List<String> userAgentExtensions;

diff --git a/src/main/java/com/twilio/TwilioOrgsTokenAuth.java b/src/main/java/com/twilio/TwilioOrgsTokenAuth.java
@@ -1,6 +1,6 @@
 package com.twilio;
 
-import com.twilio.annotations.Preview;
+import com.twilio.annotations.Beta;
 import com.twilio.exception.AuthenticationException;
 import com.twilio.http.bearertoken.BearerTokenTwilioRestClient;
 import lombok.Getter;
@@ -12,7 +12,7 @@
 import com.twilio.http.bearertoken.TokenManager;
 import com.twilio.http.bearertoken.OrgsTokenManager;
 
-@Preview
+@Beta
 public class TwilioOrgsTokenAuth {
     private static String accessToken;
     @Getter

diff --git a/src/main/java/com/twilio/annotations/Preview.java b/src/main/java/com/twilio/annotations/Preview.java
diff --git a/src/main/java/com/twilio/auth_strategy/AuthStrategy.java b/src/main/java/com/twilio/auth_strategy/AuthStrategy.java
@@ -0,0 +1,17 @@
+package com.twilio.auth_strategy;
+
+import com.twilio.constant.EnumConstants;
+import lombok.Getter;
+
+public abstract class AuthStrategy {
+    @Getter
+    private EnumConstants.AuthType authType;
+
+    public AuthStrategy(EnumConstants.AuthType authType) {
+        this.authType = authType;
+    }
+    public abstract String getAuthString();
+
+    public abstract boolean requiresAuthentication(); 
+
+}
diff --git a/src/main/java/com/twilio/auth_strategy/BasicAuthStrategy.java b/src/main/java/com/twilio/auth_strategy/BasicAuthStrategy.java
@@ -0,0 +1,44 @@
+package com.twilio.auth_strategy;
+
+import com.twilio.constant.EnumConstants;
+
+import java.nio.charset.StandardCharsets;
+import java.util.Base64;
+import java.util.Objects;
+
+public class BasicAuthStrategy extends AuthStrategy {
+    private String username;
+    private String password;
+
+    public BasicAuthStrategy(String username, String password) {
+        super(EnumConstants.AuthType.BASIC);
+        this.username = username;
+        this.password = password;
+    }
+
+    @Override
+    public String getAuthString() {
+        String credentials = username + ":" + password;
+        String encoded = Base64.getEncoder().encodeToString(credentials.getBytes(StandardCharsets.US_ASCII));
+        return "Basic " + encoded;
+    }
+
+    @Override
+    public boolean requiresAuthentication() {
+        return true;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        BasicAuthStrategy that = (BasicAuthStrategy) o;
+        return Objects.equals(username, that.username) && 
+                Objects.equals(password, that.password);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(username, password);
+    }
+}
diff --git a/src/main/java/com/twilio/auth_strategy/NoAuthStrategy.java b/src/main/java/com/twilio/auth_strategy/NoAuthStrategy.java
@@ -0,0 +1,20 @@
+package com.twilio.auth_strategy;
+
+import com.twilio.constant.EnumConstants;
+
+public class NoAuthStrategy extends AuthStrategy {
+
+    public NoAuthStrategy(String token) {
+        super(EnumConstants.AuthType.NO_AUTH);
+    }
+
+    @Override
+    public String getAuthString() {
+        return "";
+    }
+
+    @Override
+    public boolean requiresAuthentication() {
+        return false;
+    }
+}
diff --git a/src/main/java/com/twilio/auth_strategy/TokenAuthStrategy.java b/src/main/java/com/twilio/auth_strategy/TokenAuthStrategy.java
@@ -0,0 +1,66 @@
+package com.twilio.auth_strategy;
+
+import com.auth0.jwt.JWT;
+import com.auth0.jwt.interfaces.DecodedJWT;
+import com.twilio.constant.EnumConstants;
+import com.twilio.http.bearertoken.TokenManager;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.Date;
+import java.util.Objects;
+
+public class TokenAuthStrategy extends AuthStrategy {
+    private String token;
+    private TokenManager tokenManager;
+    private static final Logger logger = LoggerFactory.getLogger(TokenAuthStrategy.class);
+    public TokenAuthStrategy(TokenManager tokenManager) {
+        super(EnumConstants.AuthType.TOKEN);
+        this.tokenManager = tokenManager;
+    }
+
+    @Override
+    public String getAuthString() {
+        fetchToken();
+        return "Bearer " + token;
+    }
+
+    @Override
+    public boolean requiresAuthentication() {
+        return true;
+    }
+
+    // Token-specific refresh logic
+    public void fetchToken() {
+        if (this.token == null || this.token.isEmpty() || isTokenExpired(this.token)) {
+            synchronized (TokenAuthStrategy.class){
+                if (this.token == null || this.token.isEmpty() || isTokenExpired(this.token)) {
+                    logger.info("Fetching new token for Apis");
+                    this.token = tokenManager.fetchAccessToken();
+                }
+            }
+        }
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        TokenAuthStrategy that = (TokenAuthStrategy) o;
+        return Objects.equals(token, that.token) && 
+                Objects.equals(tokenManager, that.tokenManager);
+    }
+    @Override
+    public int hashCode() {
+        return Objects.hash(token, tokenManager);
+    }
+
+    public boolean isTokenExpired(final String token) {
+        DecodedJWT jwt = JWT.decode(token);
+        Date expiresAt = jwt.getExpiresAt();
+        // Add a buffer of 30 seconds
+        long bufferMilliseconds = 30 * 1000;
+        Date bufferExpiresAt = new Date(expiresAt.getTime() - bufferMilliseconds);
+        return bufferExpiresAt.before(new Date());
+    }
+}
diff --git a/src/main/java/com/twilio/constant/EnumConstants.java b/src/main/java/com/twilio/constant/EnumConstants.java
@@ -14,4 +14,16 @@ public enum ContentType {
 
         private final String value;
     }
+
+    @Getter
+    @RequiredArgsConstructor
+    public enum AuthType {
+        NO_AUTH("noauth"),
+        BASIC("basic"),
+        TOKEN("token"),
+        API_KEY("api_key"),
+        CLIENT_CREDENTIALS("client_credentials");
+
+        private final String value;
+    }
 }