From 9961535d92d4554ceafc09227a9d095c70adf78d Mon Sep 17 00:00:00 2001
From: Parth Gandhi <parthgandhi22@u.nus.edu>
Date: Fri, 19 Sep 2025 23:32:00 +0800
Subject: [PATCH] Add Pong Response for Unity

---
 comms/middleware/main.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/comms/middleware/main.go b/comms/middleware/main.go
index 7cfb466..6d7cda5 100644
--- a/comms/middleware/main.go
+++ b/comms/middleware/main.go
@@ -170,6 +170,26 @@ func (c *WSClient) readPump() {
 			Sender:    c,
 			EventType: arEvent.EventType,
 		}
+
+		// Send acknowledgment for debug.ping messages
+		if arEvent.EventType == "debug.ping" {
+			ackEvent := AREvent{
+				EventType: "debug.pong",
+				UserID:    c.UserID,
+				SessionID: c.SessionID,
+				Timestamp: time.Now().UnixMilli(),
+				Data: map[string]interface{}{
+					"originalTimestamp": arEvent.Timestamp,
+					"message":           "ping received",
+				},
+			}
+			ackMessage, _ := json.Marshal(ackEvent)
+			select {
+			case c.Send <- ackMessage:
+			default:
+				// If the send channel is full, we skip the acknowledgment
+			}
+		}
 	}
 }