apk/src/com/android/healthconnect/controller/dataentries/units/UnitsFragment.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.healthconnect.controller.dataentries.units

import android.os.Bundle
import androidx.annotation.StringRes
import androidx.preference.ListPreference
import com.android.healthconnect.controller.R
import com.android.healthconnect.controller.dataentries.units.DistanceUnit.*
import com.android.healthconnect.controller.dataentries.units.EnergyUnit.*
import com.android.healthconnect.controller.dataentries.units.HeightUnit.*
import com.android.healthconnect.controller.dataentries.units.TemperatureUnit.CELSIUS
import com.android.healthconnect.controller.dataentries.units.TemperatureUnit.FAHRENHEIT
import com.android.healthconnect.controller.dataentries.units.TemperatureUnit.KELVIN
import com.android.healthconnect.controller.dataentries.units.UnitPreferences.Companion.DISTANCE_UNIT_PREF_KEY
import com.android.healthconnect.controller.dataentries.units.UnitPreferences.Companion.ENERGY_UNIT_PREF_KEY
import com.android.healthconnect.controller.dataentries.units.UnitPreferences.Companion.HEIGHT_UNIT_PREF_KEY
import com.android.healthconnect.controller.dataentries.units.UnitPreferences.Companion.TEMPERATURE_UNIT_PREF_KEY
import com.android.healthconnect.controller.dataentries.units.UnitPreferences.Companion.WEIGHT_UNIT_PREF_KEY
import com.android.healthconnect.controller.dataentries.units.UnitPreferencesStrings.getUnitLabel
import com.android.healthconnect.controller.dataentries.units.WeightUnit.KILOGRAM
import com.android.healthconnect.controller.dataentries.units.WeightUnit.POUND
import com.android.healthconnect.controller.dataentries.units.WeightUnit.STONE
import com.android.healthconnect.controller.shared.preference.HealthPreferenceFragment
import com.android.healthconnect.controller.utils.logging.ElementName
import com.android.healthconnect.controller.utils.logging.HealthConnectLogger
import com.android.healthconnect.controller.utils.logging.PageName
import com.android.healthconnect.controller.utils.logging.UnitsElement
import dagger.hilt.android.AndroidEntryPoint
import javax.inject.Inject

@AndroidEntryPoint(HealthPreferenceFragment::class)
class UnitsFragment : Hilt_UnitsFragment() {

    @Inject lateinit var logger: HealthConnectLogger
    @Inject lateinit var unitsPreferences: UnitPreferences
    init {
        this.setPageName(PageName.UNITS_PAGE)
    }

    override fun onCreatePreferences(savedInstanceState: Bundle?, rootKey: String?) {
        setPreferencesFromResource(R.xml.units_screen, rootKey)

        val height =
            createUnitPreference(
                HEIGHT_UNIT_PREF_KEY,
                UnitsElement.CHANGE_UNITS_HEIGHT_BUTTON,
                R.string.height_unit_title,
                unitsPreferences.getHeightUnit().toString()) { newUnit ->
                    val newHeightUnit = HeightUnit.valueOf(newUnit)
                    val logName =
                        when (newHeightUnit) {
                            CENTIMETERS -> UnitsElement.CENTIMETERS_BUTTON
                            FEET -> UnitsElement.FEET_AND_INCHES_BUTTON
                        }
                    logger.logInteraction(logName)
                    unitsPreferences.setHeightUnit(newHeightUnit)
                }
        val weight =
            createUnitPreference(
                WEIGHT_UNIT_PREF_KEY,
                UnitsElement.CHANGE_UNITS_WEIGHT_BUTTON,
                R.string.weight_unit_title,
                unitsPreferences.getWeightUnit().toString()) { newUnit ->
                    val newWeightUnit = WeightUnit.valueOf(newUnit)
                    val logName =
                        when (newWeightUnit) {
                            POUND -> UnitsElement.POUNDS_BUTTON
                            KILOGRAM -> UnitsElement.KILOGRAMS_BUTTON
                            STONE -> UnitsElement.STONES_BUTTON
                        }
                    logger.logInteraction(logName)
                    unitsPreferences.setWeightUnit(newWeightUnit)
                }
        val distance =
            createUnitPreference(
                DISTANCE_UNIT_PREF_KEY,
                UnitsElement.CHANGE_UNITS_DISTANCE_BUTTON,
                R.string.distance_unit_title,
                unitsPreferences.getDistanceUnit().toString()) { newUnit ->
                    val newDistanceUnit = DistanceUnit.valueOf(newUnit)
                    val logName =
                        when (newDistanceUnit) {
                            KILOMETERS -> UnitsElement.KILOMETERS_BUTTON
                            MILES -> UnitsElement.MILES_BUTTON
                        }
                    logger.logInteraction(logName)
                    unitsPreferences.setDistanceUnit(newDistanceUnit)
                }
        val energy =
            createUnitPreference(
                ENERGY_UNIT_PREF_KEY,
                UnitsElement.CHANGE_UNITS_ENERGY_BUTTON,
                R.string.energy_unit_title,
                unitsPreferences.getEnergyUnit().toString()) { newUnit ->
                    val newEnergyUnit = EnergyUnit.valueOf(newUnit)
                    val logName =
                        when (newEnergyUnit) {
                            CALORIE -> UnitsElement.CALORIES_BUTTON
                            KILOJOULE -> UnitsElement.KILOJOULES_BUTTON
                        }
                    logger.logInteraction(logName)
                    unitsPreferences.setEnergyUnit(newEnergyUnit)
                }
        val temperature =
            createUnitPreference(
                TEMPERATURE_UNIT_PREF_KEY,
                UnitsElement.CHANGE_UNITS_TEMPERATURE_BUTTON,
                R.string.temperature_unit_title,
                unitsPreferences.getTemperatureUnit().toString()) { newUnit ->
                    val newTemperatureUnit = TemperatureUnit.valueOf(newUnit)
                    val logName =
                        when (newTemperatureUnit) {
                            CELSIUS -> UnitsElement.CELSIUS_BUTTON
                            FAHRENHEIT -> UnitsElement.FAHRENHEIT_BUTTON
                            KELVIN -> UnitsElement.KELVIN_BUTTON
                        }
                    logger.logInteraction(logName)
                    unitsPreferences.setTemperatureUnit(newTemperatureUnit)
                }
        preferenceScreen.addPreference(height)
        preferenceScreen.addPreference(weight)
        preferenceScreen.addPreference(distance)
        preferenceScreen.addPreference(energy)
        preferenceScreen.addPreference(temperature)
    }

    private fun createUnitPreference(
        key: String,
        logName: ElementName,
        @StringRes title: Int,
        unitValue: String,
        onNewValue: (String) -> Unit
    ): ListPreference {
        val listPreference = ListPreference(context)
        logger.logImpression(logName)

        with(listPreference) {
            isPersistent = false
            isIconSpaceReserved = false
            value = unitValue
            setKey(key)
            setTitle(title)
            setDialogTitle(title)
            setEntries(getEntries(key))
            setEntryValues(getEntriesValues(key))
            setSummary("%s")
            setNegativeButtonText(R.string.units_cancel)
        }
        listPreference.setOnPreferenceChangeListener { _, newValue ->
            onNewValue(newValue.toString())
            true
        }

        return listPreference
    }

    private fun getEntries(key: String): Array<String> {
        val entries = getUnits(key)
        return entries.map { getString(getUnitLabel(it)) }.toTypedArray()
    }

    private fun getEntriesValues(key: String): Array<String> {
        val entries = getUnits(key)
        return entries.map { it.toString() }.toTypedArray()
    }

    private fun getUnits(key: String): Array<UnitPreference> {
        return when (key) {
            DISTANCE_UNIT_PREF_KEY -> arrayOf(KILOMETERS, MILES)
            HEIGHT_UNIT_PREF_KEY -> arrayOf(CENTIMETERS, FEET)
            WEIGHT_UNIT_PREF_KEY -> arrayOf(POUND, KILOGRAM, STONE)
            ENERGY_UNIT_PREF_KEY -> arrayOf(CALORIE, KILOJOULE)
            TEMPERATURE_UNIT_PREF_KEY -> arrayOf(CELSIUS, FAHRENHEIT, KELVIN)
            else -> emptyArray()
        }
    }
}