Add socket connection status event listeners

Introduced onConnect, onDisconnect, and onConnectError functions to allow subscribing to socket connection status events. This enhances the ability to handle connection lifecycle events in the chat feature.

Author: DongDuong2001

lib/features/chat/socket.ts

@@ -21,7 +21,7 @@ export const initSocket = () => {
       console.error('Socket.io connection error:', error)
     })
   }
-  
+
   return socket
 }
 
@@ -167,6 +167,28 @@ export const onUserLeftRoom = (callback: (data: { userId: string; email: string;
   }
 }
 
+// Connection status listeners
+export const onConnect = (callback: () => void) => {
+  const sock = getSocket()
+  if (sock) {
+    sock.on('connect', callback)
+  }
+}
+
+export const onDisconnect = (callback: () => void) => {
+  const sock = getSocket()
+  if (sock) {
+    sock.on('disconnect', callback)
+  }
+}
+
+export const onConnectError = (callback: (error: Error) => void) => {
+  const sock = getSocket()
+  if (sock) {
+    sock.on('connect_error', callback)
+  }
+}
+
 // Cleanup function to remove all listeners
 export const removeAllListeners = () => {
   const sock = getSocket()