Merge pull request #20 from matthiasbeyer/feat/add-annotation

Feat/add annotation

1 files changed, 17 insertions, 0 deletions

diff --git a/src/task.rs b/src/task.rs
index b67cb94..8aaae17 100644
--- a/src/task.rs
+++ b/src/task.rs
@@ -117,6 +117,23 @@ impl Task {
         self.annotations.as_ref()
     }
 
+    pub fn add_annotation(&mut self, an: Annotation) {
+        if self.annotations.is_none() {
+            self.annotations = Some(vec![an]);
+        } else {
+            match self.annotations.as_mut() {
+                Some(ref mut anno) => anno.push(an),
+                _ => unreachable!(),
+            }
+        }
+    }
+
+    pub fn add_annotations<I: Iterator<Item = Annotation>>(&mut self, i: I) {
+        for item in i {
+            self.add_annotation(item)
+        }
+    }
+
     pub fn depends(&self) -> Option<&String> {
         self.depends.as_ref()
     }