ap-relay 0.3.121

A simple activitypub relay
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

use crate::{
    data::MediaCache,
    error::Error,
    requests::{BreakerStrategy, Requests},
    stream::limit_stream,
};
use actix_web::{body::BodyStream, web, HttpResponse};
use uuid::Uuid;

// 16 MB
const IMAGE_SIZE_LIMIT: usize = 16 * 1024 * 1024;

#[tracing::instrument(name = "Media", skip(media, requests))]
pub(crate) async fn route(
    media: web::Data<MediaCache>,
    requests: web::Data<Requests>,
    uuid: web::Path<Uuid>,
) -> Result<HttpResponse, Error> {
    let uuid = uuid.into_inner();

    if let Some(url) = media.get_url(uuid).await? {
        let res = requests
            .fetch_response(&url, BreakerStrategy::Allow404AndBelow)
            .await?;

        let mut response = HttpResponse::build(crate::http1::status_to_http02(res.status()));

        for (name, value) in res.headers().iter().filter(|(h, _)| *h != "connection") {
            response.insert_header((
                crate::http1::name_to_http02(name),
                crate::http1::value_to_http02(value),
            ));
        }

        return Ok(response.body(BodyStream::new(limit_stream(
            res.bytes_stream(),
            IMAGE_SIZE_LIMIT,
        ))));
    }

    Ok(HttpResponse::NotFound().finish())
}