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
|
/// Match a VariantMaps field and return it's contents if successfull
///
/// # Example
///
/// ```
/// use libquassel::match_variant;
/// use libquassel::primitive::{VariantMap, Variant};
///
/// let var = Variant::String("test string".to_string());
/// let result = match_variant!(var, Variant::String);
/// ```
#[macro_export]
macro_rules! match_variant {
( $values:expr, $x:path ) => {
match &$values {
$x(x) => Ok(x.clone()),
err => Err(err),
}
.unwrap()
};
}
/// Prepend the length of `buf` to `buf`
pub fn prepend_byte_len(buf: &mut Vec<u8>) {
let len: i32 = buf.len().try_into().unwrap();
let ulen: &[u8] = &len.to_be_bytes();
buf.insert(0, ulen[3]);
buf.insert(0, ulen[2]);
buf.insert(0, ulen[1]);
buf.insert(0, ulen[0]);
}
/// Insert a bytes `input` into `buf` at position `pos`
pub fn insert_bytes(pos: usize, buf: &mut Vec<u8>, input: &mut [u8]) {
input.reverse();
for i in input {
buf.insert(pos, *i)
}
}
#[macro_export]
macro_rules! map {
// map-like
($($k:expr => $v:expr),* $(,)?) => {
std::iter::Iterator::collect(std::array::IntoIter::new([$(($k, $v),)*]))
};
// set-like
($($v:expr),* $(,)?) => {
std::iter::Iterator::collect(std::array::IntoIter::new([$($v,)*]))
};
}
#[macro_export]
macro_rules! s {
($values:expr) => {
std::string::String::from($values)
};
}
#[macro_export]
macro_rules! get_param {
( $msg:expr ) => {
$msg.params.remove(0).try_into().unwrap()
};
}
|
