all: implement WebSocket notification when attack finished

When the attack finished, the WebSocket server will broadcast the
result to all clients as message "/_trunks/api/attack/result".

1 files changed, 30 insertions, 3 deletions

diff --git a/websocket_server.go b/websocket_server.go
index 80d49d2..4079ed1 100644
--- a/websocket_server.go
+++ b/websocket_server.go
@@ -10,15 +10,42 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"log"
 	"net/http"
 
 	liberrors "github.com/shuLhan/share/lib/errors"
 	"github.com/shuLhan/share/lib/websocket"
 )
 
-const (
-	apiAttackHttp = "/_trunks/api/attack/http"
-)
+func (trunks *Trunks) wsBroadcastAttackFinish(result *AttackResult) {
+	if result == nil {
+		return
+	}
+
+	logp := "wsBroadcastAttackFinish"
+
+	jsonb, err := json.Marshal(result)
+	if err != nil {
+		log.Printf("%s: %s", logp, err)
+		return
+	}
+
+	packet, err := websocket.NewBroadcast(
+		apiAttackResult,
+		base64.StdEncoding.EncodeToString(jsonb),
+	)
+	if err != nil {
+		log.Printf("%s: %s", logp, err)
+		return
+	}
+
+	for _, conn := range trunks.Wsd.Clients.All() {
+		err = websocket.Send(conn, packet)
+		if err != nil {
+			log.Printf("%s: %s", logp, err)
+		}
+	}
+}
 
 func (trunks *Trunks) initWebSocketServer() (err error) {
 	opts := &websocket.ServerOptions{