//! Arbor CLI - Command-line interface for Arbor
//!
//! This is the main entry point for users interacting with Arbor.
//! It provides commands for indexing, querying, and serving the code graph.
use clap::{Parser, Subcommand};
use colored::Colorize;
use std::path::PathBuf;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
mod commands;
#[derive(Parser)]
#[command(name = "arbor")]
#[command(author = "Arbor Contributors")]
#[command(version)]
#[command(about = "The Graph-Native Intelligence Layer for Code", long_about = None)]
struct Cli {
/// Enable verbose output
#[arg(short, long, global = true)]
verbose: bool,
#[command(subcommand)]
command: Commands,
}
#[derive(Subcommand)]
enum Commands {
/// Initialize Arbor in the current directory
Init {
/// Path to initialize (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
},
/// Index the codebase and build the graph
Index {
/// Path to index (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
/// Output file for the graph JSON
#[arg(short, long)]
output: Option<PathBuf>,
/// Follow symbolic links when walking directories
#[arg(long)]
follow_symlinks: bool,
/// Disable caching (force full re-index)
#[arg(long)]
no_cache: bool,
},
/// Search the code graph
Query {
/// Search query
query: String,
/// Maximum results to return
#[arg(short, long, default_value = "10")]
limit: usize,
},
/// Start the Arbor server
Serve {
/// Port to listen on
#[arg(short, long, default_value = "7432")]
port: u16,
/// Headless mode: bind to 0.0.0.0 for remote access (WSL/Docker/Server)
#[arg(long)]
headless: bool,
/// Path to index (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
/// Follow symbolic links when walking directories
#[arg(long)]
follow_symlinks: bool,
},
/// Export the graph to JSON
Export {
/// Output file
#[arg(short, long, default_value = "arbor-graph.json")]
output: PathBuf,
/// Path to index (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
},
/// Show index status and statistics
Status {
/// Path to check (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
/// List all indexed files
#[arg(long)]
files: bool,
},
/// Start the Arbor Visualizer
Viz {
/// Path to visualize (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
/// Follow symbolic links when walking directories
#[arg(long)]
follow_symlinks: bool,
},
/// Start the Agentic Bridge (MCP + Viz)
Bridge {
/// Path to index (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
/// Also launch the Flutter visualizer
#[arg(long)]
viz: bool,
/// Follow symbolic links when walking directories
#[arg(long)]
follow_symlinks: bool,
},
/// Check system health and environment
#[command(hide = true)]
CheckHealth,
/// Preview blast radius before refactoring a node
Refactor {
/// The node to analyze (function name, class name, or qualified path)
target: String,
/// Maximum depth to search (default: 5)
#[arg(short, long, default_value = "5")]
depth: usize,
/// Show detailed reasoning for each affected node
#[arg(long)]
why: bool,
/// Output as JSON instead of formatted text
#[arg(long)]
json: bool,
},
/// Explain code using graph-backed context
Explain {
/// The question or code path to explain
question: String,
/// Maximum tokens for context (default: 4000)
#[arg(short, long, default_value = "4000")]
tokens: usize,
/// Show detailed reasoning for context selection
#[arg(long)]
why: bool,
/// Output as JSON instead of formatted text
#[arg(long)]
json: bool,
},
/// Launch the graphical interface
Gui {
/// Path to analyze (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
},
/// Generate a PR summary for refactored symbols
PrSummary {
/// Symbols that were changed (comma-separated)
symbols: String,
/// Path to analyze (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
},
/// Watch for file changes and re-index automatically
Watch {
/// Path to watch (defaults to current directory)
#[arg(default_value = ".")]
path: PathBuf,
},
}
#[tokio::main]
async fn main() {
let cli = Cli::parse();
// Set up logging
let filter = if cli.verbose { "debug" } else { "info" };
tracing_subscriber::registry()
.with(
tracing_subscriber::fmt::layer()
.with_writer(std::io::stderr)
.with_target(false),
)
.with(tracing_subscriber::EnvFilter::new(filter))
.init();
let result = match cli.command {
Commands::Init { path } => commands::init(&path),
Commands::Index {
path,
output,
follow_symlinks,
no_cache,
} => commands::index(&path, output.as_deref(), follow_symlinks, no_cache),
Commands::Query { query, limit } => commands::query(&query, limit),
Commands::Serve {
port,
headless,
path,
follow_symlinks,
} => commands::serve(port, headless, &path, follow_symlinks).await,
Commands::Export { output, path } => commands::export(&path, &output),
Commands::Status { path, files } => commands::status(&path, files),
Commands::Viz {
path,
follow_symlinks,
} => commands::viz(&path, follow_symlinks).await,
Commands::Bridge {
path,
viz,
follow_symlinks,
} => commands::bridge(&path, viz, follow_symlinks).await,
Commands::CheckHealth => commands::check_health().await,
Commands::Refactor {
target,
depth,
why,
json,
} => commands::refactor(&target, depth, why, json),
Commands::Explain {
question,
tokens,
why,
json,
} => commands::explain(&question, tokens, why, json),
Commands::Gui { path } => commands::gui(&path),
Commands::PrSummary { symbols, path } => commands::pr_summary(&symbols, &path),
Commands::Watch { path } => commands::watch(&path).await,
};
if let Err(e) = result {
eprintln!("{} {}", "error:".red().bold(), e);
std::process::exit(1);
}
}
