From 14dabc831915bc5b4aee4c4c8b30cf1b1136a8d2 Mon Sep 17 00:00:00 2001
From: Brayan Oliveira <69634269+brayandso@users.noreply.github.com>
Date: Sat, 22 Jun 2024 05:39:14 -0300
Subject: [PATCH] feat(new reviewer): counts

---
 .../ui/windows/reviewer/ReviewerFragment.kt   | 27 ++++++++++++++
 .../ui/windows/reviewer/ReviewerViewModel.kt  |  3 ++
 AnkiDroid/src/main/res/layout/reviewer2.xml   | 35 ++++++++++++++++++-
 3 files changed, 64 insertions(+), 1 deletion(-)

diff --git a/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerFragment.kt b/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerFragment.kt
index 3e46527dc926..e24b9b57d7bd 100644
--- a/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerFragment.kt
+++ b/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerFragment.kt
@@ -18,6 +18,8 @@ package com.ichi2.anki.ui.windows.reviewer
 import android.content.Context
 import android.content.Intent
 import android.os.Bundle
+import android.text.SpannableString
+import android.text.style.UnderlineSpan
 import android.view.Menu
 import android.view.MenuItem
 import android.view.View
@@ -33,6 +35,7 @@ import androidx.lifecycle.flowWithLifecycle
 import androidx.lifecycle.lifecycleScope
 import com.google.android.material.appbar.MaterialToolbar
 import com.google.android.material.button.MaterialButton
+import com.google.android.material.textview.MaterialTextView
 import com.ichi2.anki.AbstractFlashcardViewer.Companion.RESULT_NO_MORE_CARDS
 import com.ichi2.anki.CollectionManager
 import com.ichi2.anki.Flag
@@ -47,6 +50,7 @@ import com.ichi2.anki.snackbar.showSnackbar
 import com.ichi2.anki.utils.ext.collectIn
 import com.ichi2.anki.utils.ext.collectLatestIn
 import com.ichi2.anki.utils.navBarNeedsScrim
+import com.ichi2.libanki.sched.Counts
 import com.ichi2.utils.increaseHorizontalPaddingOfOverflowMenuIcons
 import kotlinx.coroutines.launch
 
@@ -70,6 +74,7 @@ class ReviewerFragment :
         super.onViewCreated(view, savedInstanceState)
 
         setupAnswerButtons(view)
+        setupCounts(view)
 
         view.findViewById<MaterialToolbar>(R.id.toolbar).apply {
             setOnMenuItemClickListener(this@ReviewerFragment)
@@ -178,6 +183,28 @@ class ReviewerFragment :
         }
     }
 
+    private fun setupCounts(view: View) {
+        val newCount = view.findViewById<MaterialTextView>(R.id.new_count)
+        val learnCount = view.findViewById<MaterialTextView>(R.id.lrn_count)
+        val reviewCount = view.findViewById<MaterialTextView>(R.id.rev_count)
+
+        viewModel.countsFlow.flowWithLifecycle(lifecycle)
+            .collectLatestIn(lifecycleScope) { (counts, countsType) ->
+                newCount.text = counts.new.toString()
+                learnCount.text = counts.lrn.toString()
+                reviewCount.text = counts.rev.toString()
+
+                val currentCount = when (countsType) {
+                    Counts.Queue.NEW -> newCount
+                    Counts.Queue.LRN -> learnCount
+                    Counts.Queue.REV -> reviewCount
+                }
+                val spannableString = SpannableString(currentCount.text)
+                spannableString.setSpan(UnderlineSpan(), 0, currentCount.text.length, 0)
+                currentCount.text = spannableString
+            }
+    }
+
     private fun setupFlagMenu(menu: Menu) {
         val submenu = menu.findItem(R.id.action_flag).subMenu
         lifecycleScope.launch {
diff --git a/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerViewModel.kt b/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerViewModel.kt
index 02679dc28456..35f823cca03f 100644
--- a/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerViewModel.kt
+++ b/AnkiDroid/src/main/java/com/ichi2/anki/ui/windows/reviewer/ReviewerViewModel.kt
@@ -42,6 +42,7 @@ import com.ichi2.libanki.ChangeManager
 import com.ichi2.libanki.hasTag
 import com.ichi2.libanki.note
 import com.ichi2.libanki.redo
+import com.ichi2.libanki.sched.Counts
 import com.ichi2.libanki.sched.CurrentQueueState
 import com.ichi2.libanki.undo
 import com.ichi2.libanki.undoableOp
@@ -71,6 +72,7 @@ class ReviewerViewModel(cardMediaPlayer: CardMediaPlayer) :
     val canSuspendNoteFlow = MutableStateFlow(true)
     val undoLabelFlow = MutableStateFlow<String?>(null)
     val redoLabelFlow = MutableStateFlow<String?>(null)
+    val countsFlow = MutableStateFlow(Counts() to Counts.Queue.NEW)
 
     override val server = AnkiServer(this).also { it.start() }
     private val stateMutationKey = TimeManager.time.intTimeMS().toString()
@@ -360,6 +362,7 @@ class ReviewerViewModel(cardMediaPlayer: CardMediaPlayer) :
         flagCodeFlow.emit(card.userFlag())
         canBuryNoteFlow.emit(isBuryNoteAvailable(card))
         canSuspendNoteFlow.emit(isSuspendNoteAvailable(card))
+        countsFlow.emit(state.counts to state.countsIndex)
     }
 
     // TODO
diff --git a/AnkiDroid/src/main/res/layout/reviewer2.xml b/AnkiDroid/src/main/res/layout/reviewer2.xml
index f2ed634dc489..5f5b525c008d 100644
--- a/AnkiDroid/src/main/res/layout/reviewer2.xml
+++ b/AnkiDroid/src/main/res/layout/reviewer2.xml
@@ -28,7 +28,40 @@
                 android:layout_height="wrap_content"
                 android:background="?attr/alternativeBackgroundColor"
                 app:menu="@menu/reviewer2"
-                />
+                >
+
+                <com.google.android.material.textview.MaterialTextView
+                    android:id="@+id/new_count"
+                    android:layout_width="wrap_content"
+                    android:layout_height="wrap_content"
+                    android:textColor="?attr/newCountColor"
+                    app:layout_constraintEnd_toStartOf="@id/lrn_count"
+                    android:paddingEnd="5dp"
+                    tools:text="27"
+                    />
+
+                <com.google.android.material.textview.MaterialTextView
+                    android:id="@+id/lrn_count"
+                    android:layout_width="wrap_content"
+                    android:layout_height="wrap_content"
+                    android:textColor="?attr/learnCountColor"
+                    app:layout_constraintStart_toEndOf="@id/new_count"
+                    app:layout_constraintEnd_toStartOf="@id/rev_count"
+                    android:paddingEnd="5dp"
+                    tools:text="81"
+                    />
+
+                <com.google.android.material.textview.MaterialTextView
+                    android:id="@+id/rev_count"
+                    android:layout_width="wrap_content"
+                    android:layout_height="wrap_content"
+                    android:textColor="?attr/reviewCountColor"
+                    app:layout_constraintStart_toEndOf="@id/lrn_count"
+                    tools:text="54"
+                    />
+
+
+            </com.google.android.material.appbar.MaterialToolbar>
         </com.google.android.material.appbar.AppBarLayout>
 
         <com.google.android.material.card.MaterialCardView