graphwalker_restful/
lib.rs

1pub mod actor;
2mod authoring_dto;
3mod authoring_rest;
4pub mod rest;
5pub mod session;
6pub mod websocket;
7
8use std::net::SocketAddr;
9
10use axum::extract::{FromRef, State, WebSocketUpgrade};
11use axum::response::IntoResponse;
12use axum::routing::{get, post, put};
13use axum::Router;
14use tower_http::cors::CorsLayer;
15
16use crate::actor::spawn_machine_thread;
17use crate::session::SessionManager;
18
19pub use authoring_rest::MAX_AUTHORING_BODY_BYTES;
20
21#[derive(Clone)]
22pub(crate) struct RestApplicationState {
23    execution: rest::RestState,
24    drafts: graphwalker_service::DraftRegistry,
25}
26
27impl FromRef<RestApplicationState> for rest::RestState {
28    fn from_ref(state: &RestApplicationState) -> Self {
29        state.execution.clone()
30    }
31}
32
33impl FromRef<RestApplicationState> for graphwalker_service::DraftRegistry {
34    fn from_ref(state: &RestApplicationState) -> Self {
35        state.drafts.clone()
36    }
37}
38
39pub async fn start_rest_server(
40    port: u16,
41    seed: Option<u64>,
42) -> Result<(), Box<dyn std::error::Error>> {
43    let app = rest_router(seed);
44    serve_rest(port, app).await
45}
46
47pub async fn start_rest_server_with_model(
48    port: u16,
49    seed: Option<u64>,
50    initial_model: Option<String>,
51) -> Result<(), Box<dyn std::error::Error>> {
52    let app = match initial_model {
53        Some(json_body) => {
54            let machine_tx = crate::actor::spawn_machine_thread_with_model(json_body, seed)
55                .map_err(|error| std::io::Error::new(std::io::ErrorKind::InvalidInput, error))?;
56            build_rest_router(RestApplicationState {
57                execution: rest::RestState {
58                    machine_tx,
59                    default_seed: seed,
60                },
61                drafts: graphwalker_service::DraftRegistry::default(),
62            })
63        }
64        None => rest_router(seed),
65    };
66    serve_rest(port, app).await
67}
68
69async fn serve_rest(port: u16, app: Router) -> Result<(), Box<dyn std::error::Error>> {
70    let addr = SocketAddr::from(([0, 0, 0, 0], port));
71    eprintln!("GraphWalker REST server listening on http://{}", addr);
72
73    let listener = tokio::net::TcpListener::bind(addr).await?;
74    axum::serve(listener, app).await?;
75    Ok(())
76}
77
78/// Build the REST application with independent legacy execution and draft state.
79pub fn rest_router(default_seed: Option<u64>) -> Router {
80    rest_router_with_drafts(default_seed, graphwalker_service::DraftRegistry::default())
81}
82
83/// Build the REST application with a caller-supplied draft registry.
84pub fn rest_router_with_drafts(
85    default_seed: Option<u64>,
86    drafts: graphwalker_service::DraftRegistry,
87) -> Router {
88    build_rest_router(RestApplicationState {
89        execution: rest::RestState {
90            machine_tx: spawn_machine_thread(),
91            default_seed,
92        },
93        drafts,
94    })
95}
96
97pub async fn start_websocket_server(port: u16) -> Result<(), Box<dyn std::error::Error>> {
98    let session_mgr = SessionManager::new();
99    let app = build_websocket_router(session_mgr);
100    let addr = SocketAddr::from(([0, 0, 0, 0], port));
101    eprintln!("GraphWalker WebSocket server listening on ws://{}", addr);
102
103    let listener = tokio::net::TcpListener::bind(addr).await?;
104    axum::serve(listener, app).await?;
105    Ok(())
106}
107
108fn build_rest_router(state: RestApplicationState) -> Router {
109    Router::new()
110        .route("/graphwalker/load", post(rest::load))
111        .route("/graphwalker/hasNext", get(rest::has_next))
112        .route("/graphwalker/getNext", get(rest::get_next))
113        .route("/graphwalker/getData", get(rest::get_data))
114        .route("/graphwalker/setData/{script}", put(rest::set_data))
115        .route("/graphwalker/restart", put(rest::restart))
116        .route("/graphwalker/getStatistics", get(rest::get_statistics))
117        .merge(authoring_rest::routes())
118        .layer(CorsLayer::permissive())
119        .with_state(state)
120}
121
122fn build_websocket_router(session_mgr: SessionManager) -> Router {
123    Router::new()
124        .route("/", get(ws_upgrade_handler))
125        .route("/graphwalker", get(ws_upgrade_handler))
126        .layer(CorsLayer::permissive())
127        .with_state(session_mgr)
128}
129
130async fn ws_upgrade_handler(
131    ws: WebSocketUpgrade,
132    State(session_mgr): State<SessionManager>,
133) -> impl IntoResponse {
134    ws.on_upgrade(move |socket| websocket::handle_socket(socket, session_mgr))
135}
graphwalker_restful/lib.rs

graphwalker_restful/
lib.rs
