ZacharyZhang-NY
a02b555e66
Sweep every first-party crate source (1956 .rs files) to the project comment guidelines: delete redundant restatements, decorative banners, change narration, and end-of-line comments; keep and tighten the crucial ones (invariants, bug rationale, SAFETY blocks, ported-source attribution). No functional code changed. Every edit is proven comment-only against the prior tree by a comment-stripping lexer (string/char/raw-string aware) plus a separate doctest-fence check. Where removing a comment made rustfmt or clippy want to re-lay-out adjacent code, the minimal triggering comment is restored so code tokens stay byte-identical. Gates green: cargo fmt --all --check (0 diffs), cargo check and cargo clippy --workspace --all-targets (0 warnings). Adds scripts/check_codegen_comment_guidelines.py — the enforcement gate for these guidelines (flags banners, end-of-line comments, change narration, and commented-out code).
115 lines
3.6 KiB
Rust
115 lines
3.6 KiB
Rust
//! ACP extension handler for session search (`kigi/session/search`).
|
|
//!
|
|
//! Exposes session full-text search as an ACP extension method.
|
|
//! The client sends a query and receives ranked results across all
|
|
//! (or workspace-filtered) past sessions.
|
|
//!
|
|
//! ```text
|
|
//! JSON-RPC -> mvp_agent.ext_method()
|
|
//! -> session_search::handle()
|
|
//! -> storage::search::execute_search()
|
|
//! -> search_fts::SessionSearchIndex (SQLite FTS5)
|
|
//! ```
|
|
|
|
use agent_client_protocol as acp;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use crate::session::storage::search::{SessionSearchRequest, SessionSearchResponse};
|
|
use crate::session::storage::search_fts::SessionSearchRow;
|
|
|
|
use super::ExtResult;
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct SearchSessionsRequest {
|
|
pub query: String,
|
|
/// Optional workspace directory to scope results to.
|
|
#[serde(default)]
|
|
pub cwd: Option<String>,
|
|
/// Maximum number of results to return. Defaults to 20.
|
|
#[serde(default = "default_limit")]
|
|
pub limit: usize,
|
|
/// Offset for pagination. Defaults to 0.
|
|
#[serde(default)]
|
|
pub offset: usize,
|
|
/// Whether to include content snippets in results.
|
|
#[serde(default)]
|
|
pub include_content: bool,
|
|
}
|
|
|
|
fn default_limit() -> usize {
|
|
20
|
|
}
|
|
|
|
#[derive(Debug, Serialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct SearchSessionsResponse {
|
|
pub results: Vec<SearchSessionHit>,
|
|
pub next_offset: Option<usize>,
|
|
pub total_estimate: Option<usize>,
|
|
/// True when the FTS5 index is still being bootstrapped.
|
|
pub bootstrapping: bool,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct SearchSessionHit {
|
|
pub session_id: String,
|
|
pub cwd: String,
|
|
/// Session title/summary for display
|
|
pub summary: String,
|
|
/// RFC 3339 formatted updated_at
|
|
pub updated_at: String,
|
|
pub score: f32,
|
|
pub matched_fields: Vec<String>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub snippet: Option<String>,
|
|
}
|
|
|
|
pub async fn handle(args: &acp::ExtRequest) -> ExtResult {
|
|
match args.method.as_ref() {
|
|
"kigi/session/search" => {
|
|
let req: SearchSessionsRequest = super::parse_params(args)?;
|
|
let internal_req = SessionSearchRequest {
|
|
query: req.query,
|
|
cwd: req.cwd,
|
|
limit: req.limit,
|
|
offset: req.offset,
|
|
include_content: req.include_content,
|
|
};
|
|
|
|
let root_dir = crate::util::kigi_home::kigi_home();
|
|
let result = crate::session::storage::search::execute_search(&root_dir, &internal_req)
|
|
.await
|
|
.map(to_response)
|
|
.map_err(|e| anyhow::anyhow!(e));
|
|
|
|
super::to_ext_response(result)
|
|
}
|
|
_ => Err(acp::Error::method_not_found()),
|
|
}
|
|
}
|
|
|
|
fn to_response(resp: SessionSearchResponse) -> SearchSessionsResponse {
|
|
SearchSessionsResponse {
|
|
results: resp
|
|
.results
|
|
.into_iter()
|
|
.map(|row: SessionSearchRow| SearchSessionHit {
|
|
session_id: row.session_id,
|
|
cwd: row.cwd,
|
|
summary: row.title,
|
|
updated_at: chrono::DateTime::from_timestamp(row.updated_at_unix, 0)
|
|
.map(|dt| dt.to_rfc3339())
|
|
.unwrap_or_default(),
|
|
score: row.score,
|
|
matched_fields: row.matched_fields,
|
|
snippet: row.snippet,
|
|
})
|
|
.collect(),
|
|
next_offset: resp.next_offset,
|
|
total_estimate: resp.total_estimate,
|
|
bootstrapping: resp.bootstrapping,
|
|
}
|
|
}
|