WIP Workspace Explorer

packages/databricks-vscode/package.json

@@ -179,14 +179,44 @@
                 "command": "databricks.wsfs.refresh",
                 "title": "Refresh workspace filesystem view",
                 "icon": "$(refresh)",
-                "enablement": "databricks.context.activated && databricks.context.loggedIn && config.databricks.sync.destinationType == workspace && databricks.feature.views.workspace && !databricks.context.remoteMode",
+                "enablement": "databricks.context.activated && databricks.context.loggedIn && config.databricks.sync.destinationType == workspace && !databricks.context.remoteMode",
                 "category": "Databricks"
             },
             {
                 "command": "databricks.wsfs.createFolder",
                 "title": "Create Folder",
                 "icon": "$(new-folder)",
-                "enablement": "databricks.context.activated && databricks.context.loggedIn && databricks.feature.views.workspace && !databricks.context.remoteMode",
+                "enablement": "databricks.context.activated && databricks.context.loggedIn && !databricks.context.remoteMode",
+                "category": "Databricks"
+            },
+            {
+                "command": "databricks.wsfs.openInBrowser",
+                "title": "Open in Browser",
+                "icon": "$(link-external)",
+                "category": "Databricks"
+            },
+            {
+                "command": "databricks.wsfs.copyPath",
+                "title": "Copy Path",
+                "icon": "$(copy)",
+                "category": "Databricks"
+            },
+            {
+                "command": "databricks.wsfs.delete",
+                "title": "Delete",
+                "icon": "$(trash)",
+                "category": "Databricks"
+            },
+            {
+                "command": "databricks.wsfs.uploadFile",
+                "title": "Upload File",
+                "icon": "$(cloud-upload)",
+                "category": "Databricks"
+            },
+            {
+                "command": "databricks.wsfs.downloadFile",
+                "title": "Download",
+                "icon": "$(cloud-download)",
                 "category": "Databricks"
             },
             {
@@ -210,7 +240,6 @@
                 "icon": "$(gear)",
                 "title": "Select a Declarative Automation Bundle target",
                 "enablement": "databricks.context.activated && !databricks.context.remoteMode",
-
                 "category": "Databricks"
             },
             {
@@ -450,8 +479,8 @@
                 },
                 {
                     "id": "workspaceFsView",
-                    "name": "Workspace explorer",
-                    "when": "databricks.feature.views.workspace && !databricks.context.remoteMode"
+                    "name": "Workspace file system",
+                    "when": "!databricks.context.remoteMode"
                 },
                 {
                     "id": "databricksDocsView",
@@ -537,6 +566,11 @@
                     "when": "view == workspaceFsView",
                     "group": "navigation@1"
                 },
+                {
+                    "command": "databricks.wsfs.uploadFile",
+                    "when": "view == workspaceFsView",
+                    "group": "navigation@1"
+                },
                 {
                     "command": "databricks.bundle.refreshRemoteState",
                     "when": "view == dabsResourceExplorerView && databricks.context.bundle.deploymentState == idle",
@@ -583,6 +617,36 @@
                 }
             ],
             "view/item/context": [
+                {
+                    "command": "databricks.wsfs.copyPath",
+                    "when": "view == workspaceFsView && viewItem =~ /^wsfs\\./",
+                    "group": "wsfs_nav@0"
+                },
+                {
+                    "command": "databricks.wsfs.openInBrowser",
+                    "when": "view == workspaceFsView && viewItem =~ /^wsfs\\./",
+                    "group": "wsfs_nav@1"
+                },
+                {
+                    "command": "databricks.wsfs.createFolder",
+                    "when": "view == workspaceFsView && (viewItem == wsfs.directory || viewItem == wsfs.repo)",
+                    "group": "wsfs_mut@0"
+                },
+                {
+                    "command": "databricks.wsfs.uploadFile",
+                    "when": "view == workspaceFsView && (viewItem == wsfs.directory || viewItem == wsfs.repo)",
+                    "group": "wsfs_mut@1"
+                },
+                {
+                    "command": "databricks.wsfs.downloadFile",
+                    "when": "view == workspaceFsView && (viewItem == wsfs.file || viewItem == wsfs.notebook)",
+                    "group": "wsfs_mut@0"
+                },
+                {
+                    "command": "databricks.wsfs.delete",
+                    "when": "view == workspaceFsView && viewItem =~ /^wsfs\\./",
+                    "group": "wsfs_danger@0"
+                },
                 {
                     "command": "databricks.utils.openExternal",
                     "when": "viewItem =~ /^databricks.*\\.(has-url).*$/ && databricks.context.bundle.deploymentState == idle",
@@ -841,6 +905,26 @@
                     "command": "databricks.wsfs.refresh",
                     "when": "config.databricks.sync.destinationType == workspace"
                 },
+                {
+                    "command": "databricks.wsfs.delete",
+                    "when": "false"
+                },
+                {
+                    "command": "databricks.wsfs.downloadFile",
+                    "when": "false"
+                },
+                {
+                    "command": "databricks.wsfs.copyPath",
+                    "when": "false"
+                },
+                {
+                    "command": "databricks.wsfs.openInBrowser",
+                    "when": "false"
+                },
+                {
+                    "command": "databricks.wsfs.uploadFile",
+                    "when": "false"
+                },
                 {
                     "command": "databricks.utils.openExternal",
                     "when": "false"

packages/databricks-vscode/src/extension.ts

@@ -29,7 +29,11 @@ import {
     UtilsCommands,
 } from "./utils";
 import {ConfigureAutocomplete} from "./language/ConfigureAutocomplete";
-import {WorkspaceFsCommands, WorkspaceFsDataProvider} from "./workspace-fs";
+import {
+    WorkspaceFsCommands,
+    WorkspaceFsDataProvider,
+    WorkspaceFsFileSystemProvider,
+} from "./workspace-fs";
 import {CustomWhenContext} from "./vscode-objs/CustomWhenContext";
 import {StateStorage} from "./vscode-objs/StateStorage";
 import path from "node:path";
@@ -265,7 +269,6 @@ export async function activate(
     // manage contexts for experimental features
     function updateFeatureContexts() {
         customWhenContext.updateShowClusterView();
-        customWhenContext.updateShowWorkspaceView();
     }
 
     function updateStrictSSLEnv() {
@@ -382,13 +385,19 @@ export async function activate(
     const workspaceFsDataProvider = new WorkspaceFsDataProvider(
         connectionManager
     );
+    const workspaceFsFsp = new WorkspaceFsFileSystemProvider(connectionManager);
     const workspaceFsCommands = new WorkspaceFsCommands(
         workspaceFolderManager,
         connectionManager,
-        workspaceFsDataProvider
+        workspaceFsDataProvider,
+        workspaceFsFsp
     );
 
     context.subscriptions.push(
+        workspace.registerFileSystemProvider("wsfs", workspaceFsFsp, {
+            isCaseSensitive: true,
+        }),
+        workspaceFsFsp,
         window.registerTreeDataProvider(
             "workspaceFsView",
             workspaceFsDataProvider
@@ -402,6 +411,31 @@ export async function activate(
             "databricks.wsfs.createFolder",
             workspaceFsCommands.createFolder,
             workspaceFsCommands
+        ),
+        telemetry.registerCommand(
+            "databricks.wsfs.openInBrowser",
+            workspaceFsCommands.openInBrowser,
+            workspaceFsCommands
+        ),
+        telemetry.registerCommand(
+            "databricks.wsfs.copyPath",
+            workspaceFsCommands.copyPath,
+            workspaceFsCommands
+        ),
+        telemetry.registerCommand(
+            "databricks.wsfs.delete",
+            workspaceFsCommands.deleteItem,
+            workspaceFsCommands
+        ),
+        telemetry.registerCommand(
+            "databricks.wsfs.uploadFile",
+            workspaceFsCommands.uploadFile,
+            workspaceFsCommands
+        ),
+        telemetry.registerCommand(
+            "databricks.wsfs.downloadFile",
+            workspaceFsCommands.downloadFile,
+            workspaceFsCommands
         )
     );
 

packages/databricks-vscode/src/sdk-extensions/wsfs/WorkspaceFsDir.ts

@@ -6,7 +6,7 @@ import {isDirectory, isFile} from "./utils";
 
 export class WorkspaceFsDir extends WorkspaceFsEntity {
     override async generateUrl(host: URL): Promise<string> {
-        return `${host.host}/browse/folders/${this.details.object_id}`;
+        return `${host.origin}/browse/folders/${this.details.object_id}`;
     }
 
     public getAbsoluteChildPath(path: string) {

packages/databricks-vscode/src/sdk-extensions/wsfs/WorkspaceFsEntity.ts

@@ -175,6 +175,14 @@ export abstract class WorkspaceFsEntity {
     get basename(): string {
         return posix.basename(this.path);
     }
+
+    @withLogContext(ExposedLoggers.SDK)
+    async delete(recursive = false, @context ctx?: Context): Promise<void> {
+        await this._workspaceFsService.delete(
+            {path: this.path, recursive},
+            ctx
+        );
+    }
 }
 
 async function entityFromObjInfo(

packages/databricks-vscode/src/sdk-extensions/wsfs/WorkspaceFsFile.ts

@@ -6,7 +6,15 @@ export class WorkspaceFsFile extends WorkspaceFsEntity {
     }
 
     override async generateUrl(host: URL): Promise<string> {
-        return `${host.host}#folder/${(await this.parent)?.id ?? ""}`;
+        return `${host.origin}/editor/files/${this.details.object_id}`;
+    }
+
+    async readContent(): Promise<Uint8Array> {
+        const result = await this._workspaceFsService.export({
+            path: this.path,
+            format: "AUTO",
+        });
+        return Buffer.from(result.content ?? "", "base64");
     }
 }
 

packages/databricks-vscode/src/test/e2e/utils/commonUtils.ts

@@ -12,7 +12,7 @@ import {
 const ViewSectionTypes = [
     "CLUSTERS",
     "CONFIGURATION",
-    "WORKSPACE EXPLORER",
+    "WORKSPACE FILE SYSTEM",
     "BUNDLE RESOURCE EXPLORER",
     "BUNDLE VARIABLES",
     "DOCUMENTATION",

packages/databricks-vscode/src/vscode-objs/CustomWhenContext.ts

@@ -85,16 +85,6 @@ export class CustomWhenContext {
         );
     }
 
-    updateShowWorkspaceView() {
-        commands.executeCommand(
-            "setContext",
-            "databricks.feature.views.workspace",
-            workspaceConfigs.experimetalFeatureOverides.includes(
-                "views.workspace"
-            )
-        );
-    }
-
     setIsActiveFileInActiveWorkspace(value: boolean) {
         commands.executeCommand(
             "setContext",