feat: data plane view support

package.nls.json

@@ -31,5 +31,5 @@
     "azure-api-center.views.api-center-treeview.controlplane.title": "Azure API Center",
     "azure-api-center.views.api-center-treeview.dataplane.title": "API Center Data View",
     "azure-api-center.commands.apiCenterWorkspace.addApis.title": "Connect to an API Center",
-    "azure-api-center.commands.apiCenterWorkspace.removeApis.title": "Remove API Center",
+    "azure-api-center.commands.apiCenterWorkspace.removeApis.title": "Remove API Center"
 }

src/commands/exportApi.ts

@@ -6,17 +6,25 @@
 import * as path from "path";
 import * as vscode from "vscode";
 import { ApiSpecExportResultFormat } from "../azure/ApiCenter/contracts";
-import { ApiCenterVersionDefinitionManagement } from "../azure/ApiCenterDefines/ApiCenterDefinition";
+import { ApiCenterVersionDefinitionDataPlane, ApiCenterVersionDefinitionManagement } from "../azure/ApiCenterDefines/ApiCenterDefinition";
+import { TreeViewType } from "../constants";
 import { ext } from "../extensionVariables";
 import { ApiVersionDefinitionTreeItem } from "../tree/ApiVersionDefinitionTreeItem";
+import { ApiServerItem } from "../tree/DataPlaneAccount";
+import { SubscriptionTreeItem } from "../tree/SubscriptionTreeItem";
+import { UiStrings } from "../uiStrings";
 import { createTemporaryFolder } from "../utils/fsUtil";
 import { GeneralUtils } from "../utils/generalUtils";
 export namespace ExportAPI {
     export async function exportApi(
         context: IActionContext,
         node?: ApiVersionDefinitionTreeItem): Promise<void> {
         if (!node) {
-            node = await ext.treeDataProvider.showTreeItemPicker<ApiVersionDefinitionTreeItem>(new RegExp(`${ApiCenterVersionDefinitionManagement.contextValue}*`), context);
+            let res = await getDefinitionTreeNode(context);
+            if (!res) {
+                return;
+            }
+            node = res;
         }
 
         const exportedSpec = await node?.apiCenterApiVersionDefinition.getDefinitions(node?.subscription!, node?.apiCenterName!, node?.apiCenterApiName!, node?.apiCenterApiVersionName!);
@@ -28,7 +36,7 @@
     }
 
     function getFilename(treeItem: ApiVersionDefinitionTreeItem): string {
         return `${treeItem.apiCenterApiVersionDefinition.getName()}`;
     }
 
     async function writeToTempFile(node: ApiVersionDefinitionTreeItem, specFormat: string, specValue: string) {
@@ -49,4 +57,33 @@
         await vscode.workspace.fs.writeFile(vscode.Uri.file(localFilePath), Buffer.from(fileContent));
         await vscode.window.showTextDocument(document);
     }
+
+    async function getDefinitionTreeNode(context: IActionContext): Promise<ApiVersionDefinitionTreeItem | null> {
+        const controlViewItem = await ext.dataPlaneTreeDataProvier.getChildren(ext.treeItem)
+        const isControlPlaneExist = controlViewItem.some(item => item.contextValue === SubscriptionTreeItem.contextValue);
+        const dataViewItem = await ext.treeDataProvider.getChildren(ext.dataPlaneTreeItem);
+        const isDataPlaneExist = dataViewItem.some(item => item.contextValue === ApiServerItem.contextValue);
+        if (!isControlPlaneExist && !isDataPlaneExist) {
+            vscode.window.showInformationMessage(UiStrings.GetTreeView);
+            return null;
+        }
+        if (!isDataPlaneExist) {
+            return await ext.treeDataProvider.showTreeItemPicker<ApiVersionDefinitionTreeItem>(new RegExp(`${ApiCenterVersionDefinitionManagement.contextValue}*`), context);
+        }
+        if (!isControlPlaneExist) {
+            return await ext.dataPlaneTreeDataProvier.showTreeItemPicker<ApiVersionDefinitionTreeItem>(new RegExp(`${ApiCenterVersionDefinitionDataPlane.contextValue}*`), context);
+        }
+        const viewType = await vscode.window.showQuickPick(Object.values(TreeViewType), { title: UiStrings.SelectItemFromTreeView, ignoreFocusOut: true });
+        if (!viewType) {
+            return null;
+        }
+        switch (viewType) {
+            case TreeViewType.controlPlaneView:
+                return await ext.treeDataProvider.showTreeItemPicker<ApiVersionDefinitionTreeItem>(new RegExp(`${ApiCenterVersionDefinitionManagement.contextValue}*`), context);
+            case TreeViewType.dataPlaneView:
+                return await ext.dataPlaneTreeDataProvier.showTreeItemPicker<ApiVersionDefinitionTreeItem>(new RegExp(`${ApiCenterVersionDefinitionDataPlane.contextValue}*`), context);
+            default:
+                return null;
+        }
+    }
 }

src/constants.ts

@@ -34,6 +34,11 @@
     cicd: UiStrings.RegisterApiOptionCicd,
 };
 
+export const TreeViewType = {
+    controlPlaneView: "API Center Management View",
+    dataPlaneView: "API Center Data Plane View",
+}
+
 export const ApiRulesetOptions = {
     default: UiStrings.ApiRulesetOptionDefault,
     azureApiGuideline: UiStrings.ApiRulesetOptionAzureApiGuideline,

src/tree/SubscriptionTreeItem.ts

@@ -14,7 +14,7 @@ export function createSubscriptionTreeItem(
     return new SubscriptionTreeItem(parent, subscription);
 }
 
-class SubscriptionTreeItem extends AzExtParentTreeItem {
+export class SubscriptionTreeItem extends AzExtParentTreeItem {
     public readonly subscriptionContext: ISubscriptionContext;
     public readonly subscriptionId: string;
     public static contextValue: string = "azureApiCenterAzureSubscription";

src/uiStrings.ts

@@ -125,4 +125,6 @@ export class UiStrings {
     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.");
+    static readonly SelectItemFromTreeView = vscode.l10n.t("Select from which tree view");
+    static readonly GetTreeView = vscode.l10n.t("Please Sign In to Azure to get management view or connect to data plane APIs");
 }

src/utils/generalUtils.ts

@@ -56,12 +56,7 @@
     }
 
     export async function fetchDataFromLink(link: string): Promise<string> {
-        try {
-            const res = await axios.get(link);
-            return JSON.stringify(res.data);
-        }
-        catch (err) {
-            throw err;
-        }
+        const res = await axios.get(link);
+        return JSON.stringify(res.data);
     }
 }