flop_cli/

finder.rs

use anyhow::{Context, Result};
use regex::Regex;
use std::fs;
use std::path::Path;
use walkdir::WalkDir;

use crate::types::Match;

pub fn find_debug_printfs(
    path: &Path,
    find_commented: bool,
    detect_all: bool,
) -> Result<Vec<Match>> {
    let mut matches = Vec::new();

    // Pattern to match C printf-like functions (multiline support with (?s))
    // Match from function name to closing paren, then optional whitespace and semicolon
    let c_functions_pattern = if detect_all {
        // Match all output functions regardless of content
        Regex::new(
            r"(?s)(printf|fprintf|sprintf|snprintf|printf_debug|dprintf|puts|fputs|fputc|putchar|fputchar|write|perror)\s*\([^)]*\)\s*;",
        )?
    } else {
        // Match only those with "debug" or "DEBUG"
        Regex::new(
            r"(?s)(printf|fprintf|sprintf|snprintf|printf_debug|dprintf|puts|fputs|fputc|putchar|fputchar|write|perror)\s*\([^)]*?(debug|DEBUG)[^)]*\)\s*;",
        )?
    };

    // Pattern to match C++ streams (multiline support with (?s))
    let cpp_stream_pattern = if detect_all {
        // Match all stream output regardless of content
        Regex::new(r"(?s)(std::cout|std::cerr|std::clog)\s*<<[^;]*?;")?
    } else {
        // Match only those with "debug" or "DEBUG"
        Regex::new(r"(?s)(std::cout|std::cerr|std::clog)\s*<<[^;]*?(debug|DEBUG)[^;]*?;")?
    };

    // Pattern to match Rust macros (multiline support with (?s))
    // Match macro_name!(...); where ... can contain anything except unbalanced parens
    let rust_macro_pattern = if detect_all {
        // Match all Rust output macros regardless of content
        Regex::new(r"(?s)(println!|eprintln!|print!|eprint!|dbg!)\s*\([^)]*\)\s*;")?
    } else {
        // Match dbg! always, and other macros only if they contain "debug" or "DEBUG"
        Regex::new(
            r"(?s)dbg!\s*\([^)]*\)\s*;|(println!|eprintln!|print!|eprint!)\s*\([^)]*?(debug|DEBUG)[^)]*\)\s*;",
        )?
    };

    // Pattern to match Java output statements (multiline support with (?s))
    let java_pattern = if detect_all {
        // Match all Java output statements regardless of content
        Regex::new(
            r"(?s)(System\.out\.println|System\.out\.printf|System\.out\.print|System\.err\.println|System\.err\.printf|System\.err\.print)\s*\([^)]*\)\s*;",
        )?
    } else {
        // Match only those with "debug" or "DEBUG"
        Regex::new(
            r"(?s)(System\.out\.println|System\.out\.printf|System\.out\.print|System\.err\.println|System\.err\.printf|System\.err\.print)\s*\([^)]*?(debug|DEBUG)[^)]*\)\s*;",
        )?
    };

    // Pattern to match Go output statements (multiline support with (?s))
    // Go doesn't require semicolons, so we match just the closing paren
    let go_pattern = if detect_all {
        // Match all Go output statements regardless of content
        Regex::new(
            r"(?s)(fmt\.Println|fmt\.Printf|fmt\.Print|fmt\.Fprintln|fmt\.Fprintf|fmt\.Fprint|log\.Println|fmt\.Printf|log\.Print|log\.Fatal|log\.Fatalf|log\.Fatalln|log\.Panic|log\.Panicf|log\.Panicln)\s*\([^)]*\)",
        )?
    } else {
        // Match only those with "debug" or "DEBUG"
        Regex::new(
            r"(?s)(fmt\.Println|fmt\.Printf|fmt\.Print|fmt\.Fprintln|fmt\.Fprintf|fmt\.Fprint|log\.Println|log\.Printf|log\.Print|log\.Fatal|log\.Fatalf|log\.Fatalln|log\.Panic|log\.Panicf|log\.Panicln)\s*\([^)]*?(debug|DEBUG)[^)]*\)",
        )?
    };

    let comment_pattern = Regex::new(r"^\s*//")?;

    let entries: Vec<_> = if path.is_file() {
        vec![path.to_path_buf()]
    } else {
        WalkDir::new(path)
            .into_iter()
            .filter_map(|e| e.ok())
            .filter(|e| {
                e.path().is_file()
                    && e.path()
                        .extension()
                        .and_then(|s| s.to_str())
                        .map(|ext| {
                            matches!(
                                ext,
                                "c" | "h" | "cpp" | "hpp" | "cc" | "cxx" | "rs" | "java" | "go"
                            )
                        })
                        .unwrap_or(false)
            })
            .map(|e| e.path().to_path_buf())
            .collect()
    };

    for file_path in entries {
        let content = fs::read_to_string(&file_path)
            .with_context(|| format!("Failed to read file: {}", file_path.display()))?;

        // Find all C-style function calls
        for cap in c_functions_pattern.find_iter(&content) {
            let match_str = cap.as_str();
            let start_offset = cap.start();
            let end_offset = cap.end();

            // Calculate line numbers from byte offsets
            // Count newlines before the start position, then add 1
            let line_number = content[..start_offset].matches('\n').count() + 1;
            // For end line, count newlines up to the end position
            let end_line_number = content[..end_offset].matches('\n').count() + 1;

            // Get the line content (for display purposes, we'll get the first line of the match)
            let line_start_offset = content[..start_offset]
                .rfind('\n')
                .map(|pos| pos + 1)
                .unwrap_or(0);
            let line_content = content[line_start_offset..]
                .lines()
                .next()
                .unwrap_or("")
                .to_string();

            // Check if commented (check the beginning of the statement)
            let is_commented = comment_pattern.is_match(&line_content);

            if is_commented == find_commented {
                // Extract original lines for multiline display
                let multiline_content: Vec<String> =
                    match_str.lines().map(|s| s.to_string()).collect();

                matches.push(Match {
                    file_path: file_path.clone(),
                    line_number,
                    end_line_number,
                    line_content: match_str.replace('\n', " ").trim().to_string(),
                    multiline_content,
                });
            }
        }

        // Find all C++ stream operations
        for cap in cpp_stream_pattern.find_iter(&content) {
            let match_str = cap.as_str();
            let start_offset = cap.start();
            let end_offset = cap.end();

            // Calculate line numbers from byte offsets
            // Count newlines before the start position, then add 1
            let line_number = content[..start_offset].matches('\n').count() + 1;
            // For end line, count newlines up to the end position
            let end_line_number = content[..end_offset].matches('\n').count() + 1;

            // Get the line content
            let line_start_offset = content[..start_offset]
                .rfind('\n')
                .map(|pos| pos + 1)
                .unwrap_or(0);
            let line_content = content[line_start_offset..]
                .lines()
                .next()
                .unwrap_or("")
                .to_string();

            // Check if commented
            let is_commented = comment_pattern.is_match(&line_content);

            if is_commented == find_commented {
                // Extract original lines for multiline display
                let multiline_content: Vec<String> =
                    match_str.lines().map(|s| s.to_string()).collect();

                matches.push(Match {
                    file_path: file_path.clone(),
                    line_number,
                    end_line_number,
                    line_content: match_str.replace('\n', " ").trim().to_string(),
                    multiline_content,
                });
            }
        }

        // Find all Rust macro calls
        for cap in rust_macro_pattern.find_iter(&content) {
            let match_str = cap.as_str();
            let start_offset = cap.start();
            let end_offset = cap.end();

            // Calculate line numbers from byte offsets
            let line_number = content[..start_offset].matches('\n').count() + 1;
            let end_line_number = content[..end_offset].matches('\n').count() + 1;

            // Get the line content
            let line_start_offset = content[..start_offset]
                .rfind('\n')
                .map(|pos| pos + 1)
                .unwrap_or(0);
            let line_content = content[line_start_offset..]
                .lines()
                .next()
                .unwrap_or("")
                .to_string();

            // Check if commented
            let is_commented = comment_pattern.is_match(&line_content);

            if is_commented == find_commented {
                // Extract original lines for multiline display
                let multiline_content: Vec<String> =
                    match_str.lines().map(|s| s.to_string()).collect();

                matches.push(Match {
                    file_path: file_path.clone(),
                    line_number,
                    end_line_number,
                    line_content: match_str.replace('\n', " ").trim().to_string(),
                    multiline_content,
                });
            }
        }

        // Find all Java output statements
        for cap in java_pattern.find_iter(&content) {
            let match_str = cap.as_str();
            let start_offset = cap.start();
            let end_offset = cap.end();

            // Calculate line numbers from byte offsets
            let line_number = content[..start_offset].matches('\n').count() + 1;
            let end_line_number = content[..end_offset].matches('\n').count() + 1;

            // Get the line content
            let line_start_offset = content[..start_offset]
                .rfind('\n')
                .map(|pos| pos + 1)
                .unwrap_or(0);
            let line_content = content[line_start_offset..]
                .lines()
                .next()
                .unwrap_or("")
                .to_string();

            // Check if commented
            let is_commented = comment_pattern.is_match(&line_content);

            if is_commented == find_commented {
                // Extract original lines for multiline display
                let multiline_content: Vec<String> =
                    match_str.lines().map(|s| s.to_string()).collect();

                matches.push(Match {
                    file_path: file_path.clone(),
                    line_number,
                    end_line_number,
                    line_content: match_str.replace('\n', " ").trim().to_string(),
                    multiline_content,
                });
            }
        }

        // Find all Go output statements
        for cap in go_pattern.find_iter(&content) {
            let match_str = cap.as_str();
            let start_offset = cap.start();
            let end_offset = cap.end();

            // Calculate line numbers from byte offsets
            let line_number = content[..start_offset].matches('\n').count() + 1;
            let end_line_number = content[..end_offset].matches('\n').count() + 1;

            // Get the line content
            let line_start_offset = content[..start_offset]
                .rfind('\n')
                .map(|pos| pos + 1)
                .unwrap_or(0);
            let line_content = content[line_start_offset..]
                .lines()
                .next()
                .unwrap_or("")
                .to_string();

            // Check if commented
            let is_commented = comment_pattern.is_match(&line_content);

            if is_commented == find_commented {
                // Extract original lines for multiline display
                let multiline_content: Vec<String> =
                    match_str.lines().map(|s| s.to_string()).collect();

                matches.push(Match {
                    file_path: file_path.clone(),
                    line_number,
                    end_line_number,
                    line_content: match_str.replace('\n', " ").trim().to_string(),
                    multiline_content,
                });
            }
        }
    }

    // Remove duplicates based on file_path and line_number
    // This can happen when the same line matches multiple patterns (e.g., printf in C and Java)
    let mut seen = std::collections::HashSet::new();
    matches.retain(|m| seen.insert((m.file_path.clone(), m.line_number)));

    Ok(matches)
}