vm: support additional disks

abiosoft · Nov 19, 2023 · e09ad4b · e09ad4b
1 parent 0ab69ea
commit e09ad4b
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 1 deletion.
diff --git a/config/config.go b/config/config.go
@@ -111,7 +111,7 @@ type Config struct {
 type Disk struct {
 	Name   string `yaml:"name"`
 	Size   string `yaml:"size"`
-	Format string `yaml:"format"`
+	Format string `yaml:"format,omitempty"`
 }
 
 // Kubernetes is kubernetes configuration

diff --git a/embedded/defaults/colima.yaml b/embedded/defaults/colima.yaml
@@ -172,6 +172,16 @@ sshConfig: true
 # Default: []
 mounts: []
 
+# Configure additional disks
+#
+# EXAMPLE
+# additionalDisks:
+#   - size: 10
+#     path: /tmp/colima/disk1.img
+#   - size: 20
+#     path: /tmp/colima/disk2.img
+additionalDisks: []
+
 # Environment variables for the virtual machine.
 #
 # EXAMPLE

diff --git a/environment/vm/lima/lima.go b/environment/vm/lima/lima.go
@@ -163,6 +163,15 @@ func (l *limaVM) resume(ctx context.Context, conf config.Config) error {
 
 	a.Add(l.writeNetworkFile)
 
+	if len(conf.Disks) > 0 {
+		for _, d := range conf.Disks {
+			log.Println("creating disk", d.Name)
+			a.Add(func() error {
+				return l.host.Run(limactl, "disk", "create", d.Name, "--size", d.Size, "--format", d.Format)
+			})
+		}
+	}
+
 	a.Stage("starting")
 	a.Add(func() error {
 		return l.host.Run(limactl, "start", config.CurrentProfile().ID)
@@ -227,6 +236,16 @@ func (l limaVM) Teardown(ctx context.Context) error {
 		return l.host.Run(limactl, "delete", "--force", config.CurrentProfile().ID)
 	})
 
+	conf, _ := limautil.InstanceConfig()
+
+	if len(conf.Disks) > 0 {
+		for _, d := range conf.Disks {
+			a.Add(func() error {
+				return l.host.Run(limactl, "disk", "delete", d.Name)
+			})
+		}
+	}
+
 	return a.Exec()
 }