Send whep-connect events via WebHook

Was lost during refactor

Author: Sean-Der

internal/server/handlers/whep.go

@@ -6,9 +6,12 @@ import (
 	"log"
 	"mime"
 	"net/http"
+	"os"
 	"strings"
 
+	"github.com/glimesh/broadcast-box/internal/environment"
 	"github.com/glimesh/broadcast-box/internal/server/helpers"
+	"github.com/glimesh/broadcast-box/internal/server/webhook"
 	"github.com/glimesh/broadcast-box/internal/webrtc"
 	"github.com/glimesh/broadcast-box/internal/webrtc/utils"
 )
@@ -56,6 +59,14 @@ func WhepHandler(responseWriter http.ResponseWriter, request *http.Request) {
 		return
 	}
 
+	if webhookUrl := os.Getenv(environment.WEBHOOK_URL); webhookUrl != "" {
+		token, err = webhook.CallWebhook(webhookUrl, webhook.WhepConnect, token, request)
+		if err != nil {
+			responseWriter.WriteHeader(http.StatusUnauthorized)
+			return
+		}
+	}
+
 	whipAnswer, sessionId, err := webrtc.WHEP(string(offer), token)
 	if err != nil {
 		log.Println("API.WHEP: Setup Error", err.Error())

internal/server/handlers/whep_test.go

@@ -0,0 +1,64 @@
+package handlers
+
+import (
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/glimesh/broadcast-box/internal/environment"
+)
+
+type whepWebhookPayload struct {
+	Action      string            `json:"action"`
+	BearerToken string            `json:"bearerToken"`
+	QueryParams map[string]string `json:"queryParams"`
+}
+
+func TestWhepHandlerCallsWebhook(t *testing.T) {
+	payloads := make(chan whepWebhookPayload, 1)
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		defer r.Body.Close()
+
+		var payload whepWebhookPayload
+		if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
+			t.Fatalf("failed to decode webhook payload: %v", err)
+		}
+
+		payloads <- payload
+		w.WriteHeader(http.StatusUnauthorized)
+	}))
+	defer server.Close()
+
+	t.Setenv(environment.WEBHOOK_URL, server.URL)
+
+	req := httptest.NewRequest(http.MethodPost, "/api/whep?viewer=1", strings.NewReader("v=0"))
+	req.Header.Set("Authorization", "Bearer test_stream_key")
+	req.Header.Set("User-Agent", "whep-handler-test")
+	req.RemoteAddr = "203.0.113.10:1234"
+
+	resp := httptest.NewRecorder()
+	WhepHandler(resp, req)
+
+	if resp.Code != http.StatusUnauthorized {
+		t.Fatalf("expected status %d, got %d", http.StatusUnauthorized, resp.Code)
+	}
+
+	select {
+	case payload := <-payloads:
+		if payload.Action != "whep-connect" {
+			t.Fatalf("expected action %q, got %q", "whep-connect", payload.Action)
+		}
+
+		if payload.BearerToken != "test_stream_key" {
+			t.Fatalf("expected bearer token %q, got %q", "test_stream_key", payload.BearerToken)
+		}
+
+		if payload.QueryParams["viewer"] != "1" {
+			t.Fatalf("expected query param %q, got %q", "1", payload.QueryParams["viewer"])
+		}
+	default:
+		t.Fatal("expected webhook to be called")
+	}
+}