From 62d372ce1ff2439f3b83385bb5752fe7ac90afa9 Mon Sep 17 00:00:00 2001
From: bluss <bluss@users.noreply.github.com>
Date: Wed, 13 Jan 2016 12:59:06 +0100
Subject: [PATCH] Add DerefMut for ArrayString

---
 src/array_string.rs | 14 +++++++++++++-
 tests/tests.rs      |  6 ++++++
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/src/array_string.rs b/src/array_string.rs
index 65a8a02..a39fbcc 100644
--- a/src/array_string.rs
+++ b/src/array_string.rs
@@ -1,7 +1,8 @@
 use std::borrow::Borrow;
 use std::fmt;
 use std::hash::{Hash, Hasher};
-use std::ops::Deref;
+use std::mem;
+use std::ops::{Deref, DerefMut};
 use std::str;
 use std::slice;
 
@@ -142,6 +143,17 @@ impl<A: Array<Item=u8>> Deref for ArrayString<A> {
     }
 }
 
+impl<A: Array<Item=u8>> DerefMut for ArrayString<A> {
+    #[inline]
+    fn deref_mut(&mut self) -> &mut str {
+        unsafe {
+            let sl = slice::from_raw_parts_mut(self.xs.as_mut_ptr(), self.len.to_usize());
+            // FIXME: Nothing but transmute to do this right now
+            mem::transmute(sl)
+        }
+    }
+}
+
 impl<A: Array<Item=u8>> PartialEq for ArrayString<A> {
     fn eq(&self, rhs: &Self) -> bool {
         **self == **rhs
diff --git a/tests/tests.rs b/tests/tests.rs
index 3dd24bd..b0b256d 100644
--- a/tests/tests.rs
+++ b/tests/tests.rs
@@ -283,6 +283,12 @@ fn test_string() {
     assert!(t.push_str(text).is_err());
     assert_eq!(&t, "");
 
+    t.push_str("ab").unwrap();
+    // DerefMut
+    let tmut: &mut str = &mut t;
+    assert_eq!(tmut, "ab");
+
+
     // Test Error trait / try
     let t = || -> Result<(), Box<Error>> {
         let mut t = ArrayString::<[_; 2]>::new();