Export CLI_UPSTREAM vars to terminals (#1215)

Export CLI_UPSTREAM vars to terminals
databricks · May 2, 2024 · 6915b4b · 6915b4b
1 parent 53d0f2c
commit 6915b4b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/packages/databricks-vscode/src/extension.ts b/packages/databricks-vscode/src/extension.ts
@@ -183,6 +183,16 @@ export async function activate(
         `${path.delimiter}${context.asAbsolutePath("./bin")}`
     );
 
+    // Export CLI_UPSTREAM vars to the terminal to see if extension users use the CLI directly
+    context.environmentVariableCollection.replace(
+        "DATABRICKS_CLI_UPSTREAM",
+        "databricks-vscode-terminal"
+    );
+    context.environmentVariableCollection.replace(
+        "DATABRICKS_CLI_UPSTREAM_VERSION",
+        packageMetadata.version
+    );
+
     // We always use bundled terraform and databricks provider.
     // Updating environment collection means that the variables will be set in all terminals.
     // If users use different CLI version in their terminal it will only pick the variables if