feat: add --apply flag

cmd/root.go

@@ -114,6 +114,7 @@ func init() {
 	rootCmd.Flags().Bool("ci", false, "Makes some modifications to behavior if is running in CI")
 	isTerminal := term.IsTerminal(int(os.Stdout.Fd()))
 	rootCmd.Flags().Bool("disable-osc-progress", !isTerminal, "Disable the GUI progress indicator")
+	rootCmd.Flags().Bool("apply", false, "Reboot if there's an update to the image")
 	rootCmd.PersistentFlags().BoolVar(&fLogJson, "json", false, "Print logs as json (used for testing)")
 	rootCmd.PersistentFlags().StringVar(&fLogFile, "log-file", "-", "File where user-facing logs will be written to")
 	rootCmd.PersistentFlags().StringVar(&fLogLevel, "log-level", "info", "Log level for user-facing logs")

cmd/update.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"log/slog"
 	"os"
+	"os/exec"
 
 	"github.com/spf13/cobra"
 	"github.com/ublue-os/uupd/checks"
@@ -55,6 +56,11 @@ func Update(cmd *cobra.Command, args []string) {
 		slog.Error("Failed to get disable-osc-progress flag", "error", err)
 		return
 	}
+	applySystem, err := cmd.Flags().GetBool("apply")
+	if err != nil {
+		slog.Error("Failed to get apply flag", "error", err)
+		return
+	}
 
 	if hwCheck {
 		err := checks.RunHwChecks()
@@ -207,4 +213,9 @@ func Update(cmd *cobra.Command, args []string) {
 	}
 
 	slog.Info("Updates Completed Successfully")
+	if applySystem && mainSystemDriverConfig.Enabled {
+		slog.Info("Applying System Update")
+		cmd := exec.Command("/usr/bin/systemctl", "reboot")
+		cmd.Start()
+	}
 }