mirror of
https://github.com/kairos-io/kairos-agent.git
synced 2025-08-18 08:17:05 +00:00
sparkles: Add reset event for sdk (#143)
This commit is contained in:
parent
e25526cfc9
commit
0d88ec39a9
@ -31,9 +31,11 @@ func optsToArgs(options map[string]string) (res []string) {
|
|||||||
for k, v := range options {
|
for k, v := range options {
|
||||||
if k != "device" && k != "cc" && k != "reboot" && k != "poweroff" {
|
if k != "device" && k != "cc" && k != "reboot" && k != "poweroff" {
|
||||||
res = append(res, fmt.Sprintf("--%s", k))
|
res = append(res, fmt.Sprintf("--%s", k))
|
||||||
|
if v != "" {
|
||||||
res = append(res, v)
|
res = append(res, v)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,19 +1,34 @@
|
|||||||
package agent
|
package agent
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/kairos-io/kairos/internal/bus"
|
||||||
"github.com/kairos-io/kairos/internal/cmd"
|
"github.com/kairos-io/kairos/internal/cmd"
|
||||||
"github.com/kairos-io/kairos/pkg/machine"
|
"github.com/kairos-io/kairos/pkg/machine"
|
||||||
"github.com/kairos-io/kairos/pkg/utils"
|
"github.com/kairos-io/kairos/pkg/utils"
|
||||||
|
sdk "github.com/kairos-io/kairos/sdk/bus"
|
||||||
|
|
||||||
|
"github.com/mudler/go-pluggable"
|
||||||
"github.com/pterm/pterm"
|
"github.com/pterm/pterm"
|
||||||
)
|
)
|
||||||
|
|
||||||
func Reset() error {
|
func Reset() error {
|
||||||
|
bus.Manager.Initialize()
|
||||||
|
|
||||||
|
options := map[string]string{}
|
||||||
|
|
||||||
|
bus.Manager.Response(sdk.EventBeforeReset, func(p *pluggable.Plugin, r *pluggable.EventResponse) {
|
||||||
|
err := json.Unmarshal([]byte(r.Data), &options)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
cmd.PrintBranding(DefaultBanner)
|
cmd.PrintBranding(DefaultBanner)
|
||||||
|
|
||||||
@ -43,7 +58,15 @@ func Reset() error {
|
|||||||
time.Sleep(60 * time.Second)
|
time.Sleep(60 * time.Second)
|
||||||
lock.Lock()
|
lock.Lock()
|
||||||
args := []string{"reset"}
|
args := []string{"reset"}
|
||||||
|
|
||||||
|
bus.Manager.Publish(sdk.EventBeforeReset, sdk.EventPayload{}) //nolint:errcheck
|
||||||
|
|
||||||
|
optsArgs := optsToArgs(options)
|
||||||
|
if len(optsArgs) > 0 {
|
||||||
|
args = append(args, optsArgs...)
|
||||||
|
} else {
|
||||||
args = append(args, "--reset-persistent")
|
args = append(args, "--reset-persistent")
|
||||||
|
}
|
||||||
|
|
||||||
cmd := exec.Command("elemental", args...)
|
cmd := exec.Command("elemental", args...)
|
||||||
cmd.Env = os.Environ()
|
cmd.Env = os.Environ()
|
||||||
@ -55,6 +78,8 @@ func Reset() error {
|
|||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bus.Manager.Publish(sdk.EventAfterReset, sdk.EventPayload{}) //nolint:errcheck
|
||||||
|
|
||||||
pterm.Info.Println("Rebooting in 60 seconds, press Enter to abort...")
|
pterm.Info.Println("Rebooting in 60 seconds, press Enter to abort...")
|
||||||
|
|
||||||
// We don't close the lock, as none of the following actions are expected to return
|
// We don't close the lock, as none of the following actions are expected to return
|
||||||
|
Loading…
Reference in New Issue
Block a user