feat: add option for shutting down pc after backup

config.go

@@ -32,6 +32,7 @@ type Config struct {
 	BackupSourceDir string      `json:"backupdir"`
 	PxarOut         string      `json:"pxarout"`
 	SMTP            *SMTPConfig `json:"smtp"`
+	Shutdown        bool        `json:"shutdown"`
 }
 
 func (c *Config) valid() bool {
@@ -76,6 +77,8 @@ func loadConfig() *Config {
 
 	configFile := flag.String("config", "", "Path to JSON config file. If this flag is provided all the others are ignored")
 
+	shutdownFlag := flag.Bool("shutdown", false, "Shutdown the system after backup")
+
 	// Parse command line flags
 	flag.Parse()
 
@@ -104,6 +107,7 @@ func loadConfig() *Config {
 	config.BackupID = *backupIDFlag
 	config.BackupSourceDir = *backupSourceDirFlag
 	config.PxarOut = *pxarOutFlag
+	config.Shutdown = *shutdownFlag
 
 	initSmtpConfigIfNeeded := func() {
 		if config.SMTP == nil {

main.go

@@ -9,6 +9,7 @@ import (
 	"fmt"
 	"hash"
 	"os"
+	"os/exec"
 	"runtime"
 	"sync/atomic"
 
@@ -130,6 +131,9 @@ func main() {
 		}
 	}
 
+	if cfg.Shutdown {
+		shutDownPc()
+	}
 }
 
 func backup(client *PBSClient, newchunk, reusechunk *atomic.Uint64, pxarOut string, backupdir string) error {
@@ -371,3 +375,15 @@ func backup(client *PBSClient, newchunk, reusechunk *atomic.Uint64, pxarOut stri
 
 	return client.Finish()
 }
+
+func shutDownPc() {
+	if runtime.GOOS == "windows" {
+		if err := exec.Command("cmd", "/C", "shutdown", "/s", "/t", "0").Run(); err != nil {
+			fmt.Println("Failed to initiate shutdown:", err)
+		}
+	} else {
+		if err := exec.Command("shutdown", "now").Run(); err != nil {
+			fmt.Println("Failed to initiate shutdown:", err)
+		}
+	}
+}