+
+
+
+
+ `;
+ }
+}
diff --git a/src/commands/components/cmds.ts b/src/commands/components/cmds.ts
new file mode 100644
index 00000000..36c28fb0
--- /dev/null
+++ b/src/commands/components/cmds.ts
@@ -0,0 +1,159 @@
+// Copyright(c) ZenML GmbH 2024. All Rights Reserved.
+// Licensed under the Apache License, Version 2.0(the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at:
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+// or implied.See the License for the specific language governing
+// permissions and limitations under the License.
+import * as vscode from 'vscode';
+
+import ZenMLStatusBar from '../../views/statusBar';
+import { LSClient } from '../../services/LSClient';
+import { showInformationMessage } from '../../utils/notifications';
+import Panels from '../../common/panels';
+import { ComponentDataProvider } from '../../views/activityBar/componentView/ComponentDataProvider';
+import { ComponentTypesResponse, Flavor, FlavorListResponse } from '../../types/StackTypes';
+import { getFlavor, getFlavorsOfType } from '../../common/api';
+import ComponentForm from './ComponentsForm';
+import { StackComponentTreeItem } from '../../views/activityBar';
+import { traceError, traceInfo } from '../../common/log/logging';
+
+/**
+ * Refreshes the stack component view.
+ */
+const refreshComponentView = async () => {
+ try {
+ await vscode.window.withProgress(
+ {
+ location: vscode.ProgressLocation.Notification,
+ title: 'Refreshing Component View...',
+ cancellable: false,
+ },
+ async progress => {
+ await ComponentDataProvider.getInstance().refresh();
+ }
+ );
+ } catch (e) {
+ vscode.window.showErrorMessage(`Failed to refresh component view: ${e}`);
+ traceError(`Failed to refresh component view: ${e}`);
+ console.error(`Failed to refresh component view: ${e}`);
+ }
+};
+
+/**
+ * Allows one to choose a component type and flavor, then opens the component
+ * form webview panel to a form specific to register a new a component of that
+ * type and flavor.
+ */
+const registerComponent = async () => {
+ const lsClient = LSClient.getInstance();
+ try {
+ const types = await lsClient.sendLsClientRequestRegister {{type}} Stack Component ({{flavor}})
+ +
+ 
{{{description}}}
+
+
+ {{#if docs_url}}
+ Documentation
+ {{/if}}
+
+ {{#if sdk_docs_url}}
+ SDK Documentation
+ {{/if}}
+
+
+
Register Stack
+ +
+
+
+
+ `;
+ }
+}
diff --git a/src/commands/stack/cmds.ts b/src/commands/stack/cmds.ts
index 9525824a..026fb78b 100644
--- a/src/commands/stack/cmds.ts
+++ b/src/commands/stack/cmds.ts
@@ -11,11 +11,15 @@
// or implied.See the License for the specific language governing
// permissions and limitations under the License.
import * as vscode from 'vscode';
-import { StackDataProvider, StackTreeItem } from '../../views/activityBar';
+import { StackComponentTreeItem, StackDataProvider, StackTreeItem } from '../../views/activityBar';
import ZenMLStatusBar from '../../views/statusBar';
import { getStackDashboardUrl, switchActiveStack } from './utils';
import { LSClient } from '../../services/LSClient';
import { showInformationMessage } from '../../utils/notifications';
+import Panels from '../../common/panels';
+import { randomUUID } from 'crypto';
+import StackForm from './StackForm';
+import { traceError, traceInfo } from '../../common/log/logging';
/**
* Refreshes the stack view.
@@ -178,6 +182,78 @@ const goToStackUrl = (node: StackTreeItem) => {
}
};
+/**
+ * Opens the stack form webview panel to a form specific to registering a new
+ * stack.
+ */
+const registerStack = () => {
+ StackForm.getInstance().registerForm();
+};
+
+/**
+ * Opens the stack form webview panel to a form specific to updating a specified stack.
+ * @param {StackTreeItem} node The specified stack to update.
+ */
+const updateStack = async (node: StackTreeItem) => {
+ const { id, label: name } = node;
+ const components: { [type: string]: string } = {};
+
+ node.children?.forEach(child => {
+ if (child instanceof StackComponentTreeItem) {
+ const { type, id } = (child as StackComponentTreeItem).component;
+ components[type] = id;
+ }
+ });
+
+ StackForm.getInstance().updateForm(id, name, components);
+};
+
+/**
+ * Deletes a specified stack.
+ *
+ * @param {StackTreeItem} node The Stack to delete
+ */
+const deleteStack = async (node: StackTreeItem) => {
+ const lsClient = LSClient.getInstance();
+
+ const answer = await vscode.window.showWarningMessage(
+ `Are you sure you want to delete ${node.label}? This cannot be undone.`,
+ { modal: true },
+ 'Delete'
+ );
+
+ if (!answer) {
+ return;
+ }
+
+ await vscode.window.withProgress(
+ {
+ location: vscode.ProgressLocation.Window,
+ title: `Deleting stack ${node.label}...`,
+ },
+ async () => {
+ const { id } = node;
+
+ try {
+ const resp = await lsClient.sendLsClientRequest('deleteStack', [id]);
+
+ if ('error' in resp) {
+ throw resp.error;
+ }
+
+ vscode.window.showInformationMessage(`${node.label} deleted`);
+ traceInfo(`${node.label} deleted`);
+
+ StackDataProvider.getInstance().refresh();
+ } catch (e) {
+ vscode.window.showErrorMessage(`Failed to delete component: ${e}`);
+ traceError(e);
+ console.error(e);
+ }
+ }
+ );
+};
+
export const stackCommands = {
refreshStackView,
refreshActiveStack,
@@ -185,4 +261,7 @@ export const stackCommands = {
copyStack,
setActiveStack,
goToStackUrl,
+ registerStack,
+ updateStack,
+ deleteStack,
};
diff --git a/src/commands/stack/registry.ts b/src/commands/stack/registry.ts
index e8973ccd..2a6cef3f 100644
--- a/src/commands/stack/registry.ts
+++ b/src/commands/stack/registry.ts
@@ -35,6 +35,14 @@ export const registerStackCommands = (context: ExtensionContext) => {
'zenml.refreshActiveStack',
async () => await stackCommands.refreshActiveStack()
),
+ registerCommand('zenml.registerStack', async () => stackCommands.registerStack()),
+ registerCommand('zenml.updateStack', async (node: StackTreeItem) =>
+ stackCommands.updateStack(node)
+ ),
+ registerCommand(
+ 'zenml.deleteStack',
+ async (node: StackTreeItem) => await stackCommands.deleteStack(node)
+ ),
registerCommand(
'zenml.renameStack',
async (node: StackTreeItem) => await stackCommands.renameStack(node)
diff --git a/src/common/WebviewBase.ts b/src/common/WebviewBase.ts
new file mode 100644
index 00000000..52530645
--- /dev/null
+++ b/src/common/WebviewBase.ts
@@ -0,0 +1,30 @@
+// Copyright(c) ZenML GmbH 2024. All Rights Reserved.
+// Licensed under the Apache License, Version 2.0(the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at:
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+// or implied.See the License for the specific language governing
+// permissions and limitations under the License.
+import * as vscode from 'vscode';
+
+/**
+ * Provides functionality to share extension context among classes that inherit
+ * from it.
+ */
+export default class WebviewBase {
+ protected static context: vscode.ExtensionContext | null = null;
+
+ /**
+ * Sets the extension context so that descendant classes can correctly
+ * path to their resources
+ * @param {vscode.ExtensionContext} context ExtensionContext
+ */
+ public static setContext(context: vscode.ExtensionContext) {
+ WebviewBase.context = context;
+ }
+}
diff --git a/src/common/api.ts b/src/common/api.ts
new file mode 100644
index 00000000..90f96e93
--- /dev/null
+++ b/src/common/api.ts
@@ -0,0 +1,114 @@
+// Copyright(c) ZenML GmbH 2024. All Rights Reserved.
+// Licensed under the Apache License, Version 2.0(the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at:
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+// or implied.See the License for the specific language governing
+// permissions and limitations under the License.
+import { LSClient } from '../services/LSClient';
+import {
+ ComponentsListResponse,
+ Flavor,
+ FlavorListResponse,
+ StackComponent,
+} from '../types/StackTypes';
+
+let flavors: Flavor[] = [];
+
+/**
+ * Gets all component flavors and caches them
+ * @returns {Flavor[]} List of flavors
+ */
+export const getAllFlavors = async (): Promise