websocat 4.0.0-alpha3

Command-line client for web sockets, like netcat/curl/socat for ws://.
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

use clap_lex::OsStrExt;
use std::{collections::HashMap, ffi::OsStr};

use super::types::SocketType;

pub struct IdentifierGenerator {
    pub varnames: HashMap<&'static str, usize>,
}

impl Default for IdentifierGenerator {
    fn default() -> Self {
        Self::new()
    }
}

impl IdentifierGenerator {
    pub fn new() -> Self {
        IdentifierGenerator {
            varnames: HashMap::with_capacity(8),
        }
    }
    pub fn getnewvarname(&mut self, prefix: &'static str) -> String {
        let e = self.varnames.entry(prefix).or_default();
        *e += 1;
        format!("{prefix}{}", *e)
    }
}

pub trait StripPrefixMany {
    fn strip_prefix_many<'a>(&'a self, prefixes: &'static [&'static str]) -> Option<&'a OsStr>;
}
impl StripPrefixMany for OsStr {
    fn strip_prefix_many<'a>(&'a self, prefixes: &'static [&'static str]) -> Option<&'a OsStr> {
        for p in prefixes {
            if let Some(x) = self.strip_prefix(p) {
                return Some(x);
            }
        }
        None
    }
}

impl SocketType {
    pub fn is_bstrm(&self) -> bool {
        *self == SocketType::ByteStream
    }
    pub fn is_dgrms(&self) -> bool {
        *self == SocketType::Datarams
    }
}