/*
 * 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 {

    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 boolean hasNetworkButton() {
        BySelector networkWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.NETWORK_PALETTE);
        return getSpectatioUiUtil().hasUiElement(networkWidgetSelector);
    }

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

    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);
    }

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

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

    private String getSecondaryUserIconText() {
        getSpectatioUiUtil().wait5Seconds();
        BySelector secondaryUserWidgetSelector =
                getUiElementFromConfig(AutomotiveConfigConstants.HOME_SECONDARY_USER_BUTTON);
        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);
        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().clickAndWait(assistantWidget);
        getSpectatioUiUtil().wait5Seconds();
    }
}