Implement From for Id<NSArray> as an example

Author: madsmtm

objc2-foundation/examples/basic_usage.rs

@@ -1,4 +1,4 @@
-use objc2::rc::autoreleasepool;
+use objc2::rc::{autoreleasepool, Id};
 use objc2_foundation::{NSArray, NSCopying, NSDictionary, NSObject, NSString};
 
 fn main() {
@@ -14,14 +14,14 @@ fn main() {
 
     // Create an NSArray from a Vec
     let objs = vec![obj, obj2];
-    let array = NSArray::from_vec(objs);
+    let array: Id<NSArray<_, _>, _> = objs.into();
     for obj in array.iter() {
         println!("{:?}", obj);
     }
     println!("{}", array.len());
 
     // Turn the NSArray back into a Vec
-    let mut objs = NSArray::into_vec(array);
+    let mut objs = Vec::from(array);
     let obj = objs.pop().unwrap();
 
     // Create an NSString from a str slice

objc2-foundation/src/array.rs

@@ -215,6 +215,37 @@ impl<T: Message> DefaultId for NSArray<T, Shared> {
     }
 }
 
+impl<T: Message, O: Ownership> From<Id<NSArray<T, O>, Owned>> for Vec<Id<T, O>> {
+    fn from(array: Id<NSArray<T, O>, Owned>) -> Self {
+        let vec: Vec<&T> = (&*array).into();
+        vec.into_iter()
+            .map(|obj| unsafe { Id::retain(obj.into()) })
+            .collect()
+    }
+}
+
+impl<T: Message, O: Ownership> From<Vec<Id<T, O>>> for Id<NSArray<T, O>, Owned> {
+    fn from(vec: Vec<Id<T, O>>) -> Self {
+        unsafe { from_refs(vec.as_slice_ref()) }
+    }
+}
+
+impl<'a, T: Message, O: Ownership> From<&'a NSArray<T, O>> for Vec<&'a T> {
+    fn from(array: &'a NSArray<T, O>) -> Self {
+        array.objects_in_range(0..array.count())
+    }
+}
+
+impl<T: Message> From<&'_ NSArray<T, Shared>> for Vec<Id<T, Shared>> {
+    fn from(array: &NSArray<T, Shared>) -> Self {
+        array
+            .objects_in_range(0..array.count())
+            .into_iter()
+            .map(|obj| unsafe { Id::retain(obj.into()) })
+            .collect()
+    }
+}
+
 impl<T: Message, O: Ownership> NSMutableArray<T, O> {
     unsafe_def_fn!(pub fn new -> Owned);
 
@@ -378,6 +409,12 @@ impl<T: Message, O: Ownership> DefaultId for NSMutableArray<T, O> {
     }
 }
 
+impl<T: Message, O: Ownership> From<Vec<Id<T, O>>> for Id<NSMutableArray<T, O>, Owned> {
+    fn from(vec: Vec<Id<T, O>>) -> Self {
+        unsafe { from_refs(vec.as_slice_ref()) }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use alloc::format;
@@ -395,7 +432,7 @@ mod tests {
         for _ in 0..len {
             vec.push(NSObject::new());
         }
-        NSArray::from_vec(vec)
+        vec.into()
     }
 
     fn sample_number_array(len: u8) -> Id<NSArray<NSValue<u8>, Shared>, Shared> {
@@ -517,7 +554,7 @@ mod tests {
     fn test_into_vec() {
         let array = sample_array(4);
 
-        let vec = NSArray::into_vec(array);
+        let vec = Vec::from(array);
         assert_eq!(vec.len(), 4);
     }
 
@@ -566,7 +603,7 @@ mod tests {
     #[test]
     fn test_sort() {
         let strings = vec![NSString::from_str("hello"), NSString::from_str("hi")];
-        let mut strings = NSMutableArray::from_vec(strings);
+        let mut strings: Id<NSMutableArray<_, _>, Owned> = strings.into();
 
         autoreleasepool(|pool| {
             strings.sort_by(|s1, s2| s1.as_str(pool).len().cmp(&s2.as_str(pool).len()));

objc2-foundation/src/enumerator.rs

@@ -167,13 +167,16 @@ impl<'a, C: NSFastEnumeration + ?Sized> Iterator for NSFastEnumerator<'a, C> {
 
 #[cfg(test)]
 mod tests {
+    use alloc::vec::Vec;
+    use objc2::rc::Id;
+
     use super::NSFastEnumeration;
     use crate::{NSArray, NSValue};
 
     #[test]
     fn test_enumerator() {
-        let vec = (0usize..4).map(NSValue::new).collect();
-        let array = NSArray::from_vec(vec);
+        let vec: Vec<_> = (0usize..4).map(NSValue::new).collect();
+        let array: Id<NSArray<_, _>, _> = vec.into();
 
         let enumerator = array.iter();
         assert_eq!(enumerator.count(), 4);
@@ -184,8 +187,8 @@ mod tests {
 
     #[test]
     fn test_fast_enumerator() {
-        let vec = (0usize..4).map(NSValue::new).collect();
-        let array = NSArray::from_vec(vec);
+        let vec: Vec<_> = (0usize..4).map(NSValue::new).collect();
+        let array: Id<NSArray<_, _>, _> = vec.into();
 
         let enumerator = array.iter_fast();
         assert_eq!(enumerator.count(), 4);