fix missing connInfo and chanInfo in presence (#221)

.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
       - name: Install dependencies
         run: yarn install
       - name: Start Centrifugo
-        run: docker run -d -p 8000:8000 centrifugo/centrifugo:latest centrifugo --client_insecure --http_stream --sse
+        run: docker run -d -p 8000:8000 -e CENTRIFUGO_PRESENCE=true centrifugo/centrifugo:latest centrifugo --client_insecure --http_stream --sse
       - name: Lint
         run: yarn lint
       - name: Test

src/centrifuge.test.ts

@@ -248,3 +248,44 @@ test.each(transportCases.slice(0, 1))("%s: not connecting on online in disconnec
   networkEventTarget.dispatchEvent(onlineEvent);
   expect(c.state).toBe(Centrifuge.State.Disconnected);
 });
+
+test.each(transportCases)("%s: subscribe and presence", async (transport, endpoint) => {
+  const c = new Centrifuge([{
+    transport: transport as TransportName,
+    endpoint: endpoint,
+  }], {
+    websocket: WebSocket,
+    fetch: fetch,
+    eventsource: EventSource,
+    readableStream: ReadableStream,
+    emulationEndpoint: 'http://localhost:8000/emulation'
+  });
+
+  c.connect();
+  await c.ready(5000);
+  const sub = c.newSubscription('test');
+
+  sub.subscribe()
+  await sub.ready(5000);
+  expect(sub.state).toBe(Centrifuge.SubscriptionState.Subscribed);
+  expect(c.state).toBe(Centrifuge.State.Connected);
+
+  const presence = await sub.presence();
+  expect(Object.keys(presence.clients).length).toBeGreaterThan(0);
+
+  const presenceStats = await sub.presenceStats();
+  expect(presenceStats.numClients).toBeGreaterThan(0)
+  expect(presenceStats.numUsers).toBeGreaterThan(0);
+
+  let disconnectCalled: any;
+  const disconnectedPromise = new Promise<DisconnectedContext>((resolve, _) => {
+    disconnectCalled = resolve;
+  })
+  c.on('disconnected', (ctx) => {
+    disconnectCalled(ctx);
+  })
+
+  c.disconnect();
+  await disconnectedPromise;
+  expect(c.state).toBe(Centrifuge.State.Disconnected);
+});

src/centrifuge.ts

@@ -325,8 +325,21 @@ export class Centrifuge extends (EventEmitter as new () => TypedEventEmitter<Cli
 
     return this._methodCall().then(function () {
       return self._callPromise(cmd, function (reply: any) {
+        const clients = reply.presence.presence;
+        for (const clientId in clients) {
+          if (clients.hasOwnProperty(clientId)) {
+            const connInfo = clients[clientId]['conn_info'];
+            const chanInfo = clients[clientId]['chan_info'];
+            if (connInfo) {
+              clients[clientId].connInfo = connInfo;
+            }
+            if (chanInfo) {
+              clients[clientId].chanInfo = chanInfo;
+            }
+          }
+        }
         return {
-          'clients': reply.presence.presence
+          'clients': clients
         };
       });
     });

src/protobuf/index.test.ts

@@ -113,3 +113,42 @@ test.each(transportCases)("%s (Protobuf): publish and receive message", async (t
   c.disconnect();
   expect(ctx.data).toStrictEqual(binary);
 });
+
+test.each(transportCases)("%s (Protobuf): subscribe and presence", async (transport, endpoint) => {
+  const c = new Centrifuge([{
+    transport: transport as TransportName,
+    endpoint: endpoint,
+  }], {
+    protocol: 'protobuf',
+    websocket: WebSocket,
+    fetch: fetch,
+    readableStream: ReadableStream,
+    emulationEndpoint: 'http://localhost:8000/emulation'
+  });
+
+  c.connect();
+  await c.ready(5000);
+
+  const sub = c.newSubscription('test');
+  sub.subscribe()
+  await sub.ready(5000);
+
+  const presence = await sub.presence();
+  expect(Object.keys(presence.clients).length).toBeGreaterThan(0);
+
+  const presenceStats = await sub.presenceStats();
+  expect(presenceStats.numClients).toBeGreaterThan(0)
+  expect(presenceStats.numUsers).toBeGreaterThan(0);
+
+  let disconnectCalled: any;
+  const disconnectedPromise = new Promise<DisconnectedContext>((resolve, _) => {
+    disconnectCalled = resolve;
+  })
+  c.on('disconnected', (ctx) => {
+    disconnectCalled(ctx);
+  })
+
+  c.disconnect();
+  await disconnectedPromise;
+  expect(c.state).toBe(Centrifuge.State.Disconnected);
+});