From 8a6881169bbd13e7990cd48da9cb609f021a20db Mon Sep 17 00:00:00 2001
From: dylni <46035563+dylni@users.noreply.github.com>
Date: Mon, 1 Mar 2021 17:58:53 -0500
Subject: [PATCH] Upgrade os_str_bytes

---
 Cargo.toml          |  2 +-
 src/parse/parser.rs | 16 ++++++++--------
 src/util/argstr.rs  | 10 +++++++---
 3 files changed, 16 insertions(+), 12 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 21535223..10f1567b 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -69,7 +69,7 @@ clap_derive = { path = "./clap_derive", version = "3.0.0-beta.2", optional = tru
 bitflags = "1.2"
 textwrap = { version = "0.13.3", default-features = false, features = [] }
 indexmap = "1.0"
-os_str_bytes = { version = "2.4", features = ["raw"] }
+os_str_bytes = { version = "3.0", features = ["raw"] }
 vec_map = "0.8"
 strsim = { version = "0.10",  optional = true }
 yaml-rust = { version = "0.4.1",  optional = true }
diff --git a/src/parse/parser.rs b/src/parse/parser.rs
index b1bf3236..efb5faf3 100644
--- a/src/parse/parser.rs
+++ b/src/parse/parser.rs
@@ -429,7 +429,7 @@ impl<'help, 'app> Parser<'help, 'app> {
                     // get the option so we can check the settings
                     needs_val_of = self.add_val_to_arg(
                         &self.app[&id],
-                        &arg_os,
+                        arg_os,
                         matcher,
                         ValueType::CommandLine,
                         true,
@@ -523,7 +523,7 @@ impl<'help, 'app> Parser<'help, 'app> {
                 // doesn't have any value. This behaviour is right because
                 // positional arguments are always present continiously.
                 let append = self.arg_have_val(matcher, p);
-                self.add_val_to_arg(p, &arg_os, matcher, ValueType::CommandLine, append);
+                self.add_val_to_arg(p, arg_os, matcher, ValueType::CommandLine, append);
 
                 // Increase occurence no matter if we are appending, Occurences
                 // of positional argument equals to number of values rather than
@@ -1174,7 +1174,7 @@ impl<'help, 'app> Parser<'help, 'app> {
                 fv,
                 fv.starts_with("=")
             );
-            self.add_val_to_arg(opt, &v, matcher, ValueType::CommandLine, false);
+            self.add_val_to_arg(opt, v, matcher, ValueType::CommandLine, false);
         } else if require_equals && !empty_vals && !min_vals_zero {
             debug!("None, but requires equals...Error");
             return Err(ClapError::empty_value(
@@ -1223,7 +1223,7 @@ impl<'help, 'app> Parser<'help, 'app> {
     fn add_val_to_arg(
         &self,
         arg: &Arg<'help>,
-        val: &ArgStr,
+        val: ArgStr,
         matcher: &mut ArgMatcher,
         ty: ValueType,
         append: bool,
@@ -1249,7 +1249,7 @@ impl<'help, 'app> Parser<'help, 'app> {
                 } else {
                     arg_split.collect()
                 };
-                let vals = vals.into_iter().map(|x| x.to_os_string()).collect();
+                let vals = vals.into_iter().map(|x| x.into_os_string()).collect();
                 self.add_multiple_vals_to_arg(arg, vals, matcher, ty, append);
                 // If there was a delimiter used or we must use the delimiter to
                 // separate the values or no more vals is needed, we're not
@@ -1426,7 +1426,7 @@ impl<'help, 'app> Parser<'help, 'app> {
                     };
 
                     if add {
-                        self.add_val_to_arg(arg, &ArgStr::new(default), matcher, ty, false);
+                        self.add_val_to_arg(arg, ArgStr::new(default), matcher, ty, false);
                         return;
                     }
                 }
@@ -1497,11 +1497,11 @@ impl<'help, 'app> Parser<'help, 'app> {
             // Use env only if the arg was not present among command line args
             if matcher.get(&a.id).map_or(true, |a| a.occurs == 0) {
                 if let Some((_, Some(ref val))) = a.env {
-                    let val = &ArgStr::new(val);
+                    let val = ArgStr::new(val);
                     if a.is_set(ArgSettings::TakesValue) {
                         self.add_val_to_arg(a, val, matcher, ValueType::EnvVariable, false);
                     } else {
-                        self.check_for_help_and_version_str(val)?;
+                        self.check_for_help_and_version_str(&val)?;
                         matcher.add_index_to(&a.id, self.cur_idx.get(), ValueType::EnvVariable);
                     }
                 }
diff --git a/src/util/argstr.rs b/src/util/argstr.rs
index e98c3c39..0eaa3db3 100644
--- a/src/util/argstr.rs
+++ b/src/util/argstr.rs
@@ -5,13 +5,13 @@ use std::{
     str,
 };
 
-use os_str_bytes::{raw, OsStrBytes, OsStringBytes};
+use os_str_bytes::{raw, OsStrBytes};
 
 pub(crate) struct ArgStr<'a>(Cow<'a, [u8]>);
 
 impl<'a> ArgStr<'a> {
     pub(crate) fn new(s: &'a OsStr) -> Self {
-        Self(s.to_bytes())
+        Self(s.to_raw_bytes())
     }
 
     pub(crate) fn starts_with(&self, s: &str) -> bool {
@@ -120,7 +120,11 @@ impl<'a> ArgStr<'a> {
     }
 
     pub(crate) fn to_os_string(&self) -> OsString {
-        OsString::from_bytes(&self.0).unwrap()
+        self.to_borrowed().into_os_string()
+    }
+
+    pub(crate) fn into_os_string(self) -> OsString {
+        OsStr::from_raw_bytes(self.0).unwrap().into_owned()
     }
 }