@datacamp/learn-vm-scripts 0.0.11 → 1.0.0

package/dist/index.cjs

@@ -92669,6 +92669,9 @@ function ensureHttps(url) {
   }
   return url;
 }
+var unreachable = (_) => {
+  throw new Error("unreachable");
+};
 
 // src/setInput.ts
 var setInput = ({
@@ -92690,7 +92693,7 @@ var log = (message) => {
 };
 
 // src/azureLogin.ts
-var optionalPartsOfLogin = async (page) => {
+var closeOnboardingPopups = async (page) => {
   const welcomeSurveyCloseButton = 'span.fxs-welcomesurvey-close-icon[role="button"]';
   await page.waitForSelector(welcomeSurveyCloseButton);
   await page.click(welcomeSurveyCloseButton);
@@ -92702,7 +92705,8 @@ async function loginToAzure({
   url,
   username,
   page,
-  password
+  password,
+  skipOnboarding = true
 }) {
   log("Navigating to Azure login page...");
   await Promise.all([
@@ -92738,9 +92742,12 @@ async function loginToAzure({
   log("Ensuring user stays signed in ...");
   await page.waitForSelector('[type="submit"]');
   await page.click('[type="submit"]');
-  void optionalPartsOfLogin(page).catch((error) => {
-    console.error("Error in optional part of Azure login: ", error);
-  });
+  if (skipOnboarding) {
+    log("Skipping onboarding...");
+    void closeOnboardingPopups(page).catch((error) => {
+      console.error("Error in onboarding of Azure login: ", error);
+    });
+  }
   log("Logged in to Azure successfully");
 }
 
@@ -92790,8 +92797,17 @@ async function loginToAWS({
 // src/databricksLogin.ts
 async function loginToDatabricks({
   url,
-  page
+  page,
+  username,
+  password
 }) {
+  await loginToAzure({
+    url: void 0,
+    username,
+    page,
+    password,
+    skipOnboarding: false
+  });
   log("Navigating to Databricks login page...");
   await page.goto(ensureHttps(url));
   await page.waitForSelector(".aad-btn");
@@ -92837,9 +92853,51 @@ var resetChrome = async ({ userDataDir }) => {
   console.log("Chrome reset!");
 };
 
+// src/snowflakeLogin.ts
+var findAsync = (elements, predicate) => {
+  const promises = elements.map(predicate);
+  return Promise.all(promises).then(
+    (results) => elements.find((_, i) => results[i])
+  );
+};
+async function loginToSnowflake({
+  url,
+  username,
+  page,
+  password
+}) {
+  log("Navigating to Snowflake login page...");
+  await page.goto(ensureHttps(url));
+  await page.waitForSelector('input[name="username"]');
+  await setInput({
+    page,
+    querySelector: 'input[name="username"]',
+    value: username
+  });
+  await page.waitForSelector('input[name="password"]');
+  await setInput({
+    page,
+    querySelector: 'input[name="password"]',
+    value: password
+  });
+  const loginButton = 'div[data-test-id="loginButton"][aria-disabled="false"]';
+  await page.waitForSelector(loginButton);
+  await page.click(loginButton);
+  log("Logged in to Snowflake!");
+  log("Skipping optional parts of login");
+  await page.waitForSelector('[data-testid="Dialog Footer"]');
+  log("Dialog Footer found");
+  const buttons = await page.$$('div[role="button"]');
+  const notNowButton = await findAsync(buttons, async (button) => {
+    const innerText = await page.evaluate((el) => el.innerText, button);
+    return innerText === "Not now";
+  });
+  await notNowButton?.click();
+}
+
 // src/startInputWatcher.ts
 var Input = objectType({
-  cloudProvider: enumType(["aws", "azure", "databricks"]),
+  cloudProvider: enumType(["aws", "azure", "databricks", "snowflake"]),
   password: stringType(),
   url: stringType(),
   username: stringType()
@@ -92851,30 +92909,15 @@ var handleInput = async ({ line, page }) => {
     );
     switch (cloudProvider) {
       case "azure":
-        await loginToAzure({ url, username, page, password });
-        break;
+        return loginToAzure({ url, username, page, password });
       case "aws":
-        if (!url) {
-          console.error("AWS URL is required");
-          process.exit(1);
-        }
-        await loginToAWS({ url, username, page, password });
-        break;
+        return loginToAWS({ url, username, page, password });
       case "databricks":
-        if (!url) {
-          console.error("Databricks URL is required");
-          process.exit(1);
-        }
-        await loginToAzure({
-          url: void 0,
-          username,
-          page,
-          password
-        });
-        await loginToDatabricks({ url, page });
-        break;
+        return loginToDatabricks({ url, page, password, username });
+      case "snowflake":
+        return loginToSnowflake({ url, username, page, password });
       default:
-        console.error("Unsupported cloud provider");
+        return unreachable(cloudProvider);
     }
   } catch (error) {
     console.error(`Error: "${error.message}" on input ${line}`);

package/dist/local.cjs

@@ -85609,13 +85609,7 @@ async function setupPuppeteerPage({
   userDataDir: userDataDir2
 }) {
   const browser = await puppeteer_core_default.launch({
-    args: [
-      "--start-maximized"
-      // '--kiosk',
-      // '--no-prompts',
-      // '--no-dialogs',
-      // '--disable-pinch',
-    ],
+    args: ["--start-maximized", "--enable-save-password-bubble"],
     executablePath: executablePath3,
     headless: false,
     defaultViewport: null,
@@ -85629,6 +85623,11 @@ async function setupPuppeteerPage({
   return browser.newPage();
 }
 
+// src/log.ts
+var log = (message) => {
+  console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] ${message}`);
+};
+
 // src/utils.ts
 var AZURE_LOGIN_URL = "https://portal.azure.com/";
 function ensureHttps(url) {
@@ -85639,11 +85638,6 @@ function ensureHttps(url) {
   return url;
 }
 
-// src/log.ts
-var log = (message) => {
-  console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] ${message}`);
-};
-
 // src/setInput.ts
 var setInput = ({
   page,
@@ -85659,7 +85653,7 @@ var setInput = ({
 };
 
 // src/azureLogin.ts
-var optionalPartsOfLogin = async (page) => {
+var closeOnboardingPopups = async (page) => {
   const welcomeSurveyCloseButton = 'span.fxs-welcomesurvey-close-icon[role="button"]';
   await page.waitForSelector(welcomeSurveyCloseButton);
   await page.click(welcomeSurveyCloseButton);
@@ -85671,7 +85665,8 @@ async function loginToAzure({
   url,
   username,
   page,
-  password
+  password,
+  skipOnboarding = true
 }) {
   log("Navigating to Azure login page...");
   await Promise.all([
@@ -85707,10 +85702,36 @@ async function loginToAzure({
   log("Ensuring user stays signed in ...");
   await page.waitForSelector('[type="submit"]');
   await page.click('[type="submit"]');
-  void optionalPartsOfLogin(page);
+  if (skipOnboarding) {
+    log("Skipping onboarding...");
+    void closeOnboardingPopups(page).catch((error) => {
+      console.error("Error in onboarding of Azure login: ", error);
+    });
+  }
   log("Logged in to Azure successfully");
 }
 
+// src/databricksLogin.ts
+async function loginToDatabricks({
+  url,
+  page,
+  username,
+  password
+}) {
+  await loginToAzure({
+    url: void 0,
+    username,
+    page,
+    password,
+    skipOnboarding: false
+  });
+  log("Navigating to Databricks login page...");
+  await page.goto(ensureHttps(url));
+  await page.waitForSelector(".aad-btn");
+  await page.click(".aad-btn");
+  log("Logged in to Databricks successfully");
+}
+
 // src/local.ts
 var executablePath2 = `/Applications/Google Chrome.app/Contents/MacOS/Google Chrome`;
 var userDataDir = `/tmp/chrome-learn-vm-scripts-data-dir`;
@@ -85719,10 +85740,10 @@ var main = async () => {
     executablePath: executablePath2,
     userDataDir
   });
-  await loginToAzure({
-    url: "https://portal.azure.com",
-    username: "student-be1288f0-5375-4105-ab58-ed0fc2ca5d2e@datacamplearn.onmicrosoft.com",
-    password: "9y4*E6kr&q)dtfiWP@5s",
+  await loginToDatabricks({
+    url: "adb-3441331393732493.13.azuredatabricks.net",
+    username: "student-759f4c3e-6920-46f2-ab12-9cf921068e4a@datacamplearn.onmicrosoft.com",
+    password: "b_ufA6HDCWtljmRfP@5s",
     page
   });
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@datacamp/learn-vm-scripts",
-  "version": "0.0.11",
+  "version": "1.0.0",
   "repository": "git@github.com:datacamp-engineering/learn-vm-scripts.git",
   "description": "Scripts used by VMs in learn.",
   "main": "./dist/index.cjs",