cooklang-chef 0.8.3

CLI to manage cooklang recipes
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

use axum::{
    extract::State,
    response::{
        sse::{Event, KeepAlive},
        Sse,
    },
};
use futures::{Stream, TryStreamExt};
use tokio_stream::wrappers::{errors::BroadcastStreamRecvError, BroadcastStream};

use crate::serve::{async_index::Update, S};

use super::clean_path;

pub async fn sse_updates(
    State(state): State<S>,
) -> Sse<impl Stream<Item = Result<Event, BroadcastStreamRecvError>>> {
    let base_path = state.base_path.clone();
    let stream = BroadcastStream::new(state.updates_stream.resubscribe()).map_ok(move |updt| {
        let e = Event::default();
        let p = |path| clean_path(path, &base_path);
        match updt {
            Update::Modified { path } => e.event("modified").data(p(&path)),
            Update::Added { path } => e.event("added").data(p(&path)),
            Update::Deleted { path } => e.event("deleted").data(p(&path)),
            Update::Renamed { from, to } => e
                .event("renamed")
                .json_data(serde_json::json!({
                    "from": p(&from),
                    "to": p(&to)
                }))
                .unwrap(),
        }
    });

    Sse::new(stream).keep_alive(KeepAlive::default())
}