harn-cli 0.7.20

CLI for the Harn programming language — run, test, REPL, format, and lint
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

use std::collections::HashMap;

use super::rpc::{error_response, A2A_VERSION_NOT_SUPPORTED};

/// The supported A2A protocol version.
const SUPPORTED_A2A_VERSION: &str = "1.0.0";

/// Parsed HTTP request including headers.
pub(super) struct ParsedRequest {
    pub(super) method: String,
    pub(super) path: String,
    pub(super) headers: HashMap<String, String>,
    pub(super) body: String,
}

/// Parse an HTTP request from raw bytes. Returns a `ParsedRequest`.
pub(super) fn parse_http_request(raw: &[u8]) -> Option<ParsedRequest> {
    let text = String::from_utf8_lossy(raw);

    let (header_section, body) = if let Some(pos) = text.find("\r\n\r\n") {
        (&text[..pos], text[pos + 4..].to_string())
    } else if let Some(pos) = text.find("\n\n") {
        (&text[..pos], text[pos + 2..].to_string())
    } else {
        return None;
    };

    let mut lines = header_section.lines();
    let request_line = lines.next()?;
    let mut parts = request_line.split_whitespace();
    let method = parts.next()?.to_string();
    let path = parts.next()?.to_string();

    let mut headers = HashMap::new();
    for line in lines {
        if let Some((key, value)) = line.split_once(':') {
            headers.insert(key.trim().to_lowercase(), value.trim().to_string());
        }
    }

    Some(ParsedRequest {
        method,
        path,
        headers,
        body,
    })
}

/// Check the A2A-Version header. Returns an error response if the version is
/// present but not supported.
pub(super) fn check_version_header(
    headers: &HashMap<String, String>,
    rpc_id: &serde_json::Value,
) -> Option<serde_json::Value> {
    if let Some(version) = headers.get("a2a-version") {
        if version != SUPPORTED_A2A_VERSION {
            return Some(error_response(
                rpc_id,
                A2A_VERSION_NOT_SUPPORTED,
                &format!(
                    "VersionNotSupportedError: requested version {version}, supported: {SUPPORTED_A2A_VERSION}"
                ),
            ));
        }
    }
    None
}