mirror of
https://github.com/kubeshark/kubeshark.git
synced 2026-04-07 03:07:25 +00:00
93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
package mizu
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/google/go-github/v37/github"
|
|
"github.com/romana/rlog"
|
|
"github.com/up9inc/mizu/shared"
|
|
"github.com/up9inc/mizu/shared/semver"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
)
|
|
|
|
func getApiVersion(port uint16) (string, error) {
|
|
versionUrl, _ := url.Parse(fmt.Sprintf("http://localhost:%d/mizu/metadata/version", port))
|
|
req := &http.Request{
|
|
Method: http.MethodGet,
|
|
URL: versionUrl,
|
|
}
|
|
statusResp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer statusResp.Body.Close()
|
|
|
|
versionResponse := &shared.VersionResponse{}
|
|
if err := json.NewDecoder(statusResp.Body).Decode(&versionResponse); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return versionResponse.SemVer, nil
|
|
}
|
|
|
|
func CheckVersionCompatibility(port uint16) (bool, error) {
|
|
apiSemVer, err := getApiVersion(port)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if semver.SemVersion(apiSemVer).Major() == semver.SemVersion(SemVer).Major() &&
|
|
semver.SemVersion(apiSemVer).Minor() == semver.SemVersion(SemVer).Minor() {
|
|
return true, nil
|
|
}
|
|
|
|
fmt.Printf(Red, fmt.Sprintf("cli version (%s) is not compatible with api version (%s)\n", SemVer, apiSemVer))
|
|
return false, nil
|
|
}
|
|
|
|
func CheckNewerVersion() {
|
|
rlog.Debugf("Checking for newer version...")
|
|
start := time.Now()
|
|
client := github.NewClient(nil)
|
|
latestRelease, _, err := client.Repositories.GetLatestRelease(context.Background(), "up9inc", "mizu")
|
|
if err != nil {
|
|
rlog.Debugf("Failed to get latest release")
|
|
return
|
|
}
|
|
|
|
versionFileUrl := ""
|
|
for _, asset := range latestRelease.Assets {
|
|
if *asset.Name == "version.txt" {
|
|
versionFileUrl = *asset.BrowserDownloadURL
|
|
break
|
|
}
|
|
}
|
|
if versionFileUrl == "" {
|
|
rlog.Debugf("Version file not found in the latest release")
|
|
return
|
|
}
|
|
|
|
res, err := http.Get(versionFileUrl)
|
|
if err != nil {
|
|
rlog.Debugf("http.Get version asset -> %v", err)
|
|
return
|
|
}
|
|
|
|
data, err := ioutil.ReadAll(res.Body)
|
|
res.Body.Close()
|
|
if err != nil {
|
|
rlog.Debugf("ioutil.ReadAll -> %v", err)
|
|
return
|
|
}
|
|
gitHubVersion := string(data)
|
|
gitHubVersion = gitHubVersion[:len(gitHubVersion)-1]
|
|
rlog.Debugf("Finished version validation, took %v", time.Since(start))
|
|
if SemVer < gitHubVersion {
|
|
fmt.Printf(Yellow, fmt.Sprintf("Update available! %v -> %v (%v)\n", SemVer, gitHubVersion, *latestRelease.HTMLURL))
|
|
}
|
|
}
|