mirror of
https://github.com/netbirdio/netbird.git
synced 2026-03-31 06:34:14 -04:00
fe9b844511
Auto-update logic moved out of the UI into a dedicated updatemanager.Manager service that runs in the connection layer. The UI no longer polls or checks for updates independently. The update manager supports three modes driven by the management server's auto-update policy: No policy set by mgm: checks GitHub for the latest version and notifies the user (previous behavior, now centralized) mgm enforces update: the "About" menu triggers installation directly instead of just downloading the file — user still initiates the action mgm forces update: installation proceeds automatically without user interaction updateManager lifecycle is now owned by daemon, giving the daemon server direct control via a new TriggerUpdate RPC Introduces EngineServices struct to group external service dependencies passed to NewEngine, reducing its argument count from 11 to 4
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
//go:build windows || darwin
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/updater/installer"
|
|
"github.com/netbirdio/netbird/util"
|
|
)
|
|
|
|
var (
|
|
updateCmd = &cobra.Command{
|
|
Use: "update",
|
|
Short: "Update the NetBird client application",
|
|
RunE: updateFunc,
|
|
}
|
|
|
|
tempDirFlag string
|
|
installerFile string
|
|
serviceDirFlag string
|
|
dryRunFlag bool
|
|
)
|
|
|
|
func init() {
|
|
updateCmd.Flags().StringVar(&tempDirFlag, "temp-dir", "", "temporary dir")
|
|
updateCmd.Flags().StringVar(&installerFile, "installer-file", "", "installer file")
|
|
updateCmd.Flags().StringVar(&serviceDirFlag, "service-dir", "", "service directory")
|
|
updateCmd.Flags().BoolVar(&dryRunFlag, "dry-run", false, "dry run the update process without making any changes")
|
|
}
|
|
|
|
// isUpdateBinary checks if the current executable is named "update" or "update.exe"
|
|
func isUpdateBinary() bool {
|
|
// Remove extension for cross-platform compatibility
|
|
execPath, err := os.Executable()
|
|
if err != nil {
|
|
return false
|
|
}
|
|
baseName := filepath.Base(execPath)
|
|
name := strings.TrimSuffix(baseName, filepath.Ext(baseName))
|
|
|
|
return name == installer.UpdaterBinaryNameWithoutExtension()
|
|
}
|
|
|
|
func updateFunc(cmd *cobra.Command, args []string) error {
|
|
if err := setupLogToFile(tempDirFlag); err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Infof("updater started: %s", serviceDirFlag)
|
|
updater := installer.NewWithDir(tempDirFlag)
|
|
if err := updater.Setup(context.Background(), dryRunFlag, installerFile, serviceDirFlag); err != nil {
|
|
log.Errorf("failed to update application: %v", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func setupLogToFile(dir string) error {
|
|
logFile := filepath.Join(dir, installer.LogFile)
|
|
|
|
if _, err := os.Stat(logFile); err == nil {
|
|
if err := os.Remove(logFile); err != nil {
|
|
log.Errorf("failed to remove existing log file: %v\n", err)
|
|
}
|
|
}
|
|
|
|
return util.InitLog(logLevel, util.LogConsole, logFile)
|
|
}
|