From 8be0f742bc0efb7b0682f0557d7491c3c4562898 Mon Sep 17 00:00:00 2001
From: Aasim Khan <aasimkhan30@gmail.com>
Date: Wed, 11 Aug 2021 17:48:44 -0700
Subject: [PATCH] Listing filtered locations for migration service public
 preview.  (#16731)

* Filtering regions for public preview

* rename var

* reverting back to const
---
 extensions/sql-migration/src/api/azure.ts | 26 ++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/extensions/sql-migration/src/api/azure.ts b/extensions/sql-migration/src/api/azure.ts
index bbf775bc86..e747cc66cf 100644
--- a/extensions/sql-migration/src/api/azure.ts
+++ b/extensions/sql-migration/src/api/azure.ts
@@ -43,7 +43,31 @@ export async function getLocations(account: azdata.Account, subscription: Subscr
 	const filteredLocations = response.locations.filter(loc => {
 		return sqlMigrationResourceLocations.includes(loc.displayName);
 	});
-	return filteredLocations;
+
+	// Only including the regions that have migration service deployed for public preview.
+	const publicPreviewLocations = [
+		'eastus',
+		'canadaeast',
+		'canadacentral',
+		'centralus',
+		'westus2',
+		'westus',
+		'southcentralus',
+		'westeurope',
+		'uksouth',
+		'australiaeast',
+		'southeastasia',
+		'japaneast',
+		'centralindia',
+		'eastus2',
+		'eastus2euap',
+		'francecentral',
+		'southindia',
+		'australiasoutheast',
+		'northcentralus'
+	];
+
+	return filteredLocations.filter(v => publicPreviewLocations.includes(v.name));
 }
 
 export type AzureProduct = azureResource.AzureGraphResource;