Add a visual timer for children who have ADHD

Author: Alan Tai

backend/src/main.py

@@ -7,7 +7,7 @@
 from fastapi.staticfiles import StaticFiles
 from sentry_sdk import init
 
-from .routers import system
+from .routers import system, websocket
 
 load_dotenv()
 
@@ -25,6 +25,7 @@ def lookup_path(self, path: str):
 
 app = FastAPI(title='BentoPi API', version='v1')
 app.include_router(system.router)
+app.include_router(websocket.router)
 app.add_middleware(CORSMiddleware, allow_headers=['*'], allow_methods=['*'], allow_origins=['*'])
 
 if argv[1] != 'dev':

backend/src/routers/websocket.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter, WebSocket, WebSocketDisconnect
+
+from ..services import DeviceManager
+
+router  = APIRouter(prefix='/ws', tags=['websocket'])
+manager = DeviceManager()
+
+
+@router.websocket('/{device_id}')
+async def websocket_endpoint(websocket: WebSocket, device_id: str) -> None:
+    await manager.connect(device_id, websocket)
+
+    try:
+        while True:
+            message   = await websocket.receive_json()
+            target_id = message.get('target_id')
+
+            if target_id:
+                await manager.send(target_id, message.get('payload'))
+    except WebSocketDisconnect:
+        manager.disconnect(device_id)

backend/src/services/__init__.py

@@ -0,0 +1 @@
+from .device_manager import DeviceManager

backend/src/services/device_manager.py

@@ -0,0 +1,27 @@
+from fastapi import WebSocket, WebSocketDisconnect
+
+
+class DeviceManager:
+    def __init__(self):
+        self.devices: dict[str, WebSocket] = {}
+
+    async def connect(self, device_id: str, websocket: WebSocket) -> None:
+        await websocket.accept()
+
+        self.devices[device_id] = websocket
+
+    def disconnect(self, device_id: str) -> None:
+        if device_id in self.devices:
+            del self.devices[device_id]
+
+    async def send(self, device_id: str, payload: dict[str, ...]) -> bool:
+        websocket = self.devices.get(device_id)
+        if websocket:
+            try:
+                await websocket.send_json(payload)
+
+                return True
+            except WebSocketDisconnect:
+                self.disconnect(device_id)
+
+        return False

frontend/package.json

@@ -38,33 +38,34 @@
         "@fortawesome/free-solid-svg-icons" : "^7.1.0",
         "@fortawesome/react-fontawesome"    : "^3.1.1",
         "@reduxjs/toolkit"                  : "^2.11.2",
-        "@sentry/react"                     : "^10.32.1",
-        "antd"                              : "^6.1.4",
+        "@sentry/react"                     : "^10.34.0",
+        "antd"                              : "^6.2.0",
         "date-fns"                          : "^4.1.0",
         "i18next"                           : "^25.7.4",
         "react"                             : "^19.2.3",
         "react-dom"                         : "^19.2.3",
         "react-gauge-component"             : "^1.2.64",
-        "react-i18next"                     : "^16.5.1",
+        "react-i18next"                     : "^16.5.3",
         "react-redux"                       : "^9.2.0",
         "react-router"                      : "^7.12.0",
-        "redux-persist"                     : "^6.0.0"
+        "redux-persist"                     : "^6.0.0",
+        "uuid"                              : "^13.0.0"
     },
     "devDependencies" : {
         "@eslint/eslintrc"           : "^3.3.3",
         "@eslint/js"                 : "^9.39.2",
-        "@stylistic/eslint-plugin"   : "^5.6.1",
+        "@stylistic/eslint-plugin"   : "^5.7.0",
         "@testing-library/jest-dom"  : "^6.9.1",
         "@testing-library/react"     : "^16.3.1",
-        "@types/node"                : "^25.0.3",
-        "@types/react"               : "^19.2.7",
+        "@types/node"                : "^25.0.9",
+        "@types/react"               : "^19.2.8",
         "@types/react-dom"           : "^19.2.3",
         "@types/react-test-renderer" : "^19.1.0",
-        "@typescript-eslint/parser"  : "^8.52.0",
+        "@typescript-eslint/parser"  : "^8.53.0",
         "@vitejs/plugin-react-swc"   : "^4.2.2",
-        "@vitest/coverage-v8"        : "^4.0.16",
-        "@vitest/ui"                 : "^4.0.16",
-        "baseline-browser-mapping"   : "^2.9.12",
+        "@vitest/coverage-v8"        : "^4.0.17",
+        "@vitest/ui"                 : "^4.0.17",
+        "baseline-browser-mapping"   : "^2.9.15",
         "eslint"                     : "^9.39.2",
         "eslint-plugin-import"       : "^2.32.0",
         "eslint-plugin-prefer-arrow" : "^1.2.3",
@@ -75,9 +76,9 @@
         "jsdom"                      : "^27.4.0",
         "prop-types"                 : "^15.8.1",
         "typescript"                 : "^5.9.3",
-        "typescript-eslint"          : "^8.52.0",
+        "typescript-eslint"          : "^8.53.0",
         "vite"                       : "^7.3.1",
-        "vitest"                     : "^4.0.16",
+        "vitest"                     : "^4.0.17",
         "vitest-fetch-mock"          : "^0.4.5"
     },
     "engines"         : {