[PECO-2153] Support OAuth on databricks.azure.cn (#271)

jackyhu-db · web-flow · commit af23aa00c57d · 2024-12-16T10:56:53.000-08:00
[PEC-2153] Support OAuth on databricks.azure.cn

Signed-off-by: Jacky Hu &lt;jacky.hu@databricks.com&gt;
diff --git a/lib/connection/auth/DatabricksOAuth/OAuthManager.ts b/lib/connection/auth/DatabricksOAuth/OAuthManager.ts
@@ -219,7 +219,7 @@ export default abstract class OAuthManager {
     }
 
     if (options.useDatabricksOAuthInAzure) {
-      const azureDomains = ['.azuredatabricks.net'];
+      const azureDomains = ['.azuredatabricks.net', '.databricks.azure.cn'];
       const isAzureDomain = azureDomains.some((domain) => host.endsWith(domain));
       if (isAzureDomain) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define

Original file line number	Diff line number	Diff line change
`@@ -219,7 +219,7 @@ export default abstract class OAuthManager {`
`219`	`219`	`}`
`220`	`220`
`221`	`221`	`if (options.useDatabricksOAuthInAzure) {`
`222`		`- const azureDomains = ['.azuredatabricks.net'];`
	`222`	`+ const azureDomains = ['.azuredatabricks.net', '.databricks.azure.cn'];`
`223`	`223`	`const isAzureDomain = azureDomains.some((domain) => host.endsWith(domain));`
`224`	`224`	`if (isAzureDomain) {`
`225`	`225`	`// eslint-disable-next-line @typescript-eslint/no-use-before-define`