add checktool endpoint and force download

Author: facchinm

download.go

@@ -86,28 +86,38 @@ func downloadFromUrl(url string) (filename string, err error) {
 	return fileName, nil
 }
 
-func spDownloadTool(name string, url string) {
+func spCheckToolVersion(name string) {
+	var outlist []string
+	dirlist, err := ioutil.ReadDir(tempToolsPath + "/")
+	if err == nil {
+		for _, element := range dirlist {
+			if element.IsDir() && strings.Contains(element.Name(), name) {
+				outlist = append(outlist, element.Name())
+			}
+		}
+	}
+	mapD := map[string][]string{"ToolVersions": outlist}
+	mapB, _ := json.Marshal(mapD)
+	h.broadcastSys <- mapB
+}
 
-	if _, err := os.Stat(tempToolsPath + "/" + name); err != nil {
+func spDownloadTool(name string, url string) {
 
-		fileName, err := downloadFromUrl(url + "/" + name + "-" + runtime.GOOS + "-" + runtime.GOARCH + ".zip")
-		if err != nil {
-			log.Error("Could not download flashing tools!")
-			mapD := map[string]string{"DownloadStatus": "Error", "Msg": err.Error()}
-			mapB, _ := json.Marshal(mapD)
-			h.broadcastSys <- mapB
-			return
-		}
-		err = UnzipWrapper(fileName, tempToolsPath)
-		if err != nil {
-			log.Error("Could not unzip flashing tools!")
-			mapD := map[string]string{"DownloadStatus": "Error", "Msg": err.Error()}
-			mapB, _ := json.Marshal(mapD)
-			h.broadcastSys <- mapB
-			return
-		}
-	} else {
-		log.Info("Tool already present, skipping download")
+	fileName, err := downloadFromUrl(url + "/" + name + "-" + runtime.GOOS + "-" + runtime.GOARCH + ".zip")
+	if err != nil {
+		log.Error("Could not download flashing tools!")
+		mapD := map[string]string{"DownloadStatus": "Error", "Msg": err.Error()}
+		mapB, _ := json.Marshal(mapD)
+		h.broadcastSys <- mapB
+		return
+	}
+	err = UnzipWrapper(fileName, tempToolsPath)
+	if err != nil {
+		log.Error("Could not unzip flashing tools!")
+		mapD := map[string]string{"DownloadStatus": "Error", "Msg": err.Error()}
+		mapB, _ := json.Marshal(mapD)
+		h.broadcastSys <- mapB
+		return
 	}
 
 	folders, _ := ioutil.ReadDir(tempToolsPath)

hub.go

@@ -189,6 +189,11 @@ func checkCmd(m []byte) {
 		if len(args) > 2 {
 			go spDownloadTool(args[1], args[2])
 		}
+	} else if strings.HasPrefix(sl, "checktool") {
+		args := strings.Split(s, " ")
+		if len(args) > 1 {
+			go spCheckToolVersion(args[1])
+		}
 	} else if strings.HasPrefix(sl, "bufferalgorithm") {
 		go spBufferAlgorithms()
 	} else if strings.HasPrefix(sl, "log") {