beszel/beszel/internal/hub/update.go

package hub

import (
	"beszel/internal/ghupdate"
	"fmt"
	"log"
	"os"
	"os/exec"

	"github.com/spf13/cobra"
)

// Update updates beszel to the latest version
func Update(_ *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"
	}

	updated, err := ghupdate.Update(ghupdate.Config{
		ArchiveExecutable: "beszel",
		DataDir:           dataDir,
	})
	if err != nil {
		log.Fatal(err)
	}
	if !updated {
		return
	}

	// 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 {
			fmt.Println("Restarting beszel service...")
			restartCmd := exec.Command("systemctl", "restart", "beszel.service")
			if err := restartCmd.Run(); err != nil {
				fmt.Printf("Warning: Failed to restart service: %v\n", err)
				fmt.Println("Please restart the service manually: sudo systemctl restart beszel")
			} else {
				fmt.Println("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 {
			fmt.Println("Restarting beszel service...")
			restartCmd := exec.Command("rc-service", "beszel", "restart")
			if err := restartCmd.Run(); err != nil {
				fmt.Printf("Warning: Failed to restart service: %v\n", err)
				fmt.Println("Please restart the service manually: sudo rc-service beszel restart")
			} else {
				fmt.Println("Service restarted successfully")
			}
			return
		}
	}

	fmt.Println("Note: Service restart not attempted. If running as a service, restart manually.")
}