From 85c8c4e590625148faf75859072fe9fae3e1f302 Mon Sep 17 00:00:00 2001
From: GnomeZworc <nicolas.boufidjeline@g3e.fr>
Date: Sun, 24 May 2026 15:44:48 +0200
Subject: [PATCH] f-32: disk: add boot multidisk handle

Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
---
 internal/api/agent/vms.go                | 23 ++++++++++++--------
 internal/dispatcher/agent/vm_commands.go | 27 +++++++++++++++---------
 internal/vm/create.go                    |  2 +-
 internal/vm/data.go                      | 20 ++++++++++++++----
 4 files changed, 48 insertions(+), 24 deletions(-)

diff --git a/internal/api/agent/vms.go b/internal/api/agent/vms.go
index 4dc6136..2ea5748 100644
--- a/internal/api/agent/vms.go
+++ b/internal/api/agent/vms.go
@@ -77,16 +77,21 @@ func (s *Server) startVM(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	disks := make([]dispatcher.VMDisk, len(req.Storage))
+	for i, s := range req.Storage {
+		disks[i] = dispatcher.VMDisk{Path: s.Path, Dev: s.Dev}
+	}
+
 	cmd := dispatcher.StartVMCommand{
-		Name:       req.Name,
-		Subnet:     primary.Subnet,
-		IP:         primary.IP,
-		VolumePath: req.Storage[0].Path,
-		Memory:     req.Memory,
-		CPUs:       req.CPUs,
-		UEFI:       req.UEFI,
-		Password:   req.Password,
-		SSHKey:     req.SSHKey,
+		Name:     req.Name,
+		Subnet:   primary.Subnet,
+		IP:       primary.IP,
+		Disks:    disks,
+		Memory:   req.Memory,
+		CPUs:     req.CPUs,
+		UEFI:     req.UEFI,
+		Password: req.Password,
+		SSHKey:   req.SSHKey,
 	}
 
 	if err := s.dispatcher.Prepare(cmd); err != nil {
diff --git a/internal/dispatcher/agent/vm_commands.go b/internal/dispatcher/agent/vm_commands.go
index 1c4bbce..bdfeaec 100644
--- a/internal/dispatcher/agent/vm_commands.go
+++ b/internal/dispatcher/agent/vm_commands.go
@@ -13,16 +13,21 @@ import (
 	"github.com/dgraph-io/badger/v4"
 )
 
+type VMDisk struct {
+	Path string
+	Dev  string
+}
+
 type StartVMCommand struct {
-	Name       string
-	Subnet     string
-	IP         string
-	VolumePath string
-	Memory     int
-	CPUs       int
-	UEFI       bool
-	Password   string
-	SSHKey     string
+	Name     string
+	Subnet   string
+	IP       string
+	Disks    []VMDisk
+	Memory   int
+	CPUs     int
+	UEFI     bool
+	Password string
+	SSHKey   string
 }
 
 func (c StartVMCommand) Prepare(db *badger.DB, _ *configuration.Config) error {
@@ -44,7 +49,9 @@ func (c StartVMCommand) Prepare(db *badger.DB, _ *configuration.Config) error {
 	kv.AddInDB(db, "vm/"+c.Name+"/subnet", c.Subnet)
 	kv.AddInDB(db, "vm/"+c.Name+"/ip", c.IP)
 	kv.AddInDB(db, "vm/"+c.Name+"/metadata_port", strconv.Itoa(port))
-	kv.AddInDB(db, "vm/"+c.Name+"/volume_path", c.VolumePath)
+	for _, d := range c.Disks {
+		kv.AddInDB(db, "vm/"+c.Name+"/disk/"+d.Dev, d.Path)
+	}
 	kv.AddInDB(db, "vm/"+c.Name+"/memory", strconv.Itoa(c.Memory))
 	kv.AddInDB(db, "vm/"+c.Name+"/cpus", strconv.Itoa(c.CPUs))
 	if c.UEFI {
diff --git a/internal/vm/create.go b/internal/vm/create.go
index 529c73c..267dab5 100644
--- a/internal/vm/create.go
+++ b/internal/vm/create.go
@@ -56,7 +56,7 @@ func StartVM(db *badger.DB, name string, cfg *configuration.Config) error {
 		Name:       name,
 		TapID:      d.tapID,
 		Mac:        d.mac,
-		VolumePath: d.volumePath,
+		VolumePath: d.disks[0].path,
 		Memory:     d.memory,
 		CPUs:       d.cpus,
 		SerialDir:  cfg.QEMU.SerialDir,
diff --git a/internal/vm/data.go b/internal/vm/data.go
index 1e2644e..3435fb1 100644
--- a/internal/vm/data.go
+++ b/internal/vm/data.go
@@ -11,6 +11,11 @@ import (
 	"github.com/dgraph-io/badger/v4"
 )
 
+type diskEntry struct {
+	path string
+	dev  string
+}
+
 type vmData struct {
 	subnetName   string
 	vpcName      string
@@ -20,7 +25,7 @@ type vmData struct {
 	ip           string
 	metadataPort string
 	mac          string
-	volumePath   string
+	disks        []diskEntry
 	memory       int
 	cpus         int
 	uefi         bool
@@ -82,11 +87,18 @@ func loadVM(db *badger.DB, name string) (vmData, error) {
 	}
 	d.mac = mac
 
-	volumePath, err := kv.GetFromDB(db, "vm/"+name+"/volume_path")
+	diskEntries, err := kv.ListByPrefix(db, "vm/"+name+"/disk/")
 	if err != nil {
-		return d, fmt.Errorf("get volume_path: %w", err)
+		return d, fmt.Errorf("list disks: %w", err)
+	}
+	if len(diskEntries) == 0 {
+		return d, fmt.Errorf("no disks found for vm %q", name)
+	}
+	diskPrefix := "vm/" + name + "/disk/"
+	for key, path := range diskEntries {
+		dev := strings.TrimPrefix(key, diskPrefix)
+		d.disks = append(d.disks, diskEntry{path: path, dev: dev})
 	}
-	d.volumePath = volumePath
 
 	memoryStr, err := kv.GetFromDB(db, "vm/"+name+"/memory")
 	if err != nil {