velaclaw 0.3.0

Protocol-driven autonomous AI agent runtime with intelligent model selection and multi-model negotiation.
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

//! Embedded Web Chat SPA served at `/chat` (VL-UI-005).
//! 在 `/chat` 提供内嵌 Web Chat SPA（VL-UI-005）。

use axum::extract::Request;
use axum::http::{header, StatusCode};
use axum::response::{IntoResponse, Response};
use rust_embed::Embed;

#[derive(Embed)]
#[folder = "ui-chat/dist/"]
struct ChatUiAssets;

fn resolve_embed_path(uri_path: &str) -> String {
    let trimmed = uri_path
        .strip_prefix("/chat")
        .unwrap_or(uri_path)
        .trim_start_matches('/');
    if trimmed.is_empty() {
        "index.html".to_string()
    } else {
        trimmed.to_string()
    }
}

fn content_type(path: &str) -> &'static str {
    if path.ends_with(".js") {
        "application/javascript"
    } else if path.ends_with(".css") {
        "text/css"
    } else if path.ends_with(".svg") {
        "image/svg+xml"
    } else if path.ends_with(".png") {
        "image/png"
    } else if path.ends_with(".woff2") {
        "font/woff2"
    } else {
        "text/html; charset=utf-8"
    }
}

fn serve_embedded(relative_path: &str) -> Response {
    let Some(file) = ChatUiAssets::get(relative_path) else {
        if relative_path != "index.html" {
            return serve_embedded("index.html");
        }
        return (
            StatusCode::NOT_FOUND,
            [(header::CONTENT_TYPE, "text/plain; charset=utf-8")],
            "chat UI asset not found",
        )
            .into_response();
    };

    let ct = content_type(relative_path);
    (
        StatusCode::OK,
        [(header::CONTENT_TYPE, ct)],
        file.data.into_owned(),
    )
        .into_response()
}

/// Serve SPA and static assets under `/chat` and `/chat/*`.
pub async fn handle_chat_ui(req: Request) -> impl IntoResponse {
    serve_embedded(&resolve_embed_path(req.uri().path()))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn embedded_index_exists() {
        assert!(ChatUiAssets::get("index.html").is_some());
    }

    #[test]
    fn maps_chat_root_to_index() {
        assert_eq!(resolve_embed_path("/chat"), "index.html");
        assert_eq!(resolve_embed_path("/chat/"), "index.html");
        assert_eq!(resolve_embed_path("/chat/assets/app.js"), "assets/app.js");
    }
}