Merge pull request #63 from arduino/killbrowser

Kill and restart the browser

Author: mastrolinux

killbrowser.go

@@ -0,0 +1,48 @@
+package main
+
+import (
+	"errors"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+func killBrowserHandler(c *gin.Context) {
+
+	var data struct {
+		Action  string `json:"action"`
+		Process string `json:"process"`
+		URL     string `json:"url"`
+	}
+
+	c.BindJSON(&data)
+
+	if data.Process != "chrome" && data.Process != "chrom" {
+		c.JSON(http.StatusBadRequest, errors.New("You can't kill the process"+data.Process))
+		return
+	}
+
+	command, err := findBrowser(data.Process)
+
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	if data.Action == "kill" || data.Action == "restart" {
+		_, err := killBrowser(data.Process)
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, err.Error())
+			return
+		}
+	}
+
+	if data.Action == "restart" {
+		_, err := startBrowser(command, data.URL)
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, err.Error())
+			return
+		}
+	}
+
+}

killbrowser_darwin.go

@@ -0,0 +1,13 @@
+package main
+
+func findBrowser(process string) ([]byte, error) {
+	return nil, nil
+}
+
+func killBrowser(process string) ([]byte, error) {
+	return nil, nil
+}
+
+func startBrowser(command []byte, url string) ([]byte, error) {
+	return nil, nil
+}

killbrowser_linux.go

@@ -0,0 +1,25 @@
+package main
+
+import (
+	"os/exec"
+	"strings"
+)
+
+func findBrowser(process string) ([]byte, error) {
+	ps := exec.Command("ps", "-A", "-o", "command")
+	grep := exec.Command("grep", process)
+	head := exec.Command("head", "-n", "1")
+
+	return pipe_commands(ps, grep, head)
+}
+
+func killBrowser(process string) ([]byte, error) {
+	cmd := exec.Command("pkill", "-9", process)
+	return cmd.Output()
+}
+
+func startBrowser(command []byte, url string) ([]byte, error) {
+	parts := strings.Split(string(command), " ")
+	cmd := exec.Command(parts[0], url)
+	return cmd.Output()
+}

killbrowser_windows.go

@@ -0,0 +1,17 @@
+package main
+
+import "os/exec"
+
+func findBrowser(process string) ([]byte, error) {
+	return []byte(process), nil
+}
+
+func killBrowser(process string) ([]byte, error) {
+	cmd := exec.Command("Taskkill", "/F", "/IM", process+".exe")
+	return cmd.Output()
+}
+
+func startBrowser(command []byte, url string) ([]byte, error) {
+	cmd := exec.Command("cmd", "/C", "start", string(command), url)
+	return cmd.Output()
+}

main.go

@@ -241,6 +241,8 @@ func main() {
 			r.Handle("WS", "/socket.io/", socketHandler)
 			r.Handle("WSS", "/socket.io/", socketHandler)
 			r.GET("/info", infoHandler)
+			r.POST("/killbrowser", killBrowserHandler)
+
 			go func() {
 				// check if certificates exist; if not, use plain http
 				if _, err := os.Stat(filepath.Join(dest, "cert.pem")); os.IsNotExist(err) {