use clap::{Parser, Subcommand};
use std::path::PathBuf;
#[derive(Parser, Debug)]
#[command(
name = "uncomment",
version,
about = "Remove comments from code files using tree-sitter parsing",
long_about = "A fast, accurate CLI tool that removes comments from source code files using tree-sitter AST parsing. Automatically preserves important comments like linting directives, documentation, and metadata."
)]
pub struct Cli {
#[command(subcommand)]
pub command: Option<Commands>,
#[command(flatten)]
pub args: ProcessArgs,
}
#[derive(Subcommand, Debug)]
pub enum Commands {
#[command(about = "Create a template configuration file")]
Init {
#[arg(short, long, default_value = ".uncommentrc.toml")]
output: PathBuf,
#[arg(short, long)]
force: bool,
#[arg(
long,
help = "Generate comprehensive config with all supported languages"
)]
comprehensive: bool,
#[arg(short, long, help = "Interactive mode to select languages and options")]
interactive: bool,
},
}
#[derive(Parser, Debug)]
pub struct ProcessArgs {
#[arg(help = "Files, directories, or glob patterns to process")]
pub paths: Vec<String>,
#[arg(short = 'r', long, help = "Remove TODO comments (normally preserved)")]
pub remove_todo: bool,
#[arg(short = 'f', long, help = "Remove FIXME comments (normally preserved)")]
pub remove_fixme: bool,
#[arg(
short = 'd',
long,
help = "Remove documentation comments and docstrings"
)]
pub remove_doc: bool,
#[arg(
short = 'i',
long = "ignore",
help = "Additional patterns to preserve (can be used multiple times)"
)]
pub ignore_patterns: Vec<String>,
#[arg(
long = "no-default-ignores",
help = "Disable built-in preservation patterns (ESLint, Clippy, etc.)"
)]
pub no_default_ignores: bool,
#[arg(short = 'n', long, help = "Show changes without modifying files")]
pub dry_run: bool,
#[arg(
long = "diff",
help = "Show line-by-line diffs for modified files (only useful with --dry-run)"
)]
pub diff: bool,
#[arg(short = 'v', long, help = "Show detailed processing information")]
pub verbose: bool,
#[arg(long = "no-gitignore", help = "Process files ignored by .gitignore")]
pub no_gitignore: bool,
#[arg(
long = "traverse-git-repos",
help = "Traverse into other git repositories (useful for monorepos)"
)]
pub traverse_git_repos: bool,
#[arg(
short = 'j',
long = "threads",
help = "Number of parallel threads (0 = auto-detect)",
default_value = "1"
)]
pub threads: usize,
#[arg(
short = 'c',
long = "config",
help = "Path to configuration file (overrides automatic discovery)"
)]
pub config: Option<PathBuf>,
}
impl ProcessArgs {
pub fn processing_options(&self) -> crate::processor::ProcessingOptions {
crate::processor::ProcessingOptions {
remove_todo: self.remove_todo,
remove_fixme: self.remove_fixme,
remove_doc: self.remove_doc,
custom_preserve_patterns: self.ignore_patterns.clone(),
use_default_ignores: !self.no_default_ignores,
dry_run: self.dry_run,
show_diff: self.diff,
respect_gitignore: !self.no_gitignore,
traverse_git_repos: self.traverse_git_repos,
}
}
}
impl Cli {
pub fn handle_init_command(
output: &PathBuf,
force: bool,
comprehensive: bool,
interactive: bool,
) -> anyhow::Result<()> {
if output.exists() && !force {
return Err(anyhow::anyhow!(
"Configuration file already exists: {}. Use --force to overwrite.",
output.display()
));
}
let (template, detected_info) = if comprehensive {
(crate::config::Config::comprehensive_template_clean(), None)
} else if interactive {
(crate::config::Config::interactive_template_clean()?, None)
} else {
let current_dir =
std::env::current_dir().unwrap_or_else(|_| std::path::PathBuf::from("."));
let (template, info) = crate::config::Config::smart_template_with_info(¤t_dir)?;
(template, Some(info))
};
std::fs::write(output, template)?;
println!("✓ Created configuration file: {}", output.display());
if comprehensive {
println!("📦 Generated comprehensive config with 15+ language configurations");
} else if interactive {
println!("🎯 Generated customized config based on your selections");
} else if let Some(info) = detected_info {
if !info.detected_languages.is_empty() {
println!(
"🔍 Detected {} file types in your project:",
info.detected_languages.len()
);
for (lang, count) in &info.detected_languages {
println!(" {count} ({lang} files)");
}
println!(
"📝 Configured {} languages with appropriate settings",
info.configured_languages
);
} else {
println!("📝 No supported files detected, generated basic template");
}
if info.total_files > 0 {
println!("📊 Scanned {} files total", info.total_files);
}
} else {
println!("📝 Generated smart config based on detected files in your project");
}
Ok(())
}
}