packages/SettingsLib/SpaPrivileged/src/com/android/settingslib/spaprivileged/model/app/AppRepository.kt

/*
 * 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.settingslib.spaprivileged.model.app

import android.content.Context
import android.content.pm.ApplicationInfo
import android.graphics.drawable.Drawable
import android.os.UserManager
import androidx.compose.runtime.Composable
import androidx.compose.runtime.State
import androidx.compose.runtime.produceState
import androidx.compose.ui.platform.LocalContext
import androidx.compose.ui.res.stringResource
import com.android.settingslib.Utils
import com.android.settingslib.spa.framework.compose.rememberContext
import com.android.settingslib.spaprivileged.R
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext

@Composable
fun rememberAppRepository(): AppRepository = rememberContext(::AppRepositoryImpl)

interface AppRepository {
    fun loadLabel(app: ApplicationInfo): String

    @Composable
    fun produceLabel(app: ApplicationInfo, isClonedAppPage: Boolean = false): State<String> {
        val context = LocalContext.current
        return produceState(initialValue = stringResource(R.string.summary_placeholder), app) {
            withContext(Dispatchers.IO) {
                if (isClonedAppPage || isCloneApp(context, app)) {
                    value = context.getString(R.string.cloned_app_info_label, loadLabel(app))
                } else {
                    value = loadLabel(app)
                }
            }
        }
    }

    private fun isCloneApp(context: Context, app: ApplicationInfo): Boolean {
        val userManager = context.getSystemService(UserManager::class.java)!!
        val userInfo = userManager.getUserInfo(app.userId)
        return userInfo != null && userInfo.isCloneProfile
    }

    @Composable
    fun produceIcon(app: ApplicationInfo): State<Drawable?>
}

internal class AppRepositoryImpl(private val context: Context) : AppRepository {
    private val packageManager = context.packageManager

    override fun loadLabel(app: ApplicationInfo): String = app.loadLabel(packageManager).toString()

    @Composable
    override fun produceIcon(app: ApplicationInfo) =
        produceState<Drawable?>(initialValue = null, app) {
            withContext(Dispatchers.Default) {
                value = Utils.getBadgedIcon(context, app)
            }
        }
}