ockam_command 0.150.0

End-to-end encryption and mutual authentication for distributed applications.
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

use crate::{Command, CommandGlobalOpts};
use async_trait::async_trait;
use clap::Args;
use ockam::udp::{RendezvousClient, UdpBindArguments, UdpBindOptions, UdpTransport, UDP};
use ockam_api::colors::color_primary;
use ockam_api::{fmt_log, DefaultAddress};
use ockam_core::{route, Address};
use ockam_node::Context;

/// Get my public UDP address from  Rendezvous service.
#[derive(Clone, Debug, Args)]
pub struct GetMyAddressCommand {
    /// The address of the Rendezvous service.
    #[arg(display_order = 900)]
    pub rendezvous_address: Option<String>,
}

#[async_trait]
impl Command for GetMyAddressCommand {
    const NAME: &'static str = "rendezvous get-my-address";

    async fn run(self, ctx: &Context, opts: CommandGlobalOpts) -> crate::Result<()> {
        let udp = UdpTransport::get_or_create(ctx)?;
        let bind = udp
            .bind(
                UdpBindArguments::new().with_bind_address("0.0.0.0:0")?,
                UdpBindOptions::new(),
            )
            .await?;
        let rendezvous_address = self
            .rendezvous_address
            .map(|address| Address::new(UDP, address.into_bytes()))
            .unwrap_or(DefaultAddress::get_rendezvous_server_address());
        let client = RendezvousClient::new(
            &bind,
            route![rendezvous_address, DefaultAddress::RENDEZVOUS_SERVICE],
        );

        let my_address = client.get_my_address(ctx).await?;

        opts.terminal.write_line(fmt_log!(
            "Your public UDP address is {}",
            color_primary(my_address),
        ))?;

        Ok(())
    }
}