update progress after edit

anilist/src/commonMain/graphql/EditMutation.graphql

@@ -12,6 +12,7 @@ mutation EditMutation(
     ) {
         progress,
         progressVolumes,
+        repeat,
         status,
         startedAt {
             year,

composeApp/src/commonMain/kotlin/dev/datlag/aniflow/ui/navigation/screen/medium/MediumComponent.kt

@@ -57,7 +57,7 @@ interface MediumComponent : ContentHolderComponent {
     val siteUrl: Flow<String>
 
     val type: Flow<MediaType>
-    val listStatus: Flow<MediaListStatus>
+    val listStatus: StateFlow<MediaListStatus>
 
     val dialog: Value<ChildSlot<DialogConfig, DialogComponent>>
 

composeApp/src/commonMain/kotlin/dev/datlag/aniflow/ui/navigation/screen/medium/MediumScreenComponent.kt

@@ -179,9 +179,12 @@ class MediumScreenComponent(
     }
 
     @OptIn(ExperimentalCoroutinesApi::class)
-    override val listStatus: Flow<MediaListStatus> = mediumSuccessState.mapLatest {
+    override val listStatus: MutableStateFlow<MediaListStatus> = mediumSuccessState.mapLatest {
         it.medium.entry?.status ?: MediaListStatus.UNKNOWN__
-    }
+    }.mutableStateIn(
+        scope = ioScope(),
+        initialValue = initialMedium.entry?.status ?: MediaListStatus.UNKNOWN__
+    )
 
     @OptIn(ExperimentalCoroutinesApi::class)
     override val chapters: Flow<Int> = mediumSuccessState.mapLatest {
@@ -221,6 +224,22 @@ class MediumScreenComponent(
         search.ifEmpty { default }
     }
 
+    @OptIn(ExperimentalCoroutinesApi::class)
+    private val watchProgress = mediumSuccessState.mapLatest {
+        it.medium.entry?.progress
+    }.mutableStateIn(
+        scope = ioScope(),
+        initialValue = initialMedium.entry?.progress
+    )
+
+    @OptIn(ExperimentalCoroutinesApi::class)
+    private val watchRepeat = mediumSuccessState.mapLatest {
+        it.medium.entry?.repeatCount
+    }.mutableStateIn(
+        scope = ioScope(),
+        initialValue = initialMedium.entry?.repeatCount
+    )
+
     private val dialogNavigation = SlotNavigation<DialogConfig>()
     @OptIn(ExperimentalCoroutinesApi::class)
     override val dialog: Value<ChildSlot<DialogConfig, DialogComponent>> = childSlot(
@@ -238,28 +257,16 @@ class MediumScreenComponent(
                 componentContext = context,
                 di = di,
                 episodes = episodes,
-                progress = mediumSuccessState.mapLatest {
-                    it.medium.entry?.progress
-                },
+                progress = watchProgress,
                 listStatus = listStatus,
-                repeatCount = mediumSuccessState.mapLatest {
-                    it.medium.entry?.repeatCount
-                },
+                repeatCount = watchRepeat,
                 episodeStartDate = mediumSuccessState.mapLatest {
                     it.medium.startDate
                 },
                 onDismiss = dialogNavigation::dismiss,
                 onSave = { status, progress, repeat ->
                     dialogNavigation.dismiss {
-                        val call = mediumRepository.updateEditCall(
-                            status = status,
-                            progress = progress,
-                            repeat = repeat
-                        ).fetchPolicy(FetchPolicy.NetworkOnly)
-
-                        launchIO {
-                            call.execute()
-                        }
+                        editSave(status, progress, repeat)
                     }
                 }
             )
@@ -301,6 +308,34 @@ class MediumScreenComponent(
         }
     }
 
+    private fun editSave(
+        state: MediaListStatus,
+        progress: Int,
+        repeat: Int,
+    ) {
+        val newData = mediumRepository
+            .updateEditCall(
+                status = state,
+                progress = progress,
+                repeat = repeat
+            )
+            .fetchPolicy(FetchPolicy.NetworkOnly)
+            .toFlow()
+            .mapNotNull {
+                it.data?.SaveMediaListEntry
+            }
+
+        launchIO {
+            listStatus.emitAll(newData.mapNotNull { it.status })
+        }
+        launchIO {
+            watchProgress.emitAll(newData.mapNotNull { it.progress })
+        }
+        launchIO {
+            watchRepeat.emitAll(newData.mapNotNull { it.repeat })
+        }
+    }
+
     override fun descriptionTranslation(text: String?) {
         translatedDescription.update { text }
     }

composeApp/src/commonMain/kotlin/dev/datlag/aniflow/ui/navigation/screen/medium/component/FABContent.kt

@@ -62,7 +62,7 @@ fun FABContent(
 
         if (!notReleased) {
             val loggedIn by component.isLoggedIn.collectAsStateWithLifecycle(false)
-            val status by component.listStatus.collectAsStateWithLifecycle(component.initialMedium.entry?.status ?: MediaListStatus.UNKNOWN__)
+            val status by component.listStatus.collectAsStateWithLifecycle()
             val type by component.type.collectAsStateWithLifecycle(component.initialMedium.type)
             val uriHandler = LocalUriHandler.current
             val speedDialFABState = rememberSpeedDialState()

composeApp/src/commonMain/kotlin/dev/datlag/aniflow/ui/navigation/screen/medium/dialog/edit/EditState.kt

@@ -112,7 +112,7 @@ class EditState(
                     MediaListStatus.CURRENT
                 }
             } else {
-                if (it == MediaListStatus.COMPLETED) {
+                if (it == MediaListStatus.COMPLETED || it == MediaListStatus.UNKNOWN__) {
                     MediaListStatus.CURRENT
                 } else {
                     it