From 87f7bd6e83d9100273b9f51408c2ed5f61641977 Mon Sep 17 00:00:00 2001
From: 0xb00bface <0xboobface@gmail.com>
Date: Sat, 7 Aug 2021 16:31:24 +0200
Subject: [PATCH] Update mark unmark model immediately in remote mode

---
 .../src/main/java/ctbrec/recorder/RemoteRecorder.java  | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/common/src/main/java/ctbrec/recorder/RemoteRecorder.java b/common/src/main/java/ctbrec/recorder/RemoteRecorder.java
index 7552fbab..cfb5fd54 100644
--- a/common/src/main/java/ctbrec/recorder/RemoteRecorder.java
+++ b/common/src/main/java/ctbrec/recorder/RemoteRecorder.java
@@ -706,17 +706,21 @@ public class RemoteRecorder implements Recorder {
 
     @Override
     public void markForLaterRecording(Model model, boolean mark) throws InvalidKeyException, NoSuchAlgorithmException, IOException {
+        model.setMarkedForLaterRecording(mark);
+        LOG.info("Marking model {} {}", model, mark);
         if (mark) {
             sendRequest("markForLater", model);
         } else {
             sendRequest("unmarkForLater", model);
         }
-        model.setMarkedForLaterRecording(mark);
-        // update cached model
+        // update locally cached model
         int index = models.indexOf(model);
         if (index >= 0) {
             Model m = models.get(index);
-            m.setSuspended(mark);
+            m.setMarkedForLaterRecording(mark);
+            models.remove(index);
+        } else {
+            models.add(model);
         }
     }
 }