Add validated image build stage graph

4 files changed, 60 insertions, 1 deletions

diff --git a/src/cli.rs b/src/cli.rs
index de9404a..c91d4fd 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -49,6 +49,10 @@ pub fn run(cli: Cli) -> Result<()> {
             println!("target: {}", plan.spec().target.name);
             println!("architecture: {}", plan.spec().target.architecture.as_str());
             println!("format: tar");
+            println!("stages:");
+            for (index, stage) in plan.stages().iter().enumerate() {
+                println!("  {}. {stage}", index + 1);
+            }
             Ok(())
         }
         Command::Build { .. } => bail!("build execution is not available yet"),
diff --git a/src/lib.rs b/src/lib.rs
index 52a875d..792e7a5 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,6 +1,8 @@
 pub mod cli;
 pub mod model;
 pub mod plan;
+pub mod stage;
 
 pub use model::{Architecture, BootSpec, ImageSpec, OutputFormat, PackageSelector, PackageSpec, Target};
 pub use plan::BuildPlan;
+pub use stage::Stage;
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()
+    }
 }
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)
+    }
+}