/*
 * Copyright (C) 2022 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.fuelgauge;

import android.content.BroadcastReceiver;
import android.content.ContentResolver;
import android.content.Context;
import android.content.Intent;
import android.provider.Settings;
import android.util.Log;

import androidx.annotation.VisibleForTesting;

import com.android.settings.fuelgauge.batterysaver.BatterySaverScheduleRadioButtonsController;
import com.android.settings.fuelgauge.datasaver.DynamicDenylistManager;
import com.android.settingslib.fuelgauge.BatterySaverUtils;

import java.util.List;

/** Execute battery settings migration tasks in the device booting stage. */
public final class BatterySettingsMigrateChecker extends BroadcastReceiver {
    private static final String TAG = "BatterySettingsMigrateChecker";

    @VisibleForTesting static BatteryOptimizeUtils sBatteryOptimizeUtils = null;

    @Override
    public void onReceive(Context context, Intent intent) {
        Log.d(TAG, "onReceive: " + intent + " owner: " + BatteryBackupHelper.isOwner());
        if (intent != null
                && Intent.ACTION_BOOT_COMPLETED.equals(intent.getAction())
                && BatteryBackupHelper.isOwner()) {
            verifyConfiguration(context);
        }
    }

    static void verifyConfiguration(Context context) {
        context = context.getApplicationContext();
        verifySaverConfiguration(context);
        verifyBatteryOptimizeModes(context);
        DynamicDenylistManager.getInstance(context).onBootComplete();
    }

    /** Avoid users set important apps into the unexpected battery optimize modes */
    static void verifyBatteryOptimizeModes(Context context) {
        Log.d(TAG, "invoke verifyOptimizationModes()");
        verifyBatteryOptimizeModeApps(
                context,
                BatteryOptimizeUtils.MODE_OPTIMIZED,
                BatteryOptimizeUtils.getForceBatteryOptimizeModeList(context));
        verifyBatteryOptimizeModeApps(
                context,
                BatteryOptimizeUtils.MODE_UNRESTRICTED,
                BatteryOptimizeUtils.getForceBatteryUnrestrictModeList(context));
    }

    @VisibleForTesting
    static void verifyBatteryOptimizeModeApps(
            Context context,
            @BatteryOptimizeUtils.OptimizationMode int optimizationMode,
            List<String> allowList) {
        allowList.forEach(
                packageName -> {
                    final BatteryOptimizeUtils batteryOptimizeUtils =
                            BatteryBackupHelper.newBatteryOptimizeUtils(
                                    context,
                                    packageName,
                                    /* testOptimizeUtils */ sBatteryOptimizeUtils);
                    if (batteryOptimizeUtils == null) {
                        return;
                    }
                    if (batteryOptimizeUtils.getAppOptimizationMode() != optimizationMode) {
                        Log.w(
                                TAG,
                                "Reset " + packageName + " battery mode into " + optimizationMode);
                        batteryOptimizeUtils.setAppUsageState(
                                optimizationMode,
                                BatteryOptimizeHistoricalLogEntry.Action.FORCE_RESET);
                    }
                });
    }

    static void verifySaverConfiguration(Context context) {
        Log.d(TAG, "invoke verifySaverConfiguration()");
        final ContentResolver resolver = context.getContentResolver();
        final int threshold =
                Settings.Global.getInt(resolver, Settings.Global.LOW_POWER_MODE_TRIGGER_LEVEL, 0);
        // Force refine the invalid scheduled battery level.
        if (threshold < BatterySaverScheduleRadioButtonsController.TRIGGER_LEVEL_MIN
                && threshold > 0) {
            Settings.Global.putInt(
                    resolver,
                    Settings.Global.LOW_POWER_MODE_TRIGGER_LEVEL,
                    BatterySaverScheduleRadioButtonsController.TRIGGER_LEVEL_MIN);
            Log.w(TAG, "Reset invalid scheduled battery level from: " + threshold);
        }
        // Force removing the 'schedule by routine' state.
        BatterySaverUtils.revertScheduleToNoneIfNeeded(context);
    }
}