From 79d3bc47b94f6b3fb590c7ddc93507c1ccf01c7a Mon Sep 17 00:00:00 2001
From: Douglas Barone <douglasvbarone@gmail.com>
Date: Thu, 17 Dec 2020 13:15:22 -0400
Subject: [PATCH] Extract method

---
 server/src/lib/wifiDevices.js | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/server/src/lib/wifiDevices.js b/server/src/lib/wifiDevices.js
index 10e6a00..28e0dfc 100644
--- a/server/src/lib/wifiDevices.js
+++ b/server/src/lib/wifiDevices.js
@@ -23,6 +23,7 @@ async function getOnlineDevices() {
   return onlineDevices
 }
 
+// TODO: Add time threshold
 async function setAllDevicesAsOffline() {
   await prisma.wifiDevice.updateMany({
     data: {
@@ -31,13 +32,16 @@ async function setAllDevicesAsOffline() {
   })
 }
 
+async function forceUserDisconnect(mac) {
+  await prisma.$executeRaw(
+    'UPDATE "WifiDevice" SET "userId" = null WHERE mac = $1',
+    mac
+  )
+}
+
 async function updateDB(onlineDevices) {
   for (const device of onlineDevices) {
-    if (!device.user)
-      await prisma.$executeRaw(
-        'UPDATE "WifiDevice" SET "userId" = null WHERE mac = $1',
-        device.mac
-      )
+    if (!device.user) await forceUserDisconnect(device.mac)
 
     const user = device.user
       ? { connect: { sAMAccountName: device.user } }