Files
Provisioning/internal/image/service.go
T
josh e2dd78d8f9
build-and-push / test (push) Successful in 35s
build-and-push / build-and-push (push) Successful in 1m8s
Embed ISO in initrd for PXE boot with loop-mount wrapper
Instead of sanboot (which can't pass kernel params for automation),
switch back to kernel/initrd boot. The ISO is embedded in the initrd
as a CPIO append. A pxe-init wrapper script loop-mounts the ISO
before handing off to the original init, so the installer finds it
as a block device. Uses rdinit=/pxe-init kernel parameter.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-05-14 12:11:18 -04:00

107 lines
2.5 KiB
Go

package image
import (
"context"
"fmt"
"io"
"os"
"path/filepath"
"regexp"
"provisioning/internal/model"
"provisioning/internal/store"
)
type Service struct {
Store *store.Images
ImageDir string
}
type UploadParams struct {
Name string
Kind string
Version string
ISO io.Reader
OnProgress ProgressFunc
}
var slugRegex = regexp.MustCompile(`^[a-z0-9][a-z0-9.-]*$`)
func (s *Service) Upload(ctx context.Context, p UploadParams) (*model.Image, error) {
if !slugRegex.MatchString(p.Name) {
return nil, fmt.Errorf("invalid name %q: must be lowercase alphanumeric with hyphens/dots", p.Name)
}
if p.Kind == "" {
p.Kind = "proxmox"
}
if p.Version == "" {
return nil, fmt.Errorf("version is required")
}
if _, err := s.Store.GetByName(ctx, p.Name); err == nil {
return nil, fmt.Errorf("image %q already exists", p.Name)
}
destDir := filepath.Join(s.ImageDir, p.Name)
if err := os.MkdirAll(destDir, 0o755); err != nil {
return nil, fmt.Errorf("create image dir: %w", err)
}
result, err := ExtractFromISOWithProgress(p.ISO, destDir, p.OnProgress)
if err != nil {
os.RemoveAll(destDir)
return nil, fmt.Errorf("extract ISO: %w", err)
}
report := func(stage, detail string) {
if p.OnProgress != nil {
p.OnProgress(stage, detail)
}
}
report("packaging", "Creating PXE boot image (embedding ISO in initrd)...")
origInitrdPath := filepath.Join(destDir, result.InitrdFilename)
origISOPath := filepath.Join(destDir, result.ISOFilename)
pxeInitrdFile := "initrd-pxe.img"
pxeInitrdPath := filepath.Join(destDir, pxeInitrdFile)
if err := CreatePXEInitrd(origInitrdPath, origISOPath, pxeInitrdPath); err != nil {
os.RemoveAll(destDir)
return nil, fmt.Errorf("create PXE initrd: %w", err)
}
kernelPath := filepath.Join(p.Name, result.KernelFilename)
initrdPath := filepath.Join(p.Name, pxeInitrdFile)
isoPath := filepath.Join(p.Name, result.ISOFilename)
id, err := s.Store.Create(ctx, model.Image{
Name: p.Name,
Kind: p.Kind,
Version: p.Version,
KernelPath: kernelPath,
InitrdPath: initrdPath,
ISOPath: isoPath,
})
if err != nil {
os.RemoveAll(destDir)
return nil, fmt.Errorf("save image record: %w", err)
}
img, err := s.Store.Get(ctx, id)
if err != nil {
return nil, err
}
return img, nil
}
func (s *Service) Delete(ctx context.Context, id int64) error {
img, err := s.Store.Get(ctx, id)
if err != nil {
return err
}
destDir := filepath.Join(s.ImageDir, img.Name)
os.RemoveAll(destDir)
return s.Store.Delete(ctx, id)
}