From 939e43c4ad068245fddb3bc4508de02ac1a898a9 Mon Sep 17 00:00:00 2001
From: Robin Diddams <robindiddams@gmail.com>
Date: Wed, 30 Apr 2025 11:16:43 -0500
Subject: [PATCH] Dont call close handler if conn is nil

---
 internal/dev/websocket.go | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/internal/dev/websocket.go b/internal/dev/websocket.go
index cbd6ec31..5d739211 100644
--- a/internal/dev/websocket.go
+++ b/internal/dev/websocket.go
@@ -373,10 +373,12 @@ func (c *Websocket) SendMessage(logger logger.Logger, msg Message) error {
 
 func (c *Websocket) Close() error {
 	c.SendMessage(c.logger, NewCloseMessage(uuid.New().String(), c.Project.Project.ProjectId))
-	closeHandler := c.conn.CloseHandler()
-	if err := closeHandler(1000, "User requested shutdown"); err != nil {
-		c.logger.Error("failed to close connection: %s", err)
-		return err
+	if c.conn != nil {
+		closeHandler := c.conn.CloseHandler()
+		if err := closeHandler(1000, "User requested shutdown"); err != nil {
+			c.logger.Error("failed to close connection: %s", err)
+			return err
+		}
 	}
 	if c.cleanup != nil {
 		c.cleanup()