oauth2_microsoft/extensions/
internal_me_endpoint.rs

1use oauth2_client::re_exports::{
2    http::header::{ACCEPT, AUTHORIZATION},
3    serde_json, thiserror, Body, Deserialize, Endpoint, HttpError, Map, Request, Response,
4    SerdeJsonError, Serialize, Value, MIME_APPLICATION_JSON,
5};
6
7// Ref https://docs.microsoft.com/en-us/graph/api/user-get
8pub const URL: &str = "https://graph.microsoft.com/v1.0/me";
9
10//
11#[derive(Debug, Clone)]
12pub struct MeEndpoint {
13    access_token: String,
14}
15impl MeEndpoint {
16    pub fn new(access_token: impl AsRef<str>) -> Self {
17        Self {
18            access_token: access_token.as_ref().to_owned(),
19        }
20    }
21}
22
23impl Endpoint for MeEndpoint {
24    type RenderRequestError = MeEndpointError;
25
26    type ParseResponseOutput = User;
27    type ParseResponseError = MeEndpointError;
28
29    fn render_request(&self) -> Result<Request<Body>, Self::RenderRequestError> {
30        let request = Request::builder()
31            .uri(URL)
32            .header(AUTHORIZATION, format!("Bearer {}", &self.access_token))
33            .header(ACCEPT, MIME_APPLICATION_JSON)
34            .body(vec![])
35            .map_err(MeEndpointError::MakeRequestFailed)?;
36
37        Ok(request)
38    }
39
40    fn parse_response(
41        &self,
42        response: Response<Body>,
43    ) -> Result<Self::ParseResponseOutput, Self::ParseResponseError> {
44        let body = serde_json::from_slice::<User>(response.body())
45            .map_err(MeEndpointError::DeResponseBodyFailed)?;
46
47        Ok(body)
48    }
49}
50
51#[derive(Deserialize, Serialize, Debug, Clone)]
52pub struct User {
53    pub id: String,
54    #[serde(rename = "userPrincipalName")]
55    pub user_principal_name: String,
56    pub mail: Option<String>,
57    //
58    #[serde(flatten, skip_serializing_if = "Option::is_none")]
59    pub _extra: Option<Map<String, Value>>,
60}
61
62#[derive(thiserror::Error, Debug)]
63pub enum MeEndpointError {
64    #[error("MakeRequestFailed {0}")]
65    MakeRequestFailed(HttpError),
66    //
67    #[error("DeResponseBodyFailed {0}")]
68    DeResponseBodyFailed(SerdeJsonError),
69}
70
71#[cfg(test)]
72mod tests {
73    use super::*;
74
75    #[test]
76    fn de_user() {
77        match serde_json::from_str::<User>(include_str!(
78            "../../tests/response_body_json_files/me.json"
79        )) {
80            Ok(user) => {
81                assert_eq!(user.id, "f26b4f162cf4fb1f");
82            }
83            Err(err) => panic!("{err}"),
84        }
85    }
86}