improve yes/no dialog design

uhabits-android/src/main/java/org/isoron/uhabits/activities/common/dialogs/CheckmarkDialog.kt

@@ -1,36 +1,63 @@
 package org.isoron.uhabits.activities.common.dialogs
 
 import android.content.Context
+import android.graphics.Color
+import android.graphics.Typeface
 import android.view.LayoutInflater
+import android.view.View
 import android.view.WindowManager.LayoutParams.SOFT_INPUT_STATE_ALWAYS_VISIBLE
-import android.widget.EditText
+import android.widget.Button
 import androidx.appcompat.app.AlertDialog
 import org.isoron.uhabits.R
+import org.isoron.uhabits.core.models.Entry.Companion.NO
+import org.isoron.uhabits.core.models.Entry.Companion.SKIP
+import org.isoron.uhabits.core.models.Entry.Companion.UNKNOWN
+import org.isoron.uhabits.core.models.Entry.Companion.YES_AUTO
+import org.isoron.uhabits.core.models.Entry.Companion.YES_MANUAL
+import org.isoron.uhabits.core.models.PaletteColor
+import org.isoron.uhabits.core.preferences.Preferences
 import org.isoron.uhabits.core.ui.screens.habits.list.ListHabitsBehavior
+import org.isoron.uhabits.databinding.CheckmarkDialogBinding
 import org.isoron.uhabits.inject.ActivityContext
+import org.isoron.uhabits.utils.InterfaceUtils
+import org.isoron.uhabits.utils.StyledResources
 import javax.inject.Inject
 
 class CheckmarkDialog
 @Inject constructor(
-    @ActivityContext private val context: Context
-) {
+    @ActivityContext private val context: Context,
+    private val preferences: Preferences,
+) : View.OnClickListener {
+
+    private lateinit var binding: CheckmarkDialogBinding
+    private lateinit var fontAwesome: Typeface
+    private val allButtons = mutableListOf<Button>()
+    private var selectedButton: Button? = null
 
     fun create(
+        value: Int,
         notes: String,
+        dateString: String,
+        paletteColor: PaletteColor,
         callback: ListHabitsBehavior.CheckMarkDialogCallback
     ): AlertDialog {
-        val inflater = LayoutInflater.from(context)
-        val view = inflater.inflate(R.layout.checkmark_dialog, null)
+        binding = CheckmarkDialogBinding.inflate(LayoutInflater.from(context))
+        fontAwesome = InterfaceUtils.getFontAwesome(context)!!
 
-        val etNotes = view.findViewById<EditText>(R.id.etNotes)
+        binding.etNotes.setText(notes)
+        setUpButtons(value, Color.parseColor(paletteColor.toCsvColor()))
 
-        etNotes.setText(notes)
         val dialog = AlertDialog.Builder(context)
-            .setView(view)
-            .setTitle(R.string.edit_notes)
+            .setView(binding.root)
+            .setTitle(dateString)
             .setPositiveButton(R.string.save) { _, _ ->
-                val note = etNotes.text.toString()
-                callback.onNotesSaved(note)
+                val newValue = when (selectedButton?.id) {
+                    R.id.yesBtn -> YES_MANUAL
+                    R.id.noBtn -> NO
+                    R.id.skippedBtn -> SKIP
+                    else -> UNKNOWN
+                }
+                callback.onNotesSaved(newValue, binding.etNotes.text.toString())
             }
             .setNegativeButton(android.R.string.cancel) { _, _ ->
                 callback.onNotesDismissed()
@@ -41,10 +68,52 @@ class CheckmarkDialog
             .create()
 
         dialog.setOnShowListener {
-            etNotes.requestFocus()
+            binding.etNotes.requestFocus()
             dialog.window?.setSoftInputMode(SOFT_INPUT_STATE_ALWAYS_VISIBLE)
         }
 
         return dialog
     }
+
+    private fun setUpButtons(value: Int, color: Int) {
+        val sres = StyledResources(context)
+        val mediumContrastColor = sres.getColor(R.attr.contrast60)
+
+        setButtonAttrs(binding.yesBtn, color)
+        setButtonAttrs(binding.noBtn, mediumContrastColor)
+
+        if (preferences.isSkipEnabled) {
+            setButtonAttrs(binding.skippedBtn, color)
+            if (value == SKIP) binding.skippedBtn.performClick()
+        }
+
+        if (preferences.areQuestionMarksEnabled) {
+            setButtonAttrs(binding.questionBtn, mediumContrastColor)
+            if (value == UNKNOWN) binding.questionBtn.performClick()
+        }
+
+        when (value) {
+            YES_MANUAL, YES_AUTO -> binding.yesBtn.performClick()
+            NO -> binding.noBtn.performClick()
+        }
+    }
+
+    private fun setButtonAttrs(button: Button, color: Int) {
+        button.apply {
+            visibility = View.VISIBLE
+            typeface = fontAwesome
+            setTextColor(color)
+            setOnClickListener(this@CheckmarkDialog)
+        }
+        allButtons.add(button)
+    }
+
+    override fun onClick(v: View?) {
+        allButtons.forEach {
+            if (v?.id == it.id) {
+                it.isSelected = true
+                selectedButton = it
+            } else it.isSelected = false
+        }
+    }
 }

uhabits-android/src/main/java/org/isoron/uhabits/activities/habits/list/ListHabitsScreen.kt

@@ -235,10 +235,13 @@ class ListHabitsScreen
     }
 
     override fun showCheckmarkDialog(
+        value: Int,
         notes: String,
+        dateString: String,
+        color: PaletteColor,
         callback: ListHabitsBehavior.CheckMarkDialogCallback
     ) {
-        checkMarkDialog.create(notes, callback).show()
+        checkMarkDialog.create(value, notes, dateString, color, callback).show()
     }
 
     private fun getExecuteString(command: Command): String? {

uhabits-android/src/main/java/org/isoron/uhabits/activities/habits/show/ShowHabitActivity.kt

@@ -39,6 +39,7 @@ import org.isoron.uhabits.activities.common.dialogs.NumberPickerFactory
 import org.isoron.uhabits.core.commands.Command
 import org.isoron.uhabits.core.commands.CommandRunner
 import org.isoron.uhabits.core.models.Habit
+import org.isoron.uhabits.core.models.PaletteColor
 import org.isoron.uhabits.core.preferences.Preferences
 import org.isoron.uhabits.core.ui.callbacks.OnConfirmedCallback
 import org.isoron.uhabits.core.ui.screens.habits.list.ListHabitsBehavior
@@ -173,10 +174,14 @@ class ShowHabitActivity : AppCompatActivity(), CommandRunner.Listener {
         }
 
         override fun showCheckmarkDialog(
+            value: Int,
             notes: String,
+            dateString: String,
+            preferences: Preferences,
+            color: PaletteColor,
             callback: ListHabitsBehavior.CheckMarkDialogCallback
         ) {
-            CheckmarkDialog(this@ShowHabitActivity).create(notes, callback).show()
+            CheckmarkDialog(this@ShowHabitActivity, preferences).create(value, notes, dateString, color, callback).show()
         }
 
         override fun showEditHabitScreen(habit: Habit) {