use kglite_mcp_server::CodeTreeHooks;
fn main() -> anyhow::Result<()> {
let hooks = CodeTreeHooks {
build: Box::new(|dir, include_docs| {
codingest::build_code_tree(dir, false, true, None, None, include_docs)
}),
build_revs: Box::new(|dir, revs, include_docs| {
let revs = codingest::dedup_revs(revs);
let graph = codingest::build_code_tree_revs(
dir,
&revs,
None,
false,
true,
None,
None,
include_docs,
)?;
Ok((graph, revs))
}),
is_code_file: Box::new(|p| codingest::language_for_path(p).is_some()),
};
kglite_mcp_server::run_with_code_tree_hooks(std::env::args_os(), Some(hooks))
}