vs-daemon 0.1.9

Daemon that owns the browser engine for vibesurfer.
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

//! Wire handlers for session/page lifecycle.

use vs_protocol::{ErrorCode, Request, ResponseHead, StateToken};

use super::helpers::{flag_value, format_daemon_error, format_error, require_session};
use crate::daemon::{
    CloseResponse, Daemon, OpenResponse, SessionCloseResponse, SessionOpenResponse,
};

pub(super) fn handle_session_open(daemon: &Daemon, req: &Request) -> String {
    let policy = flag_value(req, "policy");
    match daemon.session_open(policy.as_deref()) {
        Ok(SessionOpenResponse { session_id, token }) => {
            let head = ResponseHead::ok(token).encode();
            format!("{head}{session_id}\n")
        }
        Err(e) => format_daemon_error(&e),
    }
}

pub(super) fn handle_session_close(daemon: &Daemon, req: &Request) -> String {
    let Some(session_id) = req.args.first().cloned() else {
        return format_error(
            ErrorCode::BadRequest,
            vec!["vs_session_close: missing session id".into()],
        );
    };
    match daemon.session_close(&session_id) {
        Ok(SessionCloseResponse) => ResponseHead::ok(StateToken::ZERO).encode(),
        Err(e) => format_daemon_error(&e),
    }
}

pub(super) fn handle_open(daemon: &Daemon, req: &Request) -> String {
    let session_id = match require_session(req) {
        Ok(s) => s,
        Err(msg) => return format_error(ErrorCode::BadRequest, vec![msg]),
    };
    let Some(url) = req.args.first().cloned() else {
        return format_error(ErrorCode::BadRequest, vec!["vs_open: missing url".into()]);
    };
    match daemon.open(&session_id, &url) {
        Ok(OpenResponse {
            page_id,
            token,
            warnings,
        }) => {
            let mut head = ResponseHead::ok(token);
            head.warnings = warnings;
            let open_wire = format!("{}{page_id}\n", head.encode());

            // M5.5 PR2: `vs_open --view` composite. Append a fresh view
            // of the just-opened page. Page id is resolved
            // automatically from the open response — agents do not pass
            // it. View is never cached; idempotent open + fresh view.
            if req.flags.contains_key("view") {
                let view_req = Request::new("vs_view")
                    .arg(page_id.clone())
                    .flag_value("session", session_id.clone());
                let view_wire = super::page_ops::handle_view(daemon, &view_req);
                return format!("{open_wire}\n{view_wire}");
            }
            open_wire
        }
        Err(e) => format_daemon_error(&e),
    }
}

pub(super) fn handle_close(daemon: &Daemon, req: &Request) -> String {
    let session_id = match require_session(req) {
        Ok(s) => s,
        Err(msg) => return format_error(ErrorCode::BadRequest, vec![msg]),
    };
    let Some(page_id) = req.args.first().cloned() else {
        return format_error(
            ErrorCode::BadRequest,
            vec!["vs_close: missing page id".into()],
        );
    };
    match daemon.close(&session_id, &page_id) {
        Ok(CloseResponse) => ResponseHead::ok(StateToken::ZERO).encode(),
        Err(e) => format_daemon_error(&e),
    }
}