Lot 7 : outils fichiers et shell exécutés par la façade

Quand la façade tient la conversation (LOKI_CHAT=rust), read, grep, glob,
write, edit, bash et ask s'exécutent dans rust/loki-rs/src/tools au lieu de
repartir au Go par /internal/v1/tool/exec : mêmes garde-fous dans le même
ordre (commande catastrophique, « & » final en mode Code, bac à sable du
mode Code, dossier mémoire interdit, lu avant d'écrire), même diff pour
l'UI, même nettoyage de la sortie du shell. Le jeton de l'API interne est
retiré de l'environnement du shell de l'agent.

L'état partagé reste au Go, en un seul exemplaire (sous-agents et tâches
le lisent aussi) : POST /internal/v1/tool/context (dossier de travail,
dossier mémoire, cible distante), /tool/tracker (lecture notée, contrôle
avant écriture), /tool/wrote (note après écriture, diagnostics LSP).
Repli sur le Go pour une cible distante ou un contexte indisponible.

Parité : internal/loki/parity_tools_test.go passe 58 appels à execTool sur
une arborescence de test et range le journal dans tests/parity_tools ;
tests/tools.rs le retrouve à l'identique avec exec_local. Écarts assumés
consignés dans docs/superpowers/specs/2026-10-15-rust-migration-lot7-outils-design.md.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_016tcGfJWefW5kUGoC2wL8cv
This commit is contained in:
Claude committed 2026-10-15 11:51:20 +00:00
1 parent bc388d7a01
commit a739534104
18 files changed
+3560 -2

No files matched your search

+292
View File
@@ -0,0 +1,292 @@
//! Outils de l'agent exécutés par la façade (lot 7 de la migration).
//!
//! Quand la façade tient la conversation (`LOKI_CHAT=rust`), les outils
//! fichiers et le shell s'exécutent ici : `read`, `grep`, `glob`, `write`,
//! `edit`, `bash`, `ask`. Les autres (mémoire, web, tâches, navigateur, MCP,
//! sous-agents, critères…) restent au Go (`/internal/v1/tool/exec`), de même
//! que `write`/`edit`/`bash` quand l'agent vise un poste distant.
//!
//! Ce que ces outils partagent avec le reste de Loki reste au Go, en un seul
//! exemplaire ([`ToolState`]) : le dossier de travail de la discussion, le
//! dossier mémoire, la cible distante, le suivi « lu avant d'écrire » (aussi
//! tenu par les sous-agents et les tâches, qui tournent là-bas) et les
//! diagnostics LSP. Les garde-fous sont ceux du Go, dans le même ordre.
pub mod diff;
pub mod files;
pub mod policy;
pub mod shell;
use std::future::Future;
use std::path::{Path, PathBuf};
use serde_json::{Map, Value, json};
use crate::chat::turn::ToolOutcome;
use crate::chat::types::AskEvent;
use crate::store::Store;
/// Ce que les outils doivent savoir avant de s'exécuter.
#[derive(Debug, Clone, Default, PartialEq)]
pub struct ToolCtx {
/// Dossier de travail (`agentCwd`).
pub cwd: PathBuf,
/// Dossier interdit au shell et à write/edit (`memoryDir`).
pub memory_dir: PathBuf,
/// Poste distant visé ("" : cette machine).
pub target: String,
/// Mode Code : chemins bornés au dossier, lu avant d'écrire.
pub code: bool,
}
/// L'état partagé avec le Go.
pub trait ToolState: Send + Sync {
/// Le contexte du moment ; `Err` : outils locaux indisponibles.
fn context(&self) -> impl Future<Output = Result<ToolCtx, String>> + Send;
/// `trackerNoteRead`.
fn note_read(&self, path: &Path) -> impl Future<Output = ()> + Send;
/// `trackerCheck` : le refus, ou `None`.
fn check(&self, path: &Path, for_write: bool) -> impl Future<Output = Option<String>> + Send;
/// Après une écriture réussie : `trackerNoteWrite` puis le bloc LSP.
fn wrote(&self, path: &Path) -> impl Future<Output = String> + Send;
}
/// Outils exécutés ici (hors cible distante pour write/edit/bash).
pub fn is_local(name: &str) -> bool {
matches!(
name,
"read" | "grep" | "glob" | "write" | "edit" | "bash" | "ask"
)
}
fn outcome(result: String) -> ToolOutcome {
ToolOutcome {
result,
..Default::default()
}
}
async fn blocking<T: Send + 'static>(f: impl FnOnce() -> T + Send + 'static) -> Result<T, String> {
tokio::task::spawn_blocking(f)
.await
.map_err(|e| format!("[erreur] outil interrompu : {e}"))
}
/// `codeWriteGuard` : en mode Code, borne de chemin puis « lu avant
/// d'écrire ». `None` : l'écriture peut avoir lieu.
async fn code_write_guard<S: ToolState>(
st: &S,
ctx: &ToolCtx,
file: &str,
for_write: bool,
) -> Option<String> {
if !ctx.code {
return None;
}
let path = policy::resolve_agent_path(&ctx.cwd, file);
if !policy::code_path_allowed(&ctx.cwd, &path) {
return Some(policy::refused_path_result(file));
}
st.check(&path, for_write).await
}
/// Exécute un outil ici ; `None` : c'est au Go de le faire (outil non porté,
/// cible distante, contexte indisponible).
pub async fn exec_local<S: ToolState>(
st: &S,
code: bool,
name: &str,
args: &Map<String, Value>,
label: &str,
) -> Option<ToolOutcome> {
if !is_local(name) {
return None;
}
let mut ctx = match st.context().await {
Ok(c) => c,
Err(e) => {
eprintln!("[outils] contexte indisponible, exécution par le Go : {e}");
return None;
}
};
ctx.code = code;
let remote = !ctx.target.is_empty();
let out = match name {
"read" => {
let (cwd, a) = (ctx.cwd.clone(), args.clone());
let r = match blocking(move || files::tool_read(&cwd, &a, code)).await {
Ok(r) => r,
Err(e) => return Some(outcome(e)),
};
if let Some(p) = &r.noted {
st.note_read(p).await;
}
outcome(r.result)
}
"grep" => {
let (cwd, a) = (ctx.cwd.clone(), args.clone());
outcome(
blocking(move || files::tool_grep(&cwd, &a, code))
.await
.unwrap_or_else(|e| e),
)
}
"glob" => {
let (cwd, a) = (ctx.cwd.clone(), args.clone());
outcome(
blocking(move || files::tool_glob(&cwd, &a))
.await
.unwrap_or_else(|e| e),
)
}
"ask" => {
let result = files::tool_ask(args);
let ask = (!result.starts_with("[erreur]")).then(|| AskEvent {
question: files::arg_str(args, "question").to_string(),
options: files::ask_options(args),
});
ToolOutcome {
result,
ask,
..Default::default()
}
}
"write" if !remote => {
let content = files::arg_str(args, "content").to_string();
if let Some(msg) = code_write_guard(st, &ctx, label, true).await {
return Some(outcome(msg));
}
let (cwd, mem, l, c) = (
ctx.cwd.clone(),
ctx.memory_dir.clone(),
label.to_string(),
content.clone(),
);
let mut result = match blocking(move || files::file_write(&cwd, &mem, &l, &c)).await {
Ok(r) => r,
Err(e) => return Some(outcome(e)),
};
let mut o = ToolOutcome::default();
if !result.starts_with("[erreur]") {
(o.diff, o.added) = diff::added_diff(&content);
result += &st.wrote(&policy::resolve_agent_path(&ctx.cwd, label)).await;
}
o.result = result;
o
}
"edit" if !remote => {
let (old, new) = (
files::arg_str(args, "old").to_string(),
files::arg_str(args, "new").to_string(),
);
if let Some(msg) = code_write_guard(st, &ctx, label, false).await {
return Some(outcome(msg));
}
let (cwd, mem, l, o2, n2) = (
ctx.cwd.clone(),
ctx.memory_dir.clone(),
label.to_string(),
old.clone(),
new.clone(),
);
let mut result =
match blocking(move || files::file_edit(&cwd, &mem, &l, &o2, &n2)).await {
Ok(r) => r,
Err(e) => return Some(outcome(e)),
};
let mut o = ToolOutcome::default();
if !result.starts_with("[erreur]") {
(o.diff, o.added, o.removed) = diff::line_diff(&old, &new);
result += &st.wrote(&policy::resolve_agent_path(&ctx.cwd, label)).await;
}
o.result = result;
o
}
"bash" if !remote || policy::dangerous_command(label).is_some() => {
let timeout = files::arg_int(args, "timeout").unwrap_or(0);
// Refusées AVANT toute exécution, cible distante comprise.
if let Some(reason) = policy::dangerous_command(label) {
return Some(outcome(policy::refused_command_result(reason)));
}
// Mode Code : « cmd & » laisse un orphelin sans sortie ni arrêt.
if code && shell::trailing_background(label) {
return Some(outcome("[refusé] commande terminée par « & » : lance-la avec bash_bg (sortie consultable avec bash_tail), pas en arrière-plan dans bash.".into()));
}
outcome(shell::run_shell(&ctx.cwd, &ctx.memory_dir, label, timeout).await)
}
_ => return None, // cible distante : le Go passe par le poste
};
Some(out)
}
/// [`ToolState`] servi par l'API interne du Go.
#[derive(Clone)]
pub struct GoToolState {
pub store: Store,
/// Capacités du tour (pour les diagnostics LSP).
pub caps: Value,
}
impl ToolState for GoToolState {
async fn context(&self) -> Result<ToolCtx, String> {
let (status, v) = self
.store
.internal_post("/internal/v1/tool/context", &json!({}))
.await?;
if !status.is_success() {
return Err(format!("contexte des outils : {status}"));
}
let s = |k: &str| v.get(k).and_then(Value::as_str).unwrap_or("").to_string();
Ok(ToolCtx {
cwd: PathBuf::from(s("cwd")),
memory_dir: PathBuf::from(s("memory_dir")),
target: s("target"),
code: false,
})
}
async fn note_read(&self, path: &Path) {
let body = json!({ "op": "read", "path": path });
if let Err(e) = self
.store
.internal_post("/internal/v1/tool/tracker", &body)
.await
{
eprintln!("[outils] lecture non notée : {e}");
}
}
async fn check(&self, path: &Path, for_write: bool) -> Option<String> {
let body = json!({ "op": "check", "path": path, "for_write": for_write });
match self
.store
.internal_post("/internal/v1/tool/tracker", &body)
.await
{
Ok((st, v)) if st.is_success() => v
.get("refusal")
.and_then(Value::as_str)
.filter(|r| !r.is_empty())
.map(str::to_string),
// Sans le suivi, écrire serait écrire à l'aveugle : refus.
Ok((st, _)) => Some(format!("[erreur] suivi des lectures indisponible : {st}")),
Err(e) => Some(format!("[erreur] suivi des lectures indisponible : {e}")),
}
}
async fn wrote(&self, path: &Path) -> String {
let body = json!({ "path": path, "caps": self.caps });
match self
.store
.internal_post("/internal/v1/tool/wrote", &body)
.await
{
Ok((st, v)) if st.is_success() => v
.get("lsp")
.and_then(Value::as_str)
.unwrap_or("")
.to_string(),
Ok(_) | Err(_) => String::new(),
}
}
}