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

use std::slice::SliceIndex;

use super::RawStr;

/// The equivalent of `SliceIndex` for `RawStr`.
///
/// # Usage
///
/// Normally, this trait is not used directly.
/// Its functionality is exposed through `Index` and the `get` methods of `RawStr`.
///
/// # Implementors
///
/// `RawStrIndex` is automatically implemented for all that implement `SliceIndex<[u8]>`:
///
///   - a `SliceIndex<[u8], Output=[u8]>` automatically implements `RawStrIndex<Output=RawStr>`, and
///   - a `SliceIndex<[u8], Output=u8>` automatically implements `RawStrIndex<Output=u8>`.
///
/// # Examples
///
/// ```
/// # use raw_string::RawStr;
/// let s = RawStr::from_str("hello world");
/// let hello: &RawStr = &s[..5]; // This is a slice.
/// let space: u8 = s[5]; // This is a single byte
/// assert_eq!(hello, "hello");
/// assert_eq!(space, b' ');
/// ```
pub trait RawStrIndex {
	/// `RawStr` (for ranges) or `u8` (for single indexes).
	type Output: ?Sized;
	/// Get the range or byte from the given `&RawStr`.
	fn get(self, s: &RawStr) -> Option<&Self::Output>;
	/// Get the (mutable) range or byte from the given `&mut RawStr`.
	fn get_mut(self, s: &mut RawStr) -> Option<&mut Self::Output>;
	/// Like `get`, but unsafe and unchecked.
	unsafe fn get_unchecked(self, s: &RawStr) -> &Self::Output;
	/// Like `get_mut`, but unsafe and unchecked.
	unsafe fn get_unchecked_mut(self, s: &mut RawStr) -> &mut Self::Output;
	/// Like `get`, but panics on failure.
	fn index(self, s: &RawStr) -> &Self::Output;
	/// Like `get_mut`, but panics on failure.
	fn index_mut(self, s: &mut RawStr) -> &mut Self::Output;
}

#[doc(hidden)]
pub trait RawStrIndexOutput {
	type Output: ?Sized;
	fn into(&self) -> &Self::Output;
	fn into_mut(&mut self) -> &mut Self::Output;
}

impl RawStrIndexOutput for [u8] {
	type Output = RawStr;
	fn into(&self) -> &RawStr {
		RawStr::from_bytes(self)
	}
	fn into_mut(&mut self) -> &mut RawStr {
		RawStr::from_bytes_mut(self)
	}
}

impl RawStrIndexOutput for u8 {
	type Output = u8;
	fn into(&self) -> &u8 {
		self
	}
	fn into_mut(&mut self) -> &mut u8 {
		self
	}
}

impl<I> RawStrIndex for I
where
	I: SliceIndex<[u8]>,
	I::Output: RawStrIndexOutput + 'static,
{
	type Output = <<I as SliceIndex<[u8]>>::Output as RawStrIndexOutput>::Output;
	fn get(self, s: &RawStr) -> Option<&Self::Output> {
		s.as_bytes().get(self).map(RawStrIndexOutput::into)
	}
	fn get_mut(self, s: &mut RawStr) -> Option<&mut Self::Output> {
		s.as_bytes_mut()
			.get_mut(self)
			.map(RawStrIndexOutput::into_mut)
	}
	unsafe fn get_unchecked(self, s: &RawStr) -> &Self::Output {
		RawStrIndexOutput::into(s.as_bytes().get_unchecked(self))
	}
	unsafe fn get_unchecked_mut(self, s: &mut RawStr) -> &mut Self::Output {
		RawStrIndexOutput::into_mut(s.as_bytes_mut().get_unchecked_mut(self))
	}
	fn index(self, s: &RawStr) -> &Self::Output {
		RawStrIndexOutput::into(&s.as_bytes()[self])
	}
	fn index_mut(self, s: &mut RawStr) -> &mut Self::Output {
		RawStrIndexOutput::into_mut(&mut s.as_bytes_mut()[self])
	}
}