Added condition to /eventsub/subscriptions addressing #227

Author: Xemdo

internal/events/websocket/mock_server/close_messages.go

@@ -6,6 +6,11 @@ type CloseMessage struct {
 }
 
 var (
+	closeClientDisconnected = &CloseMessage{
+		code:    1000,
+		message: "client disconnected",
+	}
+
 	closeInternalServerError = &CloseMessage{
 		code:    4000,
 		message: "internal server error",

internal/events/websocket/mock_server/manager.go

@@ -24,15 +24,15 @@ import (
 
 type ServerManager struct {
 	serverList       *util.List[WebSocketServer]
-	reconnectTesting bool
-	primaryServer    string
-	ip               string
-	port             int
-	debugEnabled     bool
-	strictMode       bool
-	sslEnabled       bool
-	protocolHttp     string
-	protocolWs       string
+	reconnectTesting bool   // Indicates if the server is in the process of running a simulation server reconnect/restart
+	primaryServer    string // The current primary server by its ID. This should be in serverList
+	ip               string // IP the server will bind to
+	port             int    // Port the server will bind to
+	debugEnabled     bool   // Indicates if the server was started with --debug
+	strictMode       bool   // Indicates if the server was started with --require-subscriptions
+	sslEnabled       bool   // Indicates if the server was started with --ssl
+	protocolHttp     string // String for the HTTP protocol URIs (http or https)
+	protocolWs       string // String for the WS protocol URIs (ws or wss)
 }
 
 var serverManager *ServerManager
@@ -269,7 +269,7 @@ func subscriptionPageHandlerGet(w http.ResponseWriter, r *http.Request) {
 					Status:    subscription.Status,
 					Type:      subscription.Type,
 					Version:   subscription.Version,
-					Condition: EmptyStruct{},
+					Condition: subscription.Conditions,
 					CreatedAt: subscription.CreatedAt,
 					Transport: SubscriptionTransport{
 						Method:    "websocket",
@@ -378,6 +378,7 @@ func subscriptionPageHandlerPost(w http.ResponseWriter, r *http.Request) {
 		CreatedAt:         time.Now().UTC().Format(time.RFC3339Nano),
 		Status:            STATUS_ENABLED, // https://dev.twitch.tv/docs/api/reference/#get-eventsub-subscriptions
 		SessionClientName: clientName,
+		Conditions:        body.Condition,
 	}
 
 	var subs []Subscription
@@ -403,7 +404,7 @@ func subscriptionPageHandlerPost(w http.ResponseWriter, r *http.Request) {
 				Status:    subscription.Status,
 				Type:      subscription.Type,
 				Version:   subscription.Version,
-				Condition: EmptyStruct{},
+				Condition: subscription.Conditions,
 				CreatedAt: subscription.CreatedAt,
 				Transport: SubscriptionTransport{
 					Method:      "websocket",

internal/events/websocket/mock_server/server.go

@@ -223,8 +223,8 @@ func (ws *WebSocketServer) WsPageHandler(w http.ResponseWriter, r *http.Request)
 			log.Printf("read err [%v]: %v", client.clientName, err)
 
 			ws.muClients.Lock()
-			client.CloseWithReason(closeNetworkError)
-			ws.handleClientConnectionClose(client, closeNetworkError)
+			client.CloseWithReason(closeClientDisconnected)
+			ws.handleClientConnectionClose(client, closeClientDisconnected)
 			ws.muClients.Unlock()
 			break
 		}
@@ -488,9 +488,9 @@ func (ws *WebSocketServer) handleClientConnectionClose(client *Client, closeReas
 	if ws.Status == 2 {
 		ws.muSubscriptions.Lock()
 		subscriptions := ws.Subscriptions[client.clientName]
-		for _, subscription := range subscriptions {
-			if subscription.Status == STATUS_ENABLED {
-				subscription.Status = getStatusFromCloseMessage(closeReason)
+		for i := range subscriptions {
+			if subscriptions[i].Status == STATUS_ENABLED {
+				subscriptions[i].Status = getStatusFromCloseMessage(closeReason)
 			}
 		}
 		ws.Subscriptions[client.clientName] = subscriptions

internal/events/websocket/mock_server/subscription.go

@@ -1,5 +1,7 @@
 package mock_server
 
+import "github.com/twitchdev/twitch-cli/internal/models"
+
 type Subscription struct {
 	SubscriptionID    string // Random GUID for the subscription
 	ClientID          string // Client ID included in headers
@@ -8,14 +10,16 @@ type Subscription struct {
 	CreatedAt         string // Timestamp of when the subscription was created
 	Status            string // Status of the subscription
 	SessionClientName string // Client name of the session this is associated with.
+
+	Conditions models.EventsubCondition // Values of the subscription's condition object
 }
 
 // Request - POST /eventsub/subscriptions
 type SubscriptionPostRequest struct {
-	Type      string      `json:"type"`
-	Version   string      `json:"version"`
-	Condition interface{} `json:"condition"`
+	Type    string `json:"type"`
+	Version string `json:"version"`
 
+	Condition models.EventsubCondition         `json:"condition"`
 	Transport SubscriptionPostRequestTransport `json:"transport"`
 }
 
@@ -44,8 +48,8 @@ type SubscriptionPostSuccessResponseBody struct {
 	CreatedAt string `json:"created_at"`
 	Cost      int    `json:"cost"`
 
-	Condition EmptyStruct           `json:"condition"`
-	Transport SubscriptionTransport `json:"transport"`
+	Condition models.EventsubCondition `json:"condition"`
+	Transport SubscriptionTransport    `json:"transport"`
 }
 
 // Response (Error) - POST /eventsub/subscriptions
@@ -69,7 +73,7 @@ type SubscriptionGetSuccessResponse struct {
 type SubscriptionTransport struct {
 	Method      string `json:"method"`
 	SessionID   string `json:"session_id"`
-	ConnectedAt string `json:"connected_at"`
+	ConnectedAt string `json:"connected_at,omitempty"`
 }
 
 // Cross-usage
@@ -112,6 +116,8 @@ func getStatusFromCloseMessage(reason *CloseMessage) string {
 	code := reason.code
 
 	switch code {
+	case 1000:
+		return STATUS_WEBSOCKET_DISCONNECTED
 	case 4000:
 		return STATUS_INTERNAL_ERROR
 	case 4001: