Merge pull request #9 from flatrun/chore/updater

feat(agent): add self-update command

Author: nfebe

cmd/agent/main.go

@@ -11,21 +11,29 @@ import (
 	"github.com/flatrun/agent/internal/api"
 	"github.com/flatrun/agent/internal/watcher"
 	"github.com/flatrun/agent/pkg/config"
+	"github.com/flatrun/agent/pkg/updater"
 	"github.com/flatrun/agent/pkg/version"
 )
 
 func main() {
+	if len(os.Args) > 1 {
+		switch os.Args[1] {
+		case "update":
+			handleUpdate(os.Args[2:])
+			return
+		case "version":
+			printVersion()
+			return
+		}
+	}
+
 	configPath := flag.String("config", "config.yml", "Path to configuration file")
 	showVersion := flag.Bool("version", false, "Print version information")
 	flag.Parse()
 
 	if *showVersion {
-		info := version.Get()
-		fmt.Printf("Flatrun Agent\n")
-		fmt.Printf("Version:    %s\n", info.Version)
-		fmt.Printf("Build Time: %s\n", info.BuildTime)
-		fmt.Printf("Git Commit: %s\n", info.GitCommit)
-		os.Exit(0)
+		printVersion()
+		return
 	}
 
 	cfg, err := config.Load(*configPath)
@@ -59,3 +67,91 @@ func main() {
 	log.Println("Shutting down Flatrun Agent...")
 	_ = apiServer.Stop()
 }
+
+func printVersion() {
+	info := version.Get()
+	fmt.Printf("Flatrun Agent\n")
+	fmt.Printf("Version:    %s\n", info.Version)
+	fmt.Printf("Build Time: %s\n", info.BuildTime)
+	fmt.Printf("Git Commit: %s\n", info.GitCommit)
+}
+
+func handleUpdate(args []string) {
+	updateFlags := flag.NewFlagSet("update", flag.ExitOnError)
+	checkOnly := updateFlags.Bool("check", false, "Check for updates without installing")
+	force := updateFlags.Bool("force", false, "Force update even if on latest version")
+	restart := updateFlags.Bool("restart", false, "Restart service after update")
+	rollback := updateFlags.Bool("rollback", false, "Rollback to previous version")
+
+	updateFlags.Usage = func() {
+		fmt.Println("Usage: flatrun-agent update [options]")
+		fmt.Println()
+		fmt.Println("Options:")
+		updateFlags.PrintDefaults()
+		fmt.Println()
+		fmt.Println("Examples:")
+		fmt.Println("  flatrun-agent update --check     Check for available updates")
+		fmt.Println("  flatrun-agent update             Download and install latest version")
+		fmt.Println("  flatrun-agent update --restart   Update and restart the service")
+		fmt.Println("  flatrun-agent update --rollback  Rollback to previous version")
+	}
+
+	if err := updateFlags.Parse(args); err != nil {
+		os.Exit(1)
+	}
+
+	if *rollback {
+		fmt.Println("Rolling back to previous version...")
+		if err := updater.Rollback(); err != nil {
+			fmt.Printf("Error: %v\n", err)
+			os.Exit(1)
+		}
+		fmt.Println("Rollback successful")
+		fmt.Println("Restart the service to apply: sudo systemctl restart flatrun-agent")
+		return
+	}
+
+	if *checkOnly {
+		result, err := updater.CheckForUpdate()
+		if err != nil {
+			fmt.Printf("Error: %v\n", err)
+			os.Exit(1)
+		}
+		fmt.Printf("Current version: %s\n", result.CurrentVersion)
+		fmt.Printf("Latest version:  %s\n", result.LatestVersion)
+		if result.UpdateAvailable {
+			fmt.Println("Update available!")
+			fmt.Println("Run 'flatrun-agent update' to install")
+		} else {
+			fmt.Println("Already up to date")
+		}
+		return
+	}
+
+	result, err := updater.Update(*force)
+	if err != nil {
+		fmt.Printf("Error: %v\n", err)
+		os.Exit(1)
+	}
+
+	if !result.UpdateAvailable && !*force {
+		fmt.Printf("Already up to date (v%s)\n", result.CurrentVersion)
+		return
+	}
+
+	if result.Installed {
+		fmt.Printf("Successfully updated to v%s\n", result.LatestVersion)
+
+		if *restart {
+			fmt.Println("Restarting service...")
+			if err := updater.RestartService(); err != nil {
+				fmt.Printf("Failed to restart service: %v\n", err)
+				fmt.Println("Please restart manually: sudo systemctl restart flatrun-agent")
+				os.Exit(1)
+			}
+			fmt.Println("Service restarted successfully")
+		} else {
+			fmt.Println("Restart the service to apply: sudo systemctl restart flatrun-agent")
+		}
+	}
+}