moduforge-rules-engine 0.5.0

use std::rc::Rc;
use std::sync::atomic::Ordering;
use std::time::Duration;

use ::serde::{Deserialize, Serialize};
use anyhow::anyhow;
use rquickjs::{async_with, CatchResultExt, Object};
use serde_json::json;

use crate::handler::function::error::FunctionResult;
use crate::handler::function::function::{Function, HandlerResponse};
use crate::handler::function::module::console::Log;
use crate::handler::function::serde::JsValue;
use crate::handler::node::{
    NodeRequest, NodeResponse, NodeResult, PartialTraceError,
};
use crate::model::{DecisionNodeKind, FunctionNodeContent};
use crate::ZEN_CONFIG;

pub(crate) mod error;
pub(crate) mod function;
pub(crate) mod listener;
pub(crate) mod module;
pub(crate) mod serde;

#[derive(Serialize, Deserialize)]
pub struct FunctionResponse {
    performance: String,
    data: Option<HandlerResponse>,
}

pub struct FunctionHandler {
    function: Rc<Function>,
    trace: bool,
    iteration: u8,
    max_depth: u8,
    max_duration: Duration,
}

impl FunctionHandler {
    pub fn new(
        function: Rc<Function>,
        trace: bool,
        iteration: u8,
        max_depth: u8,
    ) -> Self {
        let max_duration_millis =
            ZEN_CONFIG.function_timeout_millis.load(Ordering::Relaxed);

        Self {
            function,
            trace,
            iteration,
            max_depth,
            max_duration: Duration::from_millis(max_duration_millis),
        }
    }

    pub async fn handle(
        &self,
        request: NodeRequest,
    ) -> NodeResult {
        let content = match &request.node.kind {
            DecisionNodeKind::FunctionNode { content } => match content {
                FunctionNodeContent::Version2(content) => Ok(content),
                _ => Err(anyhow!("Unexpected node type")),
            },
            _ => Err(anyhow!("Unexpected node type")),
        }?;

        let start = std::time::Instant::now();
        if content.omit_nodes {
            request.input.dot_remove("$nodes");
        }

        let module_name = self
            .function
            .suggest_module_name(request.node.id.as_str(), &content.source);

        let max_duration = self.max_duration.clone();
        let interrupt_handler =
            Box::new(move || start.elapsed() > max_duration);
        self.function
            .runtime()
            .set_interrupt_handler(Some(interrupt_handler))
            .await;

        self.attach_globals().await.map_err(|e| anyhow!(e.to_string()))?;

        self.function
            .register_module(&module_name, content.source.as_str())
            .await
            .map_err(|e| anyhow!(e.to_string()))?;

        let response_result = self
            .function
            .call_handler(&module_name, JsValue(request.input.clone()))
            .await;

        match response_result {
            Ok(response) => {
                self.function.runtime().set_interrupt_handler(None).await;

                Ok(NodeResponse {
                    output: response.data,
                    trace_data: self
                        .trace
                        .then(|| json!({ "log": response.logs })),
                })
            },
            Err(e) => {
                let mut log = self.function.extract_logs().await;
                log.push(Log {
                    lines: vec![json!(e.to_string()).to_string()],
                    ms_since_run: start.elapsed().as_millis() as usize,
                });

                Err(anyhow!(PartialTraceError {
                    message: e.to_string(),
                    trace: Some(json!({ "log": log })),
                }))
            },
        }
    }

    async fn attach_globals(&self) -> FunctionResult {
        async_with!(self.function.context() => |ctx| {
            let config = Object::new(ctx.clone()).catch(&ctx)?;

            config.prop("iteration", self.iteration).catch(&ctx)?;
            config.prop("maxDepth", self.max_depth).catch(&ctx)?;
            config.prop("trace", self.trace).catch(&ctx)?;

            ctx.globals().set("config", config).catch(&ctx)?;

            Ok(())
        })
        .await
    }
}