Move socket operations to device manager

Sources/App/DeviceManager.swift

@@ -15,7 +15,7 @@ final class DeviceManager {
     private let remoteKey: Data
 
     /// Indicate that the socket is fully initialized with an authorized device
-    var deviceIsAuthenticated = false
+    private var deviceIsAuthenticated = false
 
     private var isOpeningNewConnection = false
 
@@ -62,7 +62,7 @@ final class DeviceManager {
     
     // MARK: API
     
-    var deviceStatus: String {
+    private var deviceStatus: String {
         deviceIsConnected ? "1" : "0"
     }
     
@@ -148,6 +148,18 @@ final class DeviceManager {
     
     func createNewDeviceConnection(_ socket: WebSocket) {
         defer { updateDeviceConnectionMetric() }
+
+        socket.onBinary { _, data in
+            self.processDeviceResponse(data)
+        }
+        socket.onText { _, text in
+            self.authenticateDevice(hash: text)
+        }
+
+        _ = socket.onClose.always { _ in
+            self.didCloseDeviceSocket()
+        }
+
         isOpeningNewConnection = true
         removeDeviceConnection()
         connection = socket

Sources/App/routes.swift

@@ -76,16 +76,6 @@ func routes(_ app: Application) throws {
      - Note: The first message from the device over the connection must be a valid auth token.
      */
     app.webSocket(RouteAPI.socket.path) { req, socket in
-        socket.onBinary { _, data in
-            deviceManager.processDeviceResponse(data)
-        }
-        socket.onText { _, text in
-            deviceManager.authenticateDevice(hash: text)
-        }
-        
-        _ = socket.onClose.always { _ in
-            deviceManager.didCloseDeviceSocket()
-        }
         deviceManager.createNewDeviceConnection(socket)
     }
 }