summaryrefslogtreecommitdiffstats
path: root/third_party/rust/clap/src/osstringext.rs
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/rust/clap/src/osstringext.rs')
-rw-r--r--third_party/rust/clap/src/osstringext.rs157
1 files changed, 157 insertions, 0 deletions
diff --git a/third_party/rust/clap/src/osstringext.rs b/third_party/rust/clap/src/osstringext.rs
new file mode 100644
index 0000000000..e9e0ba33a4
--- /dev/null
+++ b/third_party/rust/clap/src/osstringext.rs
@@ -0,0 +1,157 @@
+#[cfg(any(target_os = "windows", target_arch = "wasm32"))]
+use INVALID_UTF8;
+use std::ffi::OsStr;
+#[cfg(not(any(target_os = "windows", target_arch = "wasm32")))]
+use std::os::unix::ffi::OsStrExt;
+
+#[cfg(any(target_os = "windows", target_arch = "wasm32"))]
+pub trait OsStrExt3 {
+ fn from_bytes(b: &[u8]) -> &Self;
+ fn as_bytes(&self) -> &[u8];
+}
+
+#[doc(hidden)]
+pub trait OsStrExt2 {
+ fn starts_with(&self, s: &[u8]) -> bool;
+ fn split_at_byte(&self, b: u8) -> (&OsStr, &OsStr);
+ fn split_at(&self, i: usize) -> (&OsStr, &OsStr);
+ fn trim_left_matches(&self, b: u8) -> &OsStr;
+ fn len_(&self) -> usize;
+ fn contains_byte(&self, b: u8) -> bool;
+ fn is_empty_(&self) -> bool;
+ fn split(&self, b: u8) -> OsSplit;
+}
+
+#[cfg(any(target_os = "windows", target_arch = "wasm32"))]
+impl OsStrExt3 for OsStr {
+ fn from_bytes(b: &[u8]) -> &Self {
+ use std::mem;
+ unsafe { mem::transmute(b) }
+ }
+ fn as_bytes(&self) -> &[u8] {
+ self.to_str().map(|s| s.as_bytes()).expect(INVALID_UTF8)
+ }
+}
+
+impl OsStrExt2 for OsStr {
+ fn starts_with(&self, s: &[u8]) -> bool {
+ self.as_bytes().starts_with(s)
+ }
+
+ fn is_empty_(&self) -> bool {
+ self.as_bytes().is_empty()
+ }
+
+ fn contains_byte(&self, byte: u8) -> bool {
+ for b in self.as_bytes() {
+ if b == &byte {
+ return true;
+ }
+ }
+ false
+ }
+
+ fn split_at_byte(&self, byte: u8) -> (&OsStr, &OsStr) {
+ for (i, b) in self.as_bytes().iter().enumerate() {
+ if b == &byte {
+ return (
+ OsStr::from_bytes(&self.as_bytes()[..i]),
+ OsStr::from_bytes(&self.as_bytes()[i + 1..]),
+ );
+ }
+ }
+ (
+ &*self,
+ OsStr::from_bytes(&self.as_bytes()[self.len_()..self.len_()]),
+ )
+ }
+
+ fn trim_left_matches(&self, byte: u8) -> &OsStr {
+ let mut found = false;
+ for (i, b) in self.as_bytes().iter().enumerate() {
+ if b != &byte {
+ return OsStr::from_bytes(&self.as_bytes()[i..]);
+ } else {
+ found = true;
+ }
+ }
+ if found {
+ return OsStr::from_bytes(&self.as_bytes()[self.len_()..]);
+ }
+ &*self
+ }
+
+ fn split_at(&self, i: usize) -> (&OsStr, &OsStr) {
+ (
+ OsStr::from_bytes(&self.as_bytes()[..i]),
+ OsStr::from_bytes(&self.as_bytes()[i..]),
+ )
+ }
+
+ fn len_(&self) -> usize {
+ self.as_bytes().len()
+ }
+
+ fn split(&self, b: u8) -> OsSplit {
+ OsSplit {
+ sep: b,
+ val: self.as_bytes(),
+ pos: 0,
+ }
+ }
+}
+
+#[doc(hidden)]
+#[derive(Clone, Debug)]
+pub struct OsSplit<'a> {
+ sep: u8,
+ val: &'a [u8],
+ pos: usize,
+}
+
+impl<'a> Iterator for OsSplit<'a> {
+ type Item = &'a OsStr;
+
+ fn next(&mut self) -> Option<&'a OsStr> {
+ debugln!("OsSplit::next: self={:?}", self);
+ if self.pos == self.val.len() {
+ return None;
+ }
+ let start = self.pos;
+ for b in &self.val[start..] {
+ self.pos += 1;
+ if *b == self.sep {
+ return Some(OsStr::from_bytes(&self.val[start..self.pos - 1]));
+ }
+ }
+ Some(OsStr::from_bytes(&self.val[start..]))
+ }
+ fn size_hint(&self) -> (usize, Option<usize>) {
+ let mut count = 0;
+ for b in &self.val[self.pos..] {
+ if *b == self.sep {
+ count += 1;
+ }
+ }
+ if count > 0 {
+ return (count, Some(count));
+ }
+ (0, None)
+ }
+}
+
+impl<'a> DoubleEndedIterator for OsSplit<'a> {
+ fn next_back(&mut self) -> Option<&'a OsStr> {
+ if self.pos == 0 {
+ return None;
+ }
+ let start = self.pos;
+ for b in self.val[..self.pos].iter().rev() {
+ self.pos -= 1;
+ if *b == self.sep {
+ return Some(OsStr::from_bytes(&self.val[self.pos + 1..start]));
+ }
+ }
+ Some(OsStr::from_bytes(&self.val[..start]))
+ }
+}