diff options
| -rw-r--r-- | src/main.rs | 87 |
1 files changed, 48 insertions, 39 deletions
diff --git a/src/main.rs b/src/main.rs index 8489c57..b9a8821 100644 --- a/src/main.rs +++ b/src/main.rs @@ -6,7 +6,7 @@ use crossterm::{ }; use ratatui::{prelude::*, widgets::*}; use std::{ - io::{self, stdout}, + io::{self, stdout, Write}, str, }; @@ -20,9 +20,14 @@ use nix::unistd::Pid; struct Args { /// Command to execute from ptrace command: String, + /// Write the output to a file instead of the standard output #[arg(short = 'f', long = "file")] file_to_print: Option<String>, + + /// If defined, it hides the TUI + #[arg(long = "no-tui", default_value_t = false)] + no_tui: bool, } struct UI { @@ -44,12 +49,6 @@ impl UI { /// Create a fork of the program and execute the process in the child. Parent gets the pid /// value and trace it. fn main() -> anyhow::Result<()> { - enable_raw_mode()?; - stdout().execute(EnterAlternateScreen)?; - let mut terminal = Terminal::new(CrosstermBackend::new(stdout()))?; - - let mut ui = UI::new(); - let args = Args::parse(); let pid = match fork() { @@ -58,40 +57,50 @@ fn main() -> anyhow::Result<()> { Err(err) => panic!("fork() failed: {err}"), }; let output = trace(pid, args.file_to_print)?; - let lines = str::from_utf8(&output)?; - ui.max_lines = lines.split('\n').count(); + let lines = str::from_utf8(&output)?.trim(); - let mut should_quit = false; - while !should_quit { - ui.height = terminal.get_frame().size().height as usize; - terminal.draw(move |frame| { - let size = frame.size(); - frame.render_widget( - Paragraph::new(lines) - .block( - Block::default() - .border_style(Style::default().fg(Color::Yellow)) - .title(format!("[{pid}]")) - .title( - block::Title::from(format!( - "[lines {}-{}]", - ui.scroll, - ui.scroll + ui.height - )) - .position(block::Position::Bottom) - .alignment(Alignment::Right), - ) - .borders(Borders::ALL), - ) - .scroll((ui.scroll as u16, 0)), - size, - ); - })?; - should_quit = handle_events(&mut ui)?; - } + if !args.no_tui { + enable_raw_mode()?; + stdout().execute(EnterAlternateScreen)?; + let mut terminal = Terminal::new(CrosstermBackend::new(stdout()))?; - disable_raw_mode()?; - stdout().execute(LeaveAlternateScreen)?; + let mut ui = UI::new(); + ui.max_lines = lines.split('\n').count() + 1; + + let mut should_quit = false; + while !should_quit { + ui.height = terminal.get_frame().size().height as usize; + terminal.draw(move |frame| { + let size = frame.size(); + frame.render_widget( + Paragraph::new(lines) + .block( + Block::default() + .border_style(Style::default().fg(Color::Yellow)) + .title(format!("[{pid}]")) + .title( + block::Title::from(format!( + "[lines {}-{}]", + ui.scroll, + ui.scroll + ui.height + )) + .position(block::Position::Bottom) + .alignment(Alignment::Right), + ) + .borders(Borders::ALL), + ) + .scroll((ui.scroll as u16, 0)), + size, + ); + })?; + should_quit = handle_events(&mut ui)?; + } + + disable_raw_mode()?; + stdout().execute(LeaveAlternateScreen)?; + } else { + writeln!(io::stdout(), "{lines}")?; + } Ok(()) } |