diff --git a/machine.go b/machine.go index 3a920f0..8fc6ba6 100644 --- a/machine.go +++ b/machine.go @@ -55,7 +55,11 @@ func handleMachine(r *http.Request, w http.ResponseWriter, hostname string, data } if m.Upgrade { log.Info(m.Hostname, "was told to upgrade zood") - fmt.Fprintln(w, "apt update") + if m.UpgradeCmd == "" { + fmt.Fprintln(w, "apt update") + } else { + fmt.Fprintln(w, m.UpgradeCmd) + } m.Upgrade = false } else { fmt.Fprintln(w, "good")