Add validated image build stage graph

1 files changed, 9 insertions, 1 deletions

diff --git a/src/plan.rs b/src/plan.rs
index 16158e5..3d28d96 100644
--- a/src/plan.rs
+++ b/src/plan.rs
@@ -1,6 +1,6 @@
 use anyhow::Result;
 
-use crate::ImageSpec;
+use crate::{ImageSpec, Stage};
 
 #[derive(Debug, Clone, PartialEq, Eq)]
 pub struct BuildPlan {
@@ -16,4 +16,12 @@ impl BuildPlan {
     pub fn spec(&self) -> &ImageSpec {
         &self.spec
     }
+
+    pub fn stages(&self) -> &'static [Stage] {
+        &Stage::ORDERED
+    }
+
+    pub fn dependencies(&self, stage: Stage) -> &'static [Stage] {
+        stage.dependencies()
+    }
 }