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


use crate::Bytes;


macro_rules! write_fn {
	($name:ident, $type:ident) => (
		write_fn!($name, $type, stringify!($type));
	);
	($name:ident, $type:ident, $type_str:expr) => {
		#[inline]
		#[doc = "Writes an `"]
		#[doc = $type_str]
		#[doc = "`."]
		/// 
		/// ## Panics
		/// If there aren't enough remaining bytes left.
		fn $name(&mut self, num: $type) {
			let bytes = num.to_be_bytes();
			self.write(&bytes);
		}
	}
}

/// Write bytes or numbers.
pub trait BytesWrite {

	/// Returns the entire slice mutably.
	fn as_mut(&mut self) -> &mut [u8];

	/// Returns the entire slice as a bytes struct
	/// setting the position of the new Bytes to `0`.
	fn as_bytes(&self) -> Bytes<'_>;

	/// Returns the remaining bytes mutably.
	fn remaining_mut(&mut self) -> &mut [u8];

	/// Writes a slice.
	/// 
	/// ## Panics
	/// If there aren't enough remaining bytes left.
	fn write(&mut self, slice: &[u8]);

	write_fn!(write_u8, u8);
	write_fn!(write_u16, u16);
	write_fn!(write_u32, u32);
	write_fn!(write_u64, u64);
	write_fn!(write_u128, u128);

	write_fn!(write_i8, i8);
	write_fn!(write_i16, i16);
	write_fn!(write_i32, i32);
	write_fn!(write_i64, i64);
	write_fn!(write_i128, i128);

	write_fn!(write_f32, f32);
	write_fn!(write_f64, f64);

}