chrome/android/features/autofill_assistant/public/java/src/org/chromium/chrome/browser/autofill_assistant/AutofillAssistantTabHelper.java

// Copyright 2021 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.chrome.browser.autofill_assistant;

import org.chromium.chrome.browser.tab.Tab;

/**
 * Instantiates a tab helper for autofill-assistant.
 */
public class AutofillAssistantTabHelper {
    private static final Class<Starter> USER_DATA_KEY = Starter.class;

    /**
     * Creates an autofill-assistant starter for the given tab. The starter will attach itself to
     * the tab as observer and connect to its native counterpart in order to fulfill startup
     * requests from either side.
     */
    public static void createForTab(Tab tab) {
        Starter starter = new Starter(tab, AssistantDependencyUtilsChrome::isGsa,
                AssistantDependencyUtilsChrome::isMakeSearchesAndBrowsingBetterSettingEnabled,
                new AssistantModuleInstallUiProviderChrome(tab));
        tab.addObserver(starter);
        tab.getUserDataHost().setUserData(USER_DATA_KEY, starter);
    }

    public static Starter get(Tab tab) {
        return tab.getUserDataHost().getUserData(USER_DATA_KEY);
    }
}