/*
 * Copyright (C) 2019 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.server.wm;

import static android.provider.Settings.Global.DEVELOPMENT_ENABLE_FREEFORM_WINDOWS_SUPPORT;
import static android.provider.Settings.Global.DEVELOPMENT_ENABLE_NON_RESIZABLE_MULTI_WINDOW;
import static android.provider.Settings.Global.DEVELOPMENT_FORCE_DESKTOP_MODE_ON_EXTERNAL_DISPLAYS;
import static android.provider.Settings.Global.DEVELOPMENT_FORCE_RESIZABLE_ACTIVITIES;
import static android.provider.Settings.Global.DEVELOPMENT_WM_DISPLAY_SETTINGS_PATH;

import static androidx.test.platform.app.InstrumentationRegistry.getInstrumentation;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.clearInvocations;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

import android.content.ContentResolver;
import android.net.Uri;
import android.platform.test.annotations.Presubmit;
import android.provider.Settings;

import androidx.test.filters.SmallTest;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;

import java.util.List;

/**
 * Test for {@link WindowManagerService.SettingsObserver}.
 *
 * Build/Install/Run:
 *  atest WmTests:WindowManagerSettingsTests
 */
@Presubmit
@SmallTest
@RunWith(WindowTestRunner.class)
public class WindowManagerSettingsTests extends WindowTestsBase {

    @Test
    public void testForceDesktopModeOnExternalDisplays() {
        try (BoolSettingsSession forceDesktopModeSession = new
                BoolSettingsSession(DEVELOPMENT_FORCE_DESKTOP_MODE_ON_EXTERNAL_DISPLAYS)) {
            final boolean forceDesktopMode = !forceDesktopModeSession.getSetting();
            final Uri forceDesktopModeUri = forceDesktopModeSession.setSetting(forceDesktopMode);
            mWm.mSettingsObserver.onChange(false, forceDesktopModeUri);

            assertEquals(mWm.mForceDesktopModeOnExternalDisplays, forceDesktopMode);
        }
    }

    @Test
    public void testFreeformWindow() {
        try (BoolSettingsSession freeformWindowSession = new
                BoolSettingsSession(DEVELOPMENT_ENABLE_FREEFORM_WINDOWS_SUPPORT)) {
            final boolean curFreeformWindow = freeformWindowSession.getSetting();
            final boolean newFreeformWindow = !curFreeformWindow;
            final Uri freeformWindowUri = freeformWindowSession.setSetting(newFreeformWindow);
            mWm.mAtmService.mSupportsFreeformWindowManagement = curFreeformWindow;
            mWm.mSettingsObserver.onChange(false, freeformWindowUri);

            assertEquals(mWm.mAtmService.mSupportsFreeformWindowManagement, newFreeformWindow);
        }
    }

    @Test
    public void testFreeformWindow_valueChanged_updatesDisplaySettings() {
        try (BoolSettingsSession freeformWindowSession = new
                BoolSettingsSession(DEVELOPMENT_ENABLE_FREEFORM_WINDOWS_SUPPORT)) {
            final boolean curFreeformWindow = freeformWindowSession.getSetting();
            final boolean newFreeformWindow = !curFreeformWindow;
            final Uri freeformWindowUri = freeformWindowSession.setSetting(newFreeformWindow);
            mWm.mAtmService.mSupportsFreeformWindowManagement = curFreeformWindow;
            clearInvocations(mWm.mRoot);
            mWm.mSettingsObserver.onChange(false, freeformWindowUri);

            // Changed value should update display settings.
            verify(mWm.mRoot).onSettingsRetrieved();

            clearInvocations(mWm.mRoot);
            mWm.mSettingsObserver.onChange(false, freeformWindowUri);

            // Unchanged value should not update display settings.
            verify(mWm.mRoot, never()).onSettingsRetrieved();
        }
    }

    @Test
    public void testForceResizableMode() {
        try (BoolSettingsSession forceResizableSession = new
                BoolSettingsSession(DEVELOPMENT_FORCE_RESIZABLE_ACTIVITIES)) {
            final boolean forceResizableMode = !forceResizableSession.getSetting();
            final Uri forceResizableUri = forceResizableSession.setSetting(forceResizableMode);

            mWm.mSettingsObserver.onChange(false, forceResizableUri);

            assertEquals(mWm.mAtmService.mForceResizableActivities, forceResizableMode);
        }
    }

    @Test
    public void testSupportsNonResizableMultiWindow() {
        try (BoolSettingsSession supportsNonResizableMultiWindowSession = new
                BoolSettingsSession(DEVELOPMENT_ENABLE_NON_RESIZABLE_MULTI_WINDOW)) {
            final boolean supportsNonResizableMultiWindow =
                    !supportsNonResizableMultiWindowSession.getSetting();
            final Uri supportsNonResizableMultiWindowUri = supportsNonResizableMultiWindowSession
                    .setSetting(supportsNonResizableMultiWindow);
            mWm.mSettingsObserver.onChange(false, supportsNonResizableMultiWindowUri);

            assertEquals(mWm.mAtmService.mDevEnableNonResizableMultiWindow,
                    supportsNonResizableMultiWindow);
        }
    }

    @Test
    public void testChangeBaseDisplaySettingsPath() {
        try (StringSettingsSession baseDisplaySettingsPathSession = new
                StringSettingsSession(DEVELOPMENT_WM_DISPLAY_SETTINGS_PATH)) {
            final String path = baseDisplaySettingsPathSession.getSetting() + "-test";
            final Uri baseDisplaySettingsPathUri = baseDisplaySettingsPathSession.setSetting(path);

            clearInvocations(mWm.mRoot);
            clearInvocations(mWm.mDisplayWindowSettings);
            clearInvocations(mWm.mDisplayWindowSettingsProvider);

            mWm.mSettingsObserver.onChange(false /* selfChange */, baseDisplaySettingsPathUri);

            ArgumentCaptor<String> pathCaptor = ArgumentCaptor.forClass(String.class);
            verify(mWm.mDisplayWindowSettingsProvider).setBaseSettingsFilePath(
                    pathCaptor.capture());
            assertEquals(path, pathCaptor.getValue());

            ArgumentCaptor<DisplayContent> captor =
                    ArgumentCaptor.forClass(DisplayContent.class);
            verify(mWm.mDisplayWindowSettings, times(mWm.mRoot.mChildren.size()))
                    .applySettingsToDisplayLocked(captor.capture());
            List<DisplayContent> configuredDisplays = captor.getAllValues();
            for (DisplayContent dc : mWm.mRoot.mChildren) {
                assertTrue(configuredDisplays.contains(dc));
            }

            verify(mWm.mRoot, atLeastOnce()).performSurfacePlacement();
        }
    }

    private class BoolSettingsSession implements AutoCloseable {

        private static final int SETTING_VALUE_OFF = 0;
        private static final int SETTING_VALUE_ON = 1;
        private final String mSettingName;
        private final int mInitialValue;

        BoolSettingsSession(String name) {
            mSettingName = name;
            mInitialValue = getSetting() ? SETTING_VALUE_ON : SETTING_VALUE_OFF;
        }

        boolean getSetting() {
            return Settings.Global.getInt(getContentResolver(), mSettingName, 0) != 0;
        }

        Uri setSetting(boolean enable) {
            Settings.Global.putInt(getContentResolver(), mSettingName,
                    enable ? SETTING_VALUE_ON : SETTING_VALUE_OFF);
            return Settings.Global.getUriFor(mSettingName);
        }

        private ContentResolver getContentResolver() {
            return getInstrumentation().getTargetContext().getContentResolver();
        }

        @Override
        public void close() {
            setSetting(mInitialValue == SETTING_VALUE_ON);
        }
    }

    private class StringSettingsSession implements AutoCloseable {
        private final String mSettingName;
        private final String mInitialValue;

        StringSettingsSession(String name) {
            mSettingName = name;
            mInitialValue = getSetting();
        }

        String getSetting() {
            return Settings.Global.getString(getContentResolver(), mSettingName);
        }

        Uri setSetting(String value) {
            Settings.Global.putString(getContentResolver(), mSettingName, value);
            return Settings.Global.getUriFor(mSettingName);
        }

        private ContentResolver getContentResolver() {
            return getInstrumentation().getTargetContext().getContentResolver();
        }

        @Override
        public void close() {
            setSetting(mInitialValue);
        }
    }
}