Add validated image build stage graph

1 files changed, 45 insertions, 0 deletions

diff --git a/src/stage.rs b/src/stage.rs
new file mode 100644
index 0000000..612b594
--- /dev/null
+++ b/src/stage.rs
@@ -0,0 +1,45 @@
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum Stage {
+    Resolve,
+    Install,
+    FinalizeRootfs,
+    BuildInitramfs,
+    Package,
+    Manifest,
+}
+
+impl Stage {
+    pub const ORDERED: [Self; 6] = [
+        Self::Resolve,
+        Self::Install,
+        Self::FinalizeRootfs,
+        Self::BuildInitramfs,
+        Self::Package,
+        Self::Manifest,
+    ];
+
+    pub fn dependencies(self) -> &'static [Self] {
+        match self {
+            Self::Resolve => &[],
+            Self::Install => &[Self::Resolve],
+            Self::FinalizeRootfs => &[Self::Install],
+            Self::BuildInitramfs => &[Self::Install],
+            Self::Package => &[Self::FinalizeRootfs, Self::BuildInitramfs],
+            Self::Manifest => &[Self::Package],
+        }
+    }
+}
+
+impl std::fmt::Display for Stage {
+    fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        let name = match self {
+            Self::Resolve => "resolve",
+            Self::Install => "install",
+            Self::FinalizeRootfs => "finalize-rootfs",
+            Self::BuildInitramfs => "initramfs",
+            Self::Package => "package",
+            Self::Manifest => "manifest",
+        };
+        formatter.write_str(name)
+    }
+}