Tweaked for generality.

1 files changed, 18 insertions, 0 deletions

diff --git a/src/context.rs b/src/context.rs
index 0b51677..6f8e04f 100644
--- a/src/context.rs
+++ b/src/context.rs
@@ -21,6 +21,24 @@ pub trait MappedList<'a> {
     fn get_index(&'a self, id: usize) -> Option<MappedValue<'a>>;
     fn for_each(&'a self, func: &dyn Fn(&MappedValue<'a>, &mut String) -> CrimResult<()>) -> CrimResult<String>;
 }
+    
+impl<'a, T: MappedStructure<'a>> MappedList<'a> for Vec<T> {
+    fn get_index(&'a self, id: usize) -> Option<MappedValue<'a>> {
+        if let Some(x) = self.get(id) {
+            Some(MappedValue::<'a>::Struct(x))
+        } else {
+            None
+        }
+    }
+
+    fn for_each(&'a self, func: &dyn Fn(&MappedValue<'a>, &mut String) -> CrimResult<()>) -> CrimResult<String> {
+        let mut buf = String::new();
+        for i in self {
+            func( &MappedValue::Struct(i), &mut buf )?;
+        }
+        Ok(buf)
+    }
+}
 
 impl<'a> MappedList<'a> for Vec<Value> {
     fn get_index(&'a self, id: usize) -> Option<MappedValue<'a>> {