Add --all-ranks CLI infrastructure and core processing logic

src/cli.rs

@@ -47,10 +47,18 @@ pub struct Cli {
     /// For inductor provenance tracking highlighter
     #[arg(short, long)]
     inductor_provenance: bool,
+    /// Parse all ranks and generate a single unified page
+    #[arg(long)]
+    all_ranks: bool,
 }
 
 fn main() -> anyhow::Result<()> {
     let cli = Cli::parse();
+
+    if cli.all_ranks {
+        return handle_all_ranks(&cli);
+    }
+
     let path = if cli.latest {
         let input_path = cli.path;
         // Path should be a directory
@@ -92,11 +100,12 @@ fn main() -> anyhow::Result<()> {
         strict: cli.strict,
         strict_compile_id: cli.strict_compile_id,
         custom_parsers: Vec::new(),
-        custom_header_html: cli.custom_header_html,
+        custom_header_html: cli.custom_header_html.clone(),
         verbose: cli.verbose,
         plain_text: cli.plain_text,
         export: cli.export,
         inductor_provenance: cli.inductor_provenance,
+        all_ranks: cli.all_ranks,
     };
 
     let output = parse_path(&path, config)?;
@@ -114,3 +123,132 @@ fn main() -> anyhow::Result<()> {
     }
     Ok(())
 }
+
+// handle_all_ranks function with placeholder landing page
+fn handle_all_ranks(cli: &Cli) -> anyhow::Result<()> {
+    let input_path = &cli.path;
+
+    if !input_path.is_dir() {
+        bail!(
+            "Input path {} must be a directory when using --all-ranks",
+            input_path.display()
+        );
+    }
+
+    let out_path = &cli.out;
+
+    if out_path.exists() {
+        if !cli.overwrite {
+            bail!(
+                "Directory {} already exists, use -o OUTDIR to write to another location or pass --overwrite to overwrite the old contents",
+                out_path.display()
+            );
+        }
+        fs::remove_dir_all(&out_path)?;
+    }
+    fs::create_dir(&out_path)?;
+
+    // Find all rank log files in the directory
+    let rank_files: Vec<_> = std::fs::read_dir(input_path)
+        .with_context(|| format!("Couldn't access directory {}", input_path.display()))?
+        .flatten()
+        .filter(|entry| {
+            let path = entry.path();
+            path.is_file()
+                && path
+                    .file_name()
+                    .and_then(|name| name.to_str())
+                    .map(|name| name.contains("rank_") && name.ends_with(".log"))
+                    .unwrap_or(false)
+        })
+        .collect();
+
+    if rank_files.is_empty() {
+        bail!(
+            "No rank log files found in directory {}",
+            input_path.display()
+        );
+    }
+
+    let mut rank_links = Vec::new();
+
+    // Process each rank file
+    for rank_file in rank_files {
+        let rank_path = rank_file.path();
+        let rank_name = rank_path
+            .file_stem()
+            .and_then(|name| name.to_str())
+            .unwrap_or("unknown");
+
+        // Extract rank number from filename
+        let rank_num = if let Some(pos) = rank_name.find("rank_") {
+            let after_rank = &rank_name[pos + 5..];
+            after_rank
+                .chars()
+                .take_while(|c| c.is_ascii_digit())
+                .collect::<String>()
+        } else {
+            "unknown".to_string()
+        };
+
+        println!(
+            "Processing rank {} from file: {}",
+            rank_num,
+            rank_path.display()
+        );
+
+        // Create subdirectory for this rank
+        let rank_out_dir = out_path.join(format!("rank_{rank_num}"));
+        fs::create_dir(&rank_out_dir)?;
+
+        let config = ParseConfig {
+            strict: cli.strict,
+            strict_compile_id: cli.strict_compile_id,
+            custom_parsers: Vec::new(),
+            custom_header_html: cli.custom_header_html.clone(),
+            verbose: cli.verbose,
+            plain_text: cli.plain_text,
+            export: cli.export,
+            inductor_provenance: cli.inductor_provenance,
+            all_ranks: false,
+        };
+
+        let output = parse_path(&rank_path, config)?;
+
+        // Write output files to rank subdirectory
+        for (filename, content) in output {
+            let out_file = rank_out_dir.join(filename);
+            if let Some(dir) = out_file.parent() {
+                fs::create_dir_all(dir)?;
+            }
+            fs::write(out_file, content)?;
+        }
+
+        // Add link to this rank's page
+        rank_links.push((rank_num.clone(), format!("rank_{rank_num}/index.html")));
+    }
+
+    // Sort rank links by rank number
+    rank_links.sort_by(|a, b| {
+        let a_num: i32 = a.0.parse().unwrap_or(999);
+        let b_num: i32 = b.0.parse().unwrap_or(999);
+        a_num.cmp(&b_num)
+    });
+
+    // Core logic complete - no HTML generation yet
+    // TODO - Add landing page HTML generation using template system
+
+    println!(
+        "Generated multi-rank report with {} ranks",
+        rank_links.len()
+    );
+    println!("Individual rank reports available in:");
+    for (rank_num, _) in &rank_links {
+        println!("  - rank_{}/index.html", rank_num);
+    }
+
+    // No browser opening since no landing page yet
+    // TODO - Generate landing page and open browser
+
+    Ok(())
+}

src/lib.rs

@@ -37,6 +37,7 @@ pub struct ParseConfig {
     pub plain_text: bool,
     pub export: bool,
     pub inductor_provenance: bool,
+    pub all_ranks: bool,
 }
 
 impl Default for ParseConfig {
@@ -50,6 +51,7 @@ impl Default for ParseConfig {
             plain_text: false,
             export: false,
             inductor_provenance: false,
+            all_ranks: false,
         }
     }
 }