From 4d61fb21e606292db97951812ebe9a6fa455eaff Mon Sep 17 00:00:00 2001
From: hitalin <hitalin@proton.me>
Date: Fri, 7 Aug 2026 14:01:32 +0900
Subject: [PATCH] =?UTF-8?q?fix(streaming):=20polling=20=E7=A8=BC=E5=83=8D?=
 =?UTF-8?q?=E4=B8=AD=E3=81=AE=20connect=20=E3=81=8C=20WebSocket=20?=
 =?UTF-8?q?=E3=82=92=E5=BE=A9=E6=B4=BB=E3=81=95=E3=81=9B=E3=81=AA=E3=81=84?=
 =?UTF-8?q?=E3=82=88=E3=81=86=E3=81=AB=E3=81=99=E3=82=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

フロントはカラムのマウントや復帰のたびに stream_connect を無条件に呼ぶため、
polling モード中でも WS 接続が黙って張り直されていた (notedeck#1004)。
connect の意味を「ストリームを現在のモードで確保する」に揃え、polling が
供給中なら Connected を emit して即 return する。WS へ戻す唯一の経路は
set_mode("realtime")。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
---
 src/streaming.rs | 49 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/src/streaming.rs b/src/streaming.rs
index c6f0e01..b7be3ae 100644
--- a/src/streaming.rs
+++ b/src/streaming.rs
@@ -472,6 +472,23 @@ impl StreamingManager {
         host: &str,
         token: &str,
     ) -> Result<(), NoteDeckError> {
+        // polling がこのアカウントのストリームを供給中なら WS を張らない。
+        // connect は「ストリームを現在のモードで確保する」であって「WS を
+        // 強制する」ではない。カラムのマウントや復帰は connect を無条件に
+        // 呼ぶため、ここで弾かないと polling モードが黙って崩れる
+        // (notedeck#1004)。WS へ戻す唯一の経路は set_mode("realtime")。
+        {
+            let polls = self.poll_connections.lock().await;
+            if polls.contains_key(account_id) {
+                self.emitter
+                    .emit(StreamEvent::Status(Box::new(StreamStatusEvent {
+                        account_id: account_id.to_string(),
+                        state: StreamConnectionState::Connected,
+                    })));
+                return Ok(());
+            }
+        }
+
         let mut conns = self.connections.lock().await;
         if let Some(handle) = conns.get(account_id) {
             // 冪等 return でも現在の実状態を emit する。フロントは復帰時に
@@ -2365,6 +2382,38 @@ mod tests {
         manager.disconnect("acc-1").await;
     }
 
+    #[tokio::test]
+    async fn connect_does_not_resurrect_websocket_while_polling() {
+        // フロントはカラムのマウントや復帰のたびに connect() を無条件に呼ぶ。
+        // polling 中に WS が復活すると、永続化されたモードと実動作が乖離する
+        // (notedeck#1004)。connect は「ストリームを現在のモードで確保する」であって
+        // 「WS を強制する」ではない。
+        let (_dir, manager, mut rx) = manager_with_dead_connection(&["acc-1"]).await;
+        manager
+            .set_mode("acc-1", "127.0.0.1:1", "token", "polling", Some(60_000))
+            .await
+            .unwrap();
+        drain_status(&mut rx);
+
+        manager
+            .connect("acc-1", "127.0.0.1:1", "token")
+            .await
+            .unwrap();
+
+        assert!(
+            manager.connections.lock().await.is_empty(),
+            "polling 中の connect は WS を張らない"
+        );
+        assert!(manager.poll_connections.lock().await.contains_key("acc-1"));
+        // polling がストリームを供給中なので Connected として報告する
+        assert!(
+            drain_status(&mut rx).contains(&StreamConnectionState::Connected),
+            "冪等 return でも現在状態を emit すること"
+        );
+
+        manager.disconnect("acc-1").await;
+    }
+
     #[tokio::test]
     async fn unsubscribe_drops_the_subscription_in_both_modes() {
         let (_dir, manager, _rx) = manager_with_dead_connection(&["acc-1"]).await;