xmpp-proxy 1.0.0

Reverse XMPP proxy.
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use anyhow::{anyhow, Result};

pub trait SliceSubsequence<T> {
    fn trim_start(&self, needle: &[T]) -> &[T];
    fn first_index_of(&self, needle: &[T]) -> Result<usize>;
    fn replace_first(self, needle: &[T], replacement: &[T]) -> Vec<T>;

    fn contains_seq(&self, needle: &[T]) -> bool {
        self.first_index_of(needle).is_ok()
    }
}

fn last_index_of<T: PartialEq>(s: &[T], needle: &[T]) -> usize {
    let mut len = 0;
    for i in s {
        if needle.contains(i) {
            len += 1;
        } else {
            break;
        }
    }
    len
}

impl<T: PartialEq + Clone> SliceSubsequence<T> for &[T] {
    fn trim_start(&self, needle: &[T]) -> &[T] {
        &self[last_index_of(self, needle)..]
    }

    fn first_index_of(&self, needle: &[T]) -> Result<usize> {
        for i in 0..self.len() - needle.len() + 1 {
            if self[i..i + needle.len()] == needle[..] {
                return Ok(i);
            }
        }
        Err(anyhow!("not found"))
    }

    fn replace_first(self, needle: &[T], replacement: &[T]) -> Vec<T> {
        self.to_vec().replace_first(needle, replacement)
    }
}

impl<T: PartialEq + Clone> SliceSubsequence<T> for Vec<T> {
    fn trim_start(&self, needle: &[T]) -> &[T] {
        &self[last_index_of(self, needle)..]
    }

    fn first_index_of(&self, needle: &[T]) -> Result<usize> {
        return (self.as_slice()).first_index_of(needle);
    }

    fn replace_first(self, needle: &[T], replacement: &[T]) -> Vec<T> {
        if let Ok(idx) = self.first_index_of(needle) {
            let second = &self[(idx + needle.len())..];
            let mut ret = Vec::with_capacity(idx + replacement.len() + second.len());
            ret.extend_from_slice(&self[..idx]);
            ret.extend_from_slice(replacement);
            ret.extend_from_slice(second);
            ret
        } else {
            self
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::slicesubsequence::*;
    const WHITESPACE: &[u8] = b" \t\n\r";

    #[test]
    fn trim_start() {
        let buf = &b"    bla"[..];
        let buf = buf.trim_start(WHITESPACE);
        assert_eq!(buf, b"bla");

        let buf = &b"\n\t\r   \rbla"[..];
        let buf = buf.trim_start(WHITESPACE);
        assert_eq!(buf, b"bla");
    }
    #[test]
    fn replace_first() {
        let buf = b"bla to='tsnhaou' bla2".replace_first(b" to=", b" from=");
        assert_eq!(buf, b"bla from='tsnhaou' bla2");

        let buf = buf.replace_first(b" from=", b" to=");
        assert_eq!(buf, b"bla to='tsnhaou' bla2");

        let buf = buf.replace_first(b" to=", b" from=");
        assert_eq!(buf, b"bla from='tsnhaou' bla2");

        let buf = b"bla to='tsnhaou' bla2".replace_first(b"bla", b"boo");
        assert_eq!(buf, b"boo to='tsnhaou' bla2");

        let buf = buf.replace_first(b"boo", b"bla");
        assert_eq!(buf, b"bla to='tsnhaou' bla2");

        let buf = buf.replace_first(b" bla2", b"");
        assert_eq!(buf, b"bla to='tsnhaou'");
    }
}