user-agent-parser 0.4.0

A parser to get the product, OS, device, cpu, and engine information from a user agent, inspired by https://github.com/faisalman/ua-parser-js and https://github.com/ua-parser/uap-core
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use std::borrow::Cow;

use rocket::{
    outcome::Outcome,
    request::{FromRequest, Outcome as OutcomeResult, Request},
};

use crate::{UserAgentParser, models::*};

const NOT_MANAGED: &str = "a `UserAgentParser` is not managed by Rocket; add \
                           `.manage(UserAgentParser::from_path(..))` when building the instance";

fn from_request_user_agent<'r>(request: &'r Request<'_>) -> UserAgent<'r> {
    let user_agent: Option<Cow<'r, str>> =
        request.headers().get("user-agent").next().map(Cow::from);

    UserAgent {
        user_agent,
    }
}

#[rocket::async_trait]
impl<'r> FromRequest<'r> for UserAgent<'r> {
    type Error = ();

    async fn from_request(request: &'r Request<'_>) -> OutcomeResult<Self, Self::Error> {
        Outcome::Success(from_request_user_agent(request))
    }
}

#[rocket::async_trait]
impl<'r> FromRequest<'r> for &UserAgent<'r> {
    type Error = ();

    async fn from_request(request: &'r Request<'_>) -> OutcomeResult<Self, Self::Error> {
        let cache = request.local_cache(|| from_request_user_agent(request).into_owned());

        Outcome::Success(cache)
    }
}

/// Implements both the owned and the cached-reference request guard of one model.
macro_rules! impl_from_request {
    ($model:ident, $parse:ident) => {
        #[rocket::async_trait]
        impl<'r> FromRequest<'r> for $model<'r> {
            type Error = ();

            async fn from_request(request: &'r Request<'_>) -> OutcomeResult<Self, Self::Error> {
                Outcome::Success(parse(request))
            }
        }

        #[rocket::async_trait]
        impl<'r> FromRequest<'r> for &$model<'r> {
            type Error = ();

            async fn from_request(request: &'r Request<'_>) -> OutcomeResult<Self, Self::Error> {
                let cache = request.local_cache_async(async { parse(request).into_owned() }).await;

                Outcome::Success(cache)
            }
        }

        fn parse<'r>(request: &'r Request<'_>) -> $model<'r> {
            let user_agent_parser = request.rocket().state::<UserAgentParser>().expect(NOT_MANAGED);

            match request.headers().get("user-agent").next() {
                Some(user_agent) => user_agent_parser.$parse(user_agent),
                None => $model::default(),
            }
        }
    };
}

// Each model gets its own module, so that the `parse` function of the macro does not clash.
mod product {
    use super::*;

    impl_from_request!(Product, parse_product);
}

mod os {
    use super::*;

    impl_from_request!(OS, parse_os);
}

mod device {
    use super::*;

    impl_from_request!(Device, parse_device);
}

mod cpu {
    use super::*;

    impl_from_request!(CPU, parse_cpu);
}

mod engine {
    use super::*;

    impl_from_request!(Engine, parse_engine);
}