Merge pull request #163 from williamjameshandley/feature/stdout-quiet-output

Add stdout output and quiet mode support

Author: ODAncona

crates/code2prompt-core/src/template.rs

@@ -68,27 +68,37 @@ pub fn render_template(
     Ok(rendered.trim().to_string())
 }
 
-/// Writes the rendered template to a specified output file.
+/// Writes the rendered template to a specified output file or stdout.
 ///
 /// # Arguments
 ///
-/// * `output_path` - The path to the output file.
+/// * `output_path` - The path to the output file, or "-" for stdout.
 /// * `rendered` - The rendered template string.
+/// * `quiet` - If true, suppress success messages.
 ///
 /// # Returns
 ///
 /// * `Result<()>` - An empty result indicating success or an error.
-pub fn write_to_file(output_path: &str, rendered: &str) -> Result<()> {
-    let file = std::fs::File::create(output_path)?;
-    let mut writer = std::io::BufWriter::new(file);
-    write!(writer, "{}", rendered)?;
-    println!(
-        "{}{}{} {}",
-        "[".bold().white(),
-        "✓".bold().green(),
-        "]".bold().white(),
-        format!("Prompt written to file: {}", output_path).green()
-    );
+pub fn write_to_file(output_path: &str, rendered: &str, quiet: bool) -> Result<()> {
+    if output_path == "-" {
+        // Write to stdout
+        print!("{}", rendered);
+        std::io::stdout().flush()?;
+    } else {
+        // Write to file
+        let file = std::fs::File::create(output_path)?;
+        let mut writer = std::io::BufWriter::new(file);
+        write!(writer, "{}", rendered)?;
+        if !quiet {
+            println!(
+                "{}{}{} {}",
+                "[".bold().white(),
+                "✓".bold().green(),
+                "]".bold().white(),
+                format!("Prompt written to file: {}", output_path).green()
+            );
+        }
+    }
     Ok(())
 }
 

crates/code2prompt/src/args.rs

@@ -98,6 +98,10 @@ pub struct Cli {
     #[clap(long)]
     pub sort: Option<String>,
 
+    /// Suppress progress and success messages
+    #[clap(short, long)]
+    pub quiet: bool,
+
     /// Display a visual token map of files (similar to disk usage tools)
     #[clap(long)]
     pub token_map: bool,

crates/code2prompt/src/main.rs

@@ -34,6 +34,10 @@ fn main() -> Result<()> {
         std::process::exit(1);
     }
 
+    // Disable clipboard when outputting to stdout (unless clipboard is explicitly enabled)
+    let no_clipboard = args.no_clipboard || 
+        args.output_file.as_ref().map_or(false, |f| f == "-");
+
     // ~~~ Clipboard Daemon ~~~
     #[cfg(target_os = "linux")]
     {
@@ -122,51 +126,73 @@ fn main() -> Result<()> {
 
     // ~~~ Code2Prompt ~~~
     let mut session = Code2PromptSession::new(configuration.build()?);
-    let spinner = setup_spinner("Traversing directory and building tree...");
+    let spinner = if !args.quiet {
+        Some(setup_spinner("Traversing directory and building tree..."))
+    } else {
+        None
+    };
 
     // ~~~ Gather Repository Data ~~~
     // Load Codebase
     session.load_codebase().unwrap_or_else(|e| {
-        spinner.finish_with_message("Failed!".red().to_string());
+        if let Some(ref s) = spinner {
+            s.finish_with_message("Failed!".red().to_string());
+        }
         error!("Failed to build directory tree: {}", e);
         std::process::exit(1);
     });
-    spinner.finish_with_message("Done!".green().to_string());
+    if let Some(ref s) = spinner {
+        s.finish_with_message("Done!".green().to_string());
+    }
 
     // ~~~ Git Related ~~~
     // Git Diff
     if session.config.diff_enabled {
-        spinner.set_message("Generating git diff...");
+        if let Some(ref s) = spinner {
+            s.set_message("Generating git diff...");
+        }
         session.load_git_diff().unwrap_or_else(|e| {
-            spinner.finish_with_message("Failed!".red().to_string());
+            if let Some(ref s) = spinner {
+                s.finish_with_message("Failed!".red().to_string());
+            }
             error!("Failed to generate git diff: {}", e);
             std::process::exit(1);
         });
     }
 
     // Load Git diff between branches if provided
     if session.config.diff_branches.is_some() {
-        spinner.set_message("Generating git diff between two branches...");
+        if let Some(ref s) = spinner {
+            s.set_message("Generating git diff between two branches...");
+        }
         session
             .load_git_diff_between_branches()
             .unwrap_or_else(|e| {
-                spinner.finish_with_message("Failed!".red().to_string());
+                if let Some(ref s) = spinner {
+                    s.finish_with_message("Failed!".red().to_string());
+                }
                 error!("Failed to generate git diff: {}", e);
                 std::process::exit(1);
             });
     }
 
     // Load Git log between branches if provided
     if session.config.log_branches.is_some() {
-        spinner.set_message("Generating git log between two branches...");
+        if let Some(ref s) = spinner {
+            s.set_message("Generating git log between two branches...");
+        }
         session.load_git_log_between_branches().unwrap_or_else(|e| {
-            spinner.finish_with_message("Failed!".red().to_string());
+            if let Some(ref s) = spinner {
+                s.finish_with_message("Failed!".red().to_string());
+            }
             error!("Failed to generate git log: {}", e);
             std::process::exit(1);
         });
     }
 
-    spinner.finish_with_message("Done!".green().to_string());
+    if let Some(ref s) = spinner {
+        s.finish_with_message("Done!".green().to_string());
+    }
 
     // ~~~ Template ~~~
 
@@ -192,14 +218,16 @@ fn main() -> Result<()> {
     };
     let model_info = rendered.model_info;
 
-    println!(
-        "{}{}{} Token count: {}, Model info: {}",
-        "[".bold().white(),
-        "i".bold().blue(),
-        "]".bold().white(),
-        formatted_token_count,
-        model_info
-    );
+    if !args.quiet {
+        println!(
+            "{}{}{} Token count: {}, Model info: {}",
+            "[".bold().white(),
+            "i".bold().blue(),
+            "]".bold().white(),
+            formatted_token_count,
+            model_info
+        );
+    }
 
     // ~~~ Token Map Display ~~~
     if args.token_map {
@@ -241,7 +269,7 @@ fn main() -> Result<()> {
     }
 
     // ~~~ Copy to Clipboard ~~~
-    if !args.no_clipboard {
+    if !no_clipboard {
         #[cfg(target_os = "linux")]
         {
             use clipboard::spawn_clipboard_daemon;
@@ -252,22 +280,27 @@ fn main() -> Result<()> {
             use crate::clipboard::copy_text_to_clipboard;
             match copy_text_to_clipboard(&rendered.prompt) {
                 Ok(_) => {
-                    println!(
-                        "{}{}{} {}",
-                        "[".bold().white(),
-                        "✓".bold().green(),
-                        "]".bold().white(),
-                        "Copied to clipboard successfully.".green()
-                    );
+                    if !args.quiet {
+                        println!(
+                            "{}{}{} {}",
+                            "[".bold().white(),
+                            "✓".bold().green(),
+                            "]".bold().white(),
+                            "Copied to clipboard successfully.".green()
+                        );
+                    }
                 }
                 Err(e) => {
-                    eprintln!(
-                        "{}{}{} {}",
-                        "[".bold().white(),
-                        "!".bold().red(),
-                        "]".bold().white(),
-                        format!("Failed to copy to clipboard: {}", e).red()
-                    );
+                    if !args.quiet {
+                        eprintln!(
+                            "{}{}{} {}",
+                            "[".bold().white(),
+                            "!".bold().red(),
+                            "]".bold().white(),
+                            format!("Failed to copy to clipboard: {}", e).red()
+                        );
+                    }
+                    // Always print the prompt if clipboard fails, regardless of quiet mode
                     println!("{}", &rendered.prompt);
                 }
             }
@@ -276,7 +309,7 @@ fn main() -> Result<()> {
 
     // ~~~ Output File ~~~
     if let Some(output_path) = &args.output_file {
-        write_to_file(output_path, &rendered.prompt)?;
+        write_to_file(output_path, &rendered.prompt, args.quiet)?;
     }
 
     Ok(())