#[derive(Debug, Deserialize)]
struct TdgArgs {
paths: Vec<String>,
#[serde(default)]
threshold: Option<f64>,
#[serde(default)]
top_files: Option<usize>,
#[serde(default)]
include_components: Option<bool>,
#[serde(default)] with_git_context: Option<bool>,
}
pub struct TdgTool;
impl TdgTool {
#[must_use]
pub fn new() -> Self {
Self
}
}
impl Default for TdgTool {
fn default() -> Self {
Self::new()
}
}
#[async_trait]
impl ToolHandler for TdgTool {
async fn handle(&self, args: Value, _extra: RequestHandlerExtra) -> Result<Value> {
debug!("Handling analyze.tdg with args: {}", args);
let params: TdgArgs = serde_json::from_value(args)
.map_err(|e| Error::validation(format!("Invalid arguments: {e}")))?;
let paths: Vec<PathBuf> = params.paths.into_iter().map(PathBuf::from).collect();
let results = tool_functions::analyze_tdg(
&paths,
params.threshold,
params.top_files,
params.include_components,
params.with_git_context, )
.await
.map_err(|e| Error::internal(format!("TDG analysis failed: {e}")))?;
Ok(results)
}
}
#[derive(Debug, Deserialize)]
struct TdgCompareArgs {
path1: String,
path2: String,
#[serde(default)] with_git_context: Option<bool>,
}
pub struct TdgCompareTool;
impl TdgCompareTool {
#[must_use]
pub fn new() -> Self {
Self
}
}
impl Default for TdgCompareTool {
fn default() -> Self {
Self::new()
}
}
#[async_trait]
impl ToolHandler for TdgCompareTool {
async fn handle(&self, args: Value, _extra: RequestHandlerExtra) -> Result<Value> {
debug!("Handling analyze.tdg_compare with args: {}", args);
let params: TdgCompareArgs = serde_json::from_value(args)
.map_err(|e| Error::validation(format!("Invalid arguments: {e}")))?;
let path1 = PathBuf::from(params.path1);
let path2 = PathBuf::from(params.path2);
let results = tool_functions::compare_tdg(&path1, &path2, params.with_git_context)
.await
.map_err(|e| Error::internal(format!("TDG comparison failed: {e}")))?;
Ok(results)
}
}