wallpaper/testing/FakeWallpaperClient.kt

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

import android.app.WallpaperColors
import android.graphics.Bitmap
import android.graphics.Point
import android.graphics.Rect
import android.os.Handler
import com.android.wallpaper.asset.Asset
import com.android.wallpaper.model.Screen
import com.android.wallpaper.model.WallpaperModelsPair
import com.android.wallpaper.module.logging.UserEventLogger.SetWallpaperEntryPoint
import com.android.wallpaper.picker.customization.data.content.WallpaperClient
import com.android.wallpaper.picker.customization.shared.model.WallpaperDestination
import com.android.wallpaper.picker.customization.shared.model.WallpaperModel
import com.android.wallpaper.picker.data.WallpaperModel.LiveWallpaperModel
import com.android.wallpaper.picker.data.WallpaperModel.StaticWallpaperModel
import com.android.wallpaper.picker.preview.shared.model.FullPreviewCropModel
import javax.inject.Inject
import javax.inject.Singleton
import kotlin.math.min
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.MutableStateFlow
import kotlinx.coroutines.flow.map

@Singleton
class FakeWallpaperClient @Inject constructor() : WallpaperClient {
    val wallpapersSet =
        mutableMapOf(
            WallpaperDestination.HOME to null as com.android.wallpaper.picker.data.WallpaperModel?,
            WallpaperDestination.LOCK to null as com.android.wallpaper.picker.data.WallpaperModel?,
        )
    private var wallpaperColors: WallpaperColors? = null

    private val _recentWallpapers =
        MutableStateFlow(
            buildMap {
                WallpaperDestination.values()
                    .filter { it != WallpaperDestination.BOTH }
                    .forEach { screen -> put(screen, INITIAL_RECENT_WALLPAPERS) }
            }
        )
    private var isPaused = false
    private var deferred = mutableListOf<(suspend () -> Unit)>()

    fun setRecentWallpapers(
        recentWallpapersByDestination: Map<WallpaperDestination, List<WallpaperModel>>
    ) {
        _recentWallpapers.value = recentWallpapersByDestination
    }

    fun pause() {
        isPaused = true
    }

    suspend fun unpause() {
        isPaused = false
        deferred.forEach { it.invoke() }
        deferred.clear()
    }

    override fun recentWallpapers(
        destination: WallpaperDestination,
        limit: Int,
    ): Flow<List<WallpaperModel>> {
        return _recentWallpapers.map { wallpapersByScreen ->
            val wallpapers =
                wallpapersByScreen[destination] ?: error("No wallpapers for screen $destination")
            if (wallpapers.size > limit) {
                wallpapers.subList(0, min(limit, wallpapers.size))
            } else {
                wallpapers
            }
        }
    }

    fun getCurrentWallpaper(destination: WallpaperDestination): WallpaperModel {
        return _recentWallpapers.value[destination]?.get(0)
            ?: error("No wallpapers for screen $destination")
    }

    override suspend fun setStaticWallpaper(
        setWallpaperEntryPoint: Int,
        destination: WallpaperDestination,
        wallpaperModel: StaticWallpaperModel,
        bitmap: Bitmap,
        wallpaperSize: Point,
        asset: Asset,
        fullPreviewCropModels: Map<Point, FullPreviewCropModel>?,
    ) {
        addToWallpapersSet(wallpaperModel, destination)
    }

    override suspend fun setLiveWallpaper(
        setWallpaperEntryPoint: Int,
        destination: WallpaperDestination,
        wallpaperModel: LiveWallpaperModel,
    ) {
        addToWallpapersSet(wallpaperModel, destination)
    }

    private fun addToWallpapersSet(
        wallpaperModel: com.android.wallpaper.picker.data.WallpaperModel,
        destination: WallpaperDestination,
    ) {
        wallpapersSet[destination] = wallpaperModel
    }

    override suspend fun setRecentWallpaper(
        @SetWallpaperEntryPoint setWallpaperEntryPoint: Int,
        destination: WallpaperDestination,
        wallpaperId: String,
        onDone: () -> Unit,
    ) {
        if (isPaused) {
            deferred.add {
                setRecentWallpaper(setWallpaperEntryPoint, destination, wallpaperId, onDone)
            }
        } else {
            _recentWallpapers.value =
                _recentWallpapers.value.toMutableMap().apply {
                    this[destination] =
                        _recentWallpapers.value[destination]?.sortedBy {
                            it.wallpaperId != wallpaperId
                        } ?: error("No wallpapers for screen $destination")
                }
            onDone.invoke()
        }
    }

    override suspend fun loadThumbnail(
        wallpaperId: String,
        destination: WallpaperDestination,
    ): Bitmap? {
        return Bitmap.createBitmap(1, 1, Bitmap.Config.ARGB_8888)
    }

    override fun areRecentsAvailable(): Boolean {
        return true
    }

    override fun getCurrentCropHints(displaySizes: List<Point>, which: Int): Map<Point, Rect>? {
        return emptyMap()
    }

    fun setWallpaperColors(wallpaperColors: WallpaperColors) {
        this.wallpaperColors = wallpaperColors
    }

    override suspend fun getWallpaperColors(
        bitmap: Bitmap,
        cropHints: Map<Point, Rect>?,
    ): WallpaperColors? {
        return wallpaperColors
    }

    override fun getWallpaperColors(screen: Screen): WallpaperColors? {
        return wallpaperColors
    }

    fun setCurrentWallpaperModels(
        homeWallpaper: com.android.wallpaper.picker.data.WallpaperModel,
        lockWallpaper: com.android.wallpaper.picker.data.WallpaperModel?,
    ) {
        wallpapersSet[WallpaperDestination.HOME] = homeWallpaper
        wallpapersSet[WallpaperDestination.LOCK] = lockWallpaper
    }

    // Getting current home wallpaper should always return non-null value
    override suspend fun getCurrentWallpaperModels(forceRefresh: Boolean): WallpaperModelsPair {
        return WallpaperModelsPair(
            wallpapersSet[WallpaperDestination.HOME]
                ?: (WallpaperModelUtils.getStaticWallpaperModel(
                        wallpaperId = "defaultWallpaperId",
                        collectionId = "defaultCollection",
                    )
                    .also { wallpapersSet[WallpaperDestination.HOME] = it }),
            wallpapersSet[WallpaperDestination.LOCK],
        )
    }

    override fun addOnColorsChangedListener(
        listener: (WallpaperColors?, Int) -> Unit,
        handler: Handler,
    ) {
        TODO("Not yet implemented")
    }

    override fun removeOnColorsChangedListener(listener: (WallpaperColors?, Int) -> Unit) {
        TODO("Not yet implemented")
    }

    companion object {
        val INITIAL_RECENT_WALLPAPERS =
            listOf(
                WallpaperModel(wallpaperId = "zero", placeholderColor = 0, title = "title1"),
                WallpaperModel(wallpaperId = "one", placeholderColor = 1, title = "title2"),
                WallpaperModel(wallpaperId = "two", placeholderColor = 2, title = "title3"),
            )
    }
}