/*
 * 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.settings.display;

import static android.app.admin.DevicePolicyResources.Strings.Settings.OTHER_OPTIONS_DISABLED_BY_ADMIN;
import static android.hardware.SensorPrivacyManager.Sensors.CAMERA;
import static android.provider.Settings.System.SCREEN_OFF_TIMEOUT;

import static com.android.settings.display.UtilsKt.isAdaptiveSleepSupported;

import android.app.admin.DevicePolicyManager;
import android.app.settings.SettingsEnums;
import android.content.BroadcastReceiver;
import android.content.Context;
import android.content.Intent;
import android.content.IntentFilter;
import android.content.res.Resources;
import android.graphics.drawable.Drawable;
import android.hardware.SensorPrivacyManager;
import android.os.PowerManager;
import android.os.UserHandle;
import android.provider.Settings;
import android.util.Log;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.preference.Preference;
import androidx.preference.PreferenceScreen;

import com.android.settings.R;
import com.android.settings.core.TogglePreferenceController;
import com.android.settings.flags.Flags;
import com.android.settings.overlay.FeatureFactory;
import com.android.settings.search.BaseSearchIndexProvider;
import com.android.settings.support.actionbar.HelpResourceProvider;
import com.android.settings.widget.RadioButtonPickerFragment;
import com.android.settings.wifi.dpp.WifiDppUtils;
import com.android.settingslib.RestrictedLockUtils;
import com.android.settingslib.RestrictedLockUtilsInternal;
import com.android.settingslib.core.instrumentation.MetricsFeatureProvider;
import com.android.settingslib.search.SearchIndexable;
import com.android.settingslib.search.SearchIndexableRaw;
import com.android.settingslib.widget.CandidateInfo;
import com.android.settingslib.widget.FooterPreference;
import com.android.settingslib.widget.SelectorWithWidgetPreference;

import com.google.common.annotations.VisibleForTesting;

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

/** Fragment that is used to control screen timeout. */
@SearchIndexable(forTarget = SearchIndexable.ALL & ~SearchIndexable.ARC)
public class ScreenTimeoutSettings extends RadioButtonPickerFragment
        implements HelpResourceProvider {
    private static final String TAG = "ScreenTimeout";

    /** If there is no setting in the provider, use this. */
    public static final int FALLBACK_SCREEN_TIMEOUT_VALUE = 30000;

    private static final int DEFAULT_ORDER_OF_LOWEST_PREFERENCE = Integer.MAX_VALUE - 1;

    private CharSequence[] mInitialEntries;
    private CharSequence[] mInitialValues;
    private FooterPreference mPrivacyPreference;
    private final MetricsFeatureProvider mMetricsFeatureProvider;
    private SensorPrivacyManager mPrivacyManager;
    private final BroadcastReceiver mReceiver =
            new BroadcastReceiver() {
                @Override
                public void onReceive(Context context, Intent intent) {
                    mAdaptiveSleepBatterySaverPreferenceController.updateVisibility();
                    if (!isCatalystEnabled()) {
                        mAdaptiveSleepController.updatePreference();
                    }
                }
            };

    private DevicePolicyManager mDevicePolicyManager;
    private SensorPrivacyManager.OnSensorPrivacyChangedListener mPrivacyChangedListener;
    private boolean mIsUserAuthenticated = false;

    @VisibleForTesting Context mContext;

    @VisibleForTesting RestrictedLockUtils.EnforcedAdmin mAdmin;

    @VisibleForTesting FooterPreference mDisableOptionsPreference;

    @VisibleForTesting
    FooterPreference mPowerConsumptionPreference;

    @VisibleForTesting
    AdaptiveSleepPermissionPreferenceController mAdaptiveSleepPermissionController;

    @VisibleForTesting
    AdaptiveSleepCameraStatePreferenceController mAdaptiveSleepCameraStatePreferenceController;

    @VisibleForTesting AdaptiveSleepPreferenceController mAdaptiveSleepController;

    @VisibleForTesting
    AdaptiveSleepBatterySaverPreferenceController mAdaptiveSleepBatterySaverPreferenceController;

    @Nullable
    TogglePreferenceController mAdditionalTogglePreferenceController;

    public ScreenTimeoutSettings() {
        super();
        mMetricsFeatureProvider = FeatureFactory.getFeatureFactory().getMetricsFeatureProvider();
    }

    @Override
    public void onAttach(Context context) {
        super.onAttach(context);
        mContext = context;
        mDevicePolicyManager = mContext.getSystemService(DevicePolicyManager.class);
        mAdaptiveSleepPermissionController =
                new AdaptiveSleepPermissionPreferenceController(context);
        mAdaptiveSleepCameraStatePreferenceController =
                new AdaptiveSleepCameraStatePreferenceController(context, getLifecycle());
        mAdaptiveSleepBatterySaverPreferenceController =
                new AdaptiveSleepBatterySaverPreferenceController(context);
        mPrivacyPreference = new FooterPreference(context);
        mPrivacyPreference.setIcon(R.drawable.ic_privacy_shield_24dp);
        mPrivacyPreference.setTitle(R.string.adaptive_sleep_privacy);
        mPrivacyPreference.setSelectable(false);
        mPrivacyPreference.setLayoutResource(
                com.android.settingslib.widget.preference.footer.R.layout.preference_footer);
        if (!isCatalystEnabled()) {
            mPrivacyManager = SensorPrivacyManager.getInstance(context);
            mAdaptiveSleepController = new AdaptiveSleepPreferenceController(context);
            mPrivacyChangedListener =
                    (sensor, enabled) -> mAdaptiveSleepController.updatePreference();
        }
        mAdditionalTogglePreferenceController = FeatureFactory.getFeatureFactory()
                .getDisplayFeatureProvider().createAdditionalPreference(context);
    }

    @Override
    protected List<? extends CandidateInfo> getCandidates() {
        mInitialEntries = getResources().getStringArray(R.array.screen_timeout_entries);
        mInitialValues = getResources().getStringArray(R.array.screen_timeout_values);

        final List<CandidateInfo> candidates = new ArrayList<>();
        final long maxTimeout = getMaxScreenTimeout(getContext());
        if (mInitialValues != null) {
            for (int i = 0; i < mInitialValues.length; ++i) {
                // Truncate mInitialEntries/Values so that they do not exceed maxTimeout
                if (Long.parseLong(mInitialValues[i].toString()) <= maxTimeout) {
                    candidates.add(
                            new TimeoutCandidateInfo(
                                    mInitialEntries[i], mInitialValues[i].toString(), true));
                }
            }
        } else {
            Log.e(TAG, "Screen timeout options do not exist.");
        }
        return candidates;
    }

    @Override
    public void onStart() {
        super.onStart();
        mAdaptiveSleepPermissionController.updateVisibility();
        mAdaptiveSleepCameraStatePreferenceController.updateVisibility();
        mAdaptiveSleepBatterySaverPreferenceController.updateVisibility();
        mContext.registerReceiver(
                mReceiver, new IntentFilter(PowerManager.ACTION_POWER_SAVE_MODE_CHANGED));
        if (!isCatalystEnabled()) {
            mAdaptiveSleepController.updatePreference();
            mPrivacyManager.addSensorPrivacyListener(CAMERA, mPrivacyChangedListener);
        }
        mIsUserAuthenticated = false;
        FeatureFactory.getFeatureFactory().getDisplayFeatureProvider().updatePreference(
                mAdditionalTogglePreferenceController);
    }

    @Override
    public void onStop() {
        super.onStop();
        mContext.unregisterReceiver(mReceiver);
        if (!isCatalystEnabled()) {
            mPrivacyManager.removeSensorPrivacyListener(CAMERA, mPrivacyChangedListener);
        }
    }

    @Override
    public void updateCandidates() {
        final String defaultKey = getDefaultKey();
        final PreferenceScreen screen = getPreferenceScreen();
        // Adaptive sleep preference is added to the screen when catalyst is enabled
        Preference adaptiveSleepPreference = screen.findPreference(AdaptiveSleepPreference.KEY);
        screen.removeAll();

        final List<? extends CandidateInfo> candidateList = getCandidates();
        if (candidateList == null) {
            return;
        }

        for (CandidateInfo info : candidateList) {
            ProtectedSelectorWithWidgetPreference pref =
                    new ProtectedSelectorWithWidgetPreference(
                            getContext(), info.getKey(), this);
            bindPreference(pref, info.getKey(), info, defaultKey);
            screen.addPreference(pref);
        }

        final long selectedTimeout = getTimeoutFromKey(defaultKey);
        final long maxTimeout = getMaxScreenTimeout(getContext());
        if (!candidateList.isEmpty() && (selectedTimeout > maxTimeout)) {
            // The selected time out value is longer than the max timeout allowed by the
            // admin/configuration. The list of candidates is already truncated so that
            // no value exceeds the max timeout value.
            // Select the largest value from the candidates list by default.
            int lastIndex = candidateList.size() - 1;
            final ProtectedSelectorWithWidgetPreference preferenceWithLargestTimeout =
                    (ProtectedSelectorWithWidgetPreference)
                            screen.getPreference(lastIndex);
            preferenceWithLargestTimeout.setChecked(true);
            // Update the system screen timeout setting to match the UI
            setCurrentSystemScreenTimeout(getContext(), candidateList.get(lastIndex).getKey());
        }

        mPrivacyPreference = new FooterPreference(mContext);
        mPrivacyPreference.setIcon(R.drawable.ic_privacy_shield_24dp);
        mPrivacyPreference.setTitle(R.string.adaptive_sleep_privacy);
        mPrivacyPreference.setSelectable(false);
        mPrivacyPreference.setLayoutResource(
                com.android.settingslib.widget.preference.footer.R.layout.preference_footer);

        FeatureFactory.getFeatureFactory().getDisplayFeatureProvider()
                .addToScreen(mAdditionalTogglePreferenceController, screen);

        if (isAdaptiveSleepSupported(getContext())) {
            mAdaptiveSleepPermissionController.addToScreen(screen);
            mAdaptiveSleepCameraStatePreferenceController.addToScreen(screen);
            if (adaptiveSleepPreference != null) {
                // reset order for appending
                adaptiveSleepPreference.setOrder(Preference.DEFAULT_ORDER);
                screen.addPreference(adaptiveSleepPreference);
            } else {
                mAdaptiveSleepController.addToScreen(screen);
            }
            mAdaptiveSleepBatterySaverPreferenceController.addToScreen(screen);
            screen.addPreference(mPrivacyPreference);
        }

        if (mAdmin != null) {
            setupDisabledFooterPreference();
            screen.addPreference(mDisableOptionsPreference);
        } else {
            setupPowerConsumptionFooterPreference();
            screen.addPreference(mPowerConsumptionPreference);
        }
    }

    boolean isUserAuthenticated() {
        return mIsUserAuthenticated;
    }

    void setUserAuthenticated(boolean isUserAuthenticated) {
        mIsUserAuthenticated = isUserAuthenticated;
    }

    @VisibleForTesting
    void setupDisabledFooterPreference() {
        final String textDisabledByAdmin =
                mDevicePolicyManager
                        .getResources()
                        .getString(
                                OTHER_OPTIONS_DISABLED_BY_ADMIN,
                                () ->
                                        getResources()
                                                .getString(R.string.admin_disabled_other_options));
        final String textMoreDetails = getResources().getString(R.string.admin_more_details);

        mDisableOptionsPreference = new FooterPreference(getContext());
        mDisableOptionsPreference.setTitle(textDisabledByAdmin);
        mDisableOptionsPreference.setSelectable(false);
        mDisableOptionsPreference.setLearnMoreText(textMoreDetails);
        mDisableOptionsPreference.setLearnMoreAction(
                v -> {
                    RestrictedLockUtils.sendShowAdminSupportDetailsIntent(getContext(), mAdmin);
                });
        mDisableOptionsPreference.setIcon(R.drawable.ic_info_outline_24dp);

        // The 'disabled by admin' preference should always be at the end of the setting page.
        mPrivacyPreference.setOrder(DEFAULT_ORDER_OF_LOWEST_PREFERENCE - 1);
        mDisableOptionsPreference.setOrder(DEFAULT_ORDER_OF_LOWEST_PREFERENCE);
    }

    @VisibleForTesting
    void setupPowerConsumptionFooterPreference() {
        mPowerConsumptionPreference = new FooterPreference(getContext());
        mPowerConsumptionPreference.setTitle(R.string.power_consumption_footer_summary);
        mPowerConsumptionPreference.setSelectable(false);
        mPowerConsumptionPreference.setIcon(R.drawable.ic_info_outline_24dp);

        // The 'Longer screen timeout' preference should always be at the end of the setting page.
        mPrivacyPreference.setOrder(DEFAULT_ORDER_OF_LOWEST_PREFERENCE - 1);
        mPowerConsumptionPreference.setOrder(DEFAULT_ORDER_OF_LOWEST_PREFERENCE);
    }

    @Override
    protected String getDefaultKey() {
        return getCurrentSystemScreenTimeout(getContext());
    }

    @Override
    protected boolean setDefaultKey(String key) {
        setCurrentSystemScreenTimeout(getContext(), key);
        return true;
    }

    @Override
    public int getMetricsCategory() {
        return SettingsEnums.SCREEN_TIMEOUT;
    }

    @Override
    protected int getPreferenceScreenResId() {
        return R.xml.screen_timeout_settings;
    }

    @Override
    public @Nullable String getPreferenceScreenBindingKey(@NonNull Context context) {
        return ScreenTimeoutScreen.KEY;
    }

    @Override
    public int getHelpResource() {
        return R.string.help_url_adaptive_sleep;
    }

    // Get the maximum screen timeout as governed by admin and/or configuration.
    // Returns the lowest timeout (admin/config) or Long.MAX_VALUE.
    private Long getMaxScreenTimeout(Context context) {
        Long adminMaxTimeout = Long.MAX_VALUE;
        Long configMaxTimeout = Long.MAX_VALUE;
        if (context == null) {
            return Long.MAX_VALUE;
        }
        final DevicePolicyManager dpm = context.getSystemService(DevicePolicyManager.class);
        if (dpm == null) {
            return Long.MAX_VALUE;
        }
        if (mAdmin == null) { // Don't overwrite mocked mAdmin
            mAdmin = RestrictedLockUtilsInternal.checkIfMaximumTimeToLockIsSet(context);
        }
        if (mAdmin != null) {
            // Get the admin max screen timeout
            adminMaxTimeout = dpm.getMaximumTimeToLock(null /* admin */, UserHandle.myUserId());
        }
        try {
            // Get the configurable max screen timeout
            configMaxTimeout = Long.valueOf(
                    context.getResources().getInteger(R.integer.config_max_screen_timeout));
        } catch (Resources.NotFoundException e) {
            // Do nothing
        }
        return Math.min(adminMaxTimeout, configMaxTimeout);
    }

    private String getCurrentSystemScreenTimeout(Context context) {
        if (context == null) {
            return Long.toString(FALLBACK_SCREEN_TIMEOUT_VALUE);
        } else {
            return Long.toString(
                    Settings.System.getLong(
                            context.getContentResolver(),
                            SCREEN_OFF_TIMEOUT,
                            FALLBACK_SCREEN_TIMEOUT_VALUE));
        }
    }

    private void setCurrentSystemScreenTimeout(Context context, String key) {
        try {
            if (context != null) {
                final long value = getTimeoutFromKey(key);
                mMetricsFeatureProvider.action(
                        context, SettingsEnums.ACTION_SCREEN_TIMEOUT_CHANGED, (int) value);
                Settings.System.putLong(context.getContentResolver(), SCREEN_OFF_TIMEOUT, value);
            }
        } catch (NumberFormatException e) {
            Log.e(TAG, "could not persist screen timeout setting", e);
        }
    }

    private static long getTimeoutFromKey(String key) {
        return Long.parseLong(key);
    }

    @VisibleForTesting
    static class TimeoutCandidateInfo extends CandidateInfo {
        private final CharSequence mLabel;
        private final String mKey;

        TimeoutCandidateInfo(CharSequence label, String key, boolean enabled) {
            super(enabled);
            mLabel = label;
            mKey = key;
        }

        @Override
        public CharSequence loadLabel() {
            return mLabel;
        }

        @Override
        public Drawable loadIcon() {
            return null;
        }

        @Override
        public String getKey() {
            return mKey;
        }
    }

    @VisibleForTesting
    static class ProtectedSelectorWithWidgetPreference
            extends SelectorWithWidgetPreference {

        private final long mTimeoutMs;
        private final ScreenTimeoutSettings mScreenTimeoutSettings;

        ProtectedSelectorWithWidgetPreference(
                Context context, String key, ScreenTimeoutSettings screenTimeoutSettings) {
            super(context);
            mTimeoutMs = getTimeoutFromKey(key);
            mScreenTimeoutSettings = screenTimeoutSettings;
        }

        @Override
        public void onClick() {
            if (Flags.protectScreenTimeoutWithAuth()
                    && !mScreenTimeoutSettings.isUserAuthenticated()
                    && !isChecked()
                    && mTimeoutMs > getTimeoutFromKey(mScreenTimeoutSettings.getDefaultKey())) {
                WifiDppUtils.showLockScreen(
                        getContext(),
                        () -> {
                            mScreenTimeoutSettings.setUserAuthenticated(true);
                            super.onClick();
                        });
            } else {
                super.onClick();
            }
        }
    }

    public static final BaseSearchIndexProvider SEARCH_INDEX_DATA_PROVIDER =
            new BaseSearchIndexProvider(R.xml.screen_timeout_settings) {
                public List<SearchIndexableRaw> getRawDataToIndex(
                        Context context, boolean enabled) {
                    if (!isAdaptiveSleepSupported(context)) {
                        return null;
                    }
                    final Resources res = context.getResources();
                    final SearchIndexableRaw data = new SearchIndexableRaw(context);
                    data.title = res.getString(R.string.adaptive_sleep_title);
                    data.key = AdaptiveSleepPreferenceController.PREFERENCE_KEY;
                    data.keywords = res.getString(R.string.adaptive_sleep_title);

                    final List<SearchIndexableRaw> result = new ArrayList<>(1);
                    result.add(data);
                    return result;
                }
            };
}