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

use serde::{Deserialize, Serialize};

/// Log metadata about this request based on the type given:
///
/// ```no_run
/// use mastodon_async::log_serde;
/// tokio_test::block_on(async {
///   let request = reqwest::get("https://example.org/").await.unwrap();
///   log::warn!(
///     status = log_serde!(request Status),
///     headers = log_serde!(request Headers);
///     "test"
///   );
/// })
/// ```
#[macro_export]
macro_rules! log_serde {
    ($response:ident $type_name:tt) => {
        log::as_serde!($crate::helpers::log::$type_name::from(&$response))
    };
}

/// Serializable form of reqwest's Status type.
#[derive(Debug, Copy, Clone, Serialize, Deserialize)]
pub struct Status {
    /// The numerical representation of the status
    pub code: u16,
    /// it's canonical reason.
    pub message: Option<&'static str>,
}

impl Status {
    /// New from reqwest's Status type (which is more useful but not
    /// serializable).
    pub fn new(status: reqwest::StatusCode) -> Self {
        Self {
            code: status.as_u16(),
            message: status.canonical_reason(),
        }
    }
}

impl From<&reqwest::Response> for Status {
    fn from(value: &reqwest::Response) -> Self {
        Self::new(value.status())
    }
}

/// Helper for logging request headers
#[derive(Debug)]
pub struct Headers(pub reqwest::header::HeaderMap);

impl Serialize for Headers {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serializer.collect_map(
            self.0
                .iter()
                .map(|(k, v)| (format!("{k:?}"), format!("{v:?}"))),
        )
    }
}

impl From<&reqwest::Response> for Headers {
    fn from(value: &reqwest::Response) -> Self {
        Headers(value.headers().clone())
    }
}