Fixing download of binaries and adding unit tests

Author: Slugsy

go.mod

@@ -4,4 +4,5 @@ go 1.14
 
 require (
 	github.com/gorilla/websocket v1.4.2
+	github.com/stretchr/testify v1.7.0
 )

go.sum

@@ -1,2 +1,12 @@
+github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
 github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

webcam.go

@@ -3,6 +3,8 @@ package camtron
 import (
 	"encoding/base64"
 	"encoding/json"
+	"errors"
+	"fmt"
 	"io"
 	"io/ioutil"
 	"log"
@@ -81,23 +83,39 @@ func Shellout(shell string, args ...string) error {
 	return err
 }
 
-func checkForElectronBinary() {
-	goos := runtime.GOOS
+func getLatestUIVersion() (string, error) {
+	url := "https://api.github.com/repos/vee2xx/camtron-ui/releases"
 
-	var filename string
-	switch goos {
-	case "windows":
-		filename = "camtron-win32-x64.zip"
-	case "darwin":
-		filename = "camtron-darwin-x64.zip"
-	case "linux":
-		filename = "camtron-linux-x64.zip"
-	default:
-		log.Fatal("Unsupported OS: %s.\n", goos)
+	res, err := http.Get(url)
+	if err != nil {
+		log.Fatal(err.Error())
 	}
+	defer res.Body.Close()
+
+	body, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		panic(err.Error())
+	}
+
+	var releases []map[string]interface{}
+	if err := json.Unmarshal(body, &releases); err != nil {
+		log.Fatal(err)
+	}
+
+	if len(releases) == 0 {
+		return "", errors.New("unable to find any versions")
+	}
+	return fmt.Sprintf("%v", releases[0]["tag_name"]), nil
+}
 
+func downloadBinary(electronBinary string) {
+	latest, err := getLatestUIVersion()
+	if err != nil {
+		log.Fatal(err.Error())
+	}
+	filename := electronBinary + ".zip"
 	if _, err := os.Stat(filename); os.IsNotExist(err) {
-		url := "https://github.com/vee2xx/camtron-ui/releases/download/v1.0.0/" + filename
+		url := fmt.Sprintf("https://github.com/vee2xx/camtron-ui/releases/download/%s/%s", latest, filename)
 
 		file, err := http.Get(url)
 		if err != nil {
@@ -116,45 +134,48 @@ func checkForElectronBinary() {
 		if err != nil {
 			log.Fatal(err)
 		}
+	}
+}
 
-		cmd := exec.Command("bash", "-c", "unzip camtron-darwin-x64.zip")
-		err = cmd.Run()
-		if err != nil {
-			log.Fatal(err)
-		}
+func UnzipBinary(electronBinary string) {
+	cmd := exec.Command("bash", "-c", "unzip "+electronBinary+".zip")
+	err := cmd.Run()
+	if err != nil {
+		log.Fatal(err)
 	}
 }
+
 func StartElectron() {
 	log.Println("INFO: starting electron")
 
-	checkForElectronBinary()
 	goos := runtime.GOOS
 
-	// _, filename, _, ok := runtime.Caller(0)
-
-	// if !ok {
-	// 	panic("No caller information")
-	// }
-
-	// goPath := path.Dir(filename)
-
 	var shell string
 	var args []string
+	var electronBinary string
 	switch goos {
 	case "windows":
 		shell = "cmd"
 		args = append(args, "/C")
 		args = append(args, "cd camtron-win32-x64 && camtron.exe")
+		electronBinary = "camtron-win32-x64"
 	case "darwin":
 		shell = "bash"
 		args = append(args, "-c")
 		args = append(args, "cd camtron-darwin-x64 && open camtron.app")
+		electronBinary = "camtron-darwin-x64"
 	case "linux":
 		shell = "bash"
 		args = append(args, "-c")
 		args = append(args, "cd camtron-linux-x64 && ./camtron")
+		electronBinary = "camtron-linux-x64"
 	default:
-		log.Println("Unsupported OS: %s.\n", goos)
+		log.Fatalf("Unsupported OS: %s.\n", goos)
+	}
+
+	if _, err := os.Stat(electronBinary); os.IsNotExist(err) {
+		downloadBinary(electronBinary)
+		UnzipBinary(electronBinary)
 	}
 
 	log.Print("Starting Electron")
@@ -170,14 +191,14 @@ func StartElectron() {
 		select {
 		case sig := <-c:
 			if goos == "windows" {
-				log.Println("Got %s signal. Its windows so gotta kill Electron\n", sig)
+				log.Printf("Got %s signal. Its windows so gotta kill Electron\n", sig)
 				cmd := exec.Command("cmd", "/C", " taskkill /f /im camtron.exe")
 				err := cmd.Run()
 				if err != nil {
 					log.Println("Couldn't kill camtron")
 				}
 			} else if goos == "darwin" {
-				log.Println("Got %s signal. Its MacOs so gotta kill Electron\n", sig)
+				log.Printf("Got %s signal. Its MacOs so gotta kill Electron\n", sig)
 				cmd := exec.Command("bash", "-c", " killall camtron")
 				err := cmd.Run()
 				if err != nil {

webcam_test.go

@@ -0,0 +1,23 @@
+package camtron
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBinaryDownload(t *testing.T) {
+	downloadBinary("camtron-linux-x64")
+	assert.FileExists(t, "camtron-linux-x64.zip")
+}
+
+func TestUnzip(t *testing.T) {
+	UnzipBinary("camtron-linux-x64")
+	assert.DirExists(t, "camtron-linux-x64")
+}
+
+func TestCleanUp(t *testing.T) {
+	_ = os.Remove("camtron-linux-x64.zip")
+	_ = os.RemoveAll("camtron-linux-x64")
+}