Add Webhook package

Add `CallWebhook` with tests. Not called by WHIP/WHEP handlers yet

Author: Neil Schark

internal/webhook/webhook.go

@@ -0,0 +1,77 @@
+package webhook
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"time"
+)
+
+type webhookPayload struct {
+	Action      string            `json:"action"`
+	IP          string            `json:"ip"`
+	BearerToken string            `json:"bearerToken"`
+	QueryParams map[string]string `json:"queryParams"`
+	UserAgent   string            `json:"userAgent"`
+}
+
+type webhookResponse struct {
+	StreamKey string `json:"streamKey"`
+}
+
+func CallWebhook(url, action, bearerToken string, timeout int, r *http.Request) (string, error) {
+	start := time.Now()
+
+	queryParams := make(map[string]string)
+	for k, v := range r.URL.Query() {
+		if len(v) > 0 {
+			queryParams[k] = v[0]
+		}
+	}
+
+	jsonPayload, err := json.Marshal(webhookPayload{
+		Action:      action,
+		IP:          getIPAddress(r),
+		BearerToken: bearerToken,
+		QueryParams: queryParams,
+		UserAgent:   r.UserAgent(),
+	})
+	if err != nil {
+		return "", fmt.Errorf("failed to marshal payload: %w", err)
+	}
+
+	client := &http.Client{
+		Timeout: time.Duration(timeout) * time.Millisecond,
+	}
+
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonPayload))
+	if err != nil {
+		return "", fmt.Errorf("failed to create request: %w", err)
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return "", fmt.Errorf("webhook request failed after %v: %w", time.Since(start), err)
+	}
+	defer resp.Body.Close() //nolint
+
+	if resp.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("webhook returned non-200 Status: %v", resp.StatusCode)
+	}
+
+	response := webhookResponse{}
+	if err = json.NewDecoder(resp.Body).Decode(&response); err != nil {
+		return "", fmt.Errorf("failed to decode response: %w", err)
+	}
+
+	return response.StreamKey, nil
+}
+
+func getIPAddress(r *http.Request) string {
+	if r.Header.Get("X-Forwarded-For") != "" {
+		return r.Header.Get("X-Forwarded-For")
+	}
+	return r.RemoteAddr
+}

internal/webhook/webhook_test.go

@@ -0,0 +1,68 @@
+package webhook
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+)
+
+func TestCallWebhook(t *testing.T) {
+	// Setup a Mock HTTP Server
+	mockServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		switch r.URL.Path {
+		case "/ok":
+			w.WriteHeader(http.StatusOK)
+			_ = json.NewEncoder(w).Encode(webhookResponse{StreamKey: "dummy_stream_key"})
+		case "/timeout":
+			time.Sleep(2 * time.Second)
+		case "/error":
+			w.WriteHeader(http.StatusInternalServerError)
+		case "/badjson":
+			w.WriteHeader(http.StatusOK)
+			_, _ = w.Write([]byte("not a json"))
+		default:
+			w.WriteHeader(http.StatusNotFound)
+		}
+	}))
+	defer mockServer.Close()
+
+	tests := []struct {
+		name        string
+		url         string
+		timeout     int
+		expectedErr bool
+		expectedKey string
+	}{
+		{"Success Case", "/ok", 1000, false, "dummy_stream_key"},
+		{"Server Timeout", "/timeout", 1000, true, ""},
+		{"Server Error", "/error", 1000, true, ""},
+		{"Malformed JSON", "/badjson", 1000, true, ""},
+		{"Not Found", "/notfound", 1000, true, ""},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			req, _ := http.NewRequest("GET", "/", nil)
+			req.RemoteAddr = "127.0.0.1"
+			req.Header.Set("User-Agent", "test-agent")
+
+			// call the function with test layers
+			result, err := CallWebhook(fmt.Sprintf("%s%s", mockServer.URL, tt.url), "action", "bearerToken", tt.timeout, req)
+
+			if tt.expectedErr && err == nil {
+				t.Fatalf("expected an error but got none")
+			}
+
+			if !tt.expectedErr && err != nil {
+				t.Fatalf("did not expect an error but got %v", err)
+			}
+
+			if result != tt.expectedKey {
+				t.Fatalf("expected stream key %s but got %s", tt.expectedKey, result)
+			}
+		})
+	}
+}