soulseek-rs-modded/soulseek-rs/src/ui/main_tui/mod.rs

mod browse;
mod downloads;
mod input;
mod render;
mod rooms;
mod search;
mod settings;

use crate::api::SessionApi;
use crate::models::AppState;
use crate::persist::{
    snapshot::{Snapshot, restore_messages, restore_searches},
    state::{PersistedMessage, StateStore},
};
use color_eyre::Result;
use ratatui::{
    DefaultTerminal,
    crossterm::event::{self, Event, KeyEventKind, poll},
};
use std::{sync::Arc, time::Duration};

pub struct MainTui {
    client: Arc<dyn SessionApi>,
    state: AppState,
    download_dir: String,
    search_timeout: Duration,
    spinner_state: usize,
    store: Option<StateStore>,
    /// Where settings changes are persisted — the config file this run
    /// resolved, so `--config` is honoured.
    config_path: Option<std::path::PathBuf>,
    /// Last snapshot written to disk, to skip no-op saves.
    saved_snapshot: Snapshot,
}

impl MainTui {
    pub fn new(
        client: Arc<dyn SessionApi>,
        download_dir: String,
        search_timeout: Duration,
        store: Option<StateStore>,
        config_path: Option<std::path::PathBuf>,
    ) -> Self {
        let mut tui = Self {
            client,
            state: AppState::new(),
            download_dir,
            search_timeout,
            spinner_state: 0,
            store,
            config_path,
            saved_snapshot: Snapshot::default(),
        };
        tui.restore_persisted_state();
        tui
    }

    /// Bring back last session's state: search history, chat rooms
    /// (rejoined on the server), and downloads — incomplete ones are
    /// re-enqueued so they resume automatically.
    fn restore_persisted_state(&mut self) {
        // Whatever the session has been collecting comes first: attached to a
        // daemon this is the conversation so far, including everything that
        // arrived while nothing was showing it. Locally it is empty and the
        // snapshot below is the whole record.
        let history: Vec<PersistedMessage> = self
            .client
            .message_history()
            .into_iter()
            .map(|message| PersistedMessage {
                peer: message.peer,
                outgoing: message.outgoing,
                text: message.text,
                at: chrono::DateTime::from_timestamp(message.at, 0)
                    .unwrap_or_default()
                    .into(),
            })
            .collect();
        restore_messages(&mut self.state, &history);

        let Some(store) = &self.store else {
            self.saved_snapshot = Snapshot::capture(&self.state);
            return;
        };

        restore_searches(&mut self.state, &store.load_search_queries());
        restore_messages(&mut self.state, &store.load_messages());
        // Restore the unread badge so messages left unread carry over. Set
        // before the baseline snapshot below so it isn't seen as a change.
        self.state.unread_messages = store.load_unread();

        for room in store.load_rooms() {
            if self.state.rooms.focus_or_open(&room)
                && let Err(e) = self.client.join_room(&room)
            {
                soulseek_rs::warn!("Could not rejoin {room}: {e}");
            }
        }

        let downloads = store.load_downloads();
        self.saved_snapshot = Snapshot::capture(&self.state);
        // Completed entries are shown as-is; the rest re-enqueue below and
        // reappear through the normal downloads channel.
        self.saved_snapshot.downloads.clone_from(&downloads);

        let sender = self.downloads_sender();
        for entry in downloads {
            if entry.completed {
                self.state.downloads.push(crate::models::DownloadEntry {
                    download: soulseek_rs::types::Download {
                        username: entry.username,
                        filename: entry.filename,
                        token: 0,
                        size: entry.size,
                        download_directory: entry.download_directory,
                        status: soulseek_rs::DownloadStatus::Completed,
                        sender: std::sync::mpsc::channel().0,
                        queue_position: None,
                        metadata: soulseek_rs::types::DownloadMetadata::default(
                        ),
                    },
                    receiver: None,
                });
            } else {
                let client = self.client.clone();
                let sender = sender.clone();
                std::thread::spawn(move || {
                    match client.download(
                        entry.filename.clone(),
                        entry.username,
                        entry.size,
                        entry.download_directory,
                    ) {
                        Ok((download, rx)) => {
                            let _ = sender.send((download, rx));
                        }
                        Err(e) => soulseek_rs::warn!(
                            "Could not resume {}: {e}",
                            entry.filename
                        ),
                    }
                });
            }
        }
    }

    /// Write state to disk when it differs from what was last saved.
    fn save_persisted_state(&mut self) {
        let Some(store) = &self.store else { return };
        let snapshot = Snapshot::capture(&self.state);
        if snapshot == self.saved_snapshot {
            return;
        }
        if snapshot.downloads != self.saved_snapshot.downloads
            && let Err(e) = store.save_downloads(&snapshot.downloads)
        {
            soulseek_rs::warn!("Could not save downloads state: {e}");
        }
        if snapshot.queries != self.saved_snapshot.queries
            && let Err(e) = store.save_search_queries(&snapshot.queries)
        {
            soulseek_rs::warn!("Could not save search history: {e}");
        }
        if snapshot.rooms != self.saved_snapshot.rooms
            && let Err(e) = store.save_rooms(&snapshot.rooms)
        {
            soulseek_rs::warn!("Could not save room state: {e}");
        }
        if snapshot.messages != self.saved_snapshot.messages
            && let Err(e) = store.save_messages(&snapshot.messages)
        {
            soulseek_rs::warn!("Could not save message history: {e}");
        }
        if snapshot.unread_messages != self.saved_snapshot.unread_messages
            && let Err(e) = store.save_unread(snapshot.unread_messages)
        {
            soulseek_rs::warn!("Could not save unread count: {e}");
        }
        self.saved_snapshot = snapshot;
    }

    pub fn run(mut self, mut terminal: DefaultTerminal) -> Result<()> {
        use ratatui::crossterm::{event::DisableMouseCapture, execute};

        // Run the event loop, then restore the terminal unconditionally: if the
        // loop returns early with an error the terminal must still be taken out
        // of raw mode / the alternate screen and mouse capture disabled, or the
        // user is left with a corrupted terminal.
        let result = self.run_event_loop(&mut terminal);
        self.save_persisted_state();

        let _ = execute!(std::io::stdout(), DisableMouseCapture);
        ratatui::restore();
        soulseek_rs::utils::logger::disable_buffering();

        result
    }

    fn run_event_loop(&mut self, terminal: &mut DefaultTerminal) -> Result<()> {
        while !self.state.should_exit {
            terminal.draw(|frame| self.render(frame))?;

            // Poll for search results updates
            self.update_search_results();

            // Poll for download updates
            self.update_downloads();

            // Poll for incoming private messages
            self.poll_private_messages();

            // Poll for a browse (shared-file listing) response
            self.poll_browse_result();

            // Poll for chat-room events
            self.poll_room_events();

            // Refresh the uploads we are serving to peers
            self.state.uploads = self.client.uploads();

            self.spinner_state = (self.spinner_state + 1) % 10;

            self.save_persisted_state();

            // Drain every queued input event before the next draw: key
            // autorepeat outpaces the frame time, and handling one event per
            // frame makes the backlog keep scrolling for seconds after the
            // key is released.
            if poll(Duration::from_millis(100))? {
                loop {
                    match event::read()? {
                        Event::Key(key) if key.kind == KeyEventKind::Press => {
                            self.handle_key_event(key);
                        }
                        Event::Mouse(mouse) => {
                            self.handle_mouse_event(mouse);
                        }
                        _ => {}
                    }
                    if self.state.should_exit || !poll(Duration::ZERO)? {
                        break;
                    }
                }
            }
        }

        Ok(())
    }
}

pub fn launch_main_tui(
    terminal: DefaultTerminal,
    client: Arc<dyn SessionApi>,
    download_dir: String,
    search_timeout: Duration,
    store: Option<StateStore>,
    config_path: Option<std::path::PathBuf>,
) -> Result<()> {
    let tui =
        MainTui::new(client, download_dir, search_timeout, store, config_path);
    tui.run(terminal)
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::daemon::proto::ChatMessageDto;
    use crate::models::{FocusedPane, MessageDirection, SearchStatus};
    use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};

    /// A stand-in for a daemon: it already holds a conversation, a transfer
    /// queue, and a search cache, none of which this window created.
    #[derive(Default)]
    struct TalkativeSession {
        history: Vec<ChatMessageDto>,
        downloads: Vec<soulseek_rs::types::Download>,
        searches: std::sync::Mutex<Vec<crate::api::SessionSearch>>,
        /// What `try_get_search_results` hands back to the window.
        results: std::sync::Mutex<Vec<soulseek_rs::SearchResult>>,
        /// How many times a window asked for the search list.
        all_searches_calls: std::sync::atomic::AtomicUsize,
        /// How many times a window pulled a full result set across. Against a
        /// daemon each pull is the whole set over the socket, so the tests
        /// count them.
        result_fetches: std::sync::atomic::AtomicUsize,
        /// Whether this session is a daemon's. Local sessions do not sync.
        shared: bool,
        /// What `set_download_directory` was last asked for, if anything.
        download_dir_set: std::sync::Mutex<Option<String>>,
        /// What `set_shared_directories` was last asked for, if anything.
        shares_set: std::sync::Mutex<Option<Vec<String>>>,
    }

    fn queued(username: &str, filename: &str) -> soulseek_rs::types::Download {
        at_status(username, filename, soulseek_rs::DownloadStatus::Queued)
    }

    fn completed(
        username: &str,
        filename: &str,
    ) -> soulseek_rs::types::Download {
        at_status(username, filename, soulseek_rs::DownloadStatus::Completed)
    }

    fn at_status(
        username: &str,
        filename: &str,
        status: soulseek_rs::DownloadStatus,
    ) -> soulseek_rs::types::Download {
        let (sender, _) = std::sync::mpsc::channel();
        soulseek_rs::types::Download {
            username: username.to_string(),
            filename: filename.to_string(),
            token: 1,
            size: 4096,
            download_directory: "/tmp".to_string(),
            status,
            sender,
            queue_position: None,
            metadata: soulseek_rs::types::DownloadMetadata::default(),
        }
    }

    fn key(ch: char) -> KeyEvent {
        KeyEvent::new(KeyCode::Char(ch), KeyModifiers::NONE)
    }

    impl SessionApi for TalkativeSession {
        fn message_history(&self) -> Vec<ChatMessageDto> {
            self.history.clone()
        }
        fn daemon_endpoint(&self) -> Option<String> {
            self.shared.then(|| "test-daemon".to_string())
        }
        fn username(&self) -> String {
            "tester".to_string()
        }
        fn listen_port(&self) -> Option<u16> {
            None
        }
        fn session_loss(&self) -> Option<soulseek_rs::SessionLoss> {
            None
        }
        fn search_with_cancel(
            &self,
            _query: &str,
            _timeout: Duration,
            _cancel: Option<Arc<std::sync::atomic::AtomicBool>>,
        ) -> soulseek_rs::Result<Vec<soulseek_rs::SearchResult>> {
            Ok(Vec::new())
        }
        fn get_search_results(
            &self,
            _key: &str,
        ) -> Vec<soulseek_rs::SearchResult> {
            Vec::new()
        }
        fn all_searches(&self) -> Vec<crate::api::SessionSearch> {
            self.all_searches_calls
                .fetch_add(1, std::sync::atomic::Ordering::Relaxed);
            self.searches.lock().expect("not poisoned").clone()
        }
        fn forget_search(&self, query: &str) -> bool {
            let mut searches = self.searches.lock().expect("not poisoned");
            let before = searches.len();
            searches.retain(|search| search.query != query);
            searches.len() != before
        }
        fn get_search_results_count(&self, _key: &str) -> usize {
            0
        }
        fn try_get_search_results(
            &self,
            _key: &str,
        ) -> Option<Vec<soulseek_rs::SearchResult>> {
            self.result_fetches
                .fetch_add(1, std::sync::atomic::Ordering::Relaxed);
            Some(self.results.lock().expect("not poisoned").clone())
        }
        fn start_wishlist_search(
            &self,
            _query: &str,
        ) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn wishlist_interval(&self) -> Duration {
            Duration::from_mins(12)
        }
        fn download(
            &self,
            _filename: String,
            _username: String,
            _size: u64,
            _directory: String,
        ) -> soulseek_rs::Result<(
            soulseek_rs::types::Download,
            std::sync::mpsc::Receiver<soulseek_rs::DownloadStatus>,
        )> {
            Err(soulseek_rs::SoulseekRs::NotConnected)
        }
        fn download_with_metadata(
            &self,
            _filename: String,
            _username: String,
            _size: u64,
            _directory: String,
            _metadata: soulseek_rs::types::DownloadMetadata,
        ) -> soulseek_rs::Result<(
            soulseek_rs::types::Download,
            std::sync::mpsc::Receiver<soulseek_rs::DownloadStatus>,
        )> {
            Err(soulseek_rs::SoulseekRs::NotConnected)
        }
        fn get_all_downloads(&self) -> Vec<soulseek_rs::types::Download> {
            self.downloads.clone()
        }
        fn pause_download(&self, _u: &str, _f: &str) -> bool {
            false
        }
        fn resume_download(&self, _u: &str, _f: &str) -> bool {
            false
        }
        fn remove_queued_download(&self, _u: &str, _f: &str) -> bool {
            false
        }
        fn remove_download(&self, _u: &str, _f: &str) -> bool {
            false
        }
        fn uploads(&self) -> Vec<soulseek_rs::UploadInfo> {
            Vec::new()
        }
        fn take_upload_events(&self) -> Vec<soulseek_rs::UploadInfo> {
            Vec::new()
        }
        fn cancel_upload(&self, _u: &str, _f: &str) -> bool {
            false
        }
        fn set_upload_slots(&self, _slots: usize) {}
        fn check_privileges(&self) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn own_privilege_seconds(&self) -> Option<u32> {
            None
        }
        fn request_room_list(&self) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn join_room(&self, _room: &str) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn leave_room(&self, _room: &str) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn say_in_room(
            &self,
            _room: &str,
            _message: &str,
        ) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn room_members(&self, _room: &str) -> Vec<String> {
            Vec::new()
        }
        fn room_member_stats(
            &self,
            _room: &str,
        ) -> Vec<soulseek_rs::types::RoomUserStats> {
            Vec::new()
        }
        fn take_room_events(&self) -> Vec<soulseek_rs::RoomEvent> {
            Vec::new()
        }
        fn send_private_message(
            &self,
            _u: &str,
            _m: &str,
        ) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn take_private_messages(&self) -> Vec<soulseek_rs::UserMessage> {
            Vec::new()
        }
        fn browse_user(&self, _username: &str) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn take_browse_result(
            &self,
            _username: &str,
        ) -> Option<Vec<soulseek_rs::SharedDirectory>> {
            None
        }
        fn request_user_info(
            &self,
            _username: &str,
        ) -> soulseek_rs::Result<()> {
            Ok(())
        }
        fn user_info(&self, _username: &str) -> Option<soulseek_rs::UserInfo> {
            None
        }
        fn shared_counts(&self) -> (u32, u32) {
            (0, 0)
        }
        fn shared_directories(&self) -> Vec<String> {
            Vec::new()
        }
        fn set_shared_directories(
            &self,
            directories: Vec<String>,
        ) -> soulseek_rs::Result<()> {
            *self.shares_set.lock().expect("not poisoned") = Some(directories);
            Ok(())
        }
        fn set_download_directory(
            &self,
            directory: String,
        ) -> soulseek_rs::Result<()> {
            *self.download_dir_set.lock().expect("not poisoned") =
                Some(directory);
            Ok(())
        }
    }

    fn tui(history: Vec<ChatMessageDto>) -> MainTui {
        with_session(TalkativeSession {
            history,
            shared: true,
            ..TalkativeSession::default()
        })
    }

    fn shared(searches: Vec<crate::api::SessionSearch>) -> TalkativeSession {
        TalkativeSession {
            searches: std::sync::Mutex::new(searches),
            shared: true,
            ..TalkativeSession::default()
        }
    }

    fn search_of(
        query: &str,
        files: usize,
        age: u64,
    ) -> crate::api::SessionSearch {
        crate::api::SessionSearch {
            query: query.to_string(),
            files,
            started_secs_ago: Some(age),
        }
    }

    fn with_session(session: TalkativeSession) -> MainTui {
        attach(Arc::new(session))
    }

    fn attach(session: Arc<TalkativeSession>) -> MainTui {
        MainTui::new(
            session,
            "/tmp".to_string(),
            Duration::from_secs(1),
            None,
            None,
        )
    }

    #[test]
    fn an_attached_session_opens_with_the_conversation_already_in_it() {
        // A TUI attached to a daemon has no state file of its own; the
        // conversation has to come from the session, or it starts blank while
        // the daemon has been collecting messages for hours.
        let tui = tui(vec![
            ChatMessageDto {
                peer: "bob".into(),
                outgoing: false,
                text: "hi".into(),
                at: 1,
            },
            ChatMessageDto {
                peer: "bob".into(),
                outgoing: true,
                text: "hello".into(),
                at: 2,
            },
        ]);

        assert_eq!(tui.state.messages.len(), 2);
        assert_eq!(tui.state.messages[0].peer, "bob");
        assert_eq!(tui.state.messages[0].direction, MessageDirection::Incoming);
        assert_eq!(
            tui.state.messages[1].direction,
            MessageDirection::Outgoing,
            "both halves of the conversation, not just what arrived"
        );
        assert_eq!(tui.state.chat_peers(), ["bob"]);
    }

    #[test]
    fn a_window_shows_transfers_it_did_not_start() {
        // Two windows on one daemon are two views of one queue. A transfer
        // queued in the other one has to appear here, or closing a window
        // would look like losing its downloads.
        let mut tui = with_session(TalkativeSession {
            downloads: vec![queued("bob", "@@x\\theirs.mp3")],
            shared: true,
            ..TalkativeSession::default()
        });
        assert!(
            tui.state.downloads.is_empty(),
            "nothing until the first poll"
        );

        tui.update_downloads();

        assert_eq!(tui.state.downloads.len(), 1);
        assert_eq!(tui.state.downloads[0].download.filename, "@@x\\theirs.mp3");
        assert!(
            tui.state.downloads[0].receiver.is_none(),
            "someone else's transfer has no channel of ours"
        );

        // Polling again must not double it up.
        tui.update_downloads();
        assert_eq!(tui.state.downloads.len(), 1);
    }

    #[test]
    fn a_transfer_the_session_forgets_leaves_every_window() {
        let mut tui = with_session(TalkativeSession {
            downloads: vec![queued("bob", "gone.mp3")],
            shared: true,
            ..TalkativeSession::default()
        });
        tui.update_downloads();
        assert_eq!(tui.state.downloads.len(), 1);

        // The session no longer has it — cancelled from the other window.
        tui.client = Arc::new(TalkativeSession {
            shared: true,
            ..TalkativeSession::default()
        });
        tui.update_downloads();
        assert!(
            tui.state.downloads.is_empty(),
            "a queue is shared in both directions"
        );
    }

    #[test]
    fn d_clears_a_finished_row_but_not_a_running_one() {
        // Finished transfers are what a list fills up with after a restart, so
        // `d` has to be able to dismiss one — restored or not. A running
        // transfer is not ours to discard, and stays put.
        let mut tui = with_session(TalkativeSession::default());
        tui.state.downloads.push(crate::models::DownloadEntry {
            download: completed("bob", "done.mp3"),
            receiver: None,
        });
        tui.state.downloads_table_state.select(Some(0));
        tui.state.focused_pane = FocusedPane::Downloads;

        tui.handle_key_event(key('d'));

        assert!(tui.state.downloads.is_empty(), "the row goes too");
        assert_eq!(tui.state.downloads_table_state.selected(), None);

        tui.state.downloads.push(crate::models::DownloadEntry {
            download: at_status(
                "bob",
                "still.mp3",
                soulseek_rs::DownloadStatus::InProgress {
                    bytes_downloaded: 1,
                    total_bytes: 2,
                    speed_bytes_per_sec: 1.0,
                },
            ),
            receiver: None,
        });
        tui.state.downloads_table_state.select(Some(0));

        tui.handle_key_event(key('d'));

        assert_eq!(tui.state.downloads.len(), 1, "it is not disposable");
    }

    #[test]
    fn a_window_shows_searches_run_in_another() {
        let mut tui =
            with_session(shared(vec![search_of("their query", 7, 9_999)]));
        tui.update_search_results();

        assert_eq!(tui.state.searches.len(), 1);
        assert_eq!(tui.state.searches[0].query, "their query");

        // And it is not added a second time on the next frame.
        tui.update_search_results();
        assert_eq!(tui.state.searches.len(), 1);
    }

    #[test]
    fn a_search_another_window_is_still_running_shows_as_active() {
        // It had said "Done" the moment it appeared, because an imported
        // search carried no notion of still collecting — so the other window
        // looked finished while results were still arriving.
        let mut tui =
            with_session(shared(vec![search_of("still going", 3, 0)]));
        tui.update_search_results();

        assert_eq!(tui.state.searches[0].status, SearchStatus::Active);
        assert_eq!(
            tui.state.searches[0].known_files, 3,
            "and its count comes across before the results do"
        );
    }

    #[test]
    fn a_search_that_has_run_its_window_shows_as_done() {
        let mut tui =
            with_session(shared(vec![search_of("finished", 12, 9_999)]));
        tui.update_search_results();
        assert_eq!(tui.state.searches[0].status, SearchStatus::Completed);
    }

    #[test]
    fn removing_a_search_removes_it_from_the_session() {
        // Removing it only from this window would last until the next sync,
        // which would put it straight back — and in a shared session it would
        // come back for every other window too.
        let mut tui =
            with_session(shared(vec![search_of("unwanted", 1, 9_999)]));
        tui.update_search_results();
        assert_eq!(tui.state.searches.len(), 1);

        tui.remove_search_at_index(0);
        tui.update_search_results();

        assert!(tui.state.searches.is_empty(), "and it stays gone");
    }

    #[test]
    fn clearing_every_search_clears_them_in_the_session_too() {
        let mut tui = with_session(shared(vec![
            search_of("one", 1, 9_999),
            search_of("two", 2, 9_999),
        ]));
        tui.update_search_results();
        assert_eq!(tui.state.searches.len(), 2);

        tui.clear_all_searches();
        tui.update_search_results();

        assert!(
            tui.state.searches.is_empty(),
            "clearing has to reach the session, or every row comes back"
        );
    }

    #[test]
    fn a_search_dismissed_in_another_window_leaves_this_one() {
        let session = shared(vec![search_of("theirs", 4, 9_999)]);
        let mut tui = with_session(session);
        tui.update_search_results();
        assert_eq!(tui.state.searches.len(), 1);

        // The other window dismissed it.
        tui.client.forget_search("theirs");
        tui.update_search_results();

        assert!(tui.state.searches.is_empty());
    }

    fn files_from_bob(names: &[&str]) -> soulseek_rs::SearchResult {
        soulseek_rs::SearchResult {
            token: 0,
            files: names
                .iter()
                .map(|name| soulseek_rs::types::File {
                    username: "bob".to_string(),
                    name: (*name).to_string(),
                    size: 42,
                    attribs: std::collections::HashMap::new(),
                })
                .collect(),
            slots: 1,
            speed: 0,
            username: "bob".to_string(),
        }
    }

    #[test]
    fn a_tick_asks_the_session_for_its_searches_once() {
        // Against a daemon every call is a round trip, and the daemon answers
        // it by walking its whole search cache — twice per frame was half the
        // cost of attaching at all.
        let session = Arc::new(shared(vec![search_of("q", 0, 9_999)]));
        let mut tui = attach(session.clone());

        tui.update_search_results();

        assert_eq!(
            session
                .all_searches_calls
                .load(std::sync::atomic::Ordering::Relaxed),
            1,
            "one sync means one listing, not one per use of it"
        );
    }

    /// An attached window that selected a two-file search and has already
    /// ticked past its one necessary fetch.
    fn window_that_pulled_two_files() -> (Arc<TalkativeSession>, MainTui) {
        let session = Arc::new(TalkativeSession {
            results: std::sync::Mutex::new(vec![files_from_bob(&[
                "@@x\\a.mp3",
                "@@x\\b.mp3",
            ])]),
            ..shared(vec![search_of("aphex twin", 2, 9_999)])
        });
        let mut tui = attach(session.clone());
        tui.state.selected_search_index = Some(0);
        tui.update_search_results();
        tui.update_search_results();
        (session, tui)
    }

    #[test]
    fn a_selected_search_with_nothing_new_is_not_pulled_across_again() {
        // The result set only accumulates, so an unchanged count means a
        // fetch would hand back what this window already holds — against a
        // daemon that is the entire result set over the socket, every frame,
        // forever.
        let (session, tui) = window_that_pulled_two_files();

        assert_eq!(tui.state.searches[0].results.len(), 2);
        assert_eq!(
            session
                .result_fetches
                .load(std::sync::atomic::Ordering::Relaxed),
            1,
            "the set crosses once; after that the unchanged count spares it"
        );
    }

    #[test]
    fn fresh_results_still_reach_a_window_that_skipped_a_fetch() {
        let (session, mut tui) = window_that_pulled_two_files();

        // A late responder lands in the session.
        *session.searches.lock().expect("not poisoned") =
            vec![search_of("aphex twin", 3, 9_999)];
        *session.results.lock().expect("not poisoned") =
            vec![files_from_bob(&["@@x\\a.mp3", "@@x\\b.mp3", "@@x\\c.mp3"])];
        tui.update_search_results();

        assert_eq!(
            tui.state.searches[0].results.len(),
            3,
            "a changed count must bring the new results across"
        );
    }

    #[test]
    fn a_local_window_fetches_results_without_counting_on_the_session() {
        // Locally nothing maintains known_files, so the count gate must stay
        // out of the way: a search whose count still reads zero has to get
        // its results anyway.
        let session = Arc::new(TalkativeSession {
            results: std::sync::Mutex::new(vec![files_from_bob(&[
                "@@x\\a.mp3",
                "@@x\\b.mp3",
            ])]),
            shared: false,
            ..TalkativeSession::default()
        });
        let mut tui = attach(session);
        tui.state.searches.push(crate::models::SearchEntry {
            query: "mine".to_string(),
            status: SearchStatus::Active,
            results: Vec::new(),
            known_files: 0,
            owned: true,
            start_time: std::time::Instant::now(),
            cancel_flag: Arc::new(std::sync::atomic::AtomicBool::new(false)),
        });

        tui.update_search_results();

        assert_eq!(
            tui.state.searches[0].results.len(),
            2,
            "a zero count matching zero held rows must not read as \
             nothing-to-fetch on a local session"
        );
    }

    #[test]
    fn a_window_with_its_own_session_is_left_alone() {
        // Nothing to share, so nothing is synced — and in particular the
        // completed transfers restored from disk, which the session has never
        // heard of, are not mistaken for cancelled and deleted.
        let mut tui = with_session(TalkativeSession {
            downloads: Vec::new(),
            shared: false,
            ..TalkativeSession::default()
        });
        tui.state.downloads.push(crate::models::DownloadEntry {
            download: queued("bob", "from-disk.mp3"),
            receiver: None,
        });

        tui.update_downloads();

        assert_eq!(
            tui.state.downloads.len(),
            1,
            "a local window's history must survive the tick"
        );
    }

    #[test]
    fn a_local_session_starts_from_its_own_snapshot_instead() {
        // Locally the session knows no history and the state file is the whole
        // record, so nothing is invented.
        assert!(tui(Vec::new()).state.messages.is_empty());
    }

    /// Attached, both folders in the settings form name paths on the
    /// *daemon's* machine: they go over the socket exactly as typed, and
    /// nothing is created or validated against this machine's filesystem.
    #[test]
    fn applying_settings_attached_pushes_both_folders_to_the_daemon() {
        let session = Arc::new(TalkativeSession {
            shared: true,
            ..TalkativeSession::default()
        });
        let scratch = tempfile::tempdir().expect("a temp dir");
        let config_path = scratch.path().join("config.toml");
        let mut tui = MainTui::new(
            session.clone(),
            "/daemon/old".to_string(),
            Duration::from_secs(1),
            None,
            Some(config_path.clone()),
        );

        let daemon_only = scratch.path().join("only-on-the-daemon");
        let daemon_dir = daemon_only.to_str().expect("utf-8").to_string();
        tui.open_settings();
        {
            let settings = tui.state.settings.as_mut().expect("open");
            settings.download_dir.clone_from(&daemon_dir);
            settings.share_dirs = vec!["/not-here-either".to_string()];
        }
        tui.apply_settings();

        assert_eq!(
            *session.download_dir_set.lock().expect("not poisoned"),
            Some(daemon_dir.clone()),
            "the folder is pushed to the daemon"
        );
        assert_eq!(
            *session.shares_set.lock().expect("not poisoned"),
            Some(vec!["/not-here-either".to_string()]),
            "shares go over as typed — the daemon's filesystem judges them"
        );
        assert!(
            !daemon_only.exists(),
            "no directory is created on this machine for a daemon's path"
        );
        assert_eq!(tui.download_dir, daemon_dir);

        let saved = crate::persist::config::FileConfig::load(&config_path)
            .expect("config readable");
        assert_eq!(
            saved.download_dir.as_deref(),
            Some(daemon_dir.as_str()),
            "the change still lands in config.toml for the next restart"
        );
    }

    /// Locally the same form works on this machine: the folder is created
    /// here and share paths that do not exist here are dropped before the
    /// session hears about them.
    #[test]
    fn applying_settings_locally_stays_on_this_machine() {
        let session = Arc::new(TalkativeSession {
            shared: false,
            ..TalkativeSession::default()
        });
        let scratch = tempfile::tempdir().expect("a temp dir");
        let config_path = scratch.path().join("config.toml");
        let share = scratch.path().join("music");
        std::fs::create_dir_all(&share).expect("share dir");
        let mut tui = MainTui::new(
            session.clone(),
            "/tmp".to_string(),
            Duration::from_secs(1),
            None,
            Some(config_path),
        );

        let downloads = scratch.path().join("downloads");
        let downloads_dir = downloads.to_str().expect("utf-8").to_string();
        tui.open_settings();
        {
            let settings = tui.state.settings.as_mut().expect("open");
            settings.download_dir.clone_from(&downloads_dir);
            settings.share_dirs = vec![
                share.to_str().expect("utf-8").to_string(),
                "/definitely-not-a-real-share".to_string(),
            ];
        }
        tui.apply_settings();

        // The seam is told in both modes; a local session accepts and takes
        // the directory per download instead of storing it.
        assert_eq!(
            *session.download_dir_set.lock().expect("not poisoned"),
            Some(downloads_dir.clone())
        );
        assert_eq!(
            *session.shares_set.lock().expect("not poisoned"),
            Some(vec![share.to_str().expect("utf-8").to_string()]),
            "paths missing on this machine are dropped before applying"
        );
        assert!(downloads.is_dir(), "the folder is created here");
        assert_eq!(tui.download_dir, downloads_dir);
    }
}