ntfy 0.9.1

Library for sending notifications over ntfy
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

// Copyright (c) 2022 Yuki Kishimoto
// Distributed under the MIT software license

#[cfg(all(feature = "socks", not(target_arch = "wasm32")))]
use ureq::Proxy;
use ureq::config::ConfigBuilder;
use ureq::http::{HeaderValue, Request, Response};
use ureq::middleware::{Middleware, MiddlewareNext};
use ureq::typestate::AgentScope;
use ureq::{Agent, Body, SendBody};
use url::Url;

use super::builder::DispatcherBuilder;
use crate::auth::Auth;
use crate::error::Error;
use crate::payload::Payload;

struct AuthMiddleware {
    header: HeaderValue,
}

impl AuthMiddleware {
    fn new(auth: Auth) -> Result<Self, Error> {
        Ok(Self {
            header: HeaderValue::from_str(&auth.header_value())?,
        })
    }
}

impl Middleware for AuthMiddleware {
    fn handle(
        &self,
        mut req: Request<SendBody>,
        next: MiddlewareNext,
    ) -> Result<Response<Body>, ureq::Error> {
        req.headers_mut()
            .insert("Authorization", self.header.clone());
        next.handle(req)
    }
}

/// Blocking dispatcher
#[derive(Debug, Clone)]
pub struct Blocking {
    client: Agent,
}

impl Blocking {
    #[inline]
    pub(crate) fn new(builder: DispatcherBuilder) -> Result<Self, Error> {
        Self::new_with_client(builder, Agent::config_builder())
    }

    pub(crate) fn new_with_client(
        builder: DispatcherBuilder,
        mut client: ConfigBuilder<AgentScope>,
    ) -> Result<Self, Error> {
        if let Some(auth) = builder.auth {
            // Construct middleware
            let middleware = AuthMiddleware::new(auth)?;

            // Set the authorization headers of every request using a middleware function
            client = client.middleware(middleware);
        }

        #[cfg(all(feature = "socks", not(target_arch = "wasm32")))]
        if let Some(proxy) = builder.proxy {
            let proxy = Proxy::new(&proxy)?;
            client = client.proxy(Some(proxy));
        }

        Ok(Self {
            client: client.build().into(),
        })
    }

    pub(crate) fn send(&self, url: &Url, payload: &Payload) -> Result<(), Error> {
        // Build request
        let builder = self.client.post(url.as_str());

        // Send request
        let res: Response<Body> = builder.send_json(payload)?;

        // Get full response text
        let text = res.into_body().read_to_string()?;

        if text.is_empty() {
            return Err(Error::EmptyResponse);
        }

        // TODO: check the text?

        Ok(())
    }
}