Fix for getting auth token for Kusto resource (#16773)

* Fix for getting auth token for Kusto resource

* Addressed comment

Co-authored-by: hhh <gfhg>

extensions/azurecore/src/account-provider/auths/azureAuth.ts

@@ -77,6 +77,10 @@ export abstract class AzureAuth implements vscode.Disposable {
 			this.resources = this.resources.concat(this.metadata.settings.azureLogAnalyticsResource);
 		}
 
+		if (this.metadata.settings.azureKustoResource) {
+			this.resources = this.resources.concat(this.metadata.settings.azureKustoResource);
+		}
+
 		this.scopes = [...this.metadata.settings.scopes];
 		this.scopesString = this.scopes.join(' ');
 	}

extensions/azurecore/src/account-provider/providerSettings.ts

@@ -19,7 +19,8 @@ const enum SettingIds {
 	vault = 'vault',
 	ado = 'ado',
 	ala = 'ala',
-	storage = 'storage'
+	storage = 'storage',
+	kusto = 'kusto'
 }
 
 const publicAzureSettings: ProviderSettings = {
@@ -81,6 +82,11 @@ const publicAzureSettings: ProviderSettings = {
 				endpointSuffix: '.core.windows.net',
 				azureResourceId: AzureResource.AzureStorage
 			},
+			azureKustoResource: {
+				id: SettingIds.kusto,
+				endpoint: 'https://api.kusto.io',
+				azureResourceId: AzureResource.AzureKusto,
+			},
 			redirectUri: 'https://vscode-redirect.azurewebsites.net/',
 			scopes: [
 				'openid', 'email', 'profile', 'offline_access',

extensions/azurecore/src/azurecore.d.ts

@@ -118,6 +118,11 @@ declare module 'azurecore' {
 		 */
 		azureDevOpsResource?: Resource;
 
+		/**
+		 * Information that describes the Azure Kusto resource
+		 */
+		 azureKustoResource?: Resource;
+
 		/**
 		 * Information that describes the Azure Log Analytics resource
 		 */

extensions/kusto/package.json

@@ -217,6 +217,7 @@
       "languageMode": "kusto",
       "notebookKernelAlias": "Kusto",
       "displayName": "%kusto.provider.displayName%",
+      "azureResource": "AzureKusto",
       "iconPath": [
         {
           "id": "kusto:cloud",

src/sql/azdata.proposed.d.ts

@@ -861,7 +861,11 @@ declare module 'azdata' {
 		/**
 		 * Azure Storage
 		 */
-		AzureStorage = 9
+		AzureStorage = 9,
+		/**
+		 * Kusto
+		 */
+		AzureKusto = 10
 	}
 
 	export interface ButtonProperties {

src/sql/platform/accounts/common/interfaces.ts

@@ -58,7 +58,9 @@ export enum AzureResource {
 	MicrosoftResourceManagement = 5,
 	AzureDevOps = 6,
 	MsGraph = 7,
-	AzureLogAnalytics = 8
+	AzureLogAnalytics = 8,
+	AzureStorage = 9,
+	AzureKusto = 10
 }
 
 export interface IAccountStore {

src/sql/workbench/api/common/sqlExtHostTypes.ts

@@ -438,7 +438,8 @@ export enum AzureResource {
 	AzureDevOps = 6,
 	MsGraph = 7,
 	AzureLogAnalytics = 8,
-	AzureStorage = 9
+	AzureStorage = 9,
+	AzureKusto = 10
 }
 
 export class TreeItem extends vsExtTypes.TreeItem {

src/sql/workbench/services/connection/browser/connectionManagementService.ts

@@ -75,7 +75,7 @@ export class ConnectionManagementService extends Disposable implements IConnecti
 
 	private static readonly CONNECTION_MEMENTO = 'ConnectionManagement';
 	private static readonly _azureResources: AzureResource[] =
-		[AzureResource.ResourceManagement, AzureResource.Sql, AzureResource.OssRdbms, AzureResource.AzureLogAnalytics];
+		[AzureResource.ResourceManagement, AzureResource.Sql, AzureResource.OssRdbms, AzureResource.AzureLogAnalytics, AzureResource.AzureKusto];
 
 	constructor(
 		@IConnectionDialogService private _connectionDialogService: IConnectionDialogService,