@datacamp/learn-vm-scripts 1.8.8 → 1.9.0

package/dist/index.cjs

@@ -366835,30 +366835,34 @@ var resetChrome = async ({ userDataDir }) => {
 };
 
 // src/snowflakeLogin.ts
+var USERNAME_SELECTOR = 'input[autocomplete="username"]';
+var PASSWORD_SELECTOR = 'input[autocomplete="current-password"]';
+var OTP_SELECTOR = 'input[autocomplete="one-time-code"]';
 async function loginToSnowflake({
   url,
   username,
   page,
-  password
+  password,
+  otp
 }) {
   log("Navigating to Snowflake login page...");
   await page.goto(ensureHttps(url));
-  await page.waitForSelector('input[autocomplete="username"]');
-  const usernameField = await page.$('input[autocomplete="username"]');
+  await page.waitForSelector(USERNAME_SELECTOR);
+  const usernameField = await page.$(USERNAME_SELECTOR);
   if (!usernameField) {
     throw new Error("Username field not found");
   }
   await usernameField.click({ clickCount: 3 });
   await page.keyboard.press("Backspace");
-  await page.type('input[autocomplete="username"]', username);
-  await page.waitForSelector('input[autocomplete="current-password"]');
-  const passwordField = await page.$('input[autocomplete="current-password"]');
+  await page.type(USERNAME_SELECTOR, username);
+  await page.waitForSelector(PASSWORD_SELECTOR);
+  const passwordField = await page.$(PASSWORD_SELECTOR);
   if (!passwordField) {
     throw new Error("Password field not found");
   }
   await passwordField.click({ clickCount: 3 });
   await page.keyboard.press("Backspace");
-  await page.type('input[autocomplete="current-password"]', password);
+  await page.type(PASSWORD_SELECTOR, password);
   await page.waitForSelector("button:not([disabled])");
   const buttons = await page.$$("button:not([disabled])");
   let signInButtonHandle;
@@ -366876,7 +366880,37 @@ async function loginToSnowflake({
     throw new Error("Could not find an enabled Sign in button on the page.");
   }
   await signInButtonHandle.click();
-  log("Logged in to Snowflake!");
+  if (otp != null) {
+    log("Handling Snowflake OTP challenge...");
+    await page.waitForSelector(OTP_SELECTOR, {
+      timeout: 15e3
+    });
+    const otpField = await page.$(OTP_SELECTOR);
+    if (!otpField) {
+      throw new Error("OTP field not found");
+    }
+    await otpField.click({ clickCount: 3 });
+    await page.keyboard.press("Backspace");
+    await page.type(OTP_SELECTOR, otp);
+    const enabledButtons = await page.$$("button:not([disabled])");
+    let continueButtonHandle;
+    for (const btn of enabledButtons) {
+      const textContent = await page.evaluate(
+        (el) => el.textContent?.trim(),
+        btn
+      );
+      if (textContent === "Continue") {
+        continueButtonHandle = btn;
+        break;
+      }
+    }
+    if (!continueButtonHandle) {
+      throw new Error("Could not find an enabled Continue button on the page.");
+    }
+    await continueButtonHandle.click();
+    log("Snowflake OTP challenge completed");
+  }
+  log("Logged in to Snowflake");
 }
 
 // src/fabricLogin.ts
@@ -366989,7 +367023,8 @@ var LoggedInProvidersSchema = objectType({
   cloudProvider: enumType(["aws", "azure", "snowflake", "n8n"]),
   username: stringType(),
   password: stringType(),
-  url: stringType()
+  url: stringType(),
+  otp: stringType().min(1).optional()
 });
 var Input = discriminatedUnionType("cloudProvider", [
   LoggedOutProvidersSchema,
@@ -367045,7 +367080,8 @@ var handleInput = async ({ line, page }) => {
           url: input.url,
           username: input.username,
           page,
-          password: input.password
+          password: input.password,
+          otp: input.otp
         });
         break;
       case "perplexity":

package/package.json

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