Merge pull request #11 from stuartleeks/sl/update-check

Add periodic update check

Author: stuartleeks

cmd/devcontainer/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"github.com/spf13/cobra"
+	"github.com/stuartleeks/devcontainer-cli/internal/pkg/update"
 )
 
 // Overridden via ldflags
@@ -14,7 +15,12 @@ var (
 
 func main() {
 
-	rootCmd := &cobra.Command{Use: "devcontainer"}
+	rootCmd := &cobra.Command{
+		Use: "devcontainer",
+		PersistentPreRun: func(cmd *cobra.Command, args []string) {
+			update.PeriodicCheckForUpdate(version)
+		},
+	}
 
 	rootCmd.AddCommand(createCompleteCommand(rootCmd))
 	rootCmd.AddCommand(createConfigCommand())

cmd/devcontainer/update.go

@@ -6,9 +6,9 @@ import (
 	"log"
 	"os"
 
-	"github.com/blang/semver"
 	"github.com/rhysd/go-github-selfupdate/selfupdate"
 	"github.com/spf13/cobra"
+	"github.com/stuartleeks/devcontainer-cli/internal/pkg/update"
 )
 
 func createUpdateCommand() *cobra.Command {
@@ -20,19 +20,18 @@ func createUpdateCommand() *cobra.Command {
 		Use:   "update",
 		Short: "update cli",
 		Long:  "Apply the latest update",
+		PersistentPreRun: func(cmd *cobra.Command, args []string) {
+			// do nothing - suppress root PersistentPreRun which does periodic update check
+		},
 		Run: func(cmd *cobra.Command, args []string) {
-			latest, found, err := selfupdate.DetectLatest("stuartleeks/devcontainer-cli")
+			latest, err := update.CheckForUpdate(version)
 			if err != nil {
-				log.Println("Error occurred while detecting version:", err)
+				log.Println("Error occurred while checking for updates:", err)
 				return
 			}
 
-			v, err := semver.Parse(version)
-			if err != nil {
-				log.Panicln(err.Error())
-			}
-			if !found || latest.Version.LTE(v) {
-				log.Println("Current version is the latest")
+			if latest == nil {
+				fmt.Println("No updates available")
 				return
 			}
 

internal/pkg/config/config.go

@@ -3,6 +3,7 @@ package config
 import (
 	"fmt"
 	"os"
+	"time"
 
 	"github.com/spf13/viper"
 )
@@ -43,6 +44,14 @@ func GetTemplateFolders() []string {
 	EnsureInitialised()
 	return viper.GetStringSlice("templatePaths")
 }
+func GetLastUpdateCheck() time.Time {
+	EnsureInitialised()
+	return viper.GetTime("lastUpdateCheck")
+}
+func SetLastUpdateCheck(t time.Time) {
+	EnsureInitialised()
+	viper.Set("lastUpdateCheck", t)
+}
 func GetAll() map[string]interface{} {
 	EnsureInitialised()
 	return viper.AllSettings()

internal/pkg/update/update.go

@@ -0,0 +1,54 @@
+package update
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/blang/semver"
+	"github.com/rhysd/go-github-selfupdate/selfupdate"
+	"github.com/stuartleeks/devcontainer-cli/internal/pkg/config"
+)
+
+func CheckForUpdate(currentVersion string) (*selfupdate.Release, error) {
+
+	latest, found, err := selfupdate.DetectLatest("stuartleeks/devcontainer-cli")
+	if err != nil {
+		return nil, fmt.Errorf("Error occurred while detecting version: %v", err)
+	}
+
+	v, err := semver.Parse(currentVersion)
+	if err != nil {
+		return nil, fmt.Errorf("Error occurred while parsing version: %v", err)
+	}
+
+	if !found || latest.Version.LTE(v) {
+		return nil, nil
+	}
+	return latest, nil
+}
+func PeriodicCheckForUpdate(currentVersion string) {
+	const checkInterval time.Duration = 24 * time.Hour
+
+	lastCheck := config.GetLastUpdateCheck()
+
+	if time.Now().Before(lastCheck.Add(checkInterval)) {
+		return
+	}
+	fmt.Println("Checking for updates...")
+	latest, err := CheckForUpdate(currentVersion)
+	if err != nil {
+		fmt.Printf("Error checking for updates: %s", err)
+	}
+
+	config.SetLastUpdateCheck(time.Now())
+	if err = config.SaveConfig(); err != nil {
+		fmt.Printf("Error saving last update check time: :%s\n", err)
+	}
+
+	if latest == nil {
+		return
+	}
+
+	fmt.Printf("\n\n UPDATE AVAILABLE: %s \n \n Release notes: %s\n", latest.Version, latest.ReleaseNotes)
+	fmt.Printf("Run `devcontainer udpate` to apply the update\n\n")
+}