Semaphore/util/upgrade.go
julian 245a033456
fix api breaking of google/go-github
google/go-github broke the api of their functions with 23d6cb9cac
This commit adds the context `context.TODO()` to the call of `gh.Repositories.ListReleases` which is meant to be used "when it's unclear which Context to use or it is not yet available (because the surrounding function has not yet been extended to accept a Context parameter)" (citing https://godoc.org/context)

Signed-off-by: julian <dev@jneureuther.de>
2017-02-21 20:44:35 +01:00

112 lines
2.1 KiB
Go

package util
import (
"errors"
"fmt"
"io"
"net/http"
"os"
"os/exec"
"path/filepath"
"runtime"
"time"
"context"
"github.com/google/go-github/github"
)
// Adapted from github.com/apex/apex
var UpdateAvailable *github.RepositoryRelease
func DoUpgrade(version string) error {
fmt.Printf("current release is v%s\n", version)
if err := CheckUpdate(version); err != nil || UpdateAvailable == nil {
return err
}
asset := findAsset(UpdateAvailable)
if asset == nil {
return errors.New("cannot find binary for your system")
}
// create tmp file
tmpPath := filepath.Join(os.TempDir(), "semaphore-upgrade")
f, err := os.OpenFile(tmpPath, os.O_CREATE|os.O_TRUNC|os.O_RDWR, 0755)
if err != nil {
return err
}
// download binary
fmt.Printf("downloading %s\n", *asset.BrowserDownloadURL)
res, err := http.Get(*asset.BrowserDownloadURL)
if err != nil {
return err
}
defer res.Body.Close()
// copy it down
_, err = io.Copy(f, res.Body)
if err != nil {
return err
}
// replace it
cmdPath := FindSemaphore()
if len(cmdPath) == 0 {
return errors.New("Cannot find semaphore binary")
}
fmt.Printf("replacing %s\n", cmdPath)
err = os.Rename(tmpPath, cmdPath)
if err != nil {
return err
}
fmt.Println("visit https://github.com/ansible-semaphore/semaphore/releases for the changelog")
go func() {
time.Sleep(time.Second * 3)
os.Exit(0)
}()
return nil
}
func FindSemaphore() string {
cmdPath, _ := exec.LookPath("semaphore")
if len(cmdPath) == 0 {
cmdPath, _ = filepath.Abs(os.Args[0])
}
return cmdPath
}
// findAsset returns the binary for this platform.
func findAsset(release *github.RepositoryRelease) *github.ReleaseAsset {
for _, asset := range release.Assets {
if *asset.Name == fmt.Sprintf("semaphore_%s_%s", runtime.GOOS, runtime.GOARCH) {
return &asset
}
}
return nil
}
func CheckUpdate(version string) error {
// fetch releases
gh := github.NewClient(nil)
releases, _, err := gh.Repositories.ListReleases(context.TODO(), "ansible-semaphore", "semaphore", nil)
if err != nil {
return err
}
UpdateAvailable = nil
if (*releases[0].TagName)[1:] != version {
UpdateAvailable = releases[0]
}
return nil
}