use anyhow::{anyhow, Context, Result};
use khive_runtime::pack::{PackRegistry, VerbRegistry, VerbRegistryBuilder, Visibility};
use khive_runtime::{KhiveRuntime, RuntimeConfig};
use serde::Serialize;
#[derive(Debug, Serialize, PartialEq, Eq)]
#[serde(rename_all = "lowercase")]
pub enum VerbVisibility {
Verb,
Subhandler,
}
impl From<&Visibility> for VerbVisibility {
fn from(v: &Visibility) -> Self {
match v {
Visibility::Verb => VerbVisibility::Verb,
Visibility::Subhandler => VerbVisibility::Subhandler,
}
}
}
#[derive(Debug, Serialize)]
pub struct VerbInfo {
pub name: String,
pub description: String,
pub visibility: VerbVisibility,
pub category: String,
}
#[derive(Debug, Serialize)]
pub struct PackInfo {
pub name: String,
pub note_kinds: Vec<String>,
pub entity_kinds: Vec<String>,
pub requires: Vec<String>,
pub verbs: Vec<VerbInfo>,
}
fn build_registry() -> Result<(VerbRegistry, KhiveRuntime)> {
let config = RuntimeConfig {
db_path: None,
default_namespace: khive_runtime::Namespace::parse("kkernel-introspect")
.unwrap_or_else(|_| khive_runtime::Namespace::local()),
embedding_model: None,
..RuntimeConfig::default()
};
let runtime = KhiveRuntime::new(config).context("building introspection runtime")?;
let mut builder = VerbRegistryBuilder::new();
let names: Vec<String> = PackRegistry::discovered_names()
.into_iter()
.map(str::to_string)
.collect();
PackRegistry::register_packs(&names, runtime.clone(), &mut builder)
.map_err(|n| anyhow!("pack {n:?} declared in inventory but factory missing"))?;
let registry = builder.build().context("building VerbRegistry")?;
Ok((registry, runtime))
}
fn pack_info_from_registry(registry: &VerbRegistry, name: &str) -> Option<PackInfo> {
let verbs = registry.pack_verbs(name)?;
Some(PackInfo {
name: name.to_string(),
note_kinds: registry
.pack_note_kinds(name)
.unwrap_or(&[])
.iter()
.map(|s| s.to_string())
.collect(),
entity_kinds: registry
.pack_entity_kinds(name)
.unwrap_or(&[])
.iter()
.map(|s| s.to_string())
.collect(),
requires: registry
.pack_requires(name)
.unwrap_or(&[])
.iter()
.map(|s| s.to_string())
.collect(),
verbs: verbs
.iter()
.map(|v| VerbInfo {
name: v.name.to_string(),
description: v.description.to_string(),
visibility: VerbVisibility::from(&v.visibility),
category: format!("{:?}", v.category),
})
.collect(),
})
}
pub fn list_packs() -> Result<Vec<PackInfo>> {
let (registry, _runtime) = build_registry()?;
let names: Vec<String> = registry
.pack_names()
.into_iter()
.map(str::to_string)
.collect();
Ok(names
.iter()
.filter_map(|n| pack_info_from_registry(®istry, n))
.collect())
}
pub fn pack_handler(name: &str) -> Result<Option<PackInfo>> {
let (registry, _runtime) = build_registry()?;
Ok(pack_info_from_registry(®istry, name))
}
#[cfg(test)]
mod tests {
use super::*;
use serial_test::serial;
#[test]
#[serial]
fn introspection_registry_builds_under_strict_mode_without_actor() {
let prev = std::env::var("KHIVE_REQUIRE_ATTRIBUTED_ACTOR").ok();
std::env::set_var("KHIVE_REQUIRE_ATTRIBUTED_ACTOR", "1");
let result = build_registry();
match prev {
Some(v) => std::env::set_var("KHIVE_REQUIRE_ATTRIBUTED_ACTOR", v),
None => std::env::remove_var("KHIVE_REQUIRE_ATTRIBUTED_ACTOR"),
}
assert!(
result.is_ok(),
"build_registry (introspection-only) must succeed under \
KHIVE_REQUIRE_ATTRIBUTED_ACTOR=1 + no actor: strict-actor enforcement \
applies only to dispatch paths, not introspection. Got: {:?}",
result.err()
);
let (registry, _runtime) = result.unwrap();
let pack_names: Vec<&str> = registry.pack_names().into_iter().collect();
assert!(
pack_names.contains(&"kg"),
"kg pack must be present in introspection registry under strict mode; \
got: {pack_names:?}"
);
}
#[test]
fn list_packs_returns_at_least_kg() {
let packs = list_packs().expect("list_packs succeeds");
assert!(!packs.is_empty(), "at least one pack must register");
let names: Vec<&str> = packs.iter().map(|p| p.name.as_str()).collect();
assert!(
names.contains(&"kg"),
"kg pack must be registered; got {names:?}"
);
}
#[test]
fn pack_handler_for_kg_returns_full_surface() {
let info = pack_handler("kg")
.expect("pack_handler succeeds")
.expect("kg pack must exist");
assert_eq!(info.name, "kg");
assert!(
!info.verbs.is_empty(),
"kg pack must expose verbs; got {:?}",
info.verbs
);
assert_eq!(
info.verbs.len(),
18,
"kg pack must expose 18 verbs; got {}: {:?}",
info.verbs.len(),
info.verbs.iter().map(|v| &v.name).collect::<Vec<_>>()
);
let create = info.verbs.iter().find(|v| v.name == "create").unwrap();
assert_eq!(
create.visibility,
VerbVisibility::Verb,
"kg create must have Verb visibility"
);
assert!(
!create.category.is_empty(),
"kg create must have a non-empty category"
);
}
#[test]
fn pack_handler_unknown_returns_none() {
let info = pack_handler("does_not_exist").unwrap();
assert!(info.is_none(), "unknown pack returns None, not Err");
}
}