/*
 * Copyright (C) 2020 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 com.android.car.pm;

import static androidx.car.app.activity.CarAppActivity.ACTION_SHOW_DIALOG;
import static androidx.car.app.activity.CarAppActivity.ACTION_START_SECOND_INSTANCE;
import static androidx.car.app.activity.CarAppActivity.SECOND_INSTANCE_TITLE;

import static com.google.common.truth.Truth.assertThat;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assume.assumeTrue;

import android.app.Activity;
import android.app.ActivityOptions;
import android.app.AlertDialog;
import android.app.UiAutomation;
import android.car.Car;
import android.car.content.pm.CarPackageManager;
import android.car.drivingstate.CarDrivingStateEvent;
import android.car.drivingstate.CarDrivingStateManager;
import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.os.Bundle;
import android.support.test.uiautomator.By;
import android.support.test.uiautomator.Configurator;
import android.support.test.uiautomator.UiDevice;
import android.support.test.uiautomator.UiObject2;
import android.support.test.uiautomator.Until;
import android.view.Display;

import androidx.car.app.activity.CarAppActivity;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import androidx.test.filters.FlakyTest;
import androidx.test.filters.MediumTest;
import androidx.test.platform.app.InstrumentationRegistry;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

@RunWith(AndroidJUnit4.class)
@MediumTest
public class CarPackageManagerServiceTest {
    private static final String ACTIVITY_BLOCKING_ACTIVITY_TEXTVIEW_ID =
            "com.android.systemui:id/blocking_text";
    private static final String ACTIVITY_BLOCKING_ACTIVITY_EXIT_BUTTON_ID =
            "com.android.systemui:id/exit_button";

    // cf_x86_auto is very slow, so uses very long timeout.
    private static final int UI_TIMEOUT_MS = 20_000;
    private static final int NOT_FOUND_UI_TIMEOUT_MS = 10_000;
    private static final long ACTIVITY_TIMEOUT_MS = 5000;
    private static final int HOME_DISPLAYED_TIMEOUT_MS = 5_000;

    private CarDrivingStateManager mCarDrivingStateManager;
    private CarPackageManager mCarPackageManager;

    private UiDevice mDevice;

    private static final CopyOnWriteArrayList<TempActivity> sTestingActivities =
            new CopyOnWriteArrayList<>();

    @Before
    public void setUp() throws Exception {
        Car car = Car.createCar(getContext());
        mCarDrivingStateManager = (CarDrivingStateManager)
                car.getCarManager(Car.CAR_DRIVING_STATE_SERVICE);
        mCarPackageManager = (CarPackageManager)
                car.getCarManager(Car.PACKAGE_SERVICE);
        assertNotNull(mCarDrivingStateManager);

        Configurator.getInstance()
                .setUiAutomationFlags(UiAutomation.FLAG_DONT_SUPPRESS_ACCESSIBILITY_SERVICES);
        mDevice = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation());
        ensureHomeIsDisplayed();
        setDrivingStateMoving();
    }

    @After
    public void tearDown() throws Exception {
        setDrivingStateParked();

        for (TempActivity testingActivity : sTestingActivities) {
            testingActivity.finishCompletely();
            sTestingActivities.remove(testingActivity);
        }
    }

    @Test
    public void testBlockingActivity_doActivity_isNotBlocked() throws Exception {
        startDoActivity(/* extra= */ null);

        assertThat(mDevice.wait(Until.findObject(By.text(
                DoActivity.class.getSimpleName())),
                UI_TIMEOUT_MS)).isNotNull();
        assertBlockingActivityNotFound();
    }

    @Test
    public void testBlockingActivity_doActivity_showingDialog_isNotBlocked() throws Exception {
        startDoActivity(DoActivity.INTENT_EXTRA_SHOW_DIALOG);

        assertThat(mDevice.wait(Until.findObject(By.text(
                DoActivity.DIALOG_TITLE)),
                UI_TIMEOUT_MS)).isNotNull();
        assertBlockingActivityNotFound();
    }

    @Test
    public void testBlockingActivity_doTemplateActivity_isNotBlocked() throws Exception {
        startActivity(toComponentName(getTestContext(), CarAppActivity.class));

        assertThat(mDevice.wait(Until.findObject(By.text(
                CarAppActivity.class.getSimpleName())),
                UI_TIMEOUT_MS)).isNotNull();
        assertBlockingActivityNotFound();
    }

    @Test
    public void testBlockingActivity_multipleDoTemplateActivity_notBlocked() throws Exception {
        startActivity(toComponentName(getTestContext(), CarAppActivity.class));
        assertThat(mDevice.wait(Until.findObject(By.text(
                CarAppActivity.class.getSimpleName())),
                UI_TIMEOUT_MS)).isNotNull();
        getContext().sendBroadcast(new Intent().setAction(ACTION_START_SECOND_INSTANCE)
                .setPackage(getTestContext().getPackageName()));
        assertThat(mDevice.wait(Until.findObject(By.text(
                SECOND_INSTANCE_TITLE)),
                UI_TIMEOUT_MS)).isNotNull();
        assertBlockingActivityNotFound();
    }

    @Test
    public void testBlockingActivity_doTemplateActivity_showingDialog_isBlocked() throws Exception {
        startActivity(toComponentName(getTestContext(), CarAppActivity.class));
        assertThat(mDevice.wait(Until.findObject(By.text(
                CarAppActivity.class.getSimpleName())),
                UI_TIMEOUT_MS)).isNotNull();
        assertBlockingActivityNotFound();
        assertThat(mCarPackageManager.isActivityDistractionOptimized(
                getTestContext().getPackageName(),
                CarAppActivity.class.getName()
        )).isTrue();

        getContext().sendBroadcast(new Intent().setAction(ACTION_SHOW_DIALOG)
                .setPackage(getTestContext().getPackageName()));

        assertThat(mDevice.wait(Until.findObject(By.res(ACTIVITY_BLOCKING_ACTIVITY_TEXTVIEW_ID)),
                UI_TIMEOUT_MS)).isNotNull();
        assertThat(mCarPackageManager.isActivityDistractionOptimized(
                getTestContext().getPackageName(),
                CarAppActivity.class.getName()
        )).isFalse();
    }

    @Test
    public void testBlockingActivity_nonDoActivity_isBlocked() throws Exception {
        testBlockingActivityShownForNdoActivity();
    }

    @Test
    public void testBlockingActivity_nonDoActivity_carIdling_isBlocked() throws Exception {
        setDrivingStateIdling();
        testBlockingActivityShownForNdoActivity();
    }

    private void testBlockingActivityShownForNdoActivity() {
        startNonDoActivity(NonDoActivity.EXTRA_DO_NOTHING);

        // The label should be 'Close app' since NonDoActivity is the root task.
        assertBlockingActivityFoundAndExit("Close app");

        // To exit ABA will close nonDoActivity.
        assertBlockingActivityNotFound();
        assertThat(mDevice.wait(Until.findObject(By.text(NonDoActivity.class.getSimpleName())),
                UI_TIMEOUT_MS)).isNull();
    }

    @Test
    public void testBlockingActivity_DoLaunchesNonDoOnCreate_isBlocked() throws Exception {
        startDoActivity(DoActivity.INTENT_EXTRA_LAUNCH_NONDO);

        // The label should be 'Back' since NonDo's root task is DO.
        assertBlockingActivityFoundAndExit("Back");

        // To exit ABA will show the root task, DoActivity.
        assertBlockingActivityNotFound();
        assertThat(mDevice.wait(Until.findObject(By.text(DoActivity.class.getSimpleName())),
                UI_TIMEOUT_MS)).isNotNull();
    }

    @FlakyTest(bugId = 338646912)
    @Test
    public void testBlockingActivity_DoLaunchesNonDo_nonDoIsKilled_noBlockingActivity()
            throws Exception {
        startDoActivity(DoActivity.INTENT_EXTRA_LAUNCH_NONDO_NEW_TASK);
        assertBlockingActivityFound();

        for (TempActivity activity : sTestingActivities) {
            if (activity instanceof NonDoActivity) {
                activity.finishCompletely();
                sTestingActivities.remove(activity);
            }
        }

        assertBlockingActivityNotFound();
        // After NonDo & ABA finishes, DoActivity will come to the top.
        assertActivityLaunched(DoActivity.class.getSimpleName());
    }

    @FlakyTest(bugId = 338646912)
    @Test
    public void testBlockingActivity_DoLaunchesNonDo_DoIsKilled_isBlocked()
            throws Exception {
        startDoActivity(DoActivity.INTENT_EXTRA_LAUNCH_NONDO_NEW_TASK);
        assertBlockingActivityFound();

        for (TempActivity activity : sTestingActivities) {
            if (activity instanceof DoActivity) {
                activity.finishCompletely();
                sTestingActivities.remove(activity);
            }
        }

        assertBlockingActivityFound();
    }

    @Test
    public void testBlockingActivity_nonDoFinishesOnCreate_noBlockingActivity()
            throws Exception {
        startNonDoActivity(NonDoActivity.EXTRA_ONCREATE_FINISH_IMMEDIATELY);

        assertBlockingActivityNotFound();
    }

    @Test
    public void testBlockingActivity_nonDoLaunchesDoOnCreate_noBlockingActivity()
            throws Exception {
        startNonDoActivity(NonDoActivity.EXTRA_ONCREATE_LAUNCH_DO_IMMEDIATELY);

        assertBlockingActivityNotFound();
    }

    @Test
    public void testBlockingActivity_nonDoFinishesOnResume_noBlockingActivity()
            throws Exception {
        startNonDoActivity(NonDoActivity.EXTRA_ONRESUME_FINISH_IMMEDIATELY);

        assertBlockingActivityNotFound();
    }

    @Test
    public void testBlockingActivity_nonDoLaunchesDoOnResume_noBlockingActivity()
            throws Exception {
        startNonDoActivity(NonDoActivity.EXTRA_ONRESUME_LAUNCH_DO_IMMEDIATELY);

        assertBlockingActivityNotFound();
    }

    @Test
    public void testIsActivityBackedBySafeActivity_notMoving_nonDoActivity_returnsTrue()
            throws Exception {
        setDrivingStateParked();

        startNonDoActivity(NonDoActivity.EXTRA_DO_NOTHING);
        assertActivityLaunched(NonDoActivity.class.getSimpleName());

        ComponentName nonDoActivity = toComponentName(getTestContext(), NonDoActivity.class);
        assertThat(mCarPackageManager.isActivityBackedBySafeActivity(nonDoActivity)).isTrue();
    }

    @Test
    public void testIsActivityBackedBySafeActivity_moving_rootNonDoActivity_returnsFalse()
            throws Exception {
        startNonDoActivity(NonDoActivity.EXTRA_DO_NOTHING);
        assertBlockingActivityFound();

        ComponentName nonDoActivity = toComponentName(getTestContext(), NonDoActivity.class);
        assertThat(mCarPackageManager.isActivityBackedBySafeActivity(nonDoActivity)).isFalse();
    }

    @Test
    public void testIsActivityBackedBySafeActivity_moving_nonDoActivityBackedByDo_returnsTrue()
            throws Exception {
        startDoActivity(DoActivity.INTENT_EXTRA_LAUNCH_NONDO);
        // DoActivity will launch NonDoActivity consecutively.
        assertBlockingActivityFound();

        ComponentName nonDoActivity = toComponentName(getTestContext(), NonDoActivity.class);
        assertThat(mCarPackageManager.isActivityBackedBySafeActivity(nonDoActivity)).isTrue();
    }

    @Test
    public void testIsActivityBackedBySafeActivity_moving_doActivity_returnsFalse()
            throws Exception {
        startDoActivity(/* extra= */ null);
        assertActivityLaunched(DoActivity.class.getSimpleName());

        ComponentName doActivity = toComponentName(getTestContext(), DoActivity.class);
        assertThat(mCarPackageManager.isActivityBackedBySafeActivity(doActivity)).isFalse();
    }

    private void assertBlockingActivityNotFound() {
        assertThat(mDevice.wait(Until.gone(By.res(ACTIVITY_BLOCKING_ACTIVITY_TEXTVIEW_ID)),
                NOT_FOUND_UI_TIMEOUT_MS)).isNotNull();
    }

    private void assertBlockingActivityFound() {
        assertThat(mDevice.wait(Until.findObject(By.res(ACTIVITY_BLOCKING_ACTIVITY_TEXTVIEW_ID)),
                UI_TIMEOUT_MS)).isNotNull();
    }

    private void assertBlockingActivityFoundAndExit(String exitLabel) {
        assertBlockingActivityFound();
        UiObject2 button = mDevice.findObject(By.res(ACTIVITY_BLOCKING_ACTIVITY_EXIT_BUTTON_ID));
        // TODO(b/200948830): Make the test not to compare the text directly.
        assertThat(button.getText()).isEqualTo(exitLabel);

        button.click();
    }

    private void assertActivityLaunched(String title) {
        assertThat(mDevice.wait(Until.findObject(By.text(title)), UI_TIMEOUT_MS)).isNotNull();
    }

    private void startActivity(ComponentName name) {
        Intent intent = new Intent();
        intent.setComponent(name);
        startActivity(intent);
    }

    private void startActivity(Intent intent) {
        intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
        ActivityOptions options = ActivityOptions.makeBasic();
        options.setLaunchDisplayId(Display.DEFAULT_DISPLAY);
        getContext().startActivity(intent, options.toBundle());
    }

    private void startNonDoActivity(int firstActionFlag) {
        ComponentName activity = toComponentName(getTestContext(), NonDoActivity.class);
        Intent intent = new Intent();
        intent.setComponent(activity);
        intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
        intent.putExtra(NonDoActivity.EXTRA_FIRST_ACTION, firstActionFlag);

        ActivityOptions options = ActivityOptions.makeBasic();
        options.setLaunchDisplayId(Display.DEFAULT_DISPLAY);

        getContext().startActivity(intent, options.toBundle());
    }

    private void startDoActivity(String extra) {
        Intent intent = new Intent()
                .setComponent(toComponentName(getTestContext(), DoActivity.class))
                .setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
        if (extra != null) {
            intent.putExtra(extra, true);
        }
        startActivity(intent);
    }

    private void ensureHomeIsDisplayed() {
        mDevice.pressHome();
        final String launcherPackage = mDevice.getLauncherPackageName();
        assertNotNull(launcherPackage);

        assumeTrue("Home is not displayed even after " + HOME_DISPLAYED_TIMEOUT_MS + "ms.",
                mDevice.wait(Until.hasObject(By.pkg(launcherPackage).depth(0)),
                        HOME_DISPLAYED_TIMEOUT_MS));
    }

    private void setDrivingStateMoving() {
        mCarDrivingStateManager.injectDrivingState(CarDrivingStateEvent.DRIVING_STATE_MOVING);
    }

    private void setDrivingStateIdling() {
        mCarDrivingStateManager.injectDrivingState(CarDrivingStateEvent.DRIVING_STATE_IDLING);
    }

    private void setDrivingStateParked() {
        mCarDrivingStateManager.injectDrivingState(CarDrivingStateEvent.DRIVING_STATE_PARKED);
    }

    private static ComponentName toComponentName(Context ctx, Class<?> cls) {
        return ComponentName.createRelative(ctx, cls.getName());
    }

    public static class NonDoActivity extends TempActivity {

        static final String EXTRA_FIRST_ACTION = "first_action";

        static final int EXTRA_DO_NOTHING = 0;
        static final int EXTRA_ONCREATE_FINISH_IMMEDIATELY = 1;
        static final int EXTRA_ONCREATE_LAUNCH_DO_IMMEDIATELY = 2;
        static final int EXTRA_ONRESUME_FINISH_IMMEDIATELY = 3;
        static final int EXTRA_ONRESUME_LAUNCH_DO_IMMEDIATELY = 4;

        @Override
        protected void onCreate(Bundle savedInstanceState) {
            super.onCreate(savedInstanceState);
            Bundle extras = getIntent().getExtras();
            if (extras != null) {
                switch (extras.getInt(EXTRA_FIRST_ACTION, EXTRA_DO_NOTHING)) {
                    case EXTRA_ONCREATE_LAUNCH_DO_IMMEDIATELY:
                        startActivity(new Intent(this, DoActivity.class));
                        finish();
                        break;
                    case EXTRA_ONCREATE_FINISH_IMMEDIATELY:
                        finish();
                        break;
                    default:
                        // do nothing
                }
            }
        }

        @Override
        protected void onResume() {
            super.onResume();
            Bundle extras = getIntent().getExtras();
            if (extras != null) {
                switch (extras.getInt(EXTRA_FIRST_ACTION, EXTRA_DO_NOTHING)) {
                    case EXTRA_ONRESUME_LAUNCH_DO_IMMEDIATELY:
                        startActivity(new Intent(this, DoActivity.class));
                        finish();
                        break;
                    case EXTRA_ONRESUME_FINISH_IMMEDIATELY:
                        finish();
                        break;
                    default:
                        // do nothing
                }
            }
        }
    }

    public static class DoActivity extends TempActivity {
        public static final String INTENT_EXTRA_SHOW_DIALOG = "SHOW_DIALOG";
        public static final String DIALOG_TITLE = "Title";

        public static final String INTENT_EXTRA_LAUNCH_NONDO = "LAUNCH_NONDO";
        public static final String INTENT_EXTRA_LAUNCH_NONDO_NEW_TASK = "LAUNCH_NONDO_NEW_TASK";

        @Override
        protected void onCreate(Bundle savedInstanceState) {
            super.onCreate(savedInstanceState);
            if (getIntent().getBooleanExtra(INTENT_EXTRA_LAUNCH_NONDO, false)) {
                startActivity(new Intent(this, NonDoActivity.class));
            }
            if (getIntent().getBooleanExtra(INTENT_EXTRA_LAUNCH_NONDO_NEW_TASK, false)) {
                startActivity(new Intent(this, NonDoActivity.class)
                        .setFlags(Intent.FLAG_ACTIVITY_NEW_TASK));
            }
            if (getIntent().getBooleanExtra(INTENT_EXTRA_SHOW_DIALOG, false)) {
                AlertDialog dialog = new AlertDialog.Builder(DoActivity.this)
                        .setTitle(DIALOG_TITLE)
                        .setMessage("Message")
                        .create();
                dialog.show();
            }
        }
    }

    /** Activity that closes itself after some timeout to clean up the screen. */
    public static class TempActivity extends Activity {
        private final CountDownLatch mDestroyed = new CountDownLatch(1);

        @Override
        protected void onCreate(Bundle savedInstanceState) {
            super.onCreate(savedInstanceState);
            setTitle(this.getClass().getSimpleName());
            sTestingActivities.add(this);
        }

        @Override
        protected void onDestroy() {
            sTestingActivities.remove(this);
            super.onDestroy();
            mDestroyed.countDown();
        }

        void finishCompletely() throws InterruptedException {
            finish();
            waitForDestroy();
        }

        boolean waitForDestroy() throws InterruptedException {
            return mDestroyed.await(ACTIVITY_TIMEOUT_MS, TimeUnit.MILLISECONDS);
        }
    }

    private Context getContext() {
        return InstrumentationRegistry.getInstrumentation().getTargetContext();
    }

    private Context getTestContext() {
        return InstrumentationRegistry.getInstrumentation().getContext();
    }
}