Add additional device naming formats to formatPartition func:

This allows for using the formatPartition func for more device types. `vd*` is a device created by virtio. `hd*` for legacy IDE hard disks. Signed-off-by: Jacob Weinstock <[email protected]>
tinkerbell · Jul 17, 2023 · 7b5ba2b · 7b5ba2b
1 parent e79eb4d
commit 7b5ba2b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/workflow/template_funcs.go b/internal/workflow/template_funcs.go
@@ -20,11 +20,12 @@ var templateFuncs = map[string]interface{}{
 //
 //	formatPartition("/dev/nvme0n1", 0) -> /dev/nvme0n1p1
 //	formatPartition("/dev/sda", 1) -> /dev/sda1
+//	formatPartition("/dev/vda", 2) -> /dev/vda2
 func formatPartition(dev string, partition int) string {
 	switch {
 	case strings.HasPrefix(dev, "/dev/nvme"):
 		return fmt.Sprintf("%vp%v", dev, partition)
-	case strings.HasPrefix(dev, "/dev/sd"):
+	case strings.HasPrefix(dev, "/dev/sd"), strings.HasPrefix(dev, "/dev/vd"), strings.HasPrefix(dev, "/dev/hd"):
 		return fmt.Sprintf("%v%v", dev, partition)
 	}
 	return dev