From 08a81696a1ef4f7fe2364e4ae4c7aa1faa830c67 Mon Sep 17 00:00:00 2001
From: Daniel-GrunbergerCA <danielg@armosec.io>
Date: Thu, 9 Sep 2021 14:58:52 +0300
Subject: [PATCH] update latest version check

---
 cmd/version.go | 15 ++-------------
 main.go        | 12 +++++++++++-
 2 files changed, 13 insertions(+), 14 deletions(-)

diff --git a/cmd/version.go b/cmd/version.go
index 818c387d..d37be0bd 100644
--- a/cmd/version.go
+++ b/cmd/version.go
@@ -12,7 +12,7 @@ import (
 var (
 	sha1ver    string
 	buildTime  string
-	run_number string
+	Run_number string
 )
 
 // versionCmd represents the version command
@@ -21,18 +21,7 @@ var versionCmd = &cobra.Command{
 	Short: "Get current version",
 	Long:  ``,
 	RunE: func(cmd *cobra.Command, args []string) error {
-		fmt.Println(sha1ver)
-		fmt.Println(buildTime)
-		latest, err := GetLatestVersion()
-		if err != nil {
-			return err
-		}
-		fmt.Println(latest)
-		if latest == run_number {
-			fmt.Println("Updated")
-		} else {
-			fmt.Println("Not updated")
-		}
+		fmt.Println("Your current version is: " + Run_number)
 		return nil
 	},
 }
diff --git a/main.go b/main.go
index 5b5cd6ea..65971460 100644
--- a/main.go
+++ b/main.go
@@ -1,10 +1,20 @@
 package main
 
 import (
+	"fmt"
+
 	"github.com/armosec/kubescape/cmd"
 )
 
 func main() {
-
+	CheckLatestVersion()
 	cmd.Execute()
 }
+
+func CheckLatestVersion() {
+	latest, _ := cmd.GetLatestVersion()
+	if latest != cmd.Run_number {
+		fmt.Println("Warning: You are not updated to the latest release: " + latest)
+	}
+
+}