Initial commit

src/websocket/index.ts

@@ -0,0 +1,104 @@
+/*
+ * server component for the TimeLimit App
+ * Copyright (C) 2019 Jonas Lochmann
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, version 3 of the License.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+import * as io from 'socket.io'
+import { ConnectedDevicesManager, VisibleConnectedDevicesManager } from '../connected-devices'
+import { Database } from '../database'
+import { deviceByAuthTokenRoom } from './rooms'
+
+export const createWebsocketHandler = ({ connectedDevicesManager, database }: {
+  connectedDevicesManager: VisibleConnectedDevicesManager
+  database: Database
+}): {
+  websocketServer: io.Server
+  websocketApi: WebsocketApi
+} => {
+  const server = io()
+
+  server.on('connection', (socket) => {
+    socket.on('devicelogin', (deviceAuthToken: any, ack: any) => {
+      socket.leaveAll()
+
+      if (typeof deviceAuthToken !== 'string') {
+        return
+      }
+
+      socket.join(deviceByAuthTokenRoom(deviceAuthToken))
+
+      ;(async () => {
+        const deviceEntryUnsafe = await database.device.findOne({
+          where: {
+            deviceAuthToken
+          },
+          attributes: ['familyId', 'deviceId']
+        })
+
+        if (deviceEntryUnsafe && socket.connected) {
+          const deviceEntry = {
+            familyId: deviceEntryUnsafe.familyId,
+            deviceId: deviceEntryUnsafe.deviceId
+          }
+
+          const key = ConnectedDevicesManager.buildKey({
+            familyId: deviceEntry.familyId,
+            deviceId: deviceEntry.deviceId
+          })
+
+          connectedDevicesManager.connectedDevicesManager.reportDeviceConnected({ key })
+          const { shutdown } = connectedDevicesManager.observeConnectedDevicesOfFamily({
+            familyId: deviceEntry.familyId,
+            listener: (connectedDevices) => {
+              socket.emit('connected devices', connectedDevices)
+            }
+          })
+
+          socket.on('disconnect', () => {
+            connectedDevicesManager.connectedDevicesManager.reportDeviceDisconnected({ key })
+            shutdown()
+          })
+        }
+      })().catch((ex) => { /* ignore */ })
+
+      if (typeof ack === 'function') {
+        ack()
+      }
+    })
+  })
+
+  const websocketApi: WebsocketApi = {
+    triggerSyncByDeviceAuthToken: ({ deviceAuthToken, isImportant }: {deviceAuthToken: string, isImportant: boolean}) => {
+      server
+        .to(deviceByAuthTokenRoom(deviceAuthToken))
+        .emit('should sync', { isImportant })
+    },
+    triggerLogoutByDeviceAuthToken: ({ deviceAuthToken }: {deviceAuthToken: string}) => {
+      server
+        .to(deviceByAuthTokenRoom(deviceAuthToken))
+        .emit('sign out')
+    }
+  }
+
+  return {
+    websocketServer: server,
+    websocketApi
+  }
+}
+
+export interface WebsocketApi {
+  triggerSyncByDeviceAuthToken: (params: {deviceAuthToken: string, isImportant: boolean}) => void
+  triggerLogoutByDeviceAuthToken: (params: {deviceAuthToken: string}) => void
+}

src/websocket/rooms.ts

@@ -0,0 +1,18 @@
+/*
+ * server component for the TimeLimit App
+ * Copyright (C) 2019 Jonas Lochmann
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, version 3 of the License.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+export const deviceByAuthTokenRoom = (deviceAuthToken: string) => 'deviceauthtoken:' + deviceAuthToken