kimun_notes/components/dialogs/

save_search_dialog.rs

use ratatui::Frame;
use ratatui::layout::{Constraint, Direction, Layout, Rect};
use ratatui::style::Style;
use ratatui::widgets::Paragraph;

use crate::components::event_state::EventState;
use crate::components::events::{AppEvent, AppTx, InputEvent, SaveSource};
use crate::components::panel::{ModalSpec, modal_chrome};
use crate::components::single_line_input::{InputOutcome, SingleLineInput};
use crate::settings::themes::Theme;

/// What submitting the dialog will do with the current name field — drives
/// the live hint line so an overwrite is never silent.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum SaveHint {
    /// The name matches the saved search the query came from (the breadcrumb
    /// provenance): submitting updates it in place.
    Update(String),
    /// The name matches a different existing saved search: submitting
    /// replaces that search's query. Rendered as a warning.
    Overwrite(String),
    /// A fresh name: submitting creates a new saved search.
    SaveNew,
    /// The name field is empty: submitting saves a new search named after the
    /// query itself (the query-as-name fallback).
    SaveNewAsQuery(String),
    /// The existing names have not loaded yet, so save-new vs overwrite is
    /// unknown (the provenance Update case never waits — it is checked
    /// synchronously). Submitting still saves.
    Pending,
}

pub struct SaveSearchDialog {
    /// The query being saved (read-only context).
    pub query: String,
    /// User-supplied name for the saved search, pre-filled with the
    /// breadcrumb provenance when the query came from a saved search.
    name: SingleLineInput,
    /// The saved-search name the query came from (breadcrumb provenance).
    provenance: Option<String>,
    /// The surface the query was sourced from; echoed on submit so the
    /// editor re-pins by identity rather than by comparing query text.
    source: SaveSource,
    /// Existing saved-search names, loaded asynchronously after open (see
    /// [`AppEvent::SavedSearchNamesLoaded`]). `None` until the load lands —
    /// the hint shows [`SaveHint::Pending`] rather than guessing "save new".
    existing: Option<Vec<String>>,
}

impl SaveSearchDialog {
    pub fn new(query: String, provenance: Option<String>, source: SaveSource) -> Self {
        let name = match &provenance {
            Some(p) => SingleLineInput::with_value(p.clone()),
            None => SingleLineInput::new(),
        };
        Self {
            query,
            name,
            provenance,
            source,
            existing: None,
        }
    }

    /// Supply the vault's existing saved-search names (async load result).
    pub fn set_existing_names(&mut self, names: Vec<String>) {
        self.existing = Some(names);
    }

    /// The name a submit would save under: the typed name, or the trimmed
    /// query when the field is empty (the query-as-name fallback).
    fn effective_name(&self) -> &str {
        let typed = self.name.value().trim();
        if typed.is_empty() {
            self.query.trim()
        } else {
            typed
        }
    }

    /// What submitting right now would do. Name matching delegates to core's
    /// `saved_search_name_matches` — the same rule `save_search` applies on
    /// write, so the hint can never disagree with the actual save outcome.
    pub fn hint(&self) -> SaveHint {
        let matches = kimun_core::saved_search_name_matches;
        let effective = self.effective_name();
        if let Some(p) = &self.provenance
            && matches(p, effective)
        {
            return SaveHint::Update(p.clone());
        }
        let Some(existing) = &self.existing else {
            return SaveHint::Pending;
        };
        if let Some(name) = existing.iter().find(|n| matches(n, effective)) {
            return SaveHint::Overwrite(name.clone());
        }
        if self.name.value().trim().is_empty() {
            SaveHint::SaveNewAsQuery(self.query.clone())
        } else {
            SaveHint::SaveNew
        }
    }

    pub fn handle_input(&mut self, event: &InputEvent, tx: &AppTx) -> EventState {
        let InputEvent::Key(key) = event else {
            return EventState::NotConsumed;
        };
        match self.name.handle_key(key) {
            InputOutcome::Submit => {
                tx.send(AppEvent::SaveSearchConfirmed {
                    name: self.effective_name().to_string(),
                    query: self.query.clone(),
                    source: self.source,
                })
                .ok();
                tx.send(AppEvent::CloseOverlay).ok();
                EventState::Consumed
            }
            InputOutcome::Cancel => {
                tx.send(AppEvent::CloseOverlay).ok();
                EventState::Consumed
            }
            InputOutcome::Changed | InputOutcome::Consumed => EventState::Consumed,
            InputOutcome::NotConsumed => EventState::NotConsumed,
        }
    }

    pub fn render(&mut self, f: &mut Frame, rect: Rect, theme: &Theme, _focused: bool) {
        let popup_area = super::fixed_centered_rect(62, 9, rect);

        let fg = theme.fg.to_ratatui();
        let gray = theme.gray.to_ratatui();
        let bg = theme.bg_panel.to_ratatui();

        let inner = modal_chrome(
            f,
            popup_area,
            theme,
            ModalSpec {
                title: Some(" Save search "),
                border: Some(Style::default().fg(gray)),
                ..Default::default()
            },
        );

        let rows = Layout::default()
            .direction(Direction::Vertical)
            .constraints([
                Constraint::Length(1), // 0: spacer
                Constraint::Length(1), // 1: query (read-only context)
                Constraint::Length(1), // 2: separator
                Constraint::Length(1), // 3: name input
                Constraint::Length(1), // 4: spacer
                Constraint::Length(1), // 5: hint
                Constraint::Min(0),    // 6: remainder
            ])
            .split(inner);

        // Row 1: read-only query context in muted style.
        f.render_widget(
            Paragraph::new(format!("  Query: {}", self.query))
                .style(Style::default().fg(gray).bg(bg)),
            rows[1],
        );

        super::render_separator(f, rows[2], gray, bg);

        // Row 3: name input with a "Name: " prefix.
        let prefix = "  Name: ";
        let prefix_len = prefix.len() as u16;
        f.render_widget(
            Paragraph::new(prefix).style(Style::default().fg(gray).bg(bg)),
            rows[3],
        );
        self.name
            .render(f, rows[3], Style::default().fg(fg).bg(bg), prefix_len, true);

        // Row 5: live hint — what Enter will do with the current name.
        // Pending renders dimmed (enter_active = false) until names load.
        let (action, warn, pending) = match self.hint() {
            SaveHint::Update(name) => (format!("Update '{name}'"), false, false),
            SaveHint::Overwrite(name) => (format!("Overwrite '{name}'"), true, false),
            SaveHint::SaveNew => ("Save new".to_string(), false, false),
            SaveHint::SaveNewAsQuery(query) => (format!("Save new: '{query}'"), false, false),
            SaveHint::Pending => ("Save".to_string(), false, true),
        };
        let enter_fg = if warn { theme.yellow.to_ratatui() } else { fg };
        super::render_confirm_hint(
            f,
            rows[5],
            &format!("  [Enter] {action}"),
            !pending,
            enter_fg,
            gray,
            bg,
        );
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::components::events::{AppEvent, InputEvent};
    use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
    use tokio::sync::mpsc::unbounded_channel;

    fn key(code: KeyCode) -> InputEvent {
        InputEvent::Key(KeyEvent::new(code, KeyModifiers::NONE))
    }

    fn dialog(query: &str, provenance: Option<&str>) -> SaveSearchDialog {
        SaveSearchDialog::new(
            query.to_string(),
            provenance.map(str::to_string),
            SaveSource::QueryPanel,
        )
    }

    /// Drain the channel and return the `SaveSearchConfirmed` payload, if any.
    fn confirmed(
        rx: &mut tokio::sync::mpsc::UnboundedReceiver<AppEvent>,
    ) -> Option<(String, String, SaveSource)> {
        let mut found = None;
        while let Ok(e) = rx.try_recv() {
            if let AppEvent::SaveSearchConfirmed {
                name,
                query,
                source,
            } = e
            {
                found = Some((name, query, source));
            }
        }
        found
    }

    #[test]
    fn submit_emits_save_event_with_typed_name() {
        let mut d = dialog("<{note}", None);
        let (tx, mut rx) = unbounded_channel();
        for ch in ['l', 'i', 'n', 'k', 's'] {
            d.handle_input(&key(KeyCode::Char(ch)), &tx);
        }
        d.handle_input(&key(KeyCode::Enter), &tx);
        let (name, query, source) = confirmed(&mut rx).expect("SaveSearchConfirmed emitted");
        assert_eq!(name, "links");
        assert_eq!(query, "<{note}");
        assert_eq!(source, SaveSource::QueryPanel);
    }

    #[test]
    fn submit_carries_the_dialog_source_through() {
        let mut d = SaveSearchDialog::new("#todo".to_string(), None, SaveSource::NoteBrowser);
        let (tx, mut rx) = unbounded_channel();
        d.handle_input(&key(KeyCode::Enter), &tx);
        let (_, _, source) = confirmed(&mut rx).expect("emitted");
        assert_eq!(source, SaveSource::NoteBrowser);
    }

    #[test]
    fn submit_empty_name_falls_back_to_query() {
        let mut d = dialog("#todo", None);
        let (tx, mut rx) = unbounded_channel();
        d.handle_input(&key(KeyCode::Enter), &tx);
        let (name, query, _) = confirmed(&mut rx).expect("emitted");
        assert_eq!(name, "#todo"); // empty → query used as name
        assert_eq!(query, "#todo");
    }

    #[test]
    fn empty_name_fallback_trims_the_query() {
        // The typed branch trims, so the fallback must too — otherwise a
        // padded query saves under a whitespace-padded, unmatchable name.
        let mut d = dialog("#todo ", None);
        let (tx, mut rx) = unbounded_channel();
        d.handle_input(&key(KeyCode::Enter), &tx);
        let (name, query, _) = confirmed(&mut rx).expect("emitted");
        assert_eq!(name, "#todo"); // trimmed
        assert_eq!(query, "#todo "); // query itself stays verbatim
    }

    #[test]
    fn provenance_prefills_name_so_plain_enter_updates() {
        let mut d = dialog("#todo and #urgent", Some("todo"));
        let (tx, mut rx) = unbounded_channel();
        d.handle_input(&key(KeyCode::Enter), &tx);
        let (name, query, _) = confirmed(&mut rx).expect("emitted");
        assert_eq!(name, "todo"); // provenance pre-filled, untouched
        assert_eq!(query, "#todo and #urgent");
    }

    #[test]
    fn hint_updates_when_name_matches_provenance_even_before_names_load() {
        // The provenance is passed synchronously, so the Update hint must
        // not wait for the async existing-names load.
        let d = dialog("#todo", Some("todo"));
        assert_eq!(d.hint(), SaveHint::Update("todo".into()));
    }

    #[test]
    fn hint_is_pending_until_names_load() {
        // Before the async load lands, the dialog cannot distinguish a fresh
        // name from an overwrite — it must say neither, not "Save new".
        let mut d = dialog("#todo", None);
        let (tx, _rx) = unbounded_channel();
        d.handle_input(&key(KeyCode::Char('x')), &tx);
        assert_eq!(d.hint(), SaveHint::Pending);
        d.set_existing_names(vec![]);
        assert_eq!(d.hint(), SaveHint::SaveNew);
    }

    #[test]
    fn hint_matches_existing_names_case_insensitively() {
        let mut d = dialog("#todo", None);
        d.set_existing_names(vec!["Todo".into()]);
        let (tx, _rx) = unbounded_channel();
        for ch in ['t', 'O', 'd', 'O'] {
            d.handle_input(&key(KeyCode::Char(ch)), &tx);
        }
        // Same rule core uses on save: ASCII case-insensitive name match.
        assert_eq!(d.hint(), SaveHint::Overwrite("Todo".into()));
    }

    #[test]
    fn hint_overwrites_when_name_matches_another_existing_search() {
        let mut d = dialog("#todo", Some("todo"));
        d.set_existing_names(vec!["todo".into(), "other".into()]);
        let (tx, _rx) = unbounded_channel();
        // Clear the pre-filled "todo" and type "other".
        for _ in 0..4 {
            d.handle_input(&key(KeyCode::Backspace), &tx);
        }
        for ch in ['o', 't', 'h', 'e', 'r'] {
            d.handle_input(&key(KeyCode::Char(ch)), &tx);
        }
        assert_eq!(d.hint(), SaveHint::Overwrite("other".into()));
    }

    #[test]
    fn hint_saves_new_for_a_fresh_name() {
        let mut d = dialog("#todo", None);
        d.set_existing_names(vec!["other".into()]);
        let (tx, _rx) = unbounded_channel();
        for ch in ['f', 'r', 'e', 's', 'h'] {
            d.handle_input(&key(KeyCode::Char(ch)), &tx);
        }
        assert_eq!(d.hint(), SaveHint::SaveNew);
    }

    #[test]
    fn hint_empty_name_shows_query_as_name_fallback() {
        let mut d = dialog("#todo", None);
        d.set_existing_names(vec![]);
        assert_eq!(d.hint(), SaveHint::SaveNewAsQuery("#todo".into()));
    }

    #[test]
    fn hint_empty_name_with_colliding_query_warns_overwrite() {
        let mut d = dialog("#todo", None);
        // A saved search literally named "#todo" exists; the query-as-name
        // fallback would overwrite it.
        d.set_existing_names(vec!["#todo".into()]);
        assert_eq!(d.hint(), SaveHint::Overwrite("#todo".into()));
    }
}