albatross-cli 2.0.3

Open source, terminal-first AI coding agent with fully transparent multi-model routing. Local (Ollama, LM Studio, MLX, llama.cpp) or cloud, your keys, one TUI.
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

use serde_json::Value;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum HookDecision {
    Block,
    Allow,
    Deny,
    Stop,
}

#[derive(Debug, Clone, Default, PartialEq)]
pub struct HookEffect {
    pub decision: Option<HookDecision>,
    pub reason: Option<String>,
    pub additional_context: Option<String>,
    pub updated_input: Option<Value>,
    pub feedback: Option<String>,
    pub warning: Option<String>,
}

pub fn parse_hook_effect(exit_code: i32, stdout: &str, stderr: &str) -> HookEffect {
    let mut effect = HookEffect::default();
    let stdout = stdout.trim();
    if !stdout.is_empty() {
        match serde_json::from_str::<Value>(stdout) {
            Ok(value) => apply_effect_json(&mut effect, &value),
            Err(e) => {
                effect.warning = Some(format!("invalid hook JSON: {e}"));
            }
        }
    }
    if exit_code == 2 && effect.decision.is_none() {
        effect.decision = Some(HookDecision::Block);
        let reason = stderr.trim();
        if !reason.is_empty() {
            effect.reason = Some(reason.to_string());
        }
    } else if exit_code != 0 && effect.warning.is_none() {
        effect.warning = Some(format!("hook exited with status {exit_code}"));
    }
    effect
}

fn apply_effect_json(effect: &mut HookEffect, value: &Value) {
    let Some(obj) = value.as_object() else {
        return;
    };
    effect.decision = obj
        .get("decision")
        .and_then(Value::as_str)
        .and_then(parse_decision);
    effect.reason = obj
        .get("reason")
        .and_then(Value::as_str)
        .map(str::to_string);
    effect.additional_context = obj
        .get("additionalContext")
        .and_then(Value::as_str)
        .map(str::to_string);
    effect.updated_input = obj.get("updatedInput").cloned();
    effect.feedback = obj
        .get("feedback")
        .and_then(Value::as_str)
        .map(str::to_string);
}

fn parse_decision(raw: &str) -> Option<HookDecision> {
    match raw {
        "block" => Some(HookDecision::Block),
        "allow" => Some(HookDecision::Allow),
        "deny" => Some(HookDecision::Deny),
        "stop" => Some(HookDecision::Stop),
        _ => None,
    }
}