mirror of
https://github.com/henrygd/beszel.git
synced 2026-03-22 05:36:15 +01:00
42da1e5a52
- Move SELinux context handling to internal/ghupdate for reuse - Make chcon a true fallback (only runs if semanage/restorecon unavailable) - Handle existing semanage rules with -m (modify) after -a (add) fails - Apply SELinux handling to both agent and hub updates - Add tests with proper skip behavior for SELinux systems --------- Co-authored-by: henrygd <hank@henrygd.me>
94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
package hub
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
|
|
"github.com/henrygd/beszel/internal/ghupdate"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// Update updates beszel to the latest version
|
|
func Update(cmd *cobra.Command, _ []string) {
|
|
dataDir := os.TempDir()
|
|
|
|
// set dataDir to ./beszel_data if it exists
|
|
if _, err := os.Stat("./beszel_data"); err == nil {
|
|
dataDir = "./beszel_data"
|
|
}
|
|
|
|
// Check if china-mirrors flag is set
|
|
useMirror, _ := cmd.Flags().GetBool("china-mirrors")
|
|
|
|
// Get the executable path before update
|
|
exePath, err := os.Executable()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
updated, err := ghupdate.Update(ghupdate.Config{
|
|
ArchiveExecutable: "beszel",
|
|
DataDir: dataDir,
|
|
UseMirror: useMirror,
|
|
})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if !updated {
|
|
return
|
|
}
|
|
|
|
// make sure the file is executable
|
|
if err := os.Chmod(exePath, 0755); err != nil {
|
|
fmt.Printf("Warning: failed to set executable permissions: %v\n", err)
|
|
}
|
|
|
|
// Fix SELinux context if necessary
|
|
if err := ghupdate.HandleSELinuxContext(exePath); err != nil {
|
|
ghupdate.ColorPrintf(ghupdate.ColorYellow, "Warning: SELinux context handling: %v", err)
|
|
}
|
|
|
|
// Try to restart the service if it's running
|
|
restartService()
|
|
}
|
|
|
|
// restartService attempts to restart the beszel service
|
|
func restartService() {
|
|
// Check if we're running as a service by looking for systemd
|
|
if _, err := exec.LookPath("systemctl"); err == nil {
|
|
// Check if beszel service exists and is active
|
|
cmd := exec.Command("systemctl", "is-active", "beszel.service")
|
|
if err := cmd.Run(); err == nil {
|
|
ghupdate.ColorPrint(ghupdate.ColorYellow, "Restarting beszel service...")
|
|
restartCmd := exec.Command("systemctl", "restart", "beszel.service")
|
|
if err := restartCmd.Run(); err != nil {
|
|
ghupdate.ColorPrintf(ghupdate.ColorYellow, "Warning: Failed to restart service: %v\n", err)
|
|
ghupdate.ColorPrint(ghupdate.ColorYellow, "Please restart the service manually: sudo systemctl restart beszel")
|
|
} else {
|
|
ghupdate.ColorPrint(ghupdate.ColorGreen, "Service restarted successfully")
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
// Check for OpenRC (Alpine Linux)
|
|
if _, err := exec.LookPath("rc-service"); err == nil {
|
|
cmd := exec.Command("rc-service", "beszel", "status")
|
|
if err := cmd.Run(); err == nil {
|
|
ghupdate.ColorPrint(ghupdate.ColorYellow, "Restarting beszel service...")
|
|
restartCmd := exec.Command("rc-service", "beszel", "restart")
|
|
if err := restartCmd.Run(); err != nil {
|
|
ghupdate.ColorPrintf(ghupdate.ColorYellow, "Warning: Failed to restart service: %v\n", err)
|
|
ghupdate.ColorPrint(ghupdate.ColorYellow, "Please restart the service manually: sudo rc-service beszel restart")
|
|
} else {
|
|
ghupdate.ColorPrint(ghupdate.ColorGreen, "Service restarted successfully")
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
ghupdate.ColorPrint(ghupdate.ColorYellow, "Service restart not attempted. If running as a service, restart manually.")
|
|
}
|