feat: Enable function for merging disks (#150)

luthermonson · Jun 4, 2024 · 215b6c1 · 215b6c1
1 parent fe3dbec
commit 215b6c1
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/virtual_machine_config.go b/virtual_machine_config.go
@@ -114,3 +114,24 @@ func (vmc *VirtualMachineConfig) MergeIPConfigs() map[string]string {
 	}
 	return vmc.IPConfigs
 }
+
+func (vmc *VirtualMachineConfig) MergeDisks() map[string]string {
+	mergedDisks := make(map[string]string)
+
+	for k, v := range vmc.MergeIDEs() {
+		mergedDisks[k] = v
+	}
+
+	for k, v := range vmc.MergeSCSIs() {
+		mergedDisks[k] = v
+	}
+
+	for k, v := range vmc.MergeSATAs() {
+		mergedDisks[k] = v
+	}
+
+	for k, v := range vmc.MergeVirtIOs() {
+		mergedDisks[k] = v
+	}
+	return mergedDisks
+}