Files
kubeshark/kubeshark/version/versionCheck.go
M. Mert Yildiran cae3b4fe17 💡 Better log messages
2022-11-29 04:52:53 +03:00

83 lines
2.1 KiB
Go

package version
import (
"context"
"fmt"
"io"
"net/http"
"runtime"
"strings"
"time"
"github.com/kubeshark/kubeshark/kubeshark"
"github.com/kubeshark/kubeshark/pkg/version"
"github.com/kubeshark/kubeshark/utils"
"github.com/rs/zerolog/log"
"github.com/google/go-github/v37/github"
)
func CheckNewerVersion() {
log.Info().Msg("Checking for a newer version...")
start := time.Now()
client := github.NewClient(nil)
latestRelease, _, err := client.Repositories.GetLatestRelease(context.Background(), "kubeshark", "kubeshark")
if err != nil {
log.Error().Msg("Failed to get latest release.")
return
}
versionFileUrl := ""
for _, asset := range latestRelease.Assets {
if *asset.Name == "version.txt" {
versionFileUrl = *asset.BrowserDownloadURL
break
}
}
if versionFileUrl == "" {
log.Error().Msg("Version file not found in the latest release.")
return
}
res, err := http.Get(versionFileUrl)
if err != nil {
log.Error().Err(err).Msg("Failed to get the version file.")
return
}
data, err := io.ReadAll(res.Body)
res.Body.Close()
if err != nil {
log.Error().Err(err).Msg("Failed to read the version file.")
return
}
gitHubVersion := string(data)
gitHubVersion = gitHubVersion[:len(gitHubVersion)-1]
greater, err := version.GreaterThen(gitHubVersion, kubeshark.Ver)
if err != nil {
log.Error().
Str("upstream-version", gitHubVersion).
Str("local-version", kubeshark.Ver).
Msg("Version is invalid!")
return
}
log.Debug().
Str("upstream-version", gitHubVersion).
Str("local-version", kubeshark.Ver).
Dur("elapsed-time", time.Since(start)).
Msg("Finished version validation.")
if greater {
var downloadCommand string
if runtime.GOOS == "windows" {
downloadCommand = fmt.Sprintf("curl -LO %v/kubeshark.exe", strings.Replace(*latestRelease.HTMLURL, "tag", "download", 1))
} else {
downloadCommand = "sh <(curl -Ls https://kubeshark.co/install)"
}
msg := fmt.Sprintf("Update available! %v -> %v run:", kubeshark.Ver, gitHubVersion)
log.Warn().Str("command", downloadCommand).Msg(fmt.Sprintf(utils.Yellow, msg))
}
}