revolt-database 0.15.1

Revolt Backend: Database Implementation
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
52
53
54
55
56

#[cfg(feature = "rocket-impl")]
pub mod rocket {
    use revolt_config::config;
    use rocket::Request;

    pub fn to_ip(request: &'_ Request<'_>) -> String {
        request
            .client_ip()
            .map(|x| x.to_string())
            .unwrap_or_default()
    }

    /// Find the actual IP of the client
    pub async fn to_real_ip(request: &'_ Request<'_>) -> String {
        if config().await.api.security.trust_cloudflare {
            request
                .headers()
                .get_one("CF-Connecting-IP")
                .map(|x| x.to_string())
                .unwrap_or_else(|| to_ip(request))
        } else {
            to_ip(request)
        }
    }
}

#[cfg(feature = "axum-impl")]
pub mod axum {
    use axum::{
        extract::ConnectInfo,
        http::request::Parts,
    };
    use revolt_config::config;
    use std::net::SocketAddr;

    pub fn to_ip(parts: &Parts) -> String {
        parts
            .extensions
            .get::<ConnectInfo<SocketAddr>>()
            .map(|info| info.ip().to_string())
            .unwrap_or_default()
    }

    /// Find the actual IP of the client
    pub async fn to_real_ip(parts: &Parts) -> String {
        if config().await.api.security.trust_cloudflare {
            parts
                .headers
                .get("CF-Connecting-IP")
                .map(|x| x.to_str().unwrap().to_string())
                .unwrap_or_else(|| to_ip(parts))
        } else {
            to_ip(parts)
        }
    }
}