mirror of
https://github.com/kairos-io/kairos-agent.git
synced 2025-07-19 18:39:34 +00:00
931c908a2e
To allow the workaround described here: https://github.com/kairos-io/kairos/issues/2304 Signed-off-by: Dimitris Karakasilis <dimitris@karakasilis.me>
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package uki
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/kairos-io/kairos-agent/v2/pkg/action"
|
|
"github.com/kairos-io/kairos-agent/v2/pkg/config"
|
|
"github.com/kairos-io/kairos-agent/v2/pkg/constants"
|
|
"github.com/kairos-io/kairos-agent/v2/pkg/elemental"
|
|
v1 "github.com/kairos-io/kairos-agent/v2/pkg/types/v1"
|
|
elementalUtils "github.com/kairos-io/kairos-agent/v2/pkg/utils"
|
|
events "github.com/kairos-io/kairos-sdk/bus"
|
|
"github.com/kairos-io/kairos-sdk/utils"
|
|
)
|
|
|
|
type ResetAction struct {
|
|
cfg *config.Config
|
|
spec *v1.ResetUkiSpec
|
|
}
|
|
|
|
func NewResetAction(cfg *config.Config, spec *v1.ResetUkiSpec) *ResetAction {
|
|
return &ResetAction{cfg: cfg, spec: spec}
|
|
}
|
|
|
|
func (r *ResetAction) Run() (err error) {
|
|
// Run pre-install stage
|
|
_ = elementalUtils.RunStage(r.cfg, "kairos-uki-reset.pre")
|
|
_ = events.RunHookScript("/usr/bin/kairos-agent.uki.reset.pre.hook")
|
|
|
|
e := elemental.NewElemental(r.cfg)
|
|
cleanup := utils.NewCleanStack()
|
|
defer func() { err = cleanup.Cleanup(err) }()
|
|
|
|
// Unmount partitions if any is already mounted before formatting
|
|
err = e.UnmountPartitions(r.spec.Partitions.PartitionsByMountPoint(true))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Reformat persistent partition
|
|
if r.spec.FormatPersistent {
|
|
persistent := r.spec.Partitions.Persistent
|
|
if persistent != nil {
|
|
err = e.FormatPartition(persistent)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// Reformat OEM
|
|
if r.spec.FormatOEM {
|
|
oem := r.spec.Partitions.OEM
|
|
if oem != nil {
|
|
err = e.FormatPartition(oem)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// REMOUNT /efi as RW (its RO by default)
|
|
umount, err := e.MountRWPartition(r.spec.Partitions.EFI)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cleanup.Push(umount)
|
|
|
|
// Copy "recovery" to "active"
|
|
err = overwriteArtifactSetRole(r.cfg.Fs, constants.UkiEfiDir, "recovery", "active", r.cfg.Logger)
|
|
if err != nil {
|
|
return fmt.Errorf("copying recovery to active: %w", err)
|
|
}
|
|
// SelectBootEntry sets the default boot entry to the selected entry
|
|
err = action.SelectBootEntry(r.cfg, "active")
|
|
// Should we fail? Or warn?
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if mnt, err := elementalUtils.IsMounted(r.cfg, r.spec.Partitions.OEM); !mnt && err == nil {
|
|
err = e.MountPartition(r.spec.Partitions.OEM)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
err = Hook(r.cfg, constants.AfterResetHook)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_ = elementalUtils.RunStage(r.cfg, "kairos-uki-reset.after")
|
|
_ = events.RunHookScript("/usr/bin/kairos-agent.uki.reset.after.hook") //nolint:errcheck
|
|
|
|
// Do not reboot/poweroff on cleanup errors
|
|
err = cleanup.Cleanup(err)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|