/*
 * Copyright (C) 2024 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.permissioncontroller.wear.permission.components.theme

import android.content.Context
import android.os.Build
import androidx.annotation.ColorRes
import androidx.annotation.RequiresApi
import androidx.compose.ui.graphics.Color
import androidx.wear.compose.material.Colors
import androidx.wear.compose.material3.ColorScheme
import com.android.permissioncontroller.wear.permission.components.R

/**
 * Creates a dynamic color maps that can be overlaid. In wear we only support dark theme for the
 * time being. If the device supports dynamic color generation these resources are updated with the
 * generated colors
 */
internal object WearComposeMaterial3ColorScheme {

    fun legacyColorScheme(): ColorScheme {
        return Colors().run {
            ColorScheme(
                background = background,
                onBackground = onBackground,
                onPrimaryContainer = primary,
                primaryDim = primaryVariant,
                onPrimary = onPrimary,
                tertiary = secondary, // Best-Fit Mapping: Refer WearMaterialBridgedTheme.
                tertiaryDim = secondaryVariant,
                onTertiary = onSecondary,
                surfaceContainer = surface,
                onSurface = onSurface,
                onSurfaceVariant = onSurfaceVariant,
            )
        }
    }

    @RequiresApi(Build.VERSION_CODES.S)
    fun tonalColorScheme(context: Context): ColorScheme {
        val tonalPalette = dynamicTonalPalette(context)
        return ColorScheme(
            background = tonalPalette.neutral0,
            onBackground = tonalPalette.neutral100,
            onPrimary = tonalPalette.primary10,
            onPrimaryContainer = tonalPalette.primary90,
            onSecondary = tonalPalette.secondary10,
            onSecondaryContainer = tonalPalette.secondary90,
            onSurface = tonalPalette.neutral95,
            onSurfaceVariant = tonalPalette.neutralVariant80,
            onTertiary = tonalPalette.tertiary10,
            onTertiaryContainer = tonalPalette.tertiary90,
            outline = tonalPalette.neutralVariant60,
            outlineVariant = tonalPalette.neutralVariant40,
            primary = tonalPalette.primary90,
            primaryContainer = tonalPalette.primary30,
            primaryDim = tonalPalette.primary80,
            secondary = tonalPalette.secondary90,
            secondaryContainer = tonalPalette.secondary30,
            secondaryDim = tonalPalette.secondary80,
            surfaceContainer = tonalPalette.neutral20,
            surfaceContainerHigh = tonalPalette.neutral30,
            tertiary = tonalPalette.tertiary90,
            tertiaryContainer = tonalPalette.tertiary30,
            tertiaryDim = tonalPalette.tertiary80,
        )
    }

    private fun Color.updatedColor(context: Context, @ColorRes colorRes: Int): Color {
        return ResourceHelper.getColor(context, colorRes) ?: this
    }

    @RequiresApi(36)
    fun dynamicColorScheme(context: Context): ColorScheme {
        val defaultColorScheme = ColorScheme()
        return ColorScheme(
            primary =
                defaultColorScheme.primary.updatedColor(
                    context,
                    R.color.wear_compose_material3_primary_fixed,
                ),
            primaryDim =
                defaultColorScheme.primaryDim.updatedColor(
                    context,
                    R.color.wear_compose_material3_primary_fixed_dim,
                ),
            primaryContainer =
                defaultColorScheme.primaryContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_primary_container_dark,
                ),
            onPrimary =
                defaultColorScheme.onPrimary.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_primary_fixed,
                ),
            onPrimaryContainer =
                defaultColorScheme.onPrimaryContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_primary_container_dark,
                ),
            secondary =
                defaultColorScheme.secondary.updatedColor(
                    context,
                    R.color.wear_compose_material3_secondary_fixed,
                ),
            secondaryDim =
                defaultColorScheme.secondaryDim.updatedColor(
                    context,
                    R.color.wear_compose_material3_secondary_fixed_dim,
                ),
            secondaryContainer =
                defaultColorScheme.secondaryContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_secondary_container_dark,
                ),
            onSecondary =
                defaultColorScheme.onSecondary.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_secondary_fixed,
                ),
            onSecondaryContainer =
                defaultColorScheme.onSecondaryContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_secondary_container_dark,
                ),
            tertiary =
                defaultColorScheme.tertiary.updatedColor(
                    context,
                    R.color.wear_compose_material3_tertiary_fixed,
                ),
            tertiaryDim =
                defaultColorScheme.tertiaryDim.updatedColor(
                    context,
                    R.color.wear_compose_material3_tertiary_fixed_dim,
                ),
            tertiaryContainer =
                defaultColorScheme.tertiaryContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_tertiary_container_dark,
                ),
            onTertiary =
                defaultColorScheme.onTertiary.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_tertiary_fixed,
                ),
            onTertiaryContainer =
                defaultColorScheme.onTertiaryContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_tertiary_container_dark,
                ),
            surfaceContainerLow =
                defaultColorScheme.surfaceContainerLow.updatedColor(
                    context,
                    R.color.wear_compose_material3_surface_container_low_dark,
                ),
            surfaceContainer =
                defaultColorScheme.surfaceContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_surface_container_dark,
                ),
            surfaceContainerHigh =
                defaultColorScheme.surfaceContainerHigh.updatedColor(
                    context,
                    R.color.wear_compose_material3_surface_container_high_dark,
                ),
            onSurface =
                defaultColorScheme.onSurface.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_surface_dark,
                ),
            onSurfaceVariant =
                defaultColorScheme.onSurfaceVariant.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_surface_variant_dark,
                ),
            outline =
                defaultColorScheme.outline.updatedColor(
                    context,
                    R.color.wear_compose_material3_outline_dark,
                ),
            outlineVariant =
                defaultColorScheme.outlineVariant.updatedColor(
                    context,
                    R.color.wear_compose_material3_outline_variant_dark,
                ),
            background =
                defaultColorScheme.background.updatedColor(
                    context,
                    R.color.wear_compose_material3_background_dark,
                ),
            onBackground =
                defaultColorScheme.onBackground.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_background_dark,
                ),
            error =
                defaultColorScheme.error.updatedColor(context, R.color.wear_compose_material3_error_dark),
            onError =
                defaultColorScheme.onError.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_error_dark,
                ),
            errorContainer =
                defaultColorScheme.errorContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_error_container_dark,
                ),
            onErrorContainer =
                defaultColorScheme.onErrorContainer.updatedColor(
                    context,
                    R.color.wear_compose_material3_on_error_container_dark,
                ),
        )
    }
}