fmt

Author: samikshya-db

databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java

@@ -242,7 +242,7 @@ public TokenSource getTokenSource() {
       return (TokenSource) headerFactory;
     }
     return new ErrorTokenSource(
-        String.format("OAuth Token not supported for current auth type %s", authType));
+            String.format("OAuth Token not supported for current auth type %s", authType));
   }
 
   public CredentialsProvider getCredentialsProvider() {
@@ -434,13 +434,17 @@ public DatabricksConfig setAzureUseMsi(boolean azureUseMsi) {
     return this;
   }
 
-  /** @deprecated Use {@link #getAzureUseMsi()} instead. */
+  /**
+   * @deprecated Use {@link #getAzureUseMsi()} instead.
+   */
   @Deprecated()
   public boolean getAzureUseMSI() {
     return azureUseMsi;
   }
 
-  /** @deprecated Use {@link #getAzureUseMsi()} instead. */
+  /**
+   * @deprecated Use {@link #getAzureUseMsi()} instead.
+   */
   @Deprecated
   public DatabricksConfig setAzureUseMSI(boolean azureUseMsi) {
     this.azureUseMsi = azureUseMsi;
@@ -657,10 +661,10 @@ public OpenIDConnectEndpoints getOidcEndpoints() throws IOException {
       }
     } catch (Exception e) {
       LOG.warn(
-          "Failed to fetch OIDC Endpoints using discovery URL: {}. Error: {}. \nDefaulting to fetch OIDC using default endpoint.",
-          discoveryUrl,
-          e.getMessage(),
-          e);
+              "Failed to fetch OIDC Endpoints using discovery URL: {}. Error: {}. \nDefaulting to fetch OIDC using default endpoint.",
+              discoveryUrl,
+              e.getMessage(),
+              e);
     }
     return fetchDefaultOidcEndpoints();
   }
@@ -691,22 +695,22 @@ private OpenIDConnectEndpoints fetchDefaultOidcEndpoints() throws IOException {
         return null;
       }
       return new OpenIDConnectEndpoints(
-          realAuthUrl.replaceAll("/authorize", "/token"), realAuthUrl);
+              realAuthUrl.replaceAll("/authorize", "/token"), realAuthUrl);
     }
     if (isAccountClient() && getAccountId() != null) {
       String prefix = getHost() + "/oidc/accounts/" + getAccountId();
       return new OpenIDConnectEndpoints(prefix + "/v1/token", prefix + "/v1/authorize");
     }
 
     ApiClient apiClient =
-        new ApiClient.Builder()
-            .withHttpClient(getHttpClient())
-            .withGetHostFunc(v -> getHost())
-            .build();
+            new ApiClient.Builder()
+                    .withHttpClient(getHttpClient())
+                    .withGetHostFunc(v -> getHost())
+                    .build();
     try {
       return apiClient.execute(
-          new Request("GET", "/oidc/.well-known/oauth-authorization-server"),
-          OpenIDConnectEndpoints.class);
+              new Request("GET", "/oidc/.well-known/oauth-authorization-server"),
+              OpenIDConnectEndpoints.class);
     } catch (IOException e) {
       throw new DatabricksException("IO error: " + e.getMessage(), e);
     }
@@ -773,18 +777,18 @@ public DatabricksConfig clone() {
 
   public DatabricksConfig newWithWorkspaceHost(String host) {
     Set<String> fieldsToSkip =
-        new HashSet<>(
-            Arrays.asList(
-                // The config for WorkspaceClient has a different host and Azure Workspace resource
-                // ID, and also omits
-                // the account ID.
-                "host",
-                "accountId",
-                "azureWorkspaceResourceId",
-                // For cloud-native OAuth, we need to reauthenticate as the audience has changed, so
-                // don't cache the
-                // header factory.
-                "headerFactory"));
+            new HashSet<>(
+                    Arrays.asList(
+                            // The config for WorkspaceClient has a different host and Azure Workspace resource
+                            // ID, and also omits
+                            // the account ID.
+                            "host",
+                            "accountId",
+                            "azureWorkspaceResourceId",
+                            // For cloud-native OAuth, we need to reauthenticate as the audience has changed, so
+                            // don't cache the
+                            // header factory.
+                            "headerFactory"));
     return clone(fieldsToSkip).setHost(host);
   }
 

databricks-sdk-java/src/main/java/com/databricks/sdk/core/oauth/OAuthClient.java

@@ -119,10 +119,7 @@ private OAuthClient(Builder b) throws IOException {
     this.hc = b.hc;
 
     DatabricksConfig config =
-        new DatabricksConfig()
-            .setHost(b.host)
-            .setAccountId(b.accountId)
-            .resolve();
+            new DatabricksConfig().setHost(b.host).setAccountId(b.accountId).resolve();
     openIDConnectEndpoints = b.openIDConnectEndpoints;
     if (openIDConnectEndpoints == null) {
       throw new DatabricksException(b.host + " does not support OAuth");
@@ -193,9 +190,9 @@ private static byte[] sha256(byte[] input) {
 
   private static String urlEncode(String urlBase, Map<String, String> params) {
     String queryParams =
-        params.entrySet().stream()
-            .map(entry -> entry.getKey() + "=" + entry.getValue())
-            .collect(Collectors.joining("&"));
+            params.entrySet().stream()
+                    .map(entry -> entry.getKey() + "=" + entry.getValue())
+                    .collect(Collectors.joining("&"));
     return urlBase + "?" + queryParams.replaceAll(" ", "%20");
   }
 
@@ -217,15 +214,15 @@ public Consent initiateConsent() throws MalformedURLException {
     String url = urlEncode(authUrl, params);
 
     return new Consent.Builder()
-        .withClientId(clientId)
-        .withClientSecret(clientSecret)
-        .withAuthUrl(url)
-        .withTokenUrl(tokenUrl)
-        .withRedirectUrl(redirectUrl)
-        .withState(state)
-        .withVerifier(verifier)
-        .withHttpClient(hc)
-        .withBrowserTimeout(browserTimeout)
-        .build();
+            .withClientId(clientId)
+            .withClientSecret(clientSecret)
+            .withAuthUrl(url)
+            .withTokenUrl(tokenUrl)
+            .withRedirectUrl(redirectUrl)
+            .withState(state)
+            .withVerifier(verifier)
+            .withHttpClient(hc)
+            .withBrowserTimeout(browserTimeout)
+            .build();
   }
 }