pub mod http_handler;
pub(crate) mod v1;
pub(crate) mod v2;
use crate::nodes::definition::NodeHandler;
use crate::nodes::function::v1::{FunctionV1NodeHandler, FunctionV1Trace};
use crate::nodes::function::v2::{FunctionV2NodeHandler, FunctionV2Trace};
use crate::nodes::result::NodeResult;
use crate::nodes::NodeContext;
use std::sync::Arc;
use zen_types::decision::{FunctionContent, FunctionNodeContent};
use zen_types::variable::Variable;
#[derive(Debug, Clone)]
pub struct FunctionNodeHandler;
pub type FunctionNodeData = FunctionNodeContent;
pub type FunctionNodeTrace = Variable;
impl NodeHandler for FunctionNodeHandler {
type NodeData = FunctionNodeData;
type TraceData = FunctionNodeTrace;
async fn handle(&self, ctx: NodeContext<Self::NodeData, Self::TraceData>) -> NodeResult {
match &ctx.node {
FunctionNodeContent::Version1(source) => {
let v1_context = NodeContext::<Arc<str>, FunctionV1Trace> {
id: ctx.id.clone(),
name: ctx.name.clone(),
input: ctx.input.clone(),
extensions: ctx.extensions.clone(),
trace: ctx.config.trace.then(|| Default::default()),
iteration: ctx.iteration,
config: ctx.config,
node: source.clone(),
};
FunctionV1NodeHandler.handle(v1_context).await
}
FunctionNodeContent::Version2(content) => {
let v2_context = NodeContext::<FunctionContent, FunctionV2Trace> {
id: ctx.id.clone(),
name: ctx.name.clone(),
input: ctx.input.clone(),
extensions: ctx.extensions.clone(),
trace: ctx.config.trace.then(|| Default::default()),
iteration: ctx.iteration,
config: ctx.config,
node: content.clone(),
};
FunctionV2NodeHandler.handle(v2_context).await
}
}
}
}