/*
 * Copyright (C) 2021 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package android.platform.helpers;

import android.app.Instrumentation;

import androidx.test.uiautomator.BySelector;
import androidx.test.uiautomator.UiObject2;

import java.util.ArrayList;
import java.util.List;

public class HomeHelperImpl extends AbstractStandardAppHelper implements IAutoHomeHelper {

    private static final int WAIT_MS = 20000;

    public HomeHelperImpl(Instrumentation instr) {
        super(instr);
    }

    /** {@inheritDoc} */
    @Override
    public String getPackage() {
        return getPackageFromConfig(AutomotiveConfigConstants.HOME_PACKAGE);
    }

    /** {@inheritDoc} */
    @Override
    public String getLauncherName() {
        throw new UnsupportedOperationException("Operation not supported.");
    }

    /** {@inheritDoc} */
    @Override
    public void dismissInitialDialogs() {
        // Nothing to dismiss
    }

    /** {@inheritDoc} */
    @Override
    public boolean hasBluetoothButton() {
        BySelector bluetoothWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.STATUS_BAR_BLUETOOTH_BUTTON);
        return getSpectatioUiUtil().hasUiElement(bluetoothWidgetSelector);
    }

    /** {@inheritDoc} */
    @Override
    public void openBluetoothButton() {
        BySelector bluetoothButtonSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.STATUS_BAR_BLUETOOTH_BUTTON);
        UiObject2 bluetoothButton = getSpectatioUiUtil().findUiObject(bluetoothButtonSelector);
        getSpectatioUiUtil()
                .validateUiObject(
                        bluetoothButton, AutomotiveConfigConstants.STATUS_BAR_BLUETOOTH_BUTTON);
        getSpectatioUiUtil().clickAndWait(bluetoothButton);
    }

    /** {@inheritDoc} */
    @Override
    public boolean hasNetworkButton() {
        BySelector networkWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.NETWORK_PALETTE);
        return getSpectatioUiUtil().hasUiElement(networkWidgetSelector);
    }

    /** {@inheritDoc} */
    @Override
    public void openNetworkPalette() {
        BySelector networkButtonSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.NETWORK_PALETTE);
        UiObject2 networkButton = getSpectatioUiUtil().findUiObject(networkButtonSelector);
        getSpectatioUiUtil()
                .validateUiObject(networkButton, AutomotiveConfigConstants.NETWORK_PALETTE);
        getSpectatioUiUtil().clickAndWait(networkButton);
    }

    /** {@inheritDoc} */
    @Override
    public boolean hasDisplayBrightness() {
        BySelector displayBrightnessWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.STATUS_BAR_BRIGHTNESS_BUTTON);
        return getSpectatioUiUtil().hasUiElement(displayBrightnessWidgetSelector);
    }

    /** {@inheritDoc} */
    @Override
    public boolean hasSoundSettings() {
        BySelector soundSettingsWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.SOUND_FACET_BUTTON);
        return getSpectatioUiUtil().hasUiElement(soundSettingsWidgetSelector);
    }

    /** {@inheritDoc} */
    @Override
    public void openSoundSettingsPalette() {
        BySelector soundSettingsSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.SOUND_FACET_BUTTON);
        UiObject2 soundSettings = getSpectatioUiUtil().findUiObject(soundSettingsSelector);
        getSpectatioUiUtil()
                .validateUiObject(soundSettings, AutomotiveConfigConstants.SOUND_FACET_BUTTON);
        getSpectatioUiUtil().clickAndWait(soundSettings);
    }

    public boolean hasAssistantWidget() {
        BySelector assistantWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_TOP_CARD);
        return (getSpectatioUiUtil().hasUiElement(assistantWidgetSelector));
    }

    public boolean hasMediaWidget() {
        BySelector mediaWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_BOTTOM_CARD);
        return (getSpectatioUiUtil().hasUiElement(mediaWidgetSelector));
    }

    /** {@inheritDoc} */
    @Override
    public void openBrightnessPalette() {
        BySelector brightnesButtonSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.STATUS_BAR_BRIGHTNESS_BUTTON);
        UiObject2 brightnessButton = getSpectatioUiUtil().findUiObject(brightnesButtonSelector);
        getSpectatioUiUtil()
                .validateUiObject(
                        brightnessButton, AutomotiveConfigConstants.STATUS_BAR_BRIGHTNESS_BUTTON);
        getSpectatioUiUtil().clickAndWait(brightnessButton);
    }

    /** {@inheritDoc} */
    @Override
    public boolean hasDisplayBrightessPalette() {
        BySelector displaybrightnessPaletteSelector =
                getUiElementFromConfig(
                        AutomotiveConfigConstants.STATUS_BAR_DISPLAY_BRIGHTNESS_PALETTE);
        return (getSpectatioUiUtil().hasUiElement(displaybrightnessPaletteSelector));
    }

    /** {@inheritDoc} */
    @Override
    public boolean hasAdaptiveBrightness() {
        BySelector adaptiveBrightnessSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.STATUS_BAR_ADAPTIVE_BRIGHTNESS);
        return (getSpectatioUiUtil().hasUiElement(adaptiveBrightnessSelector));
    }

    @Override
    public void openMediaWidget() {
        getSpectatioUiUtil().pressHome();
        getSpectatioUiUtil().waitForIdle();
        BySelector mediaWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_BOTTOM_CARD);
        UiObject2 mediaWidget = getSpectatioUiUtil().findUiObject(mediaWidgetSelector);
        getSpectatioUiUtil()
                .validateUiObject(mediaWidget, AutomotiveConfigConstants.HOME_BOTTOM_CARD);
        getSpectatioUiUtil().clickAndWait(mediaWidget);
    }

    /** {@inheritDoc} */
    @Override
    public String getUserProfileName() {
        BySelector profileGuestIconWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_PROFILE_GUEST_ICON);
        UiObject2 guestIconButtonLink =
                getSpectatioUiUtil().findUiObject(profileGuestIconWidgetSelector);
        getSpectatioUiUtil()
                .validateUiObject(
                        guestIconButtonLink, AutomotiveConfigConstants.HOME_PROFILE_GUEST_ICON);
        String profileText = guestIconButtonLink.getText();
        return profileText;
    }

    /** {@inheritDoc} */
    @Override
    public void openStatusBarProfiles() {
        BySelector profileWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_PROFILE_ICON_BUTTON);
        UiObject2 profileButtonLink = getSpectatioUiUtil().findUiObject(profileWidgetSelector);
        getSpectatioUiUtil()
                .validateUiObject(
                        profileButtonLink, AutomotiveConfigConstants.HOME_PROFILE_ICON_BUTTON);
        getSpectatioUiUtil().clickAndWait(profileButtonLink);
    }

    @Override
    public List<String> getProfileNamesFromQuickControls() {
        List<String> profileNamesText = new ArrayList<>();
        BySelector profileNamesSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_PROFILE_NAMES);
        List<UiObject2> profileNames = getSpectatioUiUtil().findUiObjects(profileNamesSelector);
        for (UiObject2 profileName : profileNames) {
            String name = profileName.getText();
            profileNamesText.add(name);
        }
        return profileNamesText;
    }

    private String getDriverIconText() {
        BySelector profileDriverIconWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_DRIVER_BUTTON);
        getSpectatioUiUtil().waitForUiObject(profileDriverIconWidgetSelector, WAIT_MS);
        UiObject2 driverIconButtonLink =
                getSpectatioUiUtil().findUiObject(profileDriverIconWidgetSelector);
        getSpectatioUiUtil()
                .validateUiObject(
                        driverIconButtonLink, AutomotiveConfigConstants.HOME_DRIVER_BUTTON);
        String driverProfileIndex = driverIconButtonLink.getText();
        return driverProfileIndex;
    }

    private String getGuestIconText() {
        BySelector profileGuestIconWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_GUEST_BUTTON);
        getSpectatioUiUtil().waitForUiObject(profileGuestIconWidgetSelector, WAIT_MS);
        UiObject2 guestIconButtonLink =
                getSpectatioUiUtil().findUiObject(profileGuestIconWidgetSelector);
        getSpectatioUiUtil()
                .validateUiObject(guestIconButtonLink, AutomotiveConfigConstants.HOME_GUEST_BUTTON);
        String guestProfileIndex = guestIconButtonLink.getText();
        return guestProfileIndex;
    }

    private String getSecondaryUserIconText() {
        BySelector secondaryUserWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_SECONDARY_USER_BUTTON);
        getSpectatioUiUtil().waitForUiObject(secondaryUserWidgetSelector, WAIT_MS);
        UiObject2 secondaryUserButtonLink =
                getSpectatioUiUtil().findUiObject(secondaryUserWidgetSelector);
        getSpectatioUiUtil()
                .validateUiObject(
                        secondaryUserButtonLink,
                        AutomotiveConfigConstants.HOME_SECONDARY_USER_BUTTON);
        String secondaryUserProfileIndex = secondaryUserButtonLink.getText();
        return secondaryUserProfileIndex;
    }

    /** {@inheritDoc} */
    @Override
    public List<String> getUserProfileNames() {
        List<String> profiles = new ArrayList<String>();
        profiles.add(getDriverIconText());
        profiles.add(getSecondaryUserIconText());
        profiles.add(getGuestIconText());
        return profiles;
    }

    /** {@inheritDoc} */
    @Override
    public void open() {
        getSpectatioUiUtil().pressHome();
        getSpectatioUiUtil().waitForIdle();
    }

    @Override
    public boolean hasTemperatureWidget() {
        getSpectatioUiUtil().pressHome();
        getSpectatioUiUtil().waitForIdle();
        BySelector temperatureSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_TEMPERATURE_BUTTON);
        List<UiObject2> temperature = getSpectatioUiUtil().findUiObjects(temperatureSelector);
        getSpectatioUiUtil()
                .validateUiObject(
                        temperature.get(0), AutomotiveConfigConstants.HOME_TEMPERATURE_BUTTON);
        getSpectatioUiUtil()
                .validateUiObject(
                        temperature.get(1), AutomotiveConfigConstants.HOME_TEMPERATURE_BUTTON);
        List<String> temperatureText = new ArrayList<>();
        for (UiObject2 uiObject : temperature) {
            String tempText = getSpectatioUiUtil().getTextForUiElement(uiObject);
            if (tempText == null) return false;
        }
        return true;
    }

    /** {@inheritDoc} */
    @Override
    public void openSystemUi() {
        getSpectatioUiUtil()
                .executeShellCommand(
                        getCommandFromConfig(AutomotiveConfigConstants.OPEN_SYSTEM_UI));
        getSpectatioUiUtil().wait15Seconds();
    }

    /** {@inheritDoc} */
    @Override
    public void openCarUi() {
        getSpectatioUiUtil()
                .executeShellCommand(getCommandFromConfig(AutomotiveConfigConstants.OPEN_CAR_UI));
        getSpectatioUiUtil().wait15Seconds();
    }
    /** {@inheritDoc} */
    @Override
    public boolean hasMapsWidget() {
        BySelector mapsWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_MAPS_WIDGET);
        getSpectatioUiUtil().waitForUiObject(mapsWidgetSelector, WAIT_MS);
        return (getSpectatioUiUtil().hasUiElement(mapsWidgetSelector));
    }

    /** {@inheritDoc} */
    @Override
    public void clickAssistantWidget() {
        BySelector assistantWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_TOP_CARD);
        UiObject2 assistantWidget = getSpectatioUiUtil().findUiObject(assistantWidgetSelector);
        getSpectatioUiUtil()
                .validateUiObject(assistantWidget, AutomotiveConfigConstants.HOME_TOP_CARD);
        getSpectatioUiUtil().clickAndWaitUntilNewWindowAppears(assistantWidget);
    }

    @Override
    public void openHVAC() {
        BySelector homeTemperatureButttonSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_TEMPERATURE_BUTTON);
        UiObject2 homeTemperatureButton =
                getSpectatioUiUtil().findUiObject(homeTemperatureButttonSelector);
        getSpectatioUiUtil().clickAndWait(homeTemperatureButton);
    }

    @Override
    public boolean isHVACOpen() {
        BySelector hvacPanelSelector = getUiElementFromConfig(AutomotiveConfigConstants.HVAC_PANEL);
        return getSpectatioUiUtil().hasUiElement(hvacPanelSelector);
    }
}