From 342c44abfb37c18a5885bb9cf39cbfca61758aa2 Mon Sep 17 00:00:00 2001
From: Dane Everitt <dane@daneeveritt.com>
Date: Sat, 20 Apr 2019 11:59:28 -0700
Subject: [PATCH] Cleanup websocket implementation

---
 websocket.go | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/websocket.go b/websocket.go
index c41e227..d042c2e 100644
--- a/websocket.go
+++ b/websocket.go
@@ -2,24 +2,24 @@ package main
 
 import (
 	"fmt"
-	"github.com/gorilla/websocket"
 	"github.com/julienschmidt/httprouter"
 	"go.uber.org/zap"
 	"net/http"
+	"strings"
 )
 
 type WebsocketMessage struct {
-	// The action to perform. Should be one of the following that are supported:
+	// The event to perform. Should be one of the following that are supported:
 	//
 	// - status : Returns the server's power state.
 	// - logs : Returns the server log data at the time of the request.
 	// - power : Performs a power action aganist the server based the data.
 	// - command : Performs a command on a server using the data field.
-	Action string
+	Event string `json:"event"`
 
 	// The data to pass along, only used by power/command currently. Other requests
-	// should either omit the field or pass an empty string value as it is ignored.
-	Data   string
+	// should either omit the field or pass an empty value as it is ignored.
+	Args []string `json:"args,omitempty"`
 }
 
 func (rt *Router) routeWebsocket(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
@@ -30,6 +30,8 @@ func (rt *Router) routeWebsocket(w http.ResponseWriter, r *http.Request, ps http
 	}
 	defer c.Close()
 
+	s := rt.Servers.Get(ps.ByName("server"))
+
 	for {
 		j := WebsocketMessage{}
 
@@ -37,14 +39,14 @@ func (rt *Router) routeWebsocket(w http.ResponseWriter, r *http.Request, ps http
 		// specific socket request. If we did a break here the client would get disconnected
 		// from the socket, which is NOT what we want to do.
 		if err := c.ReadJSON(&j); err != nil {
-			continue
+			break
 		}
 
-		fmt.Printf("%s sent: %s = %s\n", c.RemoteAddr(), j.Action, j.Data)
+		fmt.Printf("%s sent: %s = %s\n", s.Uuid, j.Event, strings.Join(j.Args, " "))
 
-		if err := c.WriteMessage(websocket.TextMessage, []byte("")); err != nil {
+		if err := c.WriteJSON(WebsocketMessage{Event: j.Event, Args: []string{}}); err != nil {
 			zap.S().Warnw("error writing JSON to socket", zap.Error(err))
-			continue
+			break
 		}
 	}
 }