Show t&c popup for private preview (#1128)

## Changes
<!-- Summary of your changes that are easy to understand -->

## Tests
<!-- How is this tested? -->

Author: kartikgupta-db

packages/databricks-vscode/src/extension.ts

@@ -5,6 +5,8 @@ import {
     extensions,
     window,
     workspace,
+    env,
+    Uri,
 } from "vscode";
 import {CliWrapper} from "./cli/CliWrapper";
 import {ConnectionCommands} from "./configuration/ConnectionCommands";
@@ -60,6 +62,7 @@ import {BundleProjectManager} from "./bundle/BundleProjectManager";
 import {TreeItemDecorationProvider} from "./ui/bundle-resource-explorer/DecorationProvider";
 import {BundleInitWizard} from "./bundle/BundleInitWizard";
 import {DatabricksDebugConfigurationProvider} from "./run/DatabricksDebugConfigurationProvider";
+import {isIntegrationTest} from "./utils/developmentUtils";
 
 const customWhenContext = new CustomWhenContext();
 
@@ -69,6 +72,43 @@ export async function activate(
     customWhenContext.setActivated(false);
     customWhenContext.setDeploymentState("idle");
 
+    const stateStorage = new StateStorage(context);
+
+    if (
+        !stateStorage.get("databricks.preview-tnc.accepted") &&
+        !isIntegrationTest()
+    ) {
+        const acceptTnc = await window.showInformationMessage(
+            `Please note that you should only be using this functionality if you are part of our private preview and have accepted our terms and conditions regarding this preview.`,
+            {
+                modal: true,
+                detail: `In order to enroll in the private preview please contact us and we will get you added`,
+            },
+            "Contact us",
+            "Continue if you are already enrolled"
+        );
+        switch (acceptTnc) {
+            case "Contact us":
+                env.openExternal(
+                    Uri.parse(
+                        "mailto:[email protected]?subject=Databricks+Extension+v2+Private+Preview+Enrollment+Request"
+                    )
+                );
+                window.showErrorMessage(
+                    "Databricks Extension is not activated"
+                );
+                return;
+            case "Continue if you are already enrolled":
+                stateStorage.set("databricks.preview-tnc.accepted", true);
+                break;
+            default:
+                window.showErrorMessage(
+                    "Databricks Extension is not activated"
+                );
+                return;
+        }
+    }
+
     if (extensions.getExtension("databricks.databricks-vscode") !== undefined) {
         await commands.executeCommand(
             "workbench.extensions.uninstallExtension",
@@ -126,7 +166,6 @@ export async function activate(
     }
 
     const workspaceUri = workspace.workspaceFolders[0].uri;
-    const stateStorage = new StateStorage(context);
 
     // Add the databricks binary to the PATH environment variable in terminals
     context.environmentVariableCollection.clear();

packages/databricks-vscode/src/telemetry/index.test.ts

@@ -40,7 +40,7 @@ describe(__filename, () => {
     });
 
     it("sets context metadata with prod env type", async () => {
-        delete process.env["TEST_DEFAULT_CLUSTER_ID"];
+        delete process.env["DATABRICKS_VSCODE_INTEGRATION_TEST"];
         telemetry.setMetadata(Metadata.CONTEXT, getContextMetadata());
         telemetry.recordEvent(Events.COMMAND_EXECUTION, {
             command: "testCommand",
@@ -58,7 +58,7 @@ describe(__filename, () => {
     });
 
     it("sets context metadata with tests env type", async () => {
-        process.env["TEST_DEFAULT_CLUSTER_ID"] = "123";
+        process.env["DATABRICKS_VSCODE_INTEGRATION_TEST"] = "true";
         telemetry.setMetadata(Metadata.CONTEXT, getContextMetadata());
         telemetry.recordEvent(Events.COMMAND_EXECUTION, {
             command: "testCommand",

packages/databricks-vscode/src/telemetry/index.ts

@@ -1,6 +1,6 @@
 import TelemetryReporter from "@vscode/extension-telemetry";
 import {DatabricksWorkspace} from "../configuration/DatabricksWorkspace";
-import {isDevExtension} from "../utils/developmentUtils";
+import {isDevExtension, isIntegrationTest} from "../utils/developmentUtils";
 import {
     DEV_APP_INSIGHTS_CONFIGURATION_KEY,
     EventProperties,
@@ -82,9 +82,7 @@ export async function toUserMetadata(
 
 export function getContextMetadata(): ExtraMetadata[Metadata.CONTEXT] {
     return {
-        environmentType: process.env["TEST_DEFAULT_CLUSTER_ID"]
-            ? "tests"
-            : "prod",
+        environmentType: isIntegrationTest() ? "tests" : "prod",
     };
 }
 

packages/databricks-vscode/src/test/e2e/wdio.conf.ts

@@ -279,6 +279,7 @@ export const config: Options.Testrunner = {
             process.env.WORKSPACE_PATH = WORKSPACE_PATH;
             process.env.TEST_REPO_PATH = repoPath;
             process.env.TEST_WORKSPACE_FOLDER_PATH = workspaceFolderPath;
+            process.env.DATABRICKS_VSCODE_INTEGRATION_TEST = "true";
         } catch (e) {
             console.error(e);
             process.exit(1);

packages/databricks-vscode/src/utils/developmentUtils.ts

@@ -3,3 +3,7 @@ export const EXTENSION_DEVELOPMENT = "EXTENSION_DEVELOPMENT";
 export function isDevExtension(): boolean {
     return process.env.EXTENSION_DEVELOPMENT === "true";
 }
+
+export function isIntegrationTest(): boolean {
+    return process.env.DATABRICKS_VSCODE_INTEGRATION_TEST === "true";
+}

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

@@ -83,6 +83,11 @@ const StorageConfigurations = {
         location: "global",
         defaultValue: "0.0.0",
     }),
+
+    "databricks.preview-tnc.accepted": withType<boolean>()({
+        location: "global",
+        defaultValue: false,
+    }),
 };
 
 type Keys = keyof typeof StorageConfigurations;