mirror of
https://github.com/kubeshark/kubeshark.git
synced 2025-07-18 02:11:57 +00:00
* Make `logger` a separate module such that don't depend on `shared` module as a whole for logging * Update `Dockerfile`
105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
package version
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/up9inc/mizu/cli/apiserver"
|
|
"github.com/up9inc/mizu/cli/mizu"
|
|
"github.com/up9inc/mizu/cli/pkg/version"
|
|
"github.com/up9inc/mizu/logger"
|
|
|
|
"github.com/google/go-github/v37/github"
|
|
"github.com/up9inc/mizu/cli/uiUtils"
|
|
)
|
|
|
|
func CheckVersionCompatibility(apiServerProvider *apiserver.Provider) (bool, error) {
|
|
apiVer, err := apiServerProvider.GetVersion()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if equals, err := version.AreEquals(apiVer, mizu.Ver); err != nil {
|
|
return false, fmt.Errorf("Failed to check version equality between mizuVer: %s and apiVer: %s, error: %w", mizu.Ver, apiVer, err)
|
|
} else if !equals {
|
|
logger.Log.Errorf(uiUtils.Red, fmt.Sprintf("cli version (%s) is not compatible with api version (%s)", mizu.Ver, apiVer))
|
|
return false, nil
|
|
}
|
|
|
|
logger.Log.Debug("cli version %s is compatible with api version %s", mizu.Ver, apiVer)
|
|
return true, nil
|
|
}
|
|
|
|
func CheckNewerVersion(versionChan chan string) {
|
|
if _, present := os.LookupEnv(mizu.DEVENVVAR); present {
|
|
versionChan <- ""
|
|
return
|
|
}
|
|
logger.Log.Debugf("Checking for newer version...")
|
|
start := time.Now()
|
|
client := github.NewClient(nil)
|
|
latestRelease, _, err := client.Repositories.GetLatestRelease(context.Background(), "up9inc", "mizu")
|
|
if err != nil {
|
|
logger.Log.Debugf("[ERROR] Failed to get latest release")
|
|
versionChan <- ""
|
|
return
|
|
}
|
|
|
|
versionFileUrl := ""
|
|
for _, asset := range latestRelease.Assets {
|
|
if *asset.Name == "version.txt" {
|
|
versionFileUrl = *asset.BrowserDownloadURL
|
|
break
|
|
}
|
|
}
|
|
if versionFileUrl == "" {
|
|
logger.Log.Debugf("[ERROR] Version file not found in the latest release")
|
|
versionChan <- ""
|
|
return
|
|
}
|
|
|
|
res, err := http.Get(versionFileUrl)
|
|
if err != nil {
|
|
logger.Log.Debugf("[ERROR] Failed to get the version file %v", err)
|
|
versionChan <- ""
|
|
return
|
|
}
|
|
|
|
data, err := ioutil.ReadAll(res.Body)
|
|
res.Body.Close()
|
|
if err != nil {
|
|
logger.Log.Debugf("[ERROR] Failed to read the version file -> %v", err)
|
|
versionChan <- ""
|
|
return
|
|
}
|
|
gitHubVersion := string(data)
|
|
gitHubVersion = gitHubVersion[:len(gitHubVersion)-1]
|
|
|
|
greater, err := version.GreaterThen(gitHubVersion, mizu.Ver)
|
|
if err != nil {
|
|
logger.Log.Debugf("[ERROR] Ver version is not valid, github version %v, current version %v", gitHubVersion, mizu.Ver)
|
|
versionChan <- ""
|
|
return
|
|
}
|
|
|
|
logger.Log.Debugf("Finished version validation, github version %v, current version %v, took %v", gitHubVersion, mizu.Ver, time.Since(start))
|
|
|
|
if greater {
|
|
var downloadMessage string
|
|
if runtime.GOOS == "windows" {
|
|
downloadMessage = fmt.Sprintf("curl -LO %v/mizu.exe", strings.Replace(*latestRelease.HTMLURL, "tag", "download", 1))
|
|
} else {
|
|
downloadMessage = fmt.Sprintf("curl -Lo mizu %v/mizu_%s_%s && chmod 755 mizu", strings.Replace(*latestRelease.HTMLURL, "tag", "download", 1), runtime.GOOS, runtime.GOARCH)
|
|
}
|
|
versionChan <- fmt.Sprintf("Update available! %v -> %v (%s)", mizu.Ver, gitHubVersion, downloadMessage)
|
|
} else {
|
|
versionChan <- ""
|
|
}
|
|
}
|