use async_trait::async_trait;
use clap::{CommandFactory, Parser};
use crate::interpreter::ExecResult;
use crate::tools::{schema_from_clap, ExecContext, ToolCtx, GlobalFlags, Tool, ToolArgs, ToolSchema};
pub struct True;
#[derive(Parser, Debug)]
#[command(name = "true", about = "Exit with success (code 0)")]
struct TrueArgs {
#[command(flatten)]
global: GlobalFlags,
#[arg(hide = true)]
rest: Vec<String>,
}
#[async_trait]
impl Tool for True {
fn name(&self) -> &str {
"true"
}
fn schema(&self) -> ToolSchema {
schema_from_clap(
&TrueArgs::command(),
"true",
"Exit with success (code 0)",
[("Always succeeds", "true")],
)
}
async fn execute(&self, args: ToolArgs, ctx: &mut dyn ToolCtx) -> ExecResult {
let Some(ctx) = ctx.as_any_mut().downcast_mut::<ExecContext>() else {
return ExecResult::failure(1, "internal error: kernel builtin requires ExecContext");
};
let parsed = match TrueArgs::try_parse_from(
std::iter::once("true".to_string()).chain(args.to_argv()),
) {
Ok(p) => p,
Err(e) => return ExecResult::failure(2, format!("true: {e}")),
};
parsed.global.apply(ctx);
ExecResult::success("")
}
}
pub struct False;
#[derive(Parser, Debug)]
#[command(name = "false", about = "Exit with failure (code 1)")]
struct FalseArgs {
#[command(flatten)]
global: GlobalFlags,
#[arg(hide = true)]
rest: Vec<String>,
}
#[async_trait]
impl Tool for False {
fn name(&self) -> &str {
"false"
}
fn schema(&self) -> ToolSchema {
schema_from_clap(
&FalseArgs::command(),
"false",
"Exit with failure (code 1)",
[("Always fails", "false")],
)
}
async fn execute(&self, args: ToolArgs, ctx: &mut dyn ToolCtx) -> ExecResult {
let Some(ctx) = ctx.as_any_mut().downcast_mut::<ExecContext>() else {
return ExecResult::failure(1, "internal error: kernel builtin requires ExecContext");
};
let parsed = match FalseArgs::try_parse_from(
std::iter::once("false".to_string()).chain(args.to_argv()),
) {
Ok(p) => p,
Err(e) => return ExecResult::failure(2, format!("false: {e}")),
};
parsed.global.apply(ctx);
ExecResult::failure(1, "")
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::ast::Value;
use crate::vfs::{MemoryFs, VfsRouter};
use std::sync::Arc;
#[tokio::test]
async fn true_returns_success() {
let mut vfs = VfsRouter::new();
vfs.mount("/", MemoryFs::new());
let mut ctx = ExecContext::new(Arc::new(vfs));
let result = True.execute(ToolArgs::new(), &mut ctx).await;
assert!(result.ok());
assert_eq!(result.code, 0);
}
#[tokio::test]
async fn true_with_args_still_succeeds() {
let mut vfs = VfsRouter::new();
vfs.mount("/", MemoryFs::new());
let mut ctx = ExecContext::new(Arc::new(vfs));
let mut args = ToolArgs::new();
args.positional.push(Value::String("ignored".into()));
let result = True.execute(args, &mut ctx).await;
assert!(result.ok());
assert_eq!(result.code, 0);
}
#[tokio::test]
async fn false_returns_failure() {
let mut vfs = VfsRouter::new();
vfs.mount("/", MemoryFs::new());
let mut ctx = ExecContext::new(Arc::new(vfs));
let result = False.execute(ToolArgs::new(), &mut ctx).await;
assert!(!result.ok());
assert_eq!(result.code, 1);
}
#[tokio::test]
async fn false_with_args_still_fails() {
let mut vfs = VfsRouter::new();
vfs.mount("/", MemoryFs::new());
let mut ctx = ExecContext::new(Arc::new(vfs));
let mut args = ToolArgs::new();
args.positional.push(Value::String("ignored".into()));
let result = False.execute(args, &mut ctx).await;
assert!(!result.ok());
assert_eq!(result.code, 1);
}
}