Pull bricks CLI binary and add a test wrapper (#68)

* Package bricks cli with VSCode for darwin_arm64

* manual package action

* manual package action

* put package github action back

* vscodeignore scripts

* update fetch script

* listen to process close

* update task

Author: kartikgupta-db

packages/databricks-vscode/.gitignore

@@ -0,0 +1 @@
+bin/**

packages/databricks-vscode/.vscode/tasks.json

@@ -2,16 +2,32 @@
 // for the documentation about the tasks.json format
 {
     "version": "2.0.0",
+    "options": {
+        "env": {
+            "BRICKS_ARCH": "darwin_arm64"
+        }
+    },
     "tasks": [
         {
+            "label": "Fetch bricks cli",
+            "type": "shell",
+            "command": "yarn",
+            "args": ["run", "package:cli:fetch"]
+        },
+        {
+            "label": "NPM watch",
             "type": "npm",
             "script": "watch",
             "problemMatcher": "$tsc-watch",
             "isBackground": true,
             "presentation": {
                 "reveal": "never",
                 "group": "watchers"
-            },
+            }
+        },
+        {
+            "label": "Build",
+            "dependsOn": ["Fetch bricks cli", "NPM watch"],
             "group": {
                 "kind": "build",
                 "isDefault": true

packages/databricks-vscode/.vscodeignore

@@ -9,3 +9,4 @@ src/**
 **/.eslintrc.json
 **/*.map
 **/*.ts
+scripts/**

packages/databricks-vscode/package.json

@@ -48,6 +48,7 @@
         "onCommand:databricks.cli.startSync",
         "onCommand:databricks.cli.startSyncFull",
         "onCommand:databricks.cli.stopSync",
+        "onCommand:databricks.cli.testBricksCli",
         "onCommand:databricks.cluster.refresh",
         "onCommand:databricks.cluster.filterByAll",
         "onCommand:databricks.cluster.filterByMe",
@@ -104,6 +105,10 @@
                 "command": "databricks.cli.stopSync",
                 "title": "Databricks: Stop synchronization"
             },
+            {
+                "command": "databricks.cli.testBricksCli",
+                "title": "Databricks: test bricks cli"
+            },
             {
                 "command": "databricks.cluster.filterByAll",
                 "title": "All"
@@ -267,6 +272,8 @@
     "scripts": {
         "vscode:prepublish": "rm -rf out && yarn run package:compile && yarn run package:copy-webview-toolkit",
         "package": "vsce package",
+        "package:cli:fetch": "bash ./scripts/fetch-latest-bricks-cli.sh $BRICKS_ARCH",
+        "package:cli:release": "bash ./scripts/fetch-latest-bricks-cli.sh $BRICKS_ARCH",
         "package:compile": "yarn run esbuild:base --minify",
         "package:copy-webview-toolkit": "cp ./node_modules/@vscode/webview-ui-toolkit/dist/toolkit.js ./out/toolkit.js",
         "esbuild:base": "esbuild ./src/extension.ts --bundle --outfile=out/extension.js --external:vscode --format=cjs --platform=node",

packages/databricks-vscode/scripts/fetch-latest-bricks-cli.sh

@@ -0,0 +1,11 @@
+BRICKS_ARCH=$1
+pushd /tmp
+rm -rf bricks_*
+gh release download -R databricks/bricks -p "*$BRICKS_ARCH.tar.gz"
+tar -xvf bricks_*_$BRICKS_ARCH.tar.gz
+
+popd
+mkdir -p bin
+cd ./bin
+rm -rf bricks
+mv /tmp/bricks ./

packages/databricks-vscode/src/cli/CliCommands.ts

@@ -1,4 +1,6 @@
+import {spawn} from "child_process";
 import {
+    ExtensionContext,
     ProcessExecution,
     Task,
     TaskExecution,
@@ -19,6 +21,29 @@ export class CliCommands {
         private cli: CliWrapper
     ) {}
 
+    testBricksCommand(context: ExtensionContext) {
+        return () => {
+            const {command, args} = this.cli.getTestBricksCommand(context);
+            const cmd = spawn(command, args);
+
+            cmd.stdout.on("data", (data) => {
+                console.log((data as Buffer).toString());
+            });
+
+            cmd.on("exit", (code) => {
+                if (code === 0) {
+                    console.log("Bricks cli found");
+                } else {
+                    console.error("Bricks cli NOT found");
+                }
+            });
+
+            cmd.on("error", () => {
+                console.error("Bricks cli NOT found");
+            });
+        };
+    }
+
     stopSyncCommand() {
         return () => {
             if (this.currentTaskExecution) {

packages/databricks-vscode/src/cli/CliWrapper.ts

@@ -1,4 +1,5 @@
 import {spawn} from "child_process";
+import {ExtensionContext} from "vscode";
 import {SyncDestination} from "../configuration/SyncDestination";
 
 interface Command {
@@ -15,6 +16,13 @@ interface Command {
 export class CliWrapper {
     constructor() {}
 
+    getTestBricksCommand(context: ExtensionContext): Command {
+        return {
+            command: context.asAbsolutePath("./bin/bricks"),
+            args: [],
+        };
+    }
+
     /**
      * Constructs the dbx sync command
      */

packages/databricks-vscode/src/extension.ts

@@ -134,6 +134,11 @@ export function activate(context: ExtensionContext) {
             "databricks.cli.stopSync",
             cliCommands.stopSyncCommand(),
             cliCommands
+        ),
+        commands.registerCommand(
+            "databricks.cli.testBricksCli",
+            cliCommands.testBricksCommand(context),
+            cliCommands
         )
     );
 }