ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-16 18:46:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

Adding Integration Runtime Page to Migration extension wizard (#14020)

Browse Source 
* - Fixed GetMigrationController
- Added createMigrationController and getControllerAuthKeys API in azure core.
- Added typings for Migration Controller
- Fixed database backup page validation logic
- Added IR page with create controller

* Fixing all the comments from the PR

* Fixed typings
This commit is contained in:
Aasim Khan
2021-01-25 14:46:39 -08:00
committed by GitHub
parent ed26938dc8
commit 42a8680738
12 changed files with 968 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
extensions/azurecore/src/azureResource/azure-resource.d.ts vendored
View File

@@ -84,6 +84,25 @@ declare module 'azureResource' {
export interface FileShare extends AzureResource { }
export interface MigrationController extends AzureResource { }
export interface MigrationControllerProperties {
name: string;
subscriptionId: string;
resourceGroup: string;
location: string;
provisioningState: string;
integrationRunTimeState?: string;
isProvisioned?: boolean;
}
export interface MigrationController {
properties: MigrationControllerProperties;
location: string;
id: string;
name: string;
error: {
code: string,
message: string
}
}
}
}

82
extensions/azurecore/src/azureResource/utils.ts
View File

@@ -7,7 +7,7 @@ import { ResourceGraphClient } from '@azure/arm-resourcegraph';
import { TokenCredentials } from '@azure/ms-rest-js';
import axios, { AxiosRequestConfig } from 'axios';
import * as azdata from 'azdata';
import { HttpGetRequestResult, GetResourceGroupsResult, GetSubscriptionsResult, ResourceQueryResult, GetBlobContainersResult, GetFileSharesResult, GetMigrationControllersResult } from 'azurecore';
import { HttpRequestResult, GetResourceGroupsResult, GetSubscriptionsResult, ResourceQueryResult, GetBlobContainersResult, GetFileSharesResult, GetMigrationControllerResult, CreateMigrationControllerResult, GetMigrationControllerAuthKeysResult } from 'azurecore';
import { azureResource } from 'azureResource';
import { EOL } from 'os';
import * as nls from 'vscode-nls';
@@ -288,11 +288,18 @@ export async function getSelectedSubscriptions(appContext: AppContext, account?:
return result;
}
enum HttpRequestType {
GET,
POST,
PUT,
DELETE
}
/**
* makes a GET request to Azure REST apis. Currently, it only supports GET ARM queries.
* Make a HTTP request to Azure REST apis.
*/
export async function makeHttpGetRequest(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, ignoreErrors: boolean = false, url: string): Promise<HttpGetRequestResult> {
const result: HttpGetRequestResult = { response: {}, errors: [] };
export async function makeHttpRequest(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, ignoreErrors: boolean = false, url: string, requestType: HttpRequestType, requestBody?: any): Promise<HttpRequestResult> {
const result: HttpRequestResult = { response: {}, errors: [] };
if (!account?.properties?.tenants || !Array.isArray(account.properties.tenants)) {
const error = new Error(invalidAzureAccount);
@@ -340,7 +347,22 @@ export async function makeHttpGetRequest(account: azdata.Account, subscription:
validateStatus: () => true // Never throw
};
const response = await axios.get(url, config);
let response;
switch (requestType) {
case HttpRequestType.GET:
response = await axios.get(url, config);
break;
case HttpRequestType.POST:
response = await axios.post(url, requestBody, config);
break;
case HttpRequestType.PUT:
response = await axios.put(url, requestBody, config);
break;
case HttpRequestType.DELETE:
response = await axios.delete(url, config);
break;
}
if (response.status !== 200) {
let errorMessage: string[] = [];
@@ -362,12 +384,8 @@ export async function makeHttpGetRequest(account: azdata.Account, subscription:
}
export async function getBlobContainers(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, storageAccount: azureResource.AzureGraphResource, ignoreErrors: boolean): Promise<GetBlobContainersResult> {
const apiEndpoint = `https://management.azure.com` +
`/subscriptions/${subscription.id}` +
`/resourceGroups/${storageAccount.resourceGroup}` +
`/providers/Microsoft.Storage/storageAccounts/${storageAccount.name}` +
`/blobServices/default/containers?api-version=2019-06-01`;
const response = await makeHttpGetRequest(account, subscription, ignoreErrors, apiEndpoint);
const apiEndpoint = `https://management.azure.com/subscriptions/${subscription.id}/resourceGroups/${storageAccount.resourceGroup}/providers/Microsoft.Storage/storageAccounts/${storageAccount.name}/blobServices/default/containers?api-version=2019-06-01`;
const response = await makeHttpRequest(account, subscription, ignoreErrors, apiEndpoint, HttpRequestType.GET);
return {
blobContainers: response?.response?.data?.value ?? [],
errors: response.errors ? response.errors : []
@@ -375,26 +393,42 @@ export async function getBlobContainers(account: azdata.Account, subscription: a
}
export async function getFileShares(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, storageAccount: azureResource.AzureGraphResource, ignoreErrors: boolean): Promise<GetFileSharesResult> {
const apiEndpoint = `https://management.azure.com` +
`/subscriptions/${subscription.id}` +
`/resourceGroups/${storageAccount.resourceGroup}` +
`/providers/Microsoft.Storage/storageAccounts/${storageAccount.name}` +
`/fileServices/default/shares?api-version=2019-06-01`;
const response = await makeHttpGetRequest(account, subscription, ignoreErrors, apiEndpoint);
const apiEndpoint = `https://management.azure.com/subscriptions/${subscription.id}/resourceGroups/${storageAccount.resourceGroup}/providers/Microsoft.Storage/storageAccounts/${storageAccount.name}/fileServices/default/shares?api-version=2019-06-01`;
const response = await makeHttpRequest(account, subscription, ignoreErrors, apiEndpoint, HttpRequestType.GET);
return {
fileShares: response?.response?.data?.value ?? [],
errors: response.errors ? response.errors : []
};
}
export async function getMigrationControllers(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, ignoreErrors: boolean): Promise<GetMigrationControllersResult> {
const apiEndpoint = `https://${regionName}.management.azure.com` +
`/subscriptions/${subscription.id}` +
`/resourceGroups/${resourceGroupName}` +
`/providers/Microsoft.DataMigration/Controllers/default/shares?api-version=2020-09-01-preview`;
const response = await makeHttpGetRequest(account, subscription, ignoreErrors, apiEndpoint);
export async function getMigrationControllers(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, controllerName: string, ignoreErrors: boolean): Promise<GetMigrationControllerResult> {
const apiEndpoint = `https://${regionName}.management.azure.com/subscriptions/${subscription.id}/resourceGroups/${resourceGroupName}/providers/Microsoft.DataMigration/Controllers/${controllerName}?api-version=2020-09-01-preview`;
const response = await makeHttpRequest(account, subscription, false, apiEndpoint, HttpRequestType.GET);
return {
controllers: response?.response?.data?.value ?? [],
controller: response?.response?.data ?? undefined,
errors: response.errors ? response.errors : []
};
}
export async function createMigrationController(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, controllerName: string, ignoreErrors: boolean): Promise<CreateMigrationControllerResult> {
const apiEndpoint = `https://${regionName}.management.azure.com/subscriptions/${subscription.id}/resourceGroups/${resourceGroupName}/providers/Microsoft.DataMigration/Controllers/${controllerName}?api-version=2020-09-01-preview`;
const requestBody = {
'location': regionName
};
const response = await makeHttpRequest(account, subscription, ignoreErrors, apiEndpoint, HttpRequestType.PUT, requestBody);
return {
controller: response?.response?.data ?? undefined,
errors: response.errors ? response.errors : []
};
}
export async function getMigrationControllerAuthKeys(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, controllerName: string, ignoreErrors: boolean): Promise<GetMigrationControllerAuthKeysResult> {
const apiEndpoint = `https://${regionName}.management.azure.com/subscriptions/${subscription.id}/resourceGroups/${resourceGroupName}/providers/Microsoft.DataMigration/Controllers/${controllerName}/ListAuthKeys?api-version=2020-09-01-preview`;
const response = await makeHttpRequest(account, subscription, ignoreErrors, apiEndpoint, HttpRequestType.POST);
return {
keyName1: response?.response?.data?.keyName1 ?? '',
keyName2: response?.response?.data?.keyName2 ?? '',
errors: response.errors ? response.errors : []
};
}

10
extensions/azurecore/src/azurecore.d.ts vendored
View File

@@ -75,7 +75,9 @@ declare module 'azurecore' {
getStorageAccounts(account: azdata.Account, subscriptions: azureResource.AzureResourceSubscription[], ignoreErrors?: boolean): Promise<GetStorageAccountResult>;
getBlobContainers(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, storageAccount: azureResource.AzureGraphResource, ignoreErrors?: boolean): Promise<GetBlobContainersResult>;
getFileShares(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, storageAccount: azureResource.AzureGraphResource, ignoreErrors?: boolean): Promise<GetFileSharesResult>;
getMigrationControllers(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, ignoreErrors?: boolean): Promise<GetMigrationControllersResult>;
getMigrationController(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, controllerName: string, ignoreErrors?: boolean): Promise<GetMigrationControllerResult>;
createMigrationController(account:azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, controllerName: string, ignoreErrors?:boolean): Promise<CreateMigrationControllerResult>;
getMigrationControllerAuthKeys(account: azdata.Account, subscription: azureResource.AzureResourceSubscription, resourceGroupName: string, regionName: string, controllerName: string, ignoreErrors?: boolean): Promise<GetMigrationControllerAuthKeysResult>;
/**
* Converts a region value (@see AzureRegion) into the localized Display Name
@@ -95,8 +97,10 @@ declare module 'azurecore' {
export type GetStorageAccountResult = { resources: azureResource.AzureGraphResource[], errors: Error[] };
export type GetBlobContainersResult = { blobContainers: azureResource.BlobContainer[], errors: Error[] };
export type GetFileSharesResult = { fileShares: azureResource.FileShare[], errors: Error[] };
export type GetMigrationControllersResult = { controllers: azureResource.MigrationController[], errors: Error[] };
export type GetMigrationControllerResult = { controller: azureResource.MigrationController | undefined, errors: Error[] };
export type CreateMigrationControllerResult = { controller: azureResource.MigrationController | undefined, errors: Error[] };
export type GetMigrationControllerAuthKeysResult = { keyName1: string, keyName2: string, errors: Error[] };
export type ResourceQueryResult<T extends azureResource.AzureGraphResource> = { resources: T[], errors: Error[] };
export type HttpGetRequestResult = { response: any, errors: Error[] };
export type HttpRequestResult = { response: any, errors: Error[] };
}

23
extensions/azurecore/src/extension.ts
View File

@@ -196,12 +196,29 @@ export async function activate(context: vscode.ExtensionContext): Promise<azurec
ignoreErrors: boolean): Promise<azurecore.GetFileSharesResult> {
return azureResourceUtils.getFileShares(account, subscription, storageAccount, ignoreErrors);
},
getMigrationControllers(account: azdata.Account,
getMigrationController(account: azdata.Account,
subscription: azureResource.AzureResourceSubscription,
resourceGroupName: string,
regionName: string,
ignoreErrors: boolean): Promise<azurecore.GetMigrationControllersResult> {
return azureResourceUtils.getMigrationControllers(account, subscription, resourceGroupName, regionName, ignoreErrors);
controllerName: string,
ignoreErrors: boolean): Promise<azurecore.GetMigrationControllerResult> {
return azureResourceUtils.getMigrationControllers(account, subscription, resourceGroupName, regionName, controllerName, ignoreErrors);
},
createMigrationController(account: azdata.Account,
subscription: azureResource.AzureResourceSubscription,
resourceGroupName: string,
regionName: string,
controllerName: string,
ignoreErrors: boolean): Promise<azurecore.CreateMigrationControllerResult> {
return azureResourceUtils.createMigrationController(account, subscription, resourceGroupName, regionName, controllerName, ignoreErrors);
},
getMigrationControllerAuthKeys(account: azdata.Account,
subscription: azureResource.AzureResourceSubscription,
resourceGroupName: string,
regionName: string,
controllerName: string,
ignoreErrors: boolean): Promise<azurecore.GetMigrationControllerAuthKeysResult> {
return azureResourceUtils.getMigrationControllerAuthKeys(account, subscription, resourceGroupName, regionName, controllerName, ignoreErrors);
},
getRegionDisplayName: utils.getRegionDisplayName,
runGraphQuery<T extends azureResource.AzureGraphResource>(account: azdata.Account,