aboutsummaryrefslogtreecommitdiff
path: root/arg_picker/src/builtin/pick_socket_attr.rs
blob: 0c0dd7182ec9fb2df93dd029ee73a7739278d12d (plain) (blame)
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
use std::net::{SocketAddr, SocketAddrV4, SocketAddrV6};

use crate::SinglePickable;

impl SinglePickable for SocketAddr {
    fn pick_single(str: Option<&str>) -> crate::PickerArgResult<Self> {
        match str {
            Some(s) => match s.parse::<SocketAddr>() {
                Ok(addr) => crate::PickerArgResult::Parsed(addr),
                Err(_) => crate::PickerArgResult::NotFound,
            },
            None => crate::PickerArgResult::NotFound,
        }
    }
}

impl SinglePickable for SocketAddrV4 {
    fn pick_single(str: Option<&str>) -> crate::PickerArgResult<Self> {
        match str {
            Some(s) => match s.parse::<SocketAddrV4>() {
                Ok(addr) => crate::PickerArgResult::Parsed(addr),
                Err(_) => crate::PickerArgResult::NotFound,
            },
            None => crate::PickerArgResult::NotFound,
        }
    }
}

impl SinglePickable for SocketAddrV6 {
    fn pick_single(str: Option<&str>) -> crate::PickerArgResult<Self> {
        match str {
            Some(s) => match s.parse::<SocketAddrV6>() {
                Ok(addr) => crate::PickerArgResult::Parsed(addr),
                Err(_) => crate::PickerArgResult::NotFound,
            },
            None => crate::PickerArgResult::NotFound,
        }
    }
}