use async_trait::async_trait;
use serde_json::Value;
use khive_runtime::pack::PackRuntime;
use khive_runtime::{KhiveRuntime, NamespaceToken, RuntimeError, SchemaPlan, VerbRegistry};
use khive_types::{EdgeEndpointRule, HandlerDef, Pack};
use crate::{handlers, SessionPack, SESSION_HANDLERS};
struct SessionPackFactory;
impl khive_runtime::PackFactory for SessionPackFactory {
fn name(&self) -> &'static str {
"session"
}
fn requires(&self) -> &'static [&'static str] {
&["kg"]
}
fn create(&self, runtime: KhiveRuntime) -> Box<dyn khive_runtime::PackRuntime> {
Box::new(SessionPack::new(runtime))
}
}
inventory::submit! { khive_runtime::PackRegistration(&SessionPackFactory) }
#[async_trait]
impl PackRuntime for SessionPack {
fn name(&self) -> &str {
<SessionPack as Pack>::NAME
}
fn note_kinds(&self) -> &'static [&'static str] {
<SessionPack as Pack>::NOTE_KINDS
}
fn entity_kinds(&self) -> &'static [&'static str] {
<SessionPack as Pack>::ENTITY_KINDS
}
fn handlers(&self) -> &'static [HandlerDef] {
&SESSION_HANDLERS
}
fn edge_rules(&self) -> &'static [EdgeEndpointRule] {
&[]
}
fn requires(&self) -> &'static [&'static str] {
<SessionPack as Pack>::REQUIRES
}
fn schema_plan(&self) -> SchemaPlan {
SchemaPlan {
pack: "session",
statements: &crate::SESSION_SCHEMA_PLAN_STMTS,
}
}
async fn warm(&self) {
let config = crate::mirror::MirrorConfig::from_env();
if !config.enabled && !config.codex_enabled {
return;
}
let runtime = self.runtime().clone();
tokio::spawn(async move {
crate::mirror::run_mirror_service(runtime, config).await;
});
}
async fn dispatch(
&self,
verb: &str,
params: Value,
_registry: &VerbRegistry,
token: &NamespaceToken,
) -> Result<Value, RuntimeError> {
let rt = self.runtime();
match verb {
"session.store" => handlers::store::handle_store(rt, token, params).await,
"session.list" => handlers::list::handle_list(rt, token, params).await,
"session.get" => handlers::get::handle_get(rt, token, params).await,
_ => Err(RuntimeError::InvalidInput(format!(
"session pack does not handle verb {verb:?}"
))),
}
}
}