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

use crate::extractor::token::AnnilClaim;
use crate::provider::AnnilProvider;
use crate::state::AnnilState;
use anni_provider::AnniProvider;
use axum::http::header::{ETAG, IF_NONE_MATCH};
use axum::http::{HeaderMap, StatusCode};
use axum::response::{IntoResponse, Response};
use axum::{Extension, Json};
use std::collections::HashSet;
use std::sync::Arc;

/// Get available albums of current annil server
pub async fn albums<P>(
    claims: AnnilClaim,
    Extension(provider): Extension<Arc<AnnilProvider<P>>>,
    Extension(data): Extension<Arc<AnnilState>>,
    headers: HeaderMap,
) -> Response
where
    P: AnniProvider + Send + Sync,
{
    match claims {
        AnnilClaim::User(_) => {
            let etag_now = data.etag.read().await.to_string();

            if let Some(Ok(mut etag)) = headers.get(IF_NONE_MATCH).map(|v| v.to_str()) {
                if etag.starts_with("W/") {
                    etag = &etag[2..];
                }
                if etag == etag_now {
                    return StatusCode::NOT_MODIFIED.into_response();
                }
            }

            // users can get real album list
            let provider = provider.read().await;
            let albums = provider.albums().await.unwrap_or(HashSet::new());
            ([(ETAG, etag_now)], Json(albums)).into_response()
        }
        AnnilClaim::Share(share) => {
            // guests can only get album list defined in jwt
            Json(share.audios.keys().collect::<Vec<_>>()).into_response()
        }
    }
}