crabmate 0.4.0

Rust AI agent: OpenAI-compatible chat/completions, function calling, HTTP serve, ops CLI
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

//! 安全相关工具：cargo audit / cargo deny

use std::path::Path;
use std::process::Command;

use super::output_util;

const MAX_OUTPUT_LINES: usize = 800;

pub fn cargo_audit(args_json: &str, workspace_root: &Path, max_output_len: usize) -> String {
    let v = match crate::cm_tools::tools::parse_args_json(args_json) {
        Ok(v) => v,
        Err(e) => return e,
    };
    if !workspace_root.join("Cargo.toml").is_file() {
        return "错误：当前工作目录未找到 Cargo.toml".to_string();
    }
    let deny_warnings = v
        .get("deny_warnings")
        .and_then(|x| x.as_bool())
        .unwrap_or(false);
    let json = v.get("json").and_then(|x| x.as_bool()).unwrap_or(false);

    let mut cmd = Command::new("cargo");
    cmd.arg("audit");
    if deny_warnings {
        cmd.arg("--deny").arg("warnings");
    }
    if json {
        cmd.arg("--json");
    }
    cmd.current_dir(workspace_root);
    run_and_format(cmd, max_output_len, "cargo audit")
}

pub fn cargo_deny(args_json: &str, workspace_root: &Path, max_output_len: usize) -> String {
    let v = match crate::cm_tools::tools::parse_args_json(args_json) {
        Ok(v) => v,
        Err(e) => return e,
    };
    if !workspace_root.join("Cargo.toml").is_file() {
        return "错误：当前工作目录未找到 Cargo.toml".to_string();
    }
    let checks = v
        .get("checks")
        .and_then(|x| x.as_str())
        .map(str::trim)
        .filter(|s| !s.is_empty())
        .unwrap_or("advisories licenses bans sources");
    let all_features = v
        .get("all_features")
        .and_then(|x| x.as_bool())
        .unwrap_or(false);

    let mut cmd = Command::new("cargo");
    cmd.arg("deny").arg("check");
    for c in checks.split_whitespace() {
        cmd.arg(c);
    }
    if all_features {
        cmd.arg("--all-features");
    }
    cmd.current_dir(workspace_root);
    let out = run_and_format(cmd, max_output_len, "cargo deny check");
    if out.contains("no such command: `deny`") {
        return "cargo deny: 未安装 cargo-deny，请先运行 `cargo install cargo-deny`".to_string();
    }
    out
}

fn run_and_format(mut cmd: Command, max_output_len: usize, title: &str) -> String {
    match cmd.output() {
        Ok(output) => {
            let status = output.status.code().unwrap_or(-1);
            let body = output_util::merge_process_output(
                &output,
                output_util::ProcessOutputMerge::ConcatStdoutStderr,
            );
            if status != 0 && body.contains("no such command: `audit`") {
                return "cargo audit: 未安装 cargo-audit，请先运行 `cargo install cargo-audit`"
                    .to_string();
            }
            output_util::format_exited_command_output(
                title,
                status,
                &body,
                max_output_len,
                MAX_OUTPUT_LINES,
            )
        }
        Err(e) => output_util::format_spawn_error_with_program(
            title,
            &e,
            output_util::CommandSpawnErrorStyle::ExecuteFailed,
            Some("cargo"),
        ),
    }
}