From da7b40c396566882f04c8579c2e5a3fa2f140498 Mon Sep 17 00:00:00 2001
From: Savanni D'Gerinel <savanni@luminescent-dreams.com>
Date: Tue, 27 Feb 2024 23:01:09 -0500
Subject: [PATCH] Ensure the game_view_model handler aborts when the view_model
 gets dropped

---
 kifu/gtk/src/view_models/game_view_model.rs | 31 ++++++++++++---------
 1 file changed, 18 insertions(+), 13 deletions(-)

diff --git a/kifu/gtk/src/view_models/game_view_model.rs b/kifu/gtk/src/view_models/game_view_model.rs
index 8a10bcf..273df4e 100644
--- a/kifu/gtk/src/view_models/game_view_model.rs
+++ b/kifu/gtk/src/view_models/game_view_model.rs
@@ -14,10 +14,7 @@ General Public License for more details.
 You should have received a copy of the GNU General Public License along with Kifu. If not, see <https://www.gnu.org/licenses/>.
 */
 
-use async_std::{
-    channel::Receiver,
-    task::{spawn, yield_now, JoinHandle},
-};
+use async_std::{channel::Receiver, task::yield_now};
 use kifu_core::{Color, Core, CoreNotification, Goban, Observable, Player};
 use std::{cell::RefCell, rc::Rc, time::Duration};
 
@@ -40,6 +37,7 @@ struct GameViewModelPrivate {
 /// The Game View Model manages the current state of the game. It shows the two player cards, the board, the current capture count, the current player, and it maintains the UI for the clock (bearing in mind that the real clock is managed in the core). This view model should only be created once the details of the game, whether a game in progress or a new game (this view model won't know the difference) is known.
 pub struct GameViewModel {
     core: Core,
+    notification_handler: glib::JoinHandle<()>,
     widget: gtk::Box,
     data: Rc<RefCell<GameViewModelPrivate>>,
 }
@@ -56,17 +54,18 @@ impl GameViewModel {
             state: game,
         }));
 
-        let s = Self {
-            core,
-            widget: gtk::Box::new(gtk::Orientation::Horizontal, 0),
-            data,
+        let notification_handler = {
+            let notifications = core.subscribe();
+            let data: Rc<RefCell<GameViewModelPrivate>> = data.clone();
+            glib::spawn_future_local(Self::listen(notifications, data))
         };
 
-        let notifications = s.core.subscribe();
-        let data = s.data.clone();
-        glib::spawn_future_local(Self::listen(notifications, data));
-
-        s
+        Self {
+            core,
+            notification_handler,
+            widget: gtk::Box::new(gtk::Orientation::Horizontal, 0),
+            data,
+        }
     }
 
     async fn listen(
@@ -84,3 +83,9 @@ impl GameViewModel {
         }
     }
 }
+
+impl Drop for GameViewModel {
+    fn drop(&mut self) {
+        self.notification_handler.abort();
+    }
+}