Wait for initial sync to be complete before executing (#142)

1. Waits for initial upload to finish before executing remote code
2. Bumps up bricks cli version



https://user-images.githubusercontent.com/88374338/198308478-4357856d-d64d-4f16-a1e6-f1b39229a471.mov

Author: shreyas-goenka

packages/databricks-vscode/package.json

@@ -482,7 +482,7 @@
     "scripts": {
         "vscode:prepublish": "rm -rf out && yarn run package:compile && yarn run package:copy-webview-toolkit",
         "package": "vsce package",
-        "package:cli:fetch": "BRICKS_VERSION=v0.0.4 && bash ./scripts/fetch-bricks-cli.sh ${BRICKS_VERSION} ${BRICKS_ARCH:-}",
+        "package:cli:fetch": "BRICKS_VERSION=v0.0.6 && bash ./scripts/fetch-bricks-cli.sh ${BRICKS_VERSION} ${BRICKS_ARCH:-}",
         "package:cli:link": "rm -f ./bin/bricks && ln -s ../../../../bricks/bricks bin",
         "package:compile": "yarn run esbuild:base --minify",
         "package:copy-webview-toolkit": "cp ./node_modules/@vscode/webview-ui-toolkit/dist/toolkit.js ./out/toolkit.js",

packages/databricks-vscode/src/cli/BricksTasks.test.ts

@@ -56,13 +56,15 @@ describe("tests for BricksSycnParser", () => {
         );
     });
 
-    it("processing empty logs transitions sync status from INACTIVE -> WATCHING_FOR_CHANGES", () => {
+    it("processing empty logs transitions sync status from STOPPED -> IN_PROGRESS and we wait for initial sync complete", () => {
         assert.equal(syncState, "STOPPED");
         bricksSycnParser.process("");
+        assert.equal(syncState, "IN_PROGRESS");
+        bricksSycnParser.process("[INFO] Initial Sync Complete");
         assert.equal(syncState, "WATCHING_FOR_CHANGES");
     });
 
-    it("processing action log transitions sync status from INACTIVE -> INPROGRESS", () => {
+    it("processing action log transitions sync status from STOPPED -> INPROGRESS", () => {
         assert.equal(syncState, "STOPPED");
         bricksSycnParser.process("Action: PUT: hello.txt");
         assert.equal(syncState, "IN_PROGRESS");
@@ -72,12 +74,14 @@ describe("tests for BricksSycnParser", () => {
         // recieving some random logs from bricks sync
         assert.equal(syncState, "STOPPED");
         bricksSycnParser.process("some random logs");
-        assert.equal(syncState, "WATCHING_FOR_CHANGES");
+        assert.equal(syncState, "IN_PROGRESS");
 
         // upload  hello.txt
         bricksSycnParser.process("Action: PUT: hello.txt");
         assert.equal(syncState, "IN_PROGRESS");
         bricksSycnParser.process("Uploaded hello.txt");
+        assert.equal(syncState, "IN_PROGRESS");
+        bricksSycnParser.process("[INFO] Initial Sync Complete");
         assert.equal(syncState, "WATCHING_FOR_CHANGES");
 
         // delete  bye.txt

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

@@ -92,6 +92,7 @@ export class SyncTask extends Task {
 export class BricksSyncParser {
     private filesBeingUploaded = new Set<string>();
     private filesBeingDeleted = new Set<string>();
+    private firstSyncDone = false;
 
     constructor(
         private syncStateCallback: (state: SyncState) => void,
@@ -190,6 +191,15 @@ export class BricksSyncParser {
         this.filesBeingDeleted.delete(filePath);
     }
 
+    // We block on execing any commands on vscode until we get a message from
+    // bricks cli that the initial sync is done
+    private parseForFirstSync(line: string) {
+        var indexOfSyncComplete = line.indexOf("Initial Sync Complete");
+        if (indexOfSyncComplete !== -1) {
+            this.firstSyncDone = true;
+        }
+    }
+
     // This function processes the stderr logs from bricks sync and parses it
     // to compute the sync state ie determine whether the remote files match
     // what we have stored locally.
@@ -201,6 +211,11 @@ export class BricksSyncParser {
             this.parseForActionsInitiated(line);
             this.parseForUploadCompleted(line);
             this.parseForDeleteCompleted(line);
+
+            if (!this.firstSyncDone) {
+                this.parseForFirstSync(line);
+            }
+
             // this.writeEmitter.fire writes to the pseudoterminal for the
             // bricks sync process
             this.writeEmitter.fire(line.trim());
@@ -216,7 +231,8 @@ export class BricksSyncParser {
         }
         if (
             this.filesBeingDeleted.size === 0 &&
-            this.filesBeingUploaded.size === 0
+            this.filesBeingUploaded.size === 0 &&
+            this.firstSyncDone
         ) {
             this.syncStateCallback("WATCHING_FOR_CHANGES");
         } else {