feat: data plane view support

src/commands/addDataPlaneApis.ts

@@ -6,27 +6,29 @@
 import { ext } from "../extensionVariables";
 import { UiStrings } from "../uiStrings";
 export async function getDataPlaneApis(context: IActionContext): Promise<any | void> {
     const endpointUrl = await vscode.window.showInputBox({ title: UiStrings.AddDataPlaneRuntimeUrl, ignoreFocusOut: true });
     if (!endpointUrl) {
         return;
     }
     const clientid = await vscode.window.showInputBox({ title: UiStrings.AddDataPlaneClientId, ignoreFocusOut: true });
     if (!clientid) {
         return;
     }
     const tenantid = await vscode.window.showInputBox({ title: UiStrings.AddDataPlaneTenantId, ignoreFocusOut: true });
     if (!tenantid) {
         return;
     }
-    // return await getSessionToken(clientid, tenantid);
+
     setAccountToExt(endpointUrl, clientid, tenantid);
     ext.workspaceItem.refresh(context);
 }
 export function setAccountToExt(domain: string, clientId: string, tenantId: string) {
     function pushIfNotExist(array: DataPlaneAccount[], element: DataPlaneAccount) {
         if (!array.some(item => item.domain === element.domain)) {
             array.push(element);
+        } else {
+            vscode.window.showInformationMessage(UiStrings.DatplaneAlreadyAdded);
         }
     }
     pushIfNotExist(ext.dataPlaneAccounts, { domain: domain, tenantId: tenantId, clientId: clientId });
 }

src/commands/registerApiSubCommands/registerStepByStep.ts

@@ -6,7 +6,7 @@
 import * as vscode from 'vscode';
 import { ApiCenterService } from "../../azure/ApiCenter/ApiCenterService";
 import { ApiCenterApi, ApiCenterApiVersion, ApiCenterApiVersionDefinition, ApiCenterApiVersionDefinitionImport, ApiKind, ApiVersionLifecycleStage, SpecificationName } from "../../azure/ApiCenter/contracts";
-import { ApiCenterVersionDefinitionManagement } from "../../azure/ApiCenterDefines/ApiCenterDefinition";
+import { ApiCenterVersionDefinitionDataPlane } from "../../azure/ApiCenterDefines/ApiCenterDefinition";
 import { ext } from "../../extensionVariables";
 import { ApiCenterTreeItem } from "../../tree/ApiCenterTreeItem";
 import { ApisTreeItem } from "../../tree/ApisTreeItem";
@@ -17,9 +17,9 @@
         const apiCenterNode = await ext.treeDataProvider.showTreeItemPicker<ApiCenterTreeItem>(ApiCenterTreeItem.contextValue, context);
         node = apiCenterNode.apisTreeItem;
     }
-    if (node.apiCenter instanceof ApiCenterVersionDefinitionManagement) {
+    if (node.apiCenter instanceof ApiCenterVersionDefinitionDataPlane) {
         return;
     }
 
     const apiTitle = await vscode.window.showInputBox({ title: UiStrings.ApiTitle, ignoreFocusOut: true, validateInput: validateInputForTitle });
     if (!apiTitle) {
@@ -64,8 +64,8 @@
         return;
     }
 
     const resourceGroupName = getResourceGroupFromId(node.apiCenter.getId());
     const apiCenterService = new ApiCenterService(node.parent?.subscription!, resourceGroupName, node.apiCenter.getName());
 
     await createApiResources(apiCenterService, apiTitle, apiKind, apiVersionTitle, apiVersionLifecycleStage,
         apiDefinitionTitle, specificationName, filePath);

src/extension.ts

@@ -11,8 +11,8 @@
 import { AzExtTreeDataProvider, AzExtTreeItem, CommandCallback, IActionContext, IParsedError, createAzExtOutputChannel, isUserCancelledError, parseError, registerCommand, registerEvent } from '@microsoft/vscode-azext-utils';
 import { AzureAccount } from "./azure/azureLogin/azureAccount";
 import { AzureSessionProviderHelper } from "./azure/azureLogin/azureSessionProvider";
 import { AzureDataSessionProviderHelper } from "./azure/azureLogin/dataSessionProvider";
 import { getDataPlaneApis } from "./commands/addDataPlaneApis";
 import { cleanupSearchResult } from './commands/cleanUpSearch';
 import { detectBreakingChange } from './commands/detectBreakingChange';
 import { showOpenApi } from './commands/editOpenApi';
@@ -21,13 +21,13 @@
 import { generateApiLibrary } from './commands/generateApiLibrary';
 import { GenerateHttpFile } from './commands/generateHttpFile';
 import { generateMarkdownDocument } from './commands/generateMarkdownDocument';
 import { handleUri } from './commands/handleUri';
 import { importOpenApi } from './commands/importOpenApi';
 import { openAPiInSwagger } from './commands/openApiInSwagger';
 import { openUrlFromTreeNode } from './commands/openUrl';
 import { refreshTree } from './commands/refreshTree';
 import { registerApi } from './commands/registerApi';
 import { removeDataplaneAPI } from './commands/removeDataplaneApi';
 import { addCustomFunction } from './commands/rules/addCustomFunction';
 import { deleteCustomFunction } from './commands/rules/deleteCustomFunction';
 import { deployRules } from './commands/rules/deployRules';
@@ -37,16 +37,15 @@
 import { renameCustomFunction } from './commands/rules/renameCustomFunction';
 import { searchApi } from './commands/searchApi';
 import { setApiRuleset } from './commands/setApiRuleset';
 import { SignInToDataPlane } from "./commands/signInToDataPlane";
 import { testInPostman } from './commands/testInPostman';
+import { ErrorProperties, TelemetryProperties } from './common/telemetryEvent';
 import { doubleClickDebounceDelay, selectedNodeKey } from './constants';
 import { ext } from './extensionVariables';
 import { ApiVersionDefinitionTreeItem } from './tree/ApiVersionDefinitionTreeItem';
 import { createAzureAccountTreeItem } from "./tree/AzureAccountTreeItem";
 import { createAzureDataAccountTreeItem } from './tree/DataPlaneAccount';
 import { OpenApiEditor } from './tree/Editors/openApi/OpenApiEditor';
-// Copilot Chat
-import { ErrorProperties, TelemetryProperties } from './common/telemetryEvent';
 export async function activate(context: vscode.ExtensionContext) {
     console.log('Congratulations, your extension "azure-api-center" is now active!');
 
@@ -60,39 +59,8 @@
     context.subscriptions.push(ext.outputChannel);
     registerAzureUtilsExtensionVariables(ext);
 
-    AzureSessionProviderHelper.activateAzureSessionProvider(context);
-    const sessionProvider = AzureSessionProviderHelper.getSessionProvider();
-
-    const azureAccountTreeItem = createAzureAccountTreeItem(sessionProvider);
-    context.subscriptions.push(azureAccountTreeItem);
-    ext.treeItem = azureAccountTreeItem;
-    // var a = ext.treeItem.subscription;
-    ext.dataPlaneAccounts = [];
-    const treeDataProvider = new AzExtTreeDataProvider(azureAccountTreeItem, "appService.loadMore");
-
-    ext.treeItem = azureAccountTreeItem;
-
-    ext.treeDataProvider = treeDataProvider;
-
-    const treeView = vscode.window.createTreeView("apiCenterTreeView", { treeDataProvider });
-    context.subscriptions.push(treeView);
-
-    AzureDataSessionProviderHelper.activateAzureSessionProvider(context);
-    const dataPlaneSessionProvider = AzureDataSessionProviderHelper.getSessionProvider();
-    const dataPlanAccountManagerTreeItem = createAzureDataAccountTreeItem(dataPlaneSessionProvider);
-    context.subscriptions.push(dataPlanAccountManagerTreeItem);
-    ext.workspaceItem = dataPlanAccountManagerTreeItem;
-
-    const workspaceTreeDataProvider = new AzExtTreeDataProvider(dataPlanAccountManagerTreeItem, "appService.loadMore");
-    ext.workspaceProvider = workspaceTreeDataProvider;
-
-    vscode.window.registerTreeDataProvider('apiCenterWorkspace', workspaceTreeDataProvider);
-
-    treeView.onDidChangeSelection((e: vscode.TreeViewSelectionChangeEvent<AzExtTreeItem>) => {
-        const selectedNode = e.selection[0];
-        ext.outputChannel.appendLine(selectedNode.id!);
-        ext.context.globalState.update(selectedNodeKey, selectedNode.id);
-    });
+    setupControlView(context);
+    setupDataTreeView(context);
 
     // Register API Center extension commands
 
@@ -156,21 +124,21 @@
     registerCommandWithTelemetry('azure-api-center.signInToAzure', AzureAccount.signInToAzure);
     registerCommandWithTelemetry('azure-api-center.selectTenant', AzureAccount.selectTenant);
     registerCommandWithTelemetry('azure-api-center.selectSubscriptions', AzureAccount.selectSubscriptions);
     registerCommandWithTelemetry('azure-api-center.openUrl', openUrlFromTreeNode);
     registerCommandWithTelemetry('azure-api-center.apiCenterWorkspace.signInToDataPlane', SignInToDataPlane);
     registerCommandWithTelemetry('azure-api-center.apiCenterWorkspace.refresh', async (context: IActionContext) => ext.workspaceItem.refresh(context));
     registerCommandWithTelemetry('azure-api-center.apiCenterWorkspace.addApis', getDataPlaneApis);
     registerCommandWithTelemetry('azure-api-center.apiCenterWorkspace.collapse', () => {
         vscode.commands.executeCommand('workbench.actions.treeView.apiCenterWorkspace.collapseAll');
     });
 
     registerCommandWithTelemetry('azure-api-center.apiCenterWorkspace.removeApi', removeDataplaneAPI);
 
     context.subscriptions.push(
         vscode.window.registerUriHandler({
             handleUri
         })
     );
 }
 
 async function registerCommandWithTelemetry(commandId: string, callback: CommandCallback, debounce?: number): Promise<void> {
@@ -203,4 +171,34 @@
     }, debounce);
 }
 
+function setupControlView(context: vscode.ExtensionContext) {
+    AzureSessionProviderHelper.activateAzureSessionProvider(context);
+    const sessionProvider = AzureSessionProviderHelper.getSessionProvider();
+    const azureAccountTreeItem = createAzureAccountTreeItem(sessionProvider);
+    context.subscriptions.push(azureAccountTreeItem);
+    ext.treeItem = azureAccountTreeItem;
+    const treeDataProvider = new AzExtTreeDataProvider(azureAccountTreeItem, "appService.loadMore");
+    ext.treeItem = azureAccountTreeItem;
+    ext.treeDataProvider = treeDataProvider;
+    const treeView = vscode.window.createTreeView("apiCenterTreeView", { treeDataProvider });
+    context.subscriptions.push(treeView);
+    treeView.onDidChangeSelection((e: vscode.TreeViewSelectionChangeEvent<AzExtTreeItem>) => {
+        const selectedNode = e.selection[0];
+        ext.outputChannel.appendLine(selectedNode.id!);
+        ext.context.globalState.update(selectedNodeKey, selectedNode.id);
+    });
+}
+
+function setupDataTreeView(context: vscode.ExtensionContext) {
+    ext.dataPlaneAccounts = [];
+    AzureDataSessionProviderHelper.activateAzureSessionProvider(context);
+    const dataPlaneSessionProvider = AzureDataSessionProviderHelper.getSessionProvider();
+    const dataPlanAccountManagerTreeItem = createAzureDataAccountTreeItem(dataPlaneSessionProvider);
+    context.subscriptions.push(dataPlanAccountManagerTreeItem);
+    ext.workspaceItem = dataPlanAccountManagerTreeItem;
+    const workspaceTreeDataProvider = new AzExtTreeDataProvider(dataPlanAccountManagerTreeItem, "appService.loadMore");
+    ext.workspaceProvider = workspaceTreeDataProvider;
+    vscode.window.registerTreeDataProvider('apiCenterWorkspace', workspaceTreeDataProvider);
+}
+
 export function deactivate() { }

src/uiStrings.ts

@@ -124,4 +124,5 @@ export class UiStrings {
     static readonly AddDataPlaneTenantId = vscode.l10n.t("Input Entra App Tenant ID");
     static readonly RequestFailedWithStatusCode = vscode.l10n.t("Request failed with status code: {0}");
     static readonly DownloadDefinitionFileWithErrorMsg = vscode.l10n.t("Download API Center Definition File error: {0}");
+    static readonly DatplaneAlreadyAdded = vscode.l10n.t("This Data Plane Runtime URL already added to Data View.");
 }