ockam_command 0.118.0

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

use crate::run::parser::building_blocks::{ArgsToCommands, ResourceNameOrMap};
use crate::run::parser::resource::traits::CommandsParser;
use crate::run::parser::resource::utils::parse_cmd_from_args;
use crate::run::parser::resource::ValuesOverrides;
use crate::tcp::outlet::create::CreateCommand;
use crate::{color_primary, tcp::outlet, Command, OckamSubcommand};
use async_trait::async_trait;
use miette::{miette, Result};
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct TcpOutlets {
    #[serde(alias = "tcp-outlets", alias = "tcp-outlet")]
    pub tcp_outlets: Option<ResourceNameOrMap>,
}

impl TcpOutlets {
    fn get_subcommand(args: &[String]) -> Result<CreateCommand> {
        if let OckamSubcommand::TcpOutlet(cmd) = parse_cmd_from_args(CreateCommand::NAME, args)? {
            if let outlet::TcpOutletSubCommand::Create(c) = cmd.subcommand {
                return Ok(c);
            }
        }
        Err(miette!(format!(
            "Failed to parse {} command",
            color_primary(CreateCommand::NAME)
        )))
    }
}

#[async_trait]
impl CommandsParser<CreateCommand> for TcpOutlets {
    fn parse_commands(self, overrides: &ValuesOverrides) -> Result<Vec<CreateCommand>> {
        match self.tcp_outlets {
            Some(c) => {
                let mut cmds = c.into_commands_with_name_arg(Self::get_subcommand, Some("from"))?;
                if let Some(node_name) = overrides.override_node_name.as_ref() {
                    for cmd in cmds.iter_mut() {
                        cmd.at = Some(node_name.clone())
                    }
                }
                Ok(cmds)
            }
            None => Ok(vec![]),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::net::SocketAddr;
    use std::str::FromStr;

    #[test]
    fn tcp_outlet_config() {
        let config = r#"
            tcp_outlets:
              to1:
                to: 6060
                at: n
              to2:
                to: 6061
                from: my_outlet
        "#;
        let parsed: TcpOutlets = serde_yaml::from_str(config).unwrap();
        let cmds = parsed.parse_commands(&ValuesOverrides::default()).unwrap();
        assert_eq!(cmds.len(), 2);
        assert_eq!(cmds[0].from.clone().unwrap(), "to1");
        assert_eq!(cmds[0].to, SocketAddr::from_str("127.0.0.1:6060").unwrap());
        assert_eq!(cmds[0].at.as_ref().unwrap(), "n");
        assert_eq!(cmds[1].from.clone().unwrap(), "my_outlet");
        assert_eq!(cmds[1].to, SocketAddr::from_str("127.0.0.1:6061").unwrap());
        assert!(cmds[1].at.is_none());
    }
}