Add: itertors over steps

1 files changed, 11 insertions, 1 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 38577d6..c73dde5 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -119,6 +119,16 @@ impl Roadmap {
         Ok(())
     }
 
+    // Get iterator over refs to steps.
+    pub fn iter(&self) -> impl Iterator<Item=&Step> {
+        self.steps.iter()
+    }
+
+    // Get iterator over mut refs to steps.
+    pub fn iter_mut(&mut self) -> impl Iterator<Item=&mut Step> { 
+        self.steps.iter_mut()
+    }
+
     /// Get a Graphviz dot language representation of a roadmap. This
     /// is the textual representation, and the caller needs to use the
     /// Graphviz dot(1) tool to create an image from it.
@@ -133,7 +143,7 @@ impl Roadmap {
             dot.push_str(&line);
         }
 
-        for step in self.steps.iter() {
+        for step in self.iter() {
             for dep in step.dependencies() {
                 let line = format!("{} -> {};\n", dep, step.name());
                 dot.push_str(&line);