From bf93c9fc43a14712ea98ae98d5ad587d7fc634b6 Mon Sep 17 00:00:00 2001
From: Douglas Barone <douglasvbarone@gmail.com>
Date: Mon, 4 Apr 2022 15:27:56 -0400
Subject: [PATCH] Fix client count

---
 server/src/resolvers/AccessPoint.js        | 12 +++++++++++-
 server/src/resolvers/Query/accessPoint.js  |  4 +++-
 server/src/resolvers/Query/accessPoints.js |  6 +++++-
 3 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/server/src/resolvers/AccessPoint.js b/server/src/resolvers/AccessPoint.js
index 32cac70..92f85b2 100644
--- a/server/src/resolvers/AccessPoint.js
+++ b/server/src/resolvers/AccessPoint.js
@@ -1,3 +1,13 @@
+import prisma from "../prisma";
+
 export const AccessPoint = {
-  updatedAt: parent => parent.updatedAt?.toISOString()
+  updatedAt: parent => parent.updatedAt?.toISOString(),
+  clients: parent => prisma.wifiDevice.count({
+    where: {
+      status: 'ONLINE',
+      accessPoint: {
+        id: parent.id
+      }
+    }
+  })
 }
diff --git a/server/src/resolvers/Query/accessPoint.js b/server/src/resolvers/Query/accessPoint.js
index 8dd3883..c2c78d1 100644
--- a/server/src/resolvers/Query/accessPoint.js
+++ b/server/src/resolvers/Query/accessPoint.js
@@ -1,7 +1,7 @@
 import prisma from '../../prisma'
 
 export async function accessPoint(parent, { id }, context, info) {
-  return prisma.accessPoint.findUnique({
+  const ap = prisma.accessPoint.findUnique({
     where: { id: parseInt(id) },
     include: {
       wifiDevices: {
@@ -11,4 +11,6 @@ export async function accessPoint(parent, { id }, context, info) {
       }
     }
   })
+  console.log(ap);
+  return ap
 }
diff --git a/server/src/resolvers/Query/accessPoints.js b/server/src/resolvers/Query/accessPoints.js
index ad9bd67..d595067 100644
--- a/server/src/resolvers/Query/accessPoints.js
+++ b/server/src/resolvers/Query/accessPoints.js
@@ -6,7 +6,11 @@ export async function accessPoints() {
       hostname: 'asc'
     },
     include: {
-      wifiDevices: true
+      wifiDevices: {
+        where: {
+          status: 'ONLINE'
+        }
+      }
     }
   })
 }