binocular/runtime/interactive/
bootstrap.rs

1use crate::app::AppEvent::{Input, LogAppend, Preview};
2use crate::app::{App, AppEvent, Mode};
3use crate::config::{LoadedAppConfig, PersistedLayout};
4use crate::infra::channel::{self, MapSender, Sender};
5use crate::infra::terminal::TerminalSessionGuard;
6use crate::output::render_selection_outputs;
7use crate::preview::{self, structured_log, PreviewRequest, PreviewSource};
8use crate::runtime::config::RunConfig;
9use crate::runtime::interactive::input::spawn_input_handler;
10use crate::runtime::interactive::r#loop::run_event_loop;
11use crate::runtime::startup::{self, StartupMode};
12use crate::search::controller::SearchController;
13use crate::search::types::SearchConfig;
14use crossterm::tty::IsTty;
15use ratatui::{backend::CrosstermBackend, Terminal};
16use std::io::{self, Write};
17
18pub fn run_interactive_with_configs(
19    run_config: RunConfig,
20    search_config: SearchConfig,
21    app_config: LoadedAppConfig,
22    persisted_layout: PersistedLayout,
23    log_max_entries: usize,
24) -> anyhow::Result<()> {
25    let prepared_input = startup::prepare_interactive_input_with_run_config(&run_config)?;
26    let mut terminal_session = TerminalSessionGuard::enter()?;
27    let mut terminal = build_terminal()?;
28    let picker = build_picker();
29
30    let (tx_main, rx_main) = channel::unbounded_default::<AppEvent>();
31    let (tx_preview_req, rx_preview_req) = channel::unbounded_default::<PreviewRequest>();
32    let (tx_cmd_noop, _rx_cmd_noop) = channel::unbounded_default();
33
34    let tx_input = MapSender::new(tx_main.clone(), Input);
35    let tx_preview_resp = MapSender::new(tx_main.clone(), |(source, content)| {
36        Preview(source, content)
37    });
38    let tx_log = MapSender::new(tx_main.clone(), |(path, entries)| LogAppend(path, entries));
39
40    let startup_mode = startup::classify_input_mode_with_run_config(&run_config);
41    let mut search_sessions = if matches!(startup_mode, StartupMode::InteractiveDirectDiff) {
42        None
43    } else {
44        Some(SearchController::from_search_config(
45            search_config.clone(),
46            prepared_input.stdin_items,
47            &tx_main,
48            !run_config.log,
49        ))
50    };
51    spawn_input_handler(tx_input);
52    preview::spawn_previewer(
53        rx_preview_req,
54        tx_preview_resp,
55        tx_log.clone(),
56        picker,
57        run_config.preview_command.clone(),
58        run_config.preview_delimiter.clone(),
59        log_max_entries,
60    );
61
62    if let Some(pipe) = prepared_input.log_pipe {
63        let _ = startup::spawn_log_stdin_reader(pipe, tx_log.clone());
64    }
65
66    if !prepared_input.log_files.is_empty() {
67        startup::spawn_log_file_watchers(&prepared_input.log_files, tx_log);
68    }
69
70    let mut app = App::from_configs(run_config, search_config, app_config);
71    initialize_app(&mut app, &persisted_layout, terminal.size()?.into());
72    prime_search_log_and_diff_state(&mut app, search_sessions.as_ref(), &tx_preview_req);
73
74    run_event_loop(
75        &mut app,
76        &mut terminal,
77        &mut terminal_session,
78        &rx_main,
79        &tx_preview_req,
80        &tx_cmd_noop,
81        &mut search_sessions,
82        log_max_entries,
83    )?;
84
85    crate::config::save_layout(&PersistedLayout {
86        panes_swapped: app.ui.layout.panes_swapped,
87        preview_percent: app.ui.layout.preview_percent,
88        search_bar_at_bottom: app.ui.layout.search_bar_at_bottom,
89        preview_hidden: app.ui.layout.preview_hidden,
90    });
91
92    if let Some(search_sessions) = search_sessions.as_mut() {
93        search_sessions.shutdown();
94    }
95    let selected_output = app.take_selected_output();
96    let rendered_output = render_selection_outputs(&selected_output, app.runtime.run.output_format);
97    drop(terminal);
98    drop(terminal_session);
99
100    if let Some(output) = rendered_output {
101        write_selection_output(&output, app.runtime.run.output_file.as_deref())?;
102    }
103
104    Ok(())
105}
106
107fn write_selection_output(output: &str, output_file: Option<&std::path::Path>) -> anyhow::Result<()> {
108    if let Some(path) = output_file {
109        std::fs::write(path, output)?;
110    } else {
111        let stdout = io::stdout();
112        let mut out = io::BufWriter::new(stdout.lock());
113        writeln!(out, "{}", output)?;
114    }
115
116    Ok(())
117}
118
119fn build_terminal() -> anyhow::Result<Terminal<CrosstermBackend<io::Stderr>>> {
120    let backend = CrosstermBackend::new(io::stderr());
121    Ok(Terminal::new(backend)?)
122}
123
124fn build_picker() -> ratatui_image::picker::Picker {
125    let mut picker = if io::stdout().is_tty() {
126        ratatui_image::picker::Picker::from_query_stdio()
127            .unwrap_or_else(|_| ratatui_image::picker::Picker::halfblocks())
128    } else {
129        ratatui_image::picker::Picker::halfblocks()
130    };
131
132    if std::env::var("TERM_PROGRAM").unwrap_or_default() == "iTerm.app"
133        || std::env::var("LC_TERMINAL").unwrap_or_default() == "iTerm2"
134    {
135        picker.set_protocol_type(ratatui_image::picker::ProtocolType::Iterm2);
136    }
137
138    picker
139}
140
141fn initialize_app(
142    app: &mut App,
143    persisted_layout: &PersistedLayout,
144    terminal_area: ratatui::layout::Rect,
145) {
146    app.ui.layout.panes_swapped = persisted_layout.panes_swapped;
147    app.ui.layout.preview_percent = persisted_layout.preview_percent;
148    app.ui.layout.search_bar_at_bottom = persisted_layout.search_bar_at_bottom;
149    app.ui.layout.preview_hidden = persisted_layout.preview_hidden;
150    app.set_terminal_area(terminal_area);
151    app.refresh_viewports();
152}
153
154fn prime_search_log_and_diff_state(
155    app: &mut App,
156    search_sessions: Option<&SearchController>,
157    tx_preview_req: &channel::DefaultSender<PreviewRequest>,
158) {
159    if app.runtime.run.log {
160        let path = if app.runtime.run.log_files.is_empty() {
161            structured_log::STDIN_STREAM_PATH.to_string()
162        } else {
163            app.runtime.run.log_files[0].display().to_string()
164        };
165        structured_log::initialize_empty_stream(app, path, structured_log::LogFormat::Jsonl);
166        return;
167    }
168
169    if let Some(diff_paths) = app.runtime.run.diff.as_ref() {
170        let [left, right] = diff_paths;
171        let left = left.display().to_string();
172        let right = right.display().to_string();
173        let source = PreviewSource::Diff {
174            left: left.clone(),
175            right: right.clone(),
176        };
177        app.preview_session.preview.source = Some(source.clone());
178        app.ui.mode = Mode::Preview;
179        app.ui.layout.preview_fullscreen = true;
180        let _ = tx_preview_req.send(PreviewRequest::Diff {
181            source,
182            left,
183            right,
184        });
185        return;
186    }
187
188    if !app.search_session.query.text.is_empty() {
189        if let Some(search_sessions) = search_sessions {
190            search_sessions.send_query(&app.search_session.query.text);
191        }
192    }
193}
binocular/runtime/interactive/bootstrap.rs

binocular/runtime/interactive/
bootstrap.rs
