Implement GetClients and bump SMEXT_CONF_VERSION to 1.0.4

Author: CallMeVerity

scripting/include/websocket/ws.inc

@@ -309,6 +309,15 @@ methodmap WebSocketServer < Handle
   */
   public native bool DisconnectClient(const char[] clientId);
 
+  /**
+  * Get handles for all connected clients
+  *
+  * @param buffer           Array to store client handles
+  * @param maxSize          Maximum size of the array
+  * @return                 Number of handles stored
+  */
+  public native int GetClients(WebSocket[] buffer, int maxSize);
+
   /**
   * Start the WebSocket server
   */

src/smsdk_config.h

@@ -3,7 +3,7 @@
 
 #define SMEXT_CONF_NAME			"SourceMod WebSocket Extension"
 #define SMEXT_CONF_DESCRIPTION	"Provide JSON and WebSocket Native"
-#define SMEXT_CONF_VERSION		"1.0.3"
+#define SMEXT_CONF_VERSION		"1.0.4"
 #define SMEXT_CONF_AUTHOR		"ProjectSky"
 #define SMEXT_CONF_URL			"https://github.com/ProjectSky/sm-ext-websocket"
 #define SMEXT_CONF_LOGTAG		"websocket"

src/ws_natives_server.cpp

@@ -265,8 +265,44 @@ static cell_t ws_IsDeflateEnabled(IPluginContext *pContext, const cell_t *params
 
 static cell_t ws_GetClients(IPluginContext *pContext, const cell_t *params)
 {
-	// TODO: Implement
-	return 1;
+	WebSocketServer *pWebsocketServer = GetWsServerPointer(pContext, params[1]);
+
+	if (!pWebsocketServer)
+	{
+		return 0;
+	}
+
+	cell_t *outputArray;
+	pContext->LocalToPhysAddr(params[2], &outputArray);
+	cell_t maxSize = params[3];
+
+	auto clients = pWebsocketServer->m_webSocketServer.getClients();
+	size_t count = 0;
+
+	HandleSecurity sec(nullptr, myself->GetIdentity());
+
+	for (const auto &client : clients)
+	{
+		if (count >= maxSize)
+			break;
+
+		WebSocketClient *pClient = new WebSocketClient(client.first.get());
+		HandleError err;
+		Handle_t handle = handlesys->CreateHandleEx(g_htWsClient, pClient, &sec, nullptr, &err);
+
+		if (handle != BAD_HANDLE)
+		{
+			pClient->m_websocket_handle = handle;
+			pClient->m_keepConnecting = true;
+			outputArray[count++] = handle;
+		}
+		else
+		{
+			delete pClient;
+		}
+	}
+
+	return count;
 }
 
 const sp_nativeinfo_t ws_natives_server[] =