Go implementation of the original bash plugin, containing all features

download/download.go

@@ -0,0 +1,73 @@
+package download
+
+import (
+	"archive/tar"
+	"compress/gzip"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"os"
+	"strings"
+
+	"github.com/pkg/errors"
+	"log"
+	"net/http"
+)
+
+var (
+	_downloadURL = "https://github.com/gohugoio/hugo/releases/download/v%s/hugo_%s_Linux-%s.tar.gz"
+)
+
+func downloadURL(version string, archType string) string {
+	return fmt.Sprintf(_downloadURL, version, version, archType)
+}
+
+func getTempFile() (string, io.WriteCloser, error) {
+	d, err := ioutil.TempDir("", "")
+	if err != nil {
+		return "", nil, errors.Wrap(err, "")
+	}
+	f, err := ioutil.TempFile(d, "")
+	return f.Name(), f, err
+}
+
+// Get will download the specified hugo verion
+func Get(ver string, archType string) (string, error) {
+	resp, err := http.Get(downloadURL(ver, archType))
+	if err != nil {
+		return "", errors.Wrap(err, "")
+	}
+	defer resp.Body.Close()
+	gz, err := gzip.NewReader(resp.Body)
+	if err != nil {
+		return "", errors.Wrap(err, "")
+	}
+	defer gz.Close()
+	targz := tar.NewReader(gz)
+
+	hugoPath, hugoBin, err := getTempFile()
+	if err != nil {
+		log.Printf("ERROR: %s", err)
+		return "", errors.Wrap(err, "")
+	}
+	defer hugoBin.Close()
+
+	for {
+		h, err := targz.Next()
+		if err == io.EOF {
+			return "", errors.New("no hugo binary found")
+		}
+		if err != nil {
+			return "", errors.Wrap(err, "")
+		}
+		if strings.HasSuffix(h.Name, "hugo") {
+			io.Copy(hugoBin, targz)
+
+			if err := os.Chmod(hugoPath, 0755); err != nil {
+				log.Fatal(err)
+			}
+
+			return hugoPath, nil
+		}
+	}
+}