mlua-swarm-server 0.1.2

HTTP + WebSocket server for mlua-swarm (task API, Blueprint store, Operator WS sessions).
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

//! HTTP surface for the `/v1/enhance/log` axis.
//!
//! - `GET /v1/enhance/log?blueprint_id=<id>` → list of `LogEntry`s tied to
//!   that BP, sorted by `ts` ascending. If `blueprint_id` is omitted, returns
//!   all entries.
//! - `GET /v1/enhance/log/:issue_id` → a single `LogEntry`. NotFound → 404.
//!
//! Shares an `EnhanceLogStore` trait object (= pass in the same instance as
//! `EnhanceApplication` via `Arc`).

use axum::{
    extract::{Path, Query, State},
    http::StatusCode,
    response::IntoResponse,
    routing::get,
    Json, Router,
};
use mlua_swarm::blueprint::store::BlueprintId;
use mlua_swarm::store::enhance_log::{EnhanceLogEntry, EnhanceLogStore, EnhanceLogStoreError};
use mlua_swarm::store::issue::IssueId;
use serde::Deserialize;
use std::sync::Arc;

/// Query params for `GET /v1/enhance/log`.
#[derive(Debug, Deserialize)]
pub struct ListQuery {
    /// When set, restricts the listing to entries tied to this Blueprint id.
    /// `None` returns all entries.
    pub blueprint_id: Option<String>,
}

/// Builds the `/v1/enhance/log*` router backed by the given `EnhanceLogStore`.
pub fn build_enhance_log_router(store: Arc<dyn EnhanceLogStore>) -> Router {
    Router::new()
        .route("/v1/enhance/log", get(list_entries))
        .route("/v1/enhance/log/:issue_id", get(get_entry))
        .with_state(store)
}

async fn list_entries(
    State(store): State<Arc<dyn EnhanceLogStore>>,
    Query(q): Query<ListQuery>,
) -> Result<Json<Vec<EnhanceLogEntry>>, (StatusCode, String)> {
    let entries = match q.blueprint_id {
        Some(bp) => store
            .list_by_blueprint(&BlueprintId::new(bp))
            .await
            .map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()))?,
        None => store
            .list_all()
            .await
            .map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()))?,
    };
    Ok(Json(entries))
}

async fn get_entry(
    State(store): State<Arc<dyn EnhanceLogStore>>,
    Path(issue_id): Path<String>,
) -> impl IntoResponse {
    match store.get(&IssueId::new(issue_id)).await {
        Ok(e) => (StatusCode::OK, Json(e)).into_response(),
        Err(EnhanceLogStoreError::NotFound(_)) => (
            StatusCode::NOT_FOUND,
            "enhance log entry not found".to_string(),
        )
            .into_response(),
        Err(e) => (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()).into_response(),
    }
}