esetres 0.1.4

A self hosted file storage server.
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
49
50
51

use std::{
    net::{IpAddr, SocketAddr},
    str::FromStr,
};

use crate::{config, jwt::validate};
use axum::{
    extract::{ConnectInfo, Request},
    middleware::Next,
    response::{IntoResponse, Response},
};
use reqwest::StatusCode;

/// Checks for token presence and validates it
pub async fn authorization(request: Request, next: Next) -> Response {
    let token = if let Some(header) = request.headers().get("Authorization") {
        // remove `Bearer ` prefix to get token
        let header = header.to_str().unwrap();
        
        let t = header.trim_start_matches("Bearer ");
        if t == "" {
            return (StatusCode::UNAUTHORIZED).into_response();
        }

        t
    } else {
        return (StatusCode::UNAUTHORIZED).into_response();
    };

    if let Err(_) = validate(token.to_string()) {
        return (StatusCode::UNAUTHORIZED).into_response();
    }

    next.run(request).await
}

pub async fn from_host(
    ConnectInfo(addr): ConnectInfo<SocketAddr>,
    request: Request,
    next: Next,
) -> Response {
    let config = config::get();

    let valid_ip = IpAddr::from_str(&config.ip).unwrap();

    if addr.ip() != valid_ip {
        return (StatusCode::UNAUTHORIZED).into_response();
    }

    next.run(request).await
}