Impl AnsiAware for AsRef<str> instead of &str

mikaelmello · Jan 4, 2024 · d21ec46 · d21ec46
1 parent 923047a
commit d21ec46
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/inquire/src/ansi.rs b/inquire/src/ansi.rs
@@ -168,19 +168,29 @@ pub trait AnsiStrippable {
     fn ansi_stripped_chars(&self) -> AnsiStrippedChars<'_>;
 }
 
-impl AnsiStrippable for &str {
+impl<T> AnsiStrippable for T
+where
+    T: AsRef<str>,
+{
     fn ansi_stripped_chars(&self) -> AnsiStrippedChars<'_> {
-        AnsiStrippedChars { input: self }
+        AnsiStrippedChars {
+            input: self.as_ref(),
+        }
     }
 }
 
 pub trait AnsiAware {
     fn ansi_aware_chars(&self) -> AnsiAwareChars<'_>;
 }
 
-impl AnsiAware for &str {
+impl<T> AnsiAware for T
+where
+    T: AsRef<str>,
+{
     fn ansi_aware_chars(&self) -> AnsiAwareChars<'_> {
-        AnsiAwareChars { input: self }
+        AnsiAwareChars {
+            input: self.as_ref(),
+        }
     }
 }