/*
 * Copyright (C) 2025 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.systemui.screencapture.common.ui.viewmodel

import android.annotation.SuppressLint
import android.content.Context
import android.graphics.drawable.Drawable
import androidx.annotation.DrawableRes
import com.android.systemui.dagger.qualifiers.UiBackground
import javax.inject.Inject
import kotlin.coroutines.CoroutineContext
import kotlinx.coroutines.withContext

/**
 * Interface to mark your ViewModels as ones capable of loading drawables. This enables it to load
 * them on [UiBackground] executor by using [DrawableLoaderViewModelImpl].
 *
 * Example:
 * ```
 * class FooViewModel(vm: DrawableLoaderViewModelImpl): DrawableLoaderViewModel by vm
 * ```
 *
 * And then in compose:
 * ```
 * @Composable
 * fun Foo(viewModel: FooViewModel) {
 *   loadIcon(viewModel, R.drawable.icon)?.let { Icon(it) }
 * }
 * ```
 */
interface DrawableLoaderViewModel {

    suspend fun loadDrawable(context: Context, @DrawableRes resId: Int): Drawable
}

class DrawableLoaderViewModelImpl
@Inject
constructor(@UiBackground private val uiBackgroundCoroutineContext: CoroutineContext) :
    DrawableLoaderViewModel {

    @SuppressLint("UseCompatLoadingForDrawables")
    override suspend fun loadDrawable(context: Context, @DrawableRes resId: Int): Drawable =
        withContext(uiBackgroundCoroutineContext) { context.getDrawable(resId)!! }
}