Add basic infrastructure for WLED support in FCS

Includes field states as a proof of concept but will clean this up later

back-end/realtime/package.json

@@ -20,6 +20,7 @@
     "@types/passport-jwt": "^3.0.9",
     "@types/passport-local": "^1.0.35",
     "@types/uuid": "^9.0.4",
+    "@types/ws": "^8.5.12",
     "@vercel/ncc": "^0.36.1",
     "typescript": "^5.1.6"
   },
@@ -34,7 +35,8 @@
     "passport-jwt": "^4.0.1",
     "passport-local": "^1.0.0",
     "socket.io": "^4.7.1",
-    "winston": "^3.10.0"
+    "winston": "^3.10.0",
+    "ws": "^8.18.0"
   },
   "optionalDependencies": {
     "bufferutil": "^4.0.7",

back-end/realtime/src/rooms/FCS.ts

@@ -9,14 +9,27 @@ import {
 } from "@toa-lib/models";
 import Room from "./Room.js";
 import Match from "./Match.js";
+import { WebSocket } from 'ws';
+import { buildWledInitializationPacket, buildWledSetColorPacket } from "../util/WLEDHelper.js";
 
 export default class FCS extends Room {
-  private readonly latestFcsStatus: FieldControlUpdatePacket = { hubs: {} };
+  private readonly latestFcsStatus: FieldControlUpdatePacket = { hubs: {}, wleds: {} };
   private fcsPackets: FcsPackets = getFcsPackets(defaultFieldOptions);
+  private wledSockets: Record<string, WebSocket> = {};
 
   public constructor(server: Server, matchRoom: Match) {
     super(server, "fcs");
 
+    // Connect to wled websocket servers if there are wleds
+    Object.entries(this.fcsPackets.init.wleds).forEach((wled) => {
+      this.wledSockets[wled[0]] = new WebSocket(wled[1].address);
+
+      // Send initialization packet
+      this.wledSockets[wled[0]].onopen = () => {
+        this.wledSockets[wled[0]].send(buildWledInitializationPacket(wled[1]));
+      }
+    });
+
     matchRoom.localEmitter.on(MatchSocketEvent.TELEOPERATED, () => {
       this.broadcastFcsUpdate(this.fcsPackets.matchStart);
     });
@@ -51,6 +64,11 @@ export default class FCS extends Room {
   private broadcastFcsUpdate(update: FieldControlUpdatePacket): void {
     this.broadcast().emit("fcs:update", update);
 
+    // Handle wleds
+    Object.entries(update.wleds).forEach((wled) => {
+      this.wledSockets[wled[0]].send(buildWledSetColorPacket(wled[1]));
+    });
+
     // Update this.latestFcsStatus AFTER sending out the new update
     for (const hubNumber in update.hubs) {
       const hubFromThisUpdate = update.hubs[hubNumber];

back-end/realtime/src/util/WLEDHelper.ts

@@ -0,0 +1,29 @@
+import { WledInitParameters, WledUpdateParameters } from "@toa-lib/models";
+
+export const buildWledInitializationPacket = (packet: WledInitParameters): string => {
+  const wledJson = {
+    transition: 0,
+    bri: 255,
+    frz: false,
+    fx: 0,
+    seg: packet.segments
+  }
+
+  return JSON.stringify(wledJson);
+}
+
+export const buildWledSetColorPacket = (packet: WledUpdateParameters): string => {
+  const wledJson = {
+    seg: packet.targetSegments?.map((segment) => ({
+      id: segment,
+      on: true,
+      frz: false,
+      fx: 0,
+      col: [
+        packet.color
+      ]
+    }))
+  }
+
+  return JSON.stringify(wledJson);
+}

front-end/src/seasons/fgc-2024/field-control.tsx

@@ -1,4 +1,4 @@
-import { FeedingTheFuture } from '@toa-lib/models';
+import { FeedingTheFuture, FieldControlUpdatePacket } from '@toa-lib/models';
 import { FieldControlCallbacks } from '..';
 import { useSocket } from 'src/api/use-socket';
 
@@ -16,18 +16,86 @@ export const useFieldControl =
     };
 
     const prepareField = () => {
+      socket?.emit('fcs:update', {
+        hubs: {},
+        wleds: {
+          center: {
+            color: 'ffff00',
+            targetSegments: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
+          },
+          red: {
+            color: 'ffff00',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          },
+          blue: {
+            color: 'ffff00',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          }
+        }
+      } satisfies FieldControlUpdatePacket);
       console.log('prepareField');
     };
 
     const startField = () => {
+      socket?.emit('fcs:update', {
+        hubs: {},
+        wleds: {
+          center: {
+            color: '000000',
+            targetSegments: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
+          },
+          red: {
+            color: '000000',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          },
+          blue: {
+            color: '000000',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          }
+        }
+      } satisfies FieldControlUpdatePacket);
       console.log('startField');
     };
 
     const abortField = () => {
+      socket?.emit('fcs:update', {
+        hubs: {},
+        wleds: {
+          center: {
+            color: 'ff0000',
+            targetSegments: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
+          },
+          red: {
+            color: 'ff0000',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          },
+          blue: {
+            color: 'ff0000',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          }
+        }
+      } satisfies FieldControlUpdatePacket);
       console.log('abortField');
     };
 
     const clearField = () => {
+      socket?.emit('fcs:update', {
+        hubs: {},
+        wleds: {
+          center: {
+            color: '00ff00',
+            targetSegments: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
+          },
+          red: {
+            color: '00ff00',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          },
+          blue: {
+            color: '00ff00',
+            targetSegments: [0, 1, 2, 3, 4, 5]
+          }
+        }
+      } satisfies FieldControlUpdatePacket);
       console.log('clearField');
     };