61 lines
2.0 KiB
Rust
61 lines
2.0 KiB
Rust
|
//! Get completion suggestions from LSP.
|
||
|
|
|
||
|
|
use crate::tools::{Tool, ToolCtx};
|
||
|
|
use anyhow::Result;
|
||
|
|
use serde_json::{json, Value};
|
||
|
|
|
||
|
|
pub struct LspCompletion;
|
||
|
|
|
||
|
|
impl Tool for LspCompletion {
|
||
|
|
fn name(&self) -> &'static str {
|
||
|
|
"lsp_completion"
|
||
|
|
}
|
||
|
|
|
||
|
|
fn description(&self) -> &'static str {
|
||
|
|
"Get completion suggestions at a position"
|
||
|
|
}
|
||
|
|
|
||
|
|
fn parameters(&self) -> Value {
|
||
|
|
json!({
|
||
|
|
"type": "object",
|
||
|
|
"properties": {
|
||
|
|
"language": {
|
||
|
|
"type": "string",
|
||
|
|
"description": "Language identifier"
|
||
|
|
},
|
||
|
|
"path": {
|
||
|
|
"type": "string",
|
||
|
|
"description": "File path"
|
||
|
|
},
|
||
|
|
"line": {
|
||
|
|
"type": "integer",
|
||
|
|
"description": "Line number (0-based)"
|
||
|
|
},
|
||
|
|
"character": {
|
||
|
|
"type": "integer",
|
||
|
|
"description": "Character offset (0-based)"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"required": ["language", "path", "line", "character"]
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
fn run(&self, ctx: &ToolCtx, args: &Value) -> Result<String> {
|
||
|
|
let language = crate::tools::arg_str(args, "language")?;
|
||
|
|
let path = crate::tools::arg_str(args, "path")?;
|
||
|
|
let line = args.get("line").and_then(|v| v.as_i64()).unwrap_or(0);
|
||
|
|
let character = args.get("character").and_then(|v| v.as_i64()).unwrap_or(0);
|
||
|
|
|
||
|
|
let manager = ctx.lsp_manager.lock().unwrap();
|
||
|
|
if let Some(client) = manager.get_client(&language) {
|
||
|
|
let result = client.send_request("textDocument/completion", &json!({
|
||
|
|
"textDocument": { "uri": format!("file://{}", path) },
|
||
|
|
"position": { "line": line, "character": character }
|
||
|
|
}))?;
|
||
|
|
Ok(serde_json::to_string_pretty(&result)?)
|
||
|
|
} else {
|
||
|
|
anyhow::bail!("no LSP client connected for '{language}'")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|