hyvä siitä tulee

network-poc/hub/src/db.rs

@@ -152,6 +152,24 @@ impl NodeDb {
         conn.last_insert_rowid()
     }
 
+    pub fn update_session_task(&self, node_id: u64, task: &str) {
+        let conn = self.conn.lock().unwrap_or_else(|e| e.into_inner());
+        let _ = conn.execute(
+            "UPDATE node_sessions SET selected_task = ?1 WHERE node_id = ?2 AND disconnected_at IS NULL",
+            params![task, node_id as i64],
+        );
+    }
+
+    /// Sulkee saman IP:n viewer-sessiot kun aktiivinen node liittyy
+    pub fn close_viewers_by_ip(&self, ip: &str) {
+        let conn = self.conn.lock().unwrap_or_else(|e| e.into_inner());
+        let now = chrono::Utc::now().to_rfc3339();
+        let _ = conn.execute(
+            "UPDATE node_sessions SET disconnected_at = ?1 WHERE ip = ?2 AND disconnected_at IS NULL AND (selected_task = 'viewer' OR selected_task = 'codelab-viewer')",
+            params![now, ip],
+        );
+    }
+
     pub fn close_session(&self, node_id: u64) {
         let conn = self.conn.lock().unwrap_or_else(|e| e.into_inner());
         let now = chrono::Utc::now().to_rfc3339();