Merge pull request #3 from zenml-io/default-view

Default view

.devcontainer/Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.11.9-slim-bookworm
+
+# Install git
+RUN apt-get -y update
+RUN apt-get -y install git
+
+# Install ZenML
+RUN pip install --no-cache-dir zenml
+RUN zenml integration install sklearn -y

.devcontainer/devcontainer.json

@@ -1,5 +1,7 @@
 {
-  "image": "mcr.microsoft.com/devcontainers/universal:2",
+  "build": {
+    "dockerfile": "Dockerfile"
+  },
   "customizations": {
     // Configure properties specific to VS Code.
     "vscode": {
@@ -22,5 +24,5 @@
   "containerEnv": {
     "PYTHONPATH": "/workspaces/vscode-quickstart-extension/zenmlQuickstart/quickstartModules"
   },
-  "onCreateCommand": "pip install zenml && zenml integration install sklearn -y && zenml init && zenml stack set default"
+  "onCreateCommand": "zenml init && zenml stack set default"
 }

package.json

@@ -11,7 +11,9 @@
   "categories": [
     "Other"
   ],
-  "activationEvents": [],
+  "activationEvents": [
+    "onStartupFinished"
+  ],
   "main": "./out/extension.js",
   "contributes": {
     "viewsContainers": {
@@ -41,7 +43,8 @@
     "pretest": "npm run compile && npm run lint",
     "lint": "eslint src --ext ts",
     "test": "vscode-test",
-    "buildExtension:replace": "vsce package && rm .devcontainer/extensions/zenml-vscode* && mv zenml-vscode* .devcontainer/extensions"
+    "buildExtension:replace": "vsce package && rm .devcontainer/extensions/zenml-vscode* && mv zenml-vscode* .devcontainer/extensions && npm run moveExtension:quickstart",
+    "moveExtension:quickstart": "rm ../vscode-quickstart/.devcontainer/extensions/zenml-vscode-quickstart* && cp .devcontainer/extensions/zenml-vscode* ../vscode-quickstart/.devcontainer/extensions"
   },
   "devDependencies": {
     "@types/mocha": "^10.0.7",

src/extension.ts

@@ -4,12 +4,12 @@ import * as vscode from "vscode";
 import quickstartMetadata from "./quickstartMetadata.json";
 import Quickstart from "./quickstart";
 import ZenmlViewProvider from "./zenmlViewProvider";
-import { setCWD, unsetCWD } from "./utils/setExtensionCWD";
+import setDirectory from "./utils/setExtensionDirectory";
 
 export async function activate(context: vscode.ExtensionContext) {
-  // if running in production set correct cwd for local devcontainer or codespace
+  // if running in production set working directory for local devcontainer or codespace
   if (context.extensionMode === vscode.ExtensionMode.Production) {
-    setCWD();
+    setDirectory();
   }
 
   const quickstart = new Quickstart(quickstartMetadata, context);
@@ -23,6 +23,9 @@ export async function activate(context: vscode.ExtensionContext) {
     )
   );
 
+  // Focuses the webview side panel
+  await vscode.commands.executeCommand("zenml.stepsView.focus");
+
   // If a user closes the terminal the extension opened we set it
   // back to undefined so we know to open a new terminal
   context.subscriptions.push(
@@ -37,7 +40,4 @@ export async function activate(context: vscode.ExtensionContext) {
 }
 
 // This method is called when your extension is deactivated
-export function deactivate() {
-  // Sets cwd back to default
-  unsetCWD();
-}
+export function deactivate() {}

src/utils/setExtensionDirectory.ts

@@ -0,0 +1,26 @@
+import * as vscode from "vscode";
+
+export default async function setDirectory() {
+  const isCodespace = process.env.CODESPACES === "true";
+
+  const targetUri = vscode.Uri.file(
+    `/root/${
+      isCodespace ? ".vscode-remote" : ".vscode-server"
+    }/extensions/zenml.zenml-vscode-quickstart-0.0.1/zenmlQuickstart/sections`
+  );
+
+  const currentWorkspace = vscode.workspace.workspaceFolders;
+
+  if (
+    !currentWorkspace ||
+    currentWorkspace[0].uri.fsPath !== targetUri.fsPath
+  ) {
+    try {
+      await vscode.commands.executeCommand("vscode.openFolder", targetUri);
+    } catch (e) {
+      vscode.window.showErrorMessage(
+        `There was a problem opening the directory: ${e}`
+      );
+    }
+  }
+}