/*
 * 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.wallpaper.config

import android.app.Flags.updateRecentsFromSystem
import android.app.WallpaperManager
import android.content.Context
import com.android.systemui.shared.Flags.clockReactiveVariants
import com.android.systemui.shared.Flags.extendedWallpaperEffects
import com.android.systemui.shared.Flags.extendibleThemeManager
import com.android.systemui.shared.Flags.lockscreenCustomClocks
import com.android.systemui.shared.Flags.newCustomizationPickerUi
import com.android.systemui.shared.customization.data.content.CustomizationProviderClient
import com.android.systemui.shared.customization.data.content.CustomizationProviderClientImpl
import com.android.systemui.shared.customization.data.content.CustomizationProviderContract as Contract
import com.android.wallpaper.Flags.adaptiveWallpaperFlag
import com.android.wallpaper.Flags.colorPickerUpdateFlag
import com.android.wallpaper.Flags.composeRefactorFlag
import com.android.wallpaper.Flags.creativeWallpaperFieldCollectionWallpaper
import com.android.wallpaper.Flags.desktopUiFlag
import com.android.wallpaper.Flags.enablePackThemeEntry
import com.android.wallpaper.Flags.fullscreenPreviewFlag
import com.android.wallpaper.Flags.newCreativeWallpaperCategory
import com.android.wallpaper.Flags.refactorIndividualPickerFlag
import com.android.wallpaper.Flags.refactorWallpaperPreviewScreenFlag
import com.android.wallpaper.Flags.wallpaperRestorerFlag
import com.android.wallpaper.R
import com.android.wallpaper.module.InjectorProvider
import com.android.wm.shell.shared.desktopmode.DesktopState
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.runBlocking

abstract class BaseFlags {
    private var customizationProviderClient: CustomizationProviderClient? = null
    private var cachedFlags: List<CustomizationProviderClient.Flag>? = null

    // local flag to gate the entry points for magic portrait
    open fun isMagicPortraitEntryPointsEnabled() = true

    open fun isStagingBackdropContentEnabled() = false

    open fun isPackThemeEnabled() = enablePackThemeEntry()

    open fun isWallpaperEffectModelDownloadEnabled() = true

    open fun isInterruptModelDownloadEnabled() = false

    // local flag to enable the refactored version of IPF2
    open fun isWallpapersFragmentEnabled() = refactorIndividualPickerFlag()

    open fun isWallpaperRestorerEnabled() = wallpaperRestorerFlag()

    open fun isNewCreativeWallpaperCategoryEnabled() = newCreativeWallpaperCategory()

    open fun isCreativeWallpaperCollectionFieldEnabled() =
        creativeWallpaperFieldCollectionWallpaper()

    open fun isExtendedWallpaperEnabled() = extendedWallpaperEffects()

    open fun isExtendibleThemeManager() = extendibleThemeManager()

    open fun isNewPickerUi() = newCustomizationPickerUi()

    open fun isClockReactiveVariantsEnabled() = clockReactiveVariants()

    open fun isComposeRefactorEnabled() = composeRefactorFlag()

    open fun isColorPickerUpdateEnabled() = colorPickerUpdateFlag()

    // Local flag to gate Compose UI under the colorPickerUpdateFlag
    open fun isColorPickerComposeEnabled() = false

    open fun isAdaptiveWallpaperEnabled() = adaptiveWallpaperFlag()

    open fun isRefactorWallpaperPreviewScreenEnabled() = refactorWallpaperPreviewScreenFlag()

    // This is just a local flag in order to ensure right behaviour in case
    // something goes wrong with PhotoPicker integration.
    open fun isPhotoPickerEnabled() = true

    open fun isKeyguardQuickAffordanceEnabled(context: Context): Boolean {
        return getCachedFlags(context)
            .firstOrNull { flag ->
                flag.name ==
                    Contract.FlagsTable.FLAG_NAME_CUSTOM_LOCK_SCREEN_QUICK_AFFORDANCES_ENABLED
            }
            ?.value == true
    }

    open fun isCustomClocksEnabled(context: Context): Boolean {
        return lockscreenCustomClocks() ||
            getCachedFlags(context)
                .firstOrNull { flag ->
                    flag.name == Contract.FlagsTable.FLAG_NAME_CUSTOM_CLOCKS_ENABLED
                }
                ?.value == true
    }

    /**
     * This flag is to for refactoring the process of setting a wallpaper from the Wallpaper Picker,
     * such as changes in WallpaperSetter, WallpaperPersister and WallpaperPreferences.
     */
    fun isRefactorSettingWallpaper(): Boolean {
        return false
    }

    open fun isPageTransitionsFeatureEnabled(context: Context): Boolean {
        return getCachedFlags(context)
            .firstOrNull { flag -> flag.name == Contract.FlagsTable.FLAG_NAME_PAGE_TRANSITIONS }
            ?.value == true
    }

    open fun isGridApplyButtonEnabled(context: Context): Boolean {
        return getCachedFlags(context)
            .firstOrNull { flag -> flag.name == Contract.FlagsTable.FLAG_NAME_GRID_APPLY_BUTTON }
            ?.value == true
    }

    open fun isPreviewLoadingAnimationEnabled(context: Context): Boolean {
        return getCachedFlags(context)
            .firstOrNull { flag ->
                flag.name == Contract.FlagsTable.FLAG_NAME_WALLPAPER_PICKER_PREVIEW_ANIMATION
            }
            ?.value == true
    }

    private fun getCustomizationProviderClient(context: Context): CustomizationProviderClient {
        return customizationProviderClient
            ?: CustomizationProviderClientImpl(context.applicationContext, Dispatchers.IO).also {
                customizationProviderClient = it
            }
    }

    open fun getCachedFlags(context: Context): List<CustomizationProviderClient.Flag> {
        return cachedFlags
            ?: runBlocking { getCustomizationProviderClient(context).queryFlags() }
                .also { cachedFlags = it }
    }

    open fun isFullscreenPreviewEnabled(context: Context): Boolean {
        return fullscreenPreviewFlag() &&
            isNewPickerUi() &&
            DesktopState.fromContext(context).canEnterDesktopMode
    }

    open fun isRecentWallpapersFromSystemEnabled(context: Context): Boolean {
        val wallpaperManager = context.getSystemService(WallpaperManager::class.java)
        try {
            wallpaperManager.javaClass.getMethod(
                "getWallpaperInstance",
                Int::class.javaPrimitiveType,
            )
            return updateRecentsFromSystem()
        } catch (e: NoSuchMethodException) {
            return false
        }
    }

    open fun shouldShowDesktopUi(context: Context): Boolean {
        // TODO: b/416024080 use a better solution than a config boolean to show desktop UI.
        return desktopUiFlag() &&
            isNewPickerUi() &&
            context.resources.getBoolean(R.bool.isDesktopUi)
    }

    companion object {
        @JvmStatic
        fun get(): BaseFlags {
            return InjectorProvider.getInjector().getFlags()
        }
    }
}