provisioning: Implement find-disk constraints

Signed-off-by: Ikey Doherty <[email protected]>

Author: ikeycode

crates/provisioning/src/commands/find_disk.rs

@@ -4,11 +4,12 @@
 
 use itertools::Itertools;
 
-use crate::Context;
+use crate::{Constraints, Context};
 
 #[derive(Debug)]
 pub struct Command {
     pub name: String,
+    pub constraints: Option<Constraints>,
 }
 
 /// Generate a command to find a disk
@@ -41,5 +42,15 @@ pub(crate) fn parse(context: Context<'_>) -> Result<super::Command, crate::Error
         }
     };
 
-    Ok(super::Command::FindDisk(Box::new(Command { name: name.to_owned() })))
+    let constraints =
+        if let Some(constraints) = context.node.iter_children().find(|n| n.name().value() == "constraints") {
+            Some(Constraints::from_kdl_node(constraints)?)
+        } else {
+            None
+        };
+
+    Ok(super::Command::FindDisk(Box::new(Command {
+        name: name.to_owned(),
+        constraints,
+    })))
 }

crates/provisioning/tests/use_whole_disk.kdl

@@ -4,8 +4,7 @@ strategy name="whole_disk" summary="Wipe and use an entire disk" {
     // The result is stored in "root_disk" (disk=..)
     find-disk "root_disk" {
         constraints {
-            min (GB)"30"
-            //- partition-table "gpt" // Require gpt for reuse
+            min (GB)30
         }
     }