implement crude tool install

Author: matteosuppo

Gopkg.lock

@@ -227,7 +227,7 @@ func (t *Tools) Download(pack, name, version, behaviour string) error {
 	// Decompress
 	t.Logger("Unpacking tool " + name)
 
-	location := path.Join( dir(), pack, correctTool.Name, correctTool.Version)
+	location := path.Join(dir(), pack, correctTool.Name, correctTool.Version)
 	err = os.RemoveAll(location)
 
 	if err != nil {
@@ -393,7 +393,7 @@ func findBaseDir(dirList []string) string {
 	return commonBaseDir
 }
 
-func  extractZip(log func(msg string) , body []byte, location string) (string, error) {
+func extractZip(log func(msg string), body []byte, location string) (string, error) {
 	path, err := utilities.SaveFileonTempDir("tooldownloaded.zip", bytes.NewReader(body))
 	r, err := zip.OpenReader(path)
 	if err != nil {
@@ -442,7 +442,7 @@ func  extractZip(log func(msg string) , body []byte, location string) (string, e
 	return location, nil
 }
 
-func extractTarGz(log func(msg string),body []byte, location string) (string, error) {
+func extractTarGz(log func(msg string), body []byte, location string) (string, error) {
 	bodyCopy := make([]byte, len(body))
 	copy(bodyCopy, body)
 	tarFile, _ := gzip.NewReader(bytes.NewReader(body))
@@ -506,8 +506,7 @@ func extractTarGz(log func(msg string),body []byte, location string) (string, er
 	return location, nil
 }
 
-
-func  extractBz2(log func(msg string),body []byte, location string) (string, error) {
+func extractBz2(log func(msg string), body []byte, location string) (string, error) {
 	bodyCopy := make([]byte, len(body))
 	copy(bodyCopy, body)
 	tarFile := bzip2.NewReader(bytes.NewReader(body))
@@ -573,7 +572,6 @@ func  extractBz2(log func(msg string),body []byte, location string) (string, err
 	return location, nil
 }
 
-
 func (t *Tools) installDrivers(location string) error {
 	OK_PRESSED := 6
 	extension := ".bat"

tools/download.go

@@ -85,26 +85,3 @@ func (c *Indexes) Remove(ctx context.Context, payload *indexes.IndexPayload) err
 	filename := url.PathEscape(payload.URL)
 	return os.RemoveAll(filepath.Join(c.Folder, filename))
 }
-
-type Index struct {
-	Packages []struct {
-		Name       string `json:"name"`
-		Maintainer string `json:"maintainer"`
-		WebsiteURL string `json:"websiteURL"`
-		Email      string `json:"email,omitempty"`
-		Help       struct {
-			Online string `json:"online"`
-		} `json:"help,omitempty"`
-		Tools []struct {
-			Name    string `json:"name"`
-			Version string `json:"version"`
-			Systems []struct {
-				Host            string `json:"host"`
-				URL             string `json:"url"`
-				ArchiveFileName string `json:"archiveFileName"`
-				Checksum        string `json:"checksum"`
-				Size            string `json:"size"`
-			} `json:"systems"`
-		} `json:"tools"`
-	} `json:"packages"`
-}

v2/pkgs/indexes.go

@@ -0,0 +1,26 @@
+package pkgs
+
+type Index struct {
+	Packages []struct {
+		Name       string `json:"name"`
+		Maintainer string `json:"maintainer"`
+		WebsiteURL string `json:"websiteURL"`
+		Email      string `json:"email,omitempty"`
+		Help       struct {
+			Online string `json:"online"`
+		} `json:"help,omitempty"`
+		Tools []Tool `json:"tools"`
+	} `json:"packages"`
+}
+
+type Tool struct {
+	Name    string `json:"name"`
+	Version string `json:"version"`
+	Systems []struct {
+		Host            string `json:"host"`
+		URL             string `json:"url"`
+		ArchiveFileName string `json:"archiveFileName"`
+		Checksum        string `json:"checksum"`
+		Size            string `json:"size"`
+	} `json:"systems"`
+}

v2/pkgs/pkgs.go

@@ -2,12 +2,17 @@ package pkgs
 
 import (
 	"context"
+	"fmt"
 	"io/ioutil"
+	"net/http"
 	"os/user"
 	"path/filepath"
+	"runtime"
 
 	"github.com/arduino/arduino-create-agent/gen/tools"
+	"github.com/codeclysm/extract"
 	"github.com/sirupsen/logrus"
+	"github.com/xrash/smetrics"
 )
 
 type Tools struct {
@@ -86,9 +91,77 @@ func (c *Tools) Installed(ctx context.Context) (tools.ToolCollection, error) {
 }
 
 func (c *Tools) Install(ctx context.Context, payload *tools.ToolPayload) error {
+	list, err := c.Indexes.List(ctx)
+	if err != nil {
+		return err
+	}
+
+	for _, url := range list {
+		index, err := c.Indexes.Get(ctx, url)
+		if err != nil {
+			return err
+		}
+
+		for _, packager := range index.Packages {
+			if packager.Name != payload.Packager {
+				continue
+			}
+
+			for _, tool := range packager.Tools {
+				if tool.Name == payload.Name &&
+					tool.Version == payload.Version {
+					return c.install(ctx, tool)
+				}
+			}
+		}
+	}
+
+	return nil
+}
+
+func (c *Tools) install(ctx context.Context, tool Tool) error {
+	i := findSystem(tool)
+
+	// Download
+	fmt.Println(tool.Systems[i].URL)
+	res, err := http.Get(tool.Systems[i].URL)
+	if err != nil {
+		return err
+	}
+	defer res.Body.Close()
+
+	err = extract.Archive(ctx, res.Body, c.Folder, nil)
+	if err != nil {
+		return err
+	}
+
 	return nil
 }
 
 func (c *Tools) Remove(ctx context.Context, payload *tools.ToolPayload) error {
 	return nil
 }
+
+func findSystem(tool Tool) int {
+	var systems = map[string]string{
+		"linuxamd64":   "x86_64-linux-gnu",
+		"linux386":     "i686-linux-gnu",
+		"darwinamd64":  "apple-darwin",
+		"windows386":   "i686-mingw32",
+		"windowsamd64": "i686-mingw32",
+		"linuxarm":     "arm-linux-gnueabihf",
+	}
+
+	var correctSystem int
+	maxSimilarity := 0.7
+
+	for i, system := range tool.Systems {
+		similarity := smetrics.Jaro(system.Host, systems[runtime.GOOS+runtime.GOARCH])
+		if similarity > maxSimilarity {
+			correctSystem = i
+			maxSimilarity = similarity
+		}
+	}
+
+	return correctSystem
+}