annil 0.2.0

A basic implementation of annil protocol.
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

use axum::body::StreamBody;
use axum::extract::Path;
use axum::http::header::{CACHE_CONTROL, CONTENT_TYPE};
use axum::http::StatusCode;
use axum::response::{IntoResponse, Response};
use axum::Extension;
use std::num::NonZeroU8;
use std::sync::Arc;

use crate::provider::AnnilProvider;
use anni_provider::AnniProvider;
use serde::Deserialize;
use tokio_util::io::ReaderStream;
use uuid::Uuid;

#[derive(Deserialize)]
pub struct CoverPath {
    album_id: Uuid,
    disc_id: Option<NonZeroU8>,
}

/// Get audio cover of an album with {album_id} and optional {disc_id}
pub async fn cover<P>(
    Path(CoverPath { album_id, disc_id }): Path<CoverPath>,
    Extension(provider): Extension<Arc<AnnilProvider<P>>>,
) -> Response
where
    P: AnniProvider + Send + Sync,
{
    let provider = provider.read().await;
    let album_id = album_id.to_string();

    if !provider.has_album(&album_id).await {
        return (StatusCode::NOT_FOUND, [(CACHE_CONTROL, "private")]).into_response();
    }

    match provider.get_cover(&album_id, disc_id).await {
        Ok(cover) => (
            ([
                (CONTENT_TYPE, "image/jpeg"),
                (CACHE_CONTROL, "public, max-age=31536000"),
            ]),
            StreamBody::new(ReaderStream::new(cover)),
        )
            .into_response(),
        Err(_) => (StatusCode::NOT_FOUND, [(CACHE_CONTROL, "private")]).into_response(),
    }
}