/*
 * Copyright (C) 2023 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.healthconnect.controller.safetycenter

import android.content.BroadcastReceiver
import android.content.ComponentName
import android.content.Context
import android.content.Intent
import android.content.Intent.ACTION_BOOT_COMPLETED
import android.content.pm.PackageManager
import android.os.UserManager
import android.safetycenter.SafetyCenterManager
import android.safetycenter.SafetyCenterManager.ACTION_REFRESH_SAFETY_SOURCES
import android.safetycenter.SafetyCenterManager.EXTRA_REFRESH_SAFETY_SOURCE_IDS
import android.safetycenter.SafetyEvent
import android.safetycenter.SafetyEvent.SAFETY_EVENT_TYPE_DEVICE_REBOOTED
import android.safetycenter.SafetyEvent.SAFETY_EVENT_TYPE_REFRESH_REQUESTED
import com.android.healthconnect.controller.safetycenter.HealthConnectSafetySource.Companion.HEALTH_CONNECT_SOURCE_ID
import com.android.healthconnect.controller.utils.DeviceInfoUtils
import dagger.hilt.android.AndroidEntryPoint
import javax.inject.Inject

@AndroidEntryPoint(BroadcastReceiver::class)
class SafetySourceBroadcastReceiver : Hilt_SafetySourceBroadcastReceiver() {

    @Inject lateinit var safetySource: HealthConnectSafetySource
    @Inject lateinit var safetyCenterManagerWrapper: SafetyCenterManagerWrapper
    @Inject lateinit var deviceInfoUtils: DeviceInfoUtils

    override fun onReceive(context: Context, intent: Intent) {
        super.onReceive(context, intent)
        tryEnableLegacySettingsEntryPoint(context)
        if (!safetyCenterManagerWrapper.isEnabled(context)) {
            return
        }
        // (b/320250695) HC doesn't support user profiles
        if ((context.getSystemService(Context.USER_SERVICE) as UserManager).isProfile) {
            return
        }
        when (intent.action) {
            ACTION_BOOT_COMPLETED -> refreshAllSafetySources(context)
            ACTION_REFRESH_SAFETY_SOURCES -> {
                val sourceIdsExtra = intent.getStringArrayExtra(EXTRA_REFRESH_SAFETY_SOURCE_IDS)
                val refreshBroadcastId =
                    intent.getStringExtra(
                        SafetyCenterManager.EXTRA_REFRESH_SAFETY_SOURCES_BROADCAST_ID)
                if (sourceIdsExtra != null &&
                    sourceIdsExtra.isNotEmpty() &&
                    refreshBroadcastId != null) {
                    val safetyEvent =
                        SafetyEvent.Builder(SAFETY_EVENT_TYPE_REFRESH_REQUESTED)
                            .setRefreshBroadcastId(refreshBroadcastId)
                            .build()
                    refreshSafetySources(context, sourceIdsExtra.toList(), safetyEvent)
                }
                return
            }
            else -> return
        }
    }

    private fun tryEnableLegacySettingsEntryPoint(context: Context) {
        val legacySettingsEntryPointComponent =
            ComponentName(context.packageName, LEGACY_SETTINGS_ACTIVITY_ALIAS)

        val componentState =
            if (shouldEnableLegacySettingsEntryPoint(context)) {
                PackageManager.COMPONENT_ENABLED_STATE_ENABLED
            } else {
                PackageManager.COMPONENT_ENABLED_STATE_DISABLED
            }

        context.packageManager.setComponentEnabledSetting(
            legacySettingsEntryPointComponent, componentState, 0)
    }

    private fun shouldEnableLegacySettingsEntryPoint(context: Context): Boolean {
        return !safetyCenterManagerWrapper.isEnabled(context) &&
            deviceInfoUtils.isHealthConnectAvailable(context)
    }

    private fun refreshSafetySources(
        context: Context,
        sourceIds: List<String>,
        safetyEvent: SafetyEvent
    ) {
        if (HEALTH_CONNECT_SOURCE_ID in sourceIds) {
            safetySource.setSafetySourceData(context, safetyEvent)
        }
    }

    private fun refreshAllSafetySources(context: Context) {
        safetySource.setSafetySourceData(
            context, SafetyEvent.Builder(SAFETY_EVENT_TYPE_DEVICE_REBOOTED).build())
    }

    companion object {
        private const val LEGACY_SETTINGS_ACTIVITY_ALIAS =
            "com.android.healthconnect.controller.LegacySettingsEntryPoint"
    }
}