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

use crate::channel::{ClosedChannel, Receiver, Sender};
use crate::interface::Service;
use crate::message::Message;

use async_trait::async_trait;

/// Serve the public IP of the server.
/// No args are required.
pub struct PublicIp;

#[async_trait]
impl Service for PublicIp {
    async fn run(
        self: Box<Self>,
        mut input: Receiver,
        output: Sender,
    ) -> Result<(), ClosedChannel> {
        loop {
            let request = input.recv().await?;
            let response = match public_ip::addr().await {
                Some(ip_addr) => Message::response(&request).body(format!("{}", ip_addr)),
                None => {
                    let msg = "Failed to get IP address";
                    log::error!("{}", msg);
                    Message::response(&request).args(["error"]).body(msg)
                }
            };
            output.send(response).await?;
        }
    }
}