Add auto-update check and notification system (#11)

* Add auto-update check and notification system

* fix lint

Author: strahe

Makefile

@@ -7,7 +7,7 @@ PACKAGE := github.com/strahe/bwh
 MAIN_PACKAGE := ./cmd/bwh
 
 # Build information
-VERSION ?= $(shell git describe --tags --always --dirty 2>/dev/null || echo "dev")
+VERSION ?= $(or $(BWH_VERSION),$(shell git describe --tags --always --dirty 2>/dev/null || echo "dev"))
 BUILD_TIME := $(shell date -u '+%Y-%m-%d_%H:%M:%S_UTC')
 COMMIT_HASH := $(shell git rev-parse --short HEAD 2>/dev/null || echo "unknown")
 

cmd/bwh/main.go

@@ -2,11 +2,15 @@ package main
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"log"
 	"os"
+	"path/filepath"
+	"time"
 
 	"github.com/strahe/bwh/internal/config"
+	"github.com/strahe/bwh/internal/updater"
 	"github.com/strahe/bwh/internal/version"
 	"github.com/urfave/cli/v3"
 )
@@ -18,6 +22,8 @@ func main() {
 		Version:               version.GetVersion(),
 		EnableShellCompletion: true,
 		ShellComplete:         shellComplete,
+		Before:                showUpdateNotificationHook,
+		After:                 checkForUpdatesHook,
 		Flags: []cli.Flag{
 			&cli.StringFlag{
 				Name:    "config",
@@ -79,3 +85,98 @@ func shellComplete(ctx context.Context, cmd *cli.Command) {
 		}
 	}
 }
+
+func showUpdateNotificationHook(ctx context.Context, cmd *cli.Command) (context.Context, error) {
+	if len(os.Args) > 1 && os.Args[1] == "update" {
+		return ctx, nil
+	}
+	showCachedUpdateNotification()
+	return ctx, nil
+}
+
+func checkForUpdatesHook(ctx context.Context, cmd *cli.Command) error {
+	if len(os.Args) > 1 && os.Args[1] == "update" {
+		return nil
+	}
+
+	if !shouldCheckForUpdates() {
+		return nil
+	}
+
+	updateLastCheckTime()
+
+	checkCtx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+
+	if info, err := updater.CheckForUpdatesWithTimeout(checkCtx, 2*time.Second); err == nil && info.HasUpdate {
+		cacheUpdateInfo(info)
+	}
+
+	return nil
+}
+
+func shouldCheckForUpdates() bool {
+	lastCheckFile := getLastCheckFilePath()
+	if stat, err := os.Stat(lastCheckFile); err == nil {
+		if time.Since(stat.ModTime()) < 24*time.Hour {
+			return false
+		}
+	}
+	return true
+}
+
+func showCachedUpdateNotification() {
+	updateCacheFile := getUpdateCacheFilePath()
+	data, err := os.ReadFile(updateCacheFile)
+	if err != nil {
+		return
+	}
+
+	var info updater.UpdateInfo
+	if err := json.Unmarshal(data, &info); err != nil {
+		os.Remove(updateCacheFile) //nolint:errcheck
+		return
+	}
+
+	// Check if user has already upgraded
+	currentVersion := version.GetVersion()
+	if updater.CompareVersions(currentVersion, info.LatestVersion) >= 0 {
+		os.Remove(updateCacheFile) //nolint:errcheck
+		return
+	}
+
+	fmt.Fprintf(os.Stderr, "\n┌─────────────────────────────────────────────────────────────┐\n")
+	fmt.Fprintf(os.Stderr, "│ 🎉 BWH CLI %s is available! Current: %-15s    │\n", info.LatestVersion, info.CurrentVersion)
+	fmt.Fprintf(os.Stderr, "│    Run 'bwh update' to upgrade                              │\n")
+	fmt.Fprintf(os.Stderr, "└─────────────────────────────────────────────────────────────┘\n\n")
+}
+
+func cacheUpdateInfo(info *updater.UpdateInfo) {
+	updateCacheFile := getUpdateCacheFilePath()
+	if err := os.MkdirAll(filepath.Dir(updateCacheFile), 0o755); err != nil {
+		return
+	}
+	data, err := json.Marshal(info)
+	if err != nil {
+		return
+	}
+	os.WriteFile(updateCacheFile, data, 0o644) //nolint:errcheck
+}
+
+func updateLastCheckTime() {
+	lastCheckFile := getLastCheckFilePath()
+	if err := os.MkdirAll(filepath.Dir(lastCheckFile), 0o755); err != nil {
+		return
+	}
+	os.WriteFile(lastCheckFile, []byte(time.Now().Format(time.RFC3339)), 0o644) //nolint:errcheck
+}
+
+func getUpdateCacheFilePath() string {
+	homeDir, _ := os.UserHomeDir()
+	return filepath.Join(homeDir, ".bwh", ".update_available")
+}
+
+func getLastCheckFilePath() string {
+	homeDir, _ := os.UserHomeDir()
+	return filepath.Join(homeDir, ".bwh", ".last_check")
+}