Update: Add systemd-boot preparation

Author: Hayao0819

alteriso5/utils/os.go

@@ -14,3 +14,29 @@ func ReadFileLine(path string) ([]string, error) {
 	lines := strings.Split(string(bytes), "\n")
 	return lines, nil
 }
+
+func GetFileSizesInDir(dir string) (map[string]int64, error) {
+	items, err := os.ReadDir(dir)
+	if err != nil {
+		return nil, err
+	}
+
+	ret := map[string]int64{}
+
+	for _, i := range items {
+		info, err := i.Info()
+		if err != nil {
+			continue
+		}
+		ret[info.Name()] = info.Size()
+	}
+	return ret, nil
+}
+
+func GetFileSize(file string) (int64, error) {
+	info, err := os.Stat(file)
+	if err != nil {
+		return 0, err
+	}
+	return info.Size(), nil
+}

alteriso5/work/airootfs/chroot.go

@@ -15,8 +15,8 @@ type Chroot struct {
 
 func GetChrootDir(dir, arch, config string) (*Chroot, error) {
 	env := Chroot{
-		Arch: arch,
-		Dir:  dir,
+		Arch:   arch,
+		Dir:    dir,
 		Config: config,
 	}
 
@@ -65,6 +65,10 @@ type kernel struct {
 	Initrd string
 }
 
+func (k *kernel) Files() []string {
+	return []string{k.Linux, k.Initrd}
+}
+
 // func (e *Chroot) FindKernels() ([]kernel, error) {
 // 	kernels := []kernel{}
 

alteriso5/work/task-bootmode-uefi.go

@@ -1,9 +1,86 @@
 package work
 
-import "log/slog"
+import (
+	"log/slog"
+	"os"
+	"path"
+
+	"github.com/FascodeNet/alterlinux/alteriso5/utils"
+	"github.com/FascodeNet/alterlinux/alteriso5/work/boot"
+	"github.com/Hayao0819/nahi/osutils"
+)
 
 // Prepare configuration files for systemd-boot
 var makeCommonSystemdBootConfig *BuildTask = NewBuildTask("makeCommonSystemdBootConfig", func(w Work) error {
+	var total int64 = 0
+	// TODO: Get ucode list
+
+	// Calculate ESP size
+
+	{
+		efiboot_files := []string{}
+
+		// For UEFI x64
+		if w.profile.HasBootMode(boot.UefiX64SystemdBootEsp) || w.profile.HasBootMode(boot.UefiX64SystemdBootElTorito) {
+			efiboot_files = append(efiboot_files,
+				path.Join(w.Dirs.Pacstrap, "usr", "lib", "systemd", "boot", "efi", "systemd-bootx64.efi"),
+				path.Join(w.Dirs.Pacstrap, "usr", "share", "edk2-shell", "x64", "Shell_Full.efi"),
+				path.Join(w.Dirs.Pacstrap, "boot", "memtest86+", "memtest.efi"),
+				path.Join(w.Dirs.Pacstrap, "usr", "share", "licenses", "spdx", "GPL-2.0-only.txt"),
+			)
+		}
+
+		// For UEFI ia32
+		if w.profile.HasBootMode(boot.UefiIa32SystemdBootEsp) || w.profile.HasBootMode(boot.UefiIa32SystemdBootElTorito) {
+			efiboot_files = append(efiboot_files,
+				path.Join(w.Dirs.Pacstrap, "usr", "lib", "systemd", "boot", "efi", "systemd-bootia32.efi"),
+				path.Join(w.Dirs.Pacstrap, "usr", "share", "edk2-shell", "ia32", "Shell_Full.efi"),
+			)
+		}
+
+		// For kernel
+		{
+			chroot, err := w.GetChroot()
+			if err != nil {
+				return err
+			}
+			kernels, err := chroot.FindKernels()
+			if err != nil {
+				return err
+			}
+			for _, k := range kernels {
+				for _, f := range k.Files() {
+					if osutils.Exists(f) {
+						efiboot_files = append(efiboot_files, f)
+					}
+				}
+			}
+		}
+
+		for _, f := range efiboot_files {
+			if s, err := utils.GetFileSize(f); err == nil {
+				total += s
+			} else {
+				slog.Warn("Failed to get file size", "file", f, "error", err)
+			}
+		}
+
+		slog.Debug("Found files for efiboot", "files", efiboot_files)
+	}
+
+	// For efiboot files
+	if err := os.MkdirAll(path.Join(w.Base, w.target.Arch, "efiboot"), 0755); err != nil {
+		return err
+	}
+	sizes, err := utils.GetFileSizesInDir(path.Join(w.Base, w.target.Arch, "efiboot"))
+	if err != nil {
+		return err
+	}
+	for _, s := range sizes {
+		total += s
+	}
+
+	slog.Debug("efiboot img size", "size", total)
 	return nil
 })
 

profile/profiledef.json

@@ -1,5 +1,9 @@
 {
   "install_dir": "alter",
-  "bootmodes": ["bios.syslinux.mbr", "bios.syslinux.eltorito"],
+  "bootmodes": [
+    "bios.syslinux.mbr",
+    "bios.syslinux.eltorito",
+    "uefi-x64.systemd-boot.esp"
+  ],
   "use_alter_syslinux": true
 }