volition-core 0.2.0

Core library for the Volition AI agent
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
114
115
116
117
118

// volition-agent-core/src/tools/git.rs

use super::CommandOutput;
use crate::utils::truncate_string; // <-- Import the helper
use anyhow::{Context, Result};
use std::path::Path;
use std::process::{Command, Stdio};
use tracing::{debug, info};

pub async fn execute_git_command(
    command_name: &str,
    command_args: &[String],
    working_dir: &Path,
) -> Result<CommandOutput> {
    let full_command_log = format!("git {} {}", command_name, command_args.join(" "));
    // Truncate command for logging
    let command_display = truncate_string(&full_command_log, 60);
    info!(
        "Executing git command: {} in {:?}",
        command_display, // <-- Use truncated version
        working_dir
    );

    let output = Command::new("git")
        .current_dir(working_dir)
        .arg(command_name)
        .args(command_args)
        .stdout(Stdio::piped())
        .stderr(Stdio::piped())
        .output()
        .with_context(|| format!("Failed to execute git command: {}", full_command_log))?;

    let stdout = String::from_utf8_lossy(&output.stdout).to_string();
    let stderr = String::from_utf8_lossy(&output.stderr).to_string();
    let status = output.status.code().unwrap_or(-1);

    debug!("git {} exit status: {}", full_command_log, status);

    Ok(CommandOutput {
        status,
        stdout,
        stderr,
    })
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::fs;
    use std::path::PathBuf;
    use std::process::Command;
    use tempfile::tempdir;
    use tokio;

    fn setup_git_repo() -> Result<PathBuf> {
        let dir = tempdir()?.into_path();
        Command::new("git").current_dir(&dir).arg("init").output()?;
        Command::new("git")
            .current_dir(&dir)
            .args(&["config", "user.email", "test@example.com"])
            .output()?;
        Command::new("git")
            .current_dir(&dir)
            .args(&["config", "user.name", "Test User"])
            .output()?;
        fs::write(dir.join("README.md"), "Initial commit")?;
        Command::new("git")
            .current_dir(&dir)
            .arg("add")
            .arg("README.md")
            .output()?;
        Command::new("git")
            .current_dir(&dir)
            .arg("commit")
            .arg("-m")
            .arg("Initial commit")
            .output()?;
        Ok(dir)
    }

    #[tokio::test]
    async fn test_execute_git_status_clean() {
        let working_dir = setup_git_repo().expect("Failed to setup git repo");
        let result = execute_git_command("status", &[], &working_dir).await;
        assert!(result.is_ok(), "git status failed: {:?}", result.err());
        let output = result.unwrap();
        println!("Output: {:?}", output);
        assert_eq!(output.status, 0);
        assert!(
            output
                .stdout
                .contains("nothing to commit, working tree clean")
        );
    }

    #[tokio::test]
    async fn test_execute_git_log_initial() {
        let working_dir = setup_git_repo().expect("Failed to setup git repo");
        let result = execute_git_command("log", &["-1".to_string()], &working_dir).await;
        assert!(result.is_ok(), "git log failed: {:?}", result.err());
        let output = result.unwrap();
        println!("Output: {:?}", output);
        assert_eq!(output.status, 0);
        assert!(output.stdout.contains("Initial commit"));
    }

    #[tokio::test]
    async fn test_execute_git_diff_fail() {
        let working_dir = setup_git_repo().expect("Failed to setup git repo");
        let result =
            execute_git_command("diff", &["nonexistentcommit".to_string()], &working_dir).await;
        assert!(result.is_ok());
        let output = result.unwrap();
        println!("Output: {:?}", output);
        assert_ne!(output.status, 0);
        assert!(output.stderr.contains("fatal: ambiguous argument"));
    }
}