websocat 4.0.0-alpha1

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

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

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

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;
        return 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
    }
}