jorm 0.1.1

A lightweight DAG execution engine with natural language processing
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use crate::core::engine::TaskResult;
use crate::core::{error::JormError, task::TaskType};
use std::process::Stdio;
use tokio::process::Command;
use tokio::time::{timeout, Duration};

pub struct JormExecutor;

impl Default for JormExecutor {
    fn default() -> Self {
        Self::new()
    }
}

impl JormExecutor {
    pub fn new() -> Self {
        Self
    }

    pub async fn execute(
        &self,
        task_name: &str,
        task_type: &TaskType,
    ) -> Result<TaskResult, JormError> {
        match task_type {
            TaskType::Jorm {
                command,
                args,
                working_dir,
            } => {
                self.execute_jorm_command(task_name, command, args.as_ref(), working_dir.as_deref())
                    .await
            }
            _ => Err(JormError::ExecutionError(format!(
                "Jorm executor cannot handle task type: {:?}",
                task_type
            ))),
        }
    }

    async fn execute_jorm_command(
        &self,
        task_name: &str,
        command: &str,
        args: Option<&Vec<String>>,
        working_dir: Option<&str>,
    ) -> Result<TaskResult, JormError> {
        if command.trim().is_empty() {
            return Err(JormError::ExecutionError("Empty jorm command".to_string()));
        }

        // Get the current executable path
        let current_exe = std::env::current_exe().map_err(|e| {
            JormError::ExecutionError(format!("Failed to get current executable path: {}", e))
        })?;

        // Create command to run jorm with the specified subcommand and args
        let mut cmd = Command::new(&current_exe);
        cmd.arg(command);

        if let Some(args_vec) = args {
            cmd.args(args_vec);
        }

        cmd.stdout(Stdio::piped()).stderr(Stdio::piped());

        // Set working directory if specified
        if let Some(dir) = working_dir {
            cmd.current_dir(dir);
        }

        // Execute with timeout (5 minutes for jorm commands as they might be complex)
        let timeout_duration = Duration::from_secs(300);

        match timeout(timeout_duration, cmd.output()).await {
            Ok(Ok(output)) => {
                let stdout = String::from_utf8_lossy(&output.stdout).to_string();
                let stderr = String::from_utf8_lossy(&output.stderr).to_string();

                let success = output.status.success();
                let error = if !stderr.is_empty() && !success {
                    Some(stderr)
                } else {
                    None
                };

                Ok(TaskResult {
                    task_name: task_name.to_string(),
                    success,
                    output: stdout,
                    error,
                })
            }
            Ok(Err(e)) => Err(JormError::ExecutionError(format!(
                "Failed to execute jorm command '{}': {}",
                command, e
            ))),
            Err(_) => Err(JormError::ExecutionError(format!(
                "Jorm command '{}' timed out after {} seconds",
                command,
                timeout_duration.as_secs()
            ))),
        }
    }
}