fencecat 0.3.0

use std::{
    cmp::Reverse,
    collections::HashSet,
    fs::{self, File},
    io::{self, Read as _},
    path::{Path, PathBuf},
};

use clap::{ArgAction, Parser};
use ignore::WalkBuilder;
use regex::RegexSet;

#[allow(clippy::struct_excessive_bools)]
#[derive(Parser, Debug)]
#[command(
    name = "fencecat",
    version,
    about = "Recursively emit Markdown code fences labeled with relative file paths.\nUseful for sharing source trees in LLM chats or other issue trackers."
)]
struct Cli {
    /// Root directories to scan OR files to emit
    #[arg(value_name = "PATHS", default_value = ".", num_args = 1..)]
    paths: Vec<PathBuf>,

    /// Copy the full output to the clipboard
    #[arg(short = 'c', long = "copy", action = ArgAction::SetTrue)]
    copy: bool,

    /// Order files by size (largest first)
    #[arg(short = 'B', long = "biggest-first", action = ArgAction::SetTrue)]
    biggest_first: bool,

    /// Only include files whose extension matches any of the given ones (comma-separated).
    /// Examples: --ext rs,ts,py   or   --ext .md,.toml
    #[arg(
        short,
        long = "ext",
        value_name = "EXT[,EXT...]",
        value_delimiter = ','
    )]
    ext: Option<Vec<String>>,

    /// Exclude files whose extension matches any of the given ones (comma-separated).
    /// These bans happen after whitelists.
    #[arg(long = "not-ext", value_name = "EXT[,EXT...]", value_delimiter = ',')]
    not_ext: Option<Vec<String>>,

    /// Only include paths matching these regexes (relative to CWD).
    /// Can be specified multiple times.
    #[arg(short, long = "regex", action = ArgAction::Append)]
    regex: Option<Vec<String>>,

    /// Exclude paths matching these regexes (relative to CWD).
    /// Can be specified multiple times.
    #[arg(long = "not-regex", action = ArgAction::Append)]
    not_regex: Option<Vec<String>>,

    /// Include hidden and gitignored files (disable ignore rules)
    #[arg(short = 'H', long = "no-ignore")]
    no_ignore: bool,

    /// Prepend a plain file listing (like `dir`) before the fences (no timestamps/metadata)
    #[arg(short = 'D', long = "dir-list", action = ArgAction::SetTrue)]
    dir_list: bool,

    /// Make file paths relative to their specific input argument rather than the CWD
    #[arg(long = "context-relative", action = ArgAction::SetTrue)]
    context_relative: bool,
}

impl Cli {
    pub fn build_walkdir(&self) -> WalkBuilder {
        let mut paths_iter = self.paths.iter();
        let first_path = paths_iter.next().expect("At least one path is required");
        let mut wb = WalkBuilder::new(first_path);

        // Add any remaining paths to the builder
        for path in paths_iter {
            wb.add(path);
        }

        if self.no_ignore {
            wb.hidden(false)
                .ignore(false)
                .git_ignore(false)
                .git_global(false)
                .git_exclude(false)
                .parents(false);
        }
        wb
    }
}

/// Heuristic: consider a file "binary" if the first few KB contain a NUL byte.
fn is_binary(path: &Path) -> io::Result<bool> {
    let mut f = File::open(path)?;
    let mut buf = [0u8; 8192];
    let n = f.read(&mut buf)?;
    Ok(buf[..n].contains(&0))
}

fn choose_fence(content: &str) -> String {
    for n in 3..=10 {
        let fence = "`".repeat(n);
        if !content.contains(&fence) {
            return fence;
        }
    }
    "````````````".to_string()
}

#[derive(Debug)]
struct FileInfo {
    path: PathBuf,
    rel: String,
    size: u64,
    arg_index: usize,
}

/// Helper to normalize extension lists (strip dots, lowercase)
fn normalize_ext_list(list: &[String]) -> HashSet<String> {
    list.iter()
        .map(|s| s.trim().trim_start_matches('.').to_ascii_lowercase())
        .filter(|s| !s.is_empty())
        .collect()
}

fn build_ext_filters(cli: &Cli) -> (Option<HashSet<String>>, Option<HashSet<String>>) {
    let allow = cli.ext.as_ref().map(|v| normalize_ext_list(v));
    let deny = cli.not_ext.as_ref().map(|v| normalize_ext_list(v));
    (allow, deny)
}

fn compile_regex_sets(cli: &Cli) -> (Option<RegexSet>, Option<RegexSet>) {
    let allow = cli
        .regex
        .as_ref()
        .map(|v| RegexSet::new(v).expect("Invalid regex in --regex"));
    let deny = cli
        .not_regex
        .as_ref()
        .map(|v| RegexSet::new(v).expect("Invalid regex in --not-regex"));
    (allow, deny)
}

fn make_fileinfo_if_included(
    path: &Path,
    root_for_rel: &Path,
    arg_index: usize,
    ext_allow: Option<&HashSet<String>>,
    ext_deny: Option<&HashSet<String>>,
    re_allow: Option<&RegexSet>,
    re_deny: Option<&RegexSet>,
) -> Option<FileInfo> {
    // 1. Check Extensions
    if ext_allow.is_some() || ext_deny.is_some() {
        let ext = path
            .extension()
            .and_then(|e| e.to_str())
            .map(str::to_ascii_lowercase);

        // If whitelist exists, file must match it
        if let Some(allow) = ext_allow
            && !ext.as_ref().is_some_and(|e| allow.contains(e))
        {
            return None;
        }
        // If blacklist exists, file must NOT match it
        if let Some(deny) = ext_deny
            && ext.as_ref().is_some_and(|e| deny.contains(e))
        {
            return None;
        }
    }

    // 2. Check Regex (against path relative to CWD)
    if re_allow.is_some() || re_deny.is_some() {
        // Calculate path relative to CWD (".").
        // If we are in `src/`, and path is `src/main.rs`, this returns `main.rs`.
        // If we are in `.`, and path is `src/main.rs`, this returns `src/main.rs`.
        let cwd_rel = fencecat::rel_string(Path::new("."), path);

        if let Some(allow) = re_allow
            && !allow.is_match(&cwd_rel)
        {
            return None;
        }
        if let Some(deny) = re_deny
            && deny.is_match(&cwd_rel)
        {
            return None;
        }
    }

    let md = match path.metadata() {
        Ok(m) => m,
        Err(e) => {
            eprintln!("skip {}: metadata error: {e}", path.display());
            return None;
        }
    };
    if md.len() == 0 {
        return None;
    }

    match is_binary(path) {
        Ok(true) => return None,
        Ok(false) => {}
        Err(e) => {
            eprintln!("skip {}: read error: {e}", path.display());
            return None;
        }
    }

    let rel = fencecat::rel_string(root_for_rel, path);

    Some(FileInfo {
        path: path.to_path_buf(),
        rel,
        size: md.len(),
        arg_index,
    })
}

fn collect_any(cli: &Cli) -> Vec<FileInfo> {
    let (ext_allow, ext_deny) = build_ext_filters(cli);
    let (re_allow, re_deny) = compile_regex_sets(cli);

    // Validate existence before walking
    for path in &cli.paths {
        if !path.exists() {
            eprintln!("No such file or directory: {}", path.display());
            std::process::exit(1);
        }
    }

    let walker = cli.build_walkdir().build();
    let mut files: Vec<FileInfo> = Vec::new();

    for dent in walker {
        let entry = match dent {
            Ok(e) => e,
            Err(err) => {
                eprintln!("walk error: {err}");
                continue;
            }
        };

        if entry.file_type().is_some_and(|ft| ft.is_file()) {
            let path = entry.path();

            // Find which input argument this file belongs to.
            // WalkBuilder yields paths that begin with the exact argument string used to add them.
            let matched_arg = cli
                .paths
                .iter()
                .enumerate()
                .find(|(_, p)| path.starts_with(p));

            let arg_index = matched_arg.map_or(0, |(i, _)| i);

            let root_for_rel = if cli.context_relative {
                matched_arg.map_or_else(
                    || PathBuf::from("."),
                    |(_, p)| {
                        if p.is_file() {
                            p.parent().unwrap_or_else(|| Path::new(".")).to_path_buf()
                        } else {
                            p.clone()
                        }
                    },
                )
            } else {
                PathBuf::from(".")
            };

            if let Some(info) = make_fileinfo_if_included(
                path,
                &root_for_rel,
                arg_index,
                ext_allow.as_ref(),
                ext_deny.as_ref(),
                re_allow.as_ref(),
                re_deny.as_ref(),
            ) {
                files.push(info);
            }
        }
    }

    if cli.biggest_first {
        // Global size sort. Tie-breakers: input argument order, then alphabetical
        files.sort_by(|a, b| {
            Reverse(a.size)
                .cmp(&Reverse(b.size))
                .then_with(|| a.arg_index.cmp(&b.arg_index))
                .then_with(|| a.rel.cmp(&b.rel))
        });
    } else {
        // Group by input argument order, then sort alphabetically within each group
        files.sort_by(|a, b| {
            a.arg_index
                .cmp(&b.arg_index)
                .then_with(|| a.rel.cmp(&b.rel))
        });
    }
    files
}

fn emit_dir_listing(files: &[FileInfo]) -> String {
    let mut s = String::new();
    s.push_str("```\n");
    for f in files {
        s.push_str(&f.rel);
        s.push('\n');
    }
    s.push_str("```\n\n");
    s
}

fn main() {
    let cli = Cli::parse();

    let files = collect_any(&cli);

    let mut out = String::new();

    if cli.dir_list {
        out.push_str(&emit_dir_listing(&files));
    }

    for f in &files {
        let bytes = match fs::read(&f.path) {
            Ok(b) => b,
            Err(e) => {
                eprintln!("skip {}: read error: {e}", f.path.display());
                continue;
            }
        };
        let content = String::from_utf8_lossy(&bytes);
        let fence = choose_fence(&content);

        out.push_str(&fence);
        out.push_str(&f.rel);
        out.push('\n');

        out.push_str(&content);
        if !content.ends_with('\n') {
            out.push('\n');
        }

        out.push('\n');
        out.push_str(&fence);
        out.push_str("\n\n");
    }

    print!("{out}");

    if cli.copy {
        match fencecat::clipboard::copy_to_clipboard_multi(&out) {
            Ok(()) => eprintln!(">> copied to clipboard"),
            Err(e) => eprintln!(">> failed to copy to clipboard: {e}"),
        }
    }
}