use object_store::path::DELIMITER;
use std::{future::Future, pin::Pin};
pub mod agent;
pub mod context;
pub mod http;
pub mod json;
pub mod model;
pub mod tool;
pub use agent::*;
pub use context::*;
pub use http::*;
pub use json::*;
pub use model::*;
pub use tool::*;
pub type BoxError = Box<dyn std::error::Error + Send + Sync>;
pub type BoxPinFut<T> = Pin<Box<dyn Future<Output = T> + Send>>;
pub type BoxFut<'a, T> = Pin<Box<dyn Future<Output = T> + Send + 'a>>;
pub fn path_lowercase(path: &Path) -> Path {
let mut raw = path.to_string();
raw.make_ascii_lowercase();
Path::parse(&raw).unwrap_or_else(|_| Path::from(raw))
}
pub fn path_join(a: &Path, b: &Path) -> Path {
path_lowercase(&Path::from_iter(a.parts().chain(b.parts())))
}
pub fn validate_path_part(part: &str) -> Result<(), BoxError> {
if part.is_empty() || part.contains(DELIMITER) || Path::from(part).as_ref() != part {
return Err(format!("invalid path part: {}", part).into());
}
Ok(())
}
pub fn validate_function_name(name: &str) -> Result<(), BoxError> {
if name.is_empty() {
return Err("empty string".into());
}
if name.len() > 64 {
return Err("string length exceeds the limit 64".into());
}
let mut iter = name.chars();
if !matches!(iter.next(), Some('a'..='z')) {
return Err("name must start with a lowercase letter".into());
}
for c in iter {
if !matches!(c, 'a'..='z' | '0'..='9' | '_' | '-') {
return Err(format!("invalid character: {}", c).into());
}
}
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_path_lowercase() {
let a = Path::from("a/Foo");
assert_eq!(path_lowercase(&a).as_ref(), "a/foo");
}
#[test]
fn test_path_join_handles_root() {
let root = Path::default();
let p = Path::from("a/b");
assert_eq!(path_join(&root, &p).as_ref(), "a/b");
assert_eq!(path_join(&p, &root).as_ref(), "a/b");
assert_eq!(path_join(&root, &root).as_ref(), "");
assert_eq!(path_join(&p, &Path::from("c")).as_ref(), "a/b/c");
}
#[test]
fn test_path_join_is_idempotent_and_does_not_double_encode() {
let ns = Path::from("agents/store");
let key = Path::from("a*b");
let joined = path_join(&ns, &key);
assert!(!joined.as_ref().contains("%25"));
let rejoined = path_join(&Path::default(), &joined);
assert_eq!(joined, rejoined);
assert_eq!(path_lowercase(&joined), joined);
}
#[test]
fn test_validate_path_part() {
assert!(validate_path_part("foo").is_ok());
assert!(validate_path_part("fOO").is_ok());
assert!(validate_path_part("").is_err());
assert!(validate_path_part("foo/").is_err());
assert!(validate_path_part("/foo").is_err());
assert!(validate_path_part("foo/bar").is_err());
assert!(validate_path_part("foo/bar/").is_err());
}
#[test]
fn test_validate_function_name() {
assert!(validate_function_name("foo").is_ok());
assert!(validate_function_name("foo_bar9").is_ok());
assert!(validate_function_name("foo-bar").is_ok());
assert!(validate_function_name("").is_err());
assert!(validate_function_name("9foo").is_err());
assert!(validate_function_name("foo.bar").is_err());
assert!(validate_function_name("foo/bar").is_err());
assert!(validate_function_name(&"a".repeat(65)).is_err());
}
}