webp_server_go/update.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"path"
	"runtime"

	log "github.com/sirupsen/logrus"
)

func autoUpdate() {
	defer func() {
		if err := recover(); err != nil {
			log.Error("Download error.", err)
		}
	}()

	var api = "https://api.github.com/repos/webp-sh/webp_server_go/releases/latest"
	type Result struct {
		TagName string `json:"tag_name"`
	}
	var res Result
	log.Debugf("Requesting to %s", api)
	resp1, _ := http.Get(api)
	data1, _ := ioutil.ReadAll(resp1.Body)
	_ = json.Unmarshal(data1, &res)
	var gitVersion = res.TagName

	if gitVersion > version {
		log.Infof("Time to update! New version %s found", gitVersion)
	} else {
		log.Debug("No new version found.")
		return
	}

	var filename = fmt.Sprintf("webp-server-%s-%s", runtime.GOOS, runtime.GOARCH)
	if runtime.GOARCH == "windows" {
		filename += ".exe"
	}
	var releaseUrl = "https://github.com/webp-sh/webp_server_go/releases/latest/download/" + filename
	log.Info("Downloading binary to update...")
	resp, _ := http.Get(releaseUrl)
	if resp.StatusCode != 200 {
		log.Debug("%s-%s not found on release.", runtime.GOOS, runtime.GOARCH)
		return
	}
	data, _ := ioutil.ReadAll(resp.Body)
	_ = os.Mkdir("update", 0755)
	// TODO: checksum
	err := ioutil.WriteFile(path.Join("update", filename), data, 0755)

	if err == nil {
		log.Info("Update complete. Please find your binary from update directory.")
	}
	_ = resp.Body.Close()
}