CompoundButtonTranslator.kt

/*
 * Copyright 2021 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 androidx.glance.appwidget.translators

import android.content.Context
import android.content.res.ColorStateList
import android.widget.RemoteViews
import androidx.compose.ui.graphics.Color
import androidx.compose.ui.graphics.toArgb
import androidx.core.widget.RemoteViewsCompat.setImageViewColorFilter
import androidx.glance.appwidget.unit.CheckableColorProvider
import androidx.glance.appwidget.unit.CheckedStateSet
import androidx.glance.appwidget.unit.CheckedUncheckedColorProvider
import androidx.glance.appwidget.unit.ResourceCheckableColorProvider
import androidx.glance.appwidget.unit.isNightMode
import androidx.glance.appwidget.unit.resolveCheckedColor

private fun CheckedUncheckedColorProvider.toColorStateList(
    context: Context,
    isNightMode: Boolean
): ColorStateList {
    return createCheckedColorStateList(
        checked = resolve(context, isNightMode, isChecked = true),
        unchecked = resolve(context, isNightMode, isChecked = false)
    )
}

internal fun CheckedUncheckedColorProvider.toDayNightColorStateList(
    context: Context
): DayNightColorStateList {
    return DayNightColorStateList(
        day = toColorStateList(context, isNightMode = false),
        night = toColorStateList(context, isNightMode = true)
    )
}

/**
 * Creates a [ColorStateList] switching between [checked] and [unchecked] depending on the checked
 * state.
 */
private fun createCheckedColorStateList(checked: Color, unchecked: Color): ColorStateList {
    return ColorStateList(
        arrayOf(CheckedStateSet, intArrayOf()),
        intArrayOf(checked.toArgb(), unchecked.toArgb())
    )
}

internal fun CheckableColorProvider.resolve(context: Context, isChecked: Boolean) = when (this) {
    is CheckedUncheckedColorProvider -> resolve(context, context.isNightMode, isChecked)
    is ResourceCheckableColorProvider -> {
        // If the user provided res id does not resolve, then resolve using the fallback
        // (which cannot fail).
        resolveCheckedColor(context, resId, isChecked)
            ?: resolveCheckedColor(context, fallback, isChecked)!!
    }
}

/**
 * Pair class holding two [ColorStateList]s corresponding to day and night alternatives for
 * [RemoteViews] APIs that accept two [ColorStateList]s for day/night.
 */
internal data class DayNightColorStateList(val day: ColorStateList, val night: ColorStateList)

internal fun RemoteViews.setImageViewColorFilter(viewId: Int, color: Color) {
    setImageViewColorFilter(viewId, color.toArgb())
}