From 56fe574940fe8334b3dbc55ff3e3b71fa410d7d0 Mon Sep 17 00:00:00 2001
From: Barnabas Balogh <bdbarni@gmail.com>
Date: Thu, 7 May 2026 19:45:09 +0200
Subject: [PATCH] feat: add watched status to MediaUiModel

---
 .../main/java/hu/bbara/purefin/core/model/MediaUiModel.kt | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/core/src/main/java/hu/bbara/purefin/core/model/MediaUiModel.kt b/core/src/main/java/hu/bbara/purefin/core/model/MediaUiModel.kt
index 144cee2f..23df0a2c 100644
--- a/core/src/main/java/hu/bbara/purefin/core/model/MediaUiModel.kt
+++ b/core/src/main/java/hu/bbara/purefin/core/model/MediaUiModel.kt
@@ -26,6 +26,7 @@ class MovieUiModel: MediaUiModel {
     override val secondaryText: String
     override val description: String
     private val prefixImageUrl: String
+    override val watched: Boolean
     override val primaryImageUrl: String
         get() {
             return ImageUrlBuilder.finishImageUrl(
@@ -49,6 +50,7 @@ class MovieUiModel: MediaUiModel {
         description = movie.synopsis
         prefixImageUrl = movie.imageUrlPrefix
         progress = (movie.progress?.toFloat() ?: 0f) / 100f
+        watched = movie.watched
     }
 
     companion object {
@@ -78,7 +80,9 @@ class SeriesUiModel : MediaUiModel {
     override val primaryText: String
     override val secondaryText: String
     override val description: String
+    override val watched: Boolean
     private val prefixImageUrl: String
+    val unwatchedEpisodeCount: Int
     override val primaryImageUrl: String
         get() {
             return ImageUrlBuilder.finishImageUrl(
@@ -100,6 +104,8 @@ class SeriesUiModel : MediaUiModel {
         secondaryText = "${series.seasonCount} seasons"
         description = series.synopsis
         prefixImageUrl = series.imageUrlPrefix
+        watched = series.unwatchedEpisodeCount == 0
+        unwatchedEpisodeCount = series.unwatchedEpisodeCount
     }
 }
 
@@ -109,6 +115,7 @@ class EpisodeUiModel : MediaUiModel {
     override val secondaryText: String
     override val description: String
     private val prefixImageUrl: String
+    override val watched: Boolean
     override val primaryImageUrl: String
         get() {
             return ImageUrlBuilder.finishImageUrl(
@@ -136,5 +143,6 @@ class EpisodeUiModel : MediaUiModel {
         progress = (episode.progress?.toFloat() ?: 0f) / 100f
         seriesId = episode.seriesId
         seasonId = episode.seasonId
+        watched = episode.watched
     }
 }
\ No newline at end of file