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
106
107
108
109
110
111
112
113

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 RustExecutor;

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

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

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

    async fn execute_cargo_command(
        &self,
        task_name: &str,
        command: &str,
        working_dir: Option<&str>,
    ) -> Result<TaskResult, JormError> {
        // Parse cargo command and arguments
        let parts: Vec<&str> = command.split_whitespace().collect();
        if parts.is_empty() {
            return Err(JormError::ExecutionError("Empty cargo command".to_string()));
        }

        // Ensure the command starts with 'cargo'
        let (program, args) = if parts[0] == "cargo" {
            ("cargo", &parts[1..])
        } else {
            // If not starting with cargo, prepend it
            ("cargo", &parts[..])
        };

        // Create command
        let mut cmd = Command::new(program);
        cmd.args(args).stdout(Stdio::piped()).stderr(Stdio::piped());

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

        // Rust compilation can take longer, so use a longer timeout
        let timeout_duration = Duration::from_secs(300); // 5 minutes for builds

        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();

                // For Rust/Cargo, stderr often contains warnings and info, not just errors
                // So we include both stdout and stderr in output, and only treat it as error if command failed
                let combined_output = if !stdout.is_empty() && !stderr.is_empty() {
                    format!("{}\n{}", stdout, stderr)
                } else if !stdout.is_empty() {
                    stdout
                } else {
                    stderr.clone()
                };

                let error = if !success {
                    Some(format!("Cargo command failed: {}", stderr))
                } else {
                    None
                };

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