diff options
Diffstat (limited to 'src/event/routes/get.rs')
| -rw-r--r-- | src/event/routes/get.rs | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/src/event/routes/get.rs b/src/event/routes/get.rs new file mode 100644 index 0000000..357845a --- /dev/null +++ b/src/event/routes/get.rs @@ -0,0 +1,83 @@ +use axum::{ + extract::State, + response::{ + self, + sse::{self, Sse}, + IntoResponse, + }, +}; +use axum_extra::extract::Query; +use futures::stream::{Stream, StreamExt as _}; + +use crate::{ + app::App, + error::{Internal, Unauthorized}, + event::{extract::LastEventId, Event, ResumePoint, Sequence, Sequenced as _}, + token::{app::ValidateError, extract::Identity}, +}; + +pub async fn handler( + State(app): State<App>, + identity: Identity, + last_event_id: Option<LastEventId<Sequence>>, + Query(query): Query<QueryParams>, +) -> Result<Response<impl Stream<Item = Event> + std::fmt::Debug>, Error> { + let resume_at = last_event_id + .map(LastEventId::into_inner) + .or(query.resume_point); + + let stream = app.events().subscribe(resume_at).await?; + let stream = app.tokens().limit_stream(identity.token, stream).await?; + + Ok(Response(stream)) +} + +#[derive(Default, serde::Deserialize)] +pub struct QueryParams { + pub resume_point: ResumePoint, +} + +#[derive(Debug)] +pub struct Response<S>(pub S); + +impl<S> IntoResponse for Response<S> +where + S: Stream<Item = Event> + Send + 'static, +{ + fn into_response(self) -> response::Response { + let Self(stream) = self; + let stream = stream.map(sse::Event::try_from); + Sse::new(stream) + .keep_alive(sse::KeepAlive::default()) + .into_response() + } +} + +impl TryFrom<Event> for sse::Event { + type Error = serde_json::Error; + + fn try_from(event: Event) -> Result<Self, Self::Error> { + let id = serde_json::to_string(&event.sequence())?; + let data = serde_json::to_string_pretty(&event)?; + + let event = Self::default().id(id).data(data); + + Ok(event) + } +} + +#[derive(Debug, thiserror::Error)] +#[error(transparent)] +pub enum Error { + Database(#[from] sqlx::Error), + Validate(#[from] ValidateError), +} + +impl IntoResponse for Error { + fn into_response(self) -> response::Response { + match self { + Self::Validate(ValidateError::InvalidToken) => Unauthorized.into_response(), + other => Internal::from(other).into_response(), + } + } +} |
