checkpoint automation changes

Author: dbajpeyi

injected/src/features/autofill-password-import.js

@@ -52,6 +52,11 @@ export default class AutofillPasswordImport extends ContentFeature {
 
     #domLoaded;
 
+    #currentLocation;
+
+    #isBookmarkModalVisible = false;
+    #isBookmarkProcessed = false;
+
     /** @type {WeakSet<Element>} */
     #tappedElements = new WeakSet();
 
@@ -422,8 +427,12 @@ export default class AutofillPasswordImport extends ContentFeature {
         }
     }
 
-    handleBookmarkImportPath(pathname) {
+    async handleBookmarkImportPath(pathname) {
         console.log('DEEP DEBUG autofill-password-import: handleBookmarkImportPath', pathname);
+        if (pathname === '/' && !this.#isBookmarkModalVisible) {
+            await this.clickDisselectAllButton();
+            await this.selectBookmark();
+        }
     }
 
     /**
@@ -508,7 +517,87 @@ export default class AutofillPasswordImport extends ContentFeature {
         this.#settingsButtonSettings = this.getFeatureSetting('settingsButton');
     }
 
+    /** Bookmark import code */
+    get disselectAllButtonSelector() {
+        return 'c-wiz[data-node-index="4;0"] button';
+    }
+
+    get bookmarkSelectButtonSelector() {
+        return 'fieldset.rcetic input';
+    }
+
+    get chromeSectionSelector() {
+        return 'c-wiz [data-id="chrome"]';
+    }
+
+    async findDisselectAllButton() {
+        return await withExponentialBackoff(() => document.querySelectorAll(this.disselectAllButtonSelector)[1]);
+    }
+    async selectBookmark() {
+        if (this.#isBookmarkProcessed) {
+            return;
+        }
+        const chromeDataButtonSelector = `${this.chromeSectionSelector} button`;
+        const chromeDataButton = /** @type HTMLButtonElement */ (
+            await withExponentialBackoff(() => document.querySelectorAll(chromeDataButtonSelector)[1], 5)
+        );
+        if (chromeDataButton != null) {
+            this.#isBookmarkModalVisible = true;
+            chromeDataButton.focus();
+            chromeDataButton.click();
+        }
+        await this.domLoaded;
+        const disselectAllButton = /** @type HTMLButtonElement */ (
+            await withExponentialBackoff(() => document.querySelectorAll('fieldset.rcetic button')[1])
+        );
+        if (disselectAllButton != null) {
+            disselectAllButton.click();
+        }
+
+        
+        const bookmarkSelectButton = /** @type HTMLInputElement */ (
+            await withExponentialBackoff(() => document.querySelectorAll(this.bookmarkSelectButtonSelector)[1])
+        );
+
+        await withExponentialBackoff(() => !bookmarkSelectButton?.checked);
+
+        if (bookmarkSelectButton != null) {
+            bookmarkSelectButton.click();
+        }
+
+        const okButton = /** @type HTMLButtonElement */ (document.querySelectorAll('div[role="button"]')[7]);
+
+        await withExponentialBackoff(() => okButton.ariaDisabled !== 'true');
+        
+        if (okButton != null) {
+            okButton.click();
+            this.#isBookmarkModalVisible = false;
+            this.#isBookmarkProcessed = true;
+        }
+    }
+
+    async clickDisselectAllButton() {
+        const element = /** @type HTMLButtonElement */ (await this.findDisselectAllButton());
+        console.log('Deep element', element);
+        if (element != null) {
+            element.click();
+        }
+
+        const chromeSectionElement = /** @type HTMLInputElement */ (
+            await withExponentialBackoff(() => document.querySelectorAll(this.chromeSectionSelector)[0].querySelector('input'))
+        );
+        console.log('DEEP chromeSectionElement', chromeSectionElement);
+
+        // First wait for the element to become unchecked (due to slow disselection)
+        await withExponentialBackoff(() => !chromeSectionElement?.checked);
+
+        chromeSectionElement.scrollIntoView({ behavior: 'smooth', block: 'center', inline: 'center' });
+
+        chromeSectionElement?.click();
+    }
+
     urlChanged() {
+        console.log('DEEP DEBUG autofill-password-import: urlChanged', window.location);
         this.handleLocation(window.location);
     }
 

injected/src/utils.js

@@ -803,22 +803,22 @@ export function legacySendMessage(messageType, options) {
 }
 
 /**
- * Takes a function that returns an element and tries to find it with exponential backoff.
+ * Takes a function that returns an element and tries to execute it until it returns a valid result or the max attempts are reached.
  * @param {number} delay
  * @param {number} [maxAttempts=4] - The maximum number of attempts to find the element.
  * @param {number} [delay=500] - The initial delay to be used to create the exponential backoff.
- * @returns {Promise<Element|HTMLElement|null>}
+ * @returns {Promise<any>}
  */
 export function withExponentialBackoff(fn, maxAttempts = 4, delay = 500) {
     return new Promise((resolve, reject) => {
         let attempts = 0;
         const tryFn = () => {
             attempts += 1;
-            const error = new Error('Element not found');
+            const error = new Error('Result is invalid');
             try {
-                const element = fn();
-                if (element) {
-                    resolve(element);
+                const result = fn();
+                if (result) {
+                    resolve(result);
                 } else if (attempts < maxAttempts) {
                     setTimeout(tryFn, delay * Math.pow(2, attempts));
                 } else {