modularize code, add dummy version of roverd

This commit is contained in:
legop3
2025-11-12 01:13:29 -05:00
parent b5ce6706d6
commit 73b1b44c35
41 changed files with 1018 additions and 401 deletions
+15
View File
@@ -0,0 +1,15 @@
require('./src/globals/logger');
require('./src/globals/config');
require('./src/globals/http');
require('./src/globals/io');
require('./src/globals/ws');
require('./src/helpers/sensorDecoder');
require('./src/services/alertService');
require('./src/services/authService');
require('./src/services/modeManager');
require('./src/services/roverManager');
require('./src/services/commandService');
require('./src/services/roverConnectionService');
require('./src/services/httpServer');
+2 -2
View File
@@ -3,8 +3,8 @@
"version": "0.1.0",
"private": true,
"scripts": {
"start": "node src/index.js",
"dev": "nodemon src/index.js"
"start": "node index.js",
"dev": "nodemon index.js"
},
"dependencies": {
"express": "^4.19.2",
-193
View File
@@ -1,193 +0,0 @@
const socket = io();
const statusEl = document.getElementById('status');
const roverSelect = document.getElementById('roverSelect');
const sensorToggleBtn = document.getElementById('sensorToggle');
const motorsStopBtn = document.getElementById('motorsStop');
const sensorOutput = document.getElementById('sensorOutput');
const alertsEl = document.getElementById('alerts');
const mediaRestartBtn = document.getElementById('mediaRestart');
let selectedRover = null;
let sensorEnabled = false;
let lastDrive = { left: 0, right: 0 };
const OI_COMMANDS = {
start: [128],
safe: [131],
full: [132],
passive: [128],
dock: [143],
};
socket.on('connect', () => {
statusEl.textContent = 'Connected';
});
socket.on('disconnect', () => {
statusEl.textContent = 'Disconnected';
});
socket.on('rovers', (list) => {
roverSelect.innerHTML = '';
list.forEach((rover) => {
const option = document.createElement('option');
option.value = rover.id;
option.textContent = `${rover.name} (${rover.id})`;
roverSelect.appendChild(option);
});
if (list.length && !selectedRover) {
roverSelect.selectedIndex = 0;
selectedRover = list[0].id;
}
});
socket.on('sensorFrame', ({ roverId, frame, sensors }) => {
if (roverId !== selectedRover) return;
sensorOutput.textContent = renderSensors(sensors, frame?.data);
});
socket.on('commandAck', ({ roverId, status, error }) => {
if (roverId !== selectedRover) return;
if (status === 'ok') {
statusEl.textContent = 'Command applied';
} else {
statusEl.textContent = `Command failed: ${error}`;
}
});
socket.on('roverEvent', ({ roverId, event, ts, data }) => {
if (!alertsEl) return;
const div = document.createElement('div');
div.className = 'alert';
const when = ts ? new Date(ts).toLocaleTimeString() : new Date().toLocaleTimeString();
const details = data && data.error ? ` (${data.error})` : '';
div.textContent = `[${when}] ${roverId}: ${event}${details}`;
alertsEl.prepend(div);
while (alertsEl.children.length > 5) {
alertsEl.removeChild(alertsEl.lastChild);
}
setTimeout(() => {
if (div.parentElement === alertsEl) {
alertsEl.removeChild(div);
}
}, 15000);
});
roverSelect.addEventListener('change', (e) => {
selectedRover = e.target.value;
});
sensorToggleBtn.addEventListener('click', () => {
if (!selectedRover) return;
sensorEnabled = !sensorEnabled;
sensorToggleBtn.textContent = sensorEnabled ? 'Disable Sensor Stream' : 'Enable Sensor Stream';
sendCommand('sensorStream', { enable: sensorEnabled });
});
motorsStopBtn.addEventListener('click', () => {
sendCommand('motors', { main: 0, side: 0, vacuum: 0 });
});
mediaRestartBtn.addEventListener('click', () => {
sendCommand('media', { action: 'restart' });
});
document.querySelectorAll('[data-mode]').forEach((btn) => {
btn.addEventListener('click', () => {
const mode = btn.dataset.mode;
if (!OI_COMMANDS[mode]) return;
sendCommand('raw', { bytes: OI_COMMANDS[mode] });
});
});
const keys = new Set();
const driveInterval = 50;
window.addEventListener('keydown', (event) => {
keys.add(event.key.toLowerCase());
});
window.addEventListener('keyup', (event) => {
keys.delete(event.key.toLowerCase());
});
setInterval(() => {
if (!selectedRover) return;
const speeds = computeDrive();
if (speeds.left === lastDrive.left && speeds.right === lastDrive.right) {
return;
}
lastDrive = speeds;
sendCommand('drive', speeds);
}, driveInterval);
function computeDrive() {
const forward = keys.has('w');
const backward = keys.has('s');
const left = keys.has('a');
const right = keys.has('d');
const fast = keys.has('shift');
const base = fast ? 300 : 150;
let leftSpeed = 0;
let rightSpeed = 0;
if (forward && !backward) {
leftSpeed += base;
rightSpeed += base;
} else if (backward && !forward) {
leftSpeed -= base;
rightSpeed -= base;
}
if (left && !right) {
leftSpeed -= base;
rightSpeed += base;
} else if (right && !left) {
leftSpeed += base;
rightSpeed -= base;
}
if (!forward && !backward && (left || right)) {
leftSpeed = left ? -base : base;
rightSpeed = left ? base : -base;
}
return {
left: clamp(leftSpeed, -500, 500),
right: clamp(rightSpeed, -500, 500),
};
}
function sendCommand(type, data = {}) {
if (!selectedRover) return;
socket.emit('command', { roverId: selectedRover, type, data });
}
function formatSensorFrame(base64) {
const buffer = Uint8Array.from(atob(base64), (c) => c.charCodeAt(0));
return Array.from(buffer)
.map((b) => b.toString(16).padStart(2, '0'))
.join(' ');
}
function clamp(value, min, max) {
return Math.max(min, Math.min(max, value));
}
function renderSensors(sensors = {}, rawBase64) {
const lines = [];
if (sensors && Object.keys(sensors).length) {
for (const [key, value] of Object.entries(sensors)) {
if (value && typeof value === 'object' && !Array.isArray(value)) {
lines.push(`${key}: ${JSON.stringify(value)}`);
} else {
lines.push(`${key}: ${value}`);
}
}
} else {
lines.push('No decoded sensor data yet.');
}
if (rawBase64) {
lines.push('', 'raw:', formatSensorFrame(rawBase64));
}
return lines.join('\n');
}
+17 -7
View File
@@ -22,18 +22,21 @@
<label for="roverSelect">Rovers</label>
<select id="roverSelect" size="10" style="width: 100%"></select>
<div class="row">
<button data-mode="start">Start OI</button>
<button data-mode="safe">Safe</button>
<button data-mode="full">Full</button>
<button data-mode="passive">Passive</button>
<button data-mode="dock">Dock</button>
<button id="requestControl">Request Control</button>
<button id="lockToggle">Lock/Unlock</button>
</div>
<div class="row">
<button data-oi="start">Start OI</button>
<button data-oi="safe">Safe</button>
<button data-oi="full">Full</button>
<button data-oi="passive">Passive</button>
<button data-oi="dock">Dock</button>
</div>
<div class="row">
<button id="sensorToggle">Enable Sensor Stream</button>
</div>
<div class="row">
<button id="motorsStop">Stop Motors</button>
<button id="mediaRestart">Restart Camera</button>
</div>
</div>
<div style="flex: 1">
@@ -44,6 +47,13 @@
</div>
</div>
<script src="/socket.io/socket.io.js"></script>
<script src="app.js" type="module"></script>
<script src="loader.js"></script>
<script src="src/globals/socket.js"></script>
<script src="src/helpers/formatters.js"></script>
<script src="src/services/state.js"></script>
<script src="src/services/alertFeed.js"></script>
<script src="src/services/roverUI.js"></script>
<script src="src/services/driveControls.js"></script>
<script src="src/main.js"></script>
</body>
</html>
+22
View File
@@ -0,0 +1,22 @@
(function () {
const modules = {};
const cache = {};
window.registerModule = function (name, factory) {
modules[name] = factory;
};
window.requireModule = function (name) {
if (cache[name]) {
return cache[name];
}
const factory = modules[name];
if (!factory) {
throw new Error(`Module ${name} not found`);
}
const exports = {};
cache[name] = exports;
factory(requireModule, exports);
return exports;
};
})();
+4
View File
@@ -0,0 +1,4 @@
registerModule('globals/socket', (require, exports) => {
const socket = io();
exports.socket = socket;
});
+11
View File
@@ -0,0 +1,11 @@
registerModule('helpers/formatters', (require, exports) => {
function formatHex(base64) {
if (!base64) return '';
const buffer = Uint8Array.from(atob(base64), (c) => c.charCodeAt(0));
return Array.from(buffer)
.map((b) => b.toString(16).padStart(2, '0'))
.join(' ');
}
exports.formatHex = formatHex;
});
+6
View File
@@ -0,0 +1,6 @@
requireModule('globals/socket');
requireModule('helpers/formatters');
requireModule('services/state');
requireModule('services/alertFeed');
requireModule('services/roverUI');
requireModule('services/driveControls');
+25
View File
@@ -0,0 +1,25 @@
registerModule('services/alertFeed', (require, exports) => {
const { socket } = require('globals/socket');
const container = document.getElementById('alerts');
if (!container) return;
function pushAlert({ color = '#2196f3', title, message, ts }) {
const div = document.createElement('div');
div.className = 'alert';
div.style.borderColor = color;
div.style.background = color + '22';
div.innerHTML = `<strong>${title || 'Notice'}</strong> - ${message || ''}`;
container.prepend(div);
while (container.children.length > 5) {
container.removeChild(container.lastChild);
}
setTimeout(() => {
if (div.parentElement === container) {
container.removeChild(div);
}
}, 15000);
}
socket.on('alert', pushAlert);
});
+122
View File
@@ -0,0 +1,122 @@
registerModule('services/driveControls', (require, exports) => {
const { socket } = require('globals/socket');
const state = require('services/state');
const motorsStopBtn = document.getElementById('motorsStop');
const sensorToggleBtn = document.getElementById('sensorToggle');
const modeButtons = document.querySelectorAll('[data-oi]');
let sensorEnabled = false;
let lastDrive = { left: 0, right: 0 };
const keys = new Set();
const driveInterval = 120;
window.addEventListener('keydown', (event) => {
keys.add(event.key.toLowerCase());
});
window.addEventListener('keyup', (event) => {
keys.delete(event.key.toLowerCase());
});
setInterval(() => {
const roverId = state.getSelected();
if (!roverId) return;
const speeds = computeDrive();
if (speeds.left === lastDrive.left && speeds.right === lastDrive.right) return;
lastDrive = speeds;
socket.emit('command', {
roverId,
type: 'drive',
data: { driveDirect: speeds },
});
}, driveInterval);
function computeDrive() {
const forward = keys.has('w');
const backward = keys.has('s');
const left = keys.has('a');
const right = keys.has('d');
const fast = keys.has('shift');
const base = fast ? 300 : 150;
let leftSpeed = 0;
let rightSpeed = 0;
if (forward && !backward) {
leftSpeed += base;
rightSpeed += base;
} else if (backward && !forward) {
leftSpeed -= base;
rightSpeed -= base;
}
if (left && !right) {
leftSpeed -= base;
rightSpeed += base;
} else if (right && !left) {
leftSpeed += base;
rightSpeed -= base;
}
if (!forward && !backward && (left || right)) {
leftSpeed = left ? -base : base;
rightSpeed = left ? base : -base;
}
return {
left: clamp(leftSpeed, -500, 500),
right: clamp(rightSpeed, -500, 500),
};
}
function clamp(value, min, max) {
return Math.max(min, Math.min(max, value));
}
motorsStopBtn?.addEventListener('click', () => {
const roverId = state.getSelected();
if (!roverId) return;
socket.emit('command', {
roverId,
type: 'motors',
data: { motorPwm: { main: 0, side: 0, vacuum: 0 } },
});
});
sensorToggleBtn?.addEventListener('click', () => {
sensorEnabled = !sensorEnabled;
sensorToggleBtn.textContent = sensorEnabled ? 'Disable Sensor Stream' : 'Enable Sensor Stream';
const roverId = state.getSelected();
socket.emit('command', {
roverId,
type: 'sensorStream',
data: { sensorStream: { enable: sensorEnabled } },
});
});
const OI_COMMANDS = {
start: [128],
safe: [131],
full: [132],
passive: [128],
dock: [143],
};
modeButtons.forEach((btn) => {
btn.addEventListener('click', () => {
const cmd = btn.dataset.oi;
const bytes = OI_COMMANDS[cmd];
if (!bytes) return;
const roverId = state.getSelected();
socket.emit('command', {
roverId,
type: 'raw',
data: { raw: bytesToBase64(bytes) },
});
});
});
function bytesToBase64(bytes) {
return btoa(String.fromCharCode(...bytes));
}
});
+68
View File
@@ -0,0 +1,68 @@
registerModule('services/roverUI', (require, exports) => {
const { socket } = require('globals/socket');
const { formatHex } = require('helpers/formatters');
const state = require('services/state');
const statusEl = document.getElementById('status');
const roverSelect = document.getElementById('roverSelect');
const sensorOutput = document.getElementById('sensorOutput');
const requestBtn = document.getElementById('requestControl');
const lockBtn = document.getElementById('lockToggle');
let roster = [];
socket.on('rovers', (list) => {
roster = list;
roverSelect.innerHTML = '';
list.forEach((rover) => {
const option = document.createElement('option');
option.value = rover.id;
option.textContent = `${rover.name}${rover.locked ? ' (locked)' : ''}`;
roverSelect.appendChild(option);
});
if (!state.getSelected() && list.length) {
state.setSelected(list[0].id);
roverSelect.value = list[0].id;
}
});
roverSelect.addEventListener('change', () => {
state.setSelected(roverSelect.value);
});
socket.on('controlGranted', ({ roverId }) => {
state.setSelected(roverId);
roverSelect.value = roverId;
statusEl.textContent = `Driving ${roverId}`;
});
socket.on('sensorFrame', ({ roverId, frame, sensors }) => {
if (roverId !== state.getSelected()) return;
const lines = [];
if (sensors) {
Object.entries(sensors).forEach(([key, value]) => {
lines.push(`${key}: ${typeof value === 'object' ? JSON.stringify(value) : value}`);
});
}
lines.push('', 'raw:', formatHex(frame?.data));
sensorOutput.textContent = lines.join('\n');
});
requestBtn?.addEventListener('click', () => {
socket.emit('requestControl', { roverId: state.getSelected() });
});
lockBtn?.addEventListener('click', () => {
const roverId = state.getSelected();
const rover = roster.find((r) => r.id === roverId);
if (!rover) return;
socket.emit('lockRover', { roverId, locked: !rover.locked });
});
socket.on('connect', () => {
statusEl.textContent = 'Connected';
});
socket.on('disconnect', () => {
statusEl.textContent = 'Disconnected';
});
});
+14
View File
@@ -0,0 +1,14 @@
registerModule('services/state', (require, exports) => {
let selectedRover = null;
function setSelected(roverId) {
selectedRover = roverId;
}
function getSelected() {
return selectedRover;
}
exports.setSelected = setSelected;
exports.getSelected = getSelected;
});
+6
View File
@@ -0,0 +1,6 @@
const path = require('path');
module.exports = {
port: process.env.PORT || 8080,
staticDir: path.join(__dirname, '..', '..', 'public'),
};
+14
View File
@@ -0,0 +1,14 @@
const path = require('path');
const http = require('http');
const express = require('express');
const morgan = require('morgan');
const config = require('./config');
const app = express();
app.use(morgan('dev'));
app.use(express.json());
app.use(express.static(config.staticDir));
const httpServer = http.createServer(app);
module.exports = { app, httpServer };
+8
View File
@@ -0,0 +1,8 @@
const { Server: SocketIOServer } = require('socket.io');
const { httpServer } = require('./http');
const io = new SocketIOServer(httpServer, {
cors: { origin: '*' },
});
module.exports = io;
+9
View File
@@ -0,0 +1,9 @@
function stamp(level, args) {
return [new Date().toISOString(), `[${level}]`, ...args];
}
module.exports = {
info: (...args) => console.log(...stamp('INFO', args)),
warn: (...args) => console.warn(...stamp('WARN', args)),
error: (...args) => console.error(...stamp('ERROR', args)),
};
+21
View File
@@ -0,0 +1,21 @@
const { WebSocketServer } = require('ws');
const { httpServer } = require('./http');
const logger = require('./logger');
const roverWSS = new WebSocketServer({ noServer: true });
httpServer.on('upgrade', (req, socket, head) => {
if (req.url.startsWith('/rover')) {
roverWSS.handleUpgrade(req, socket, head, (ws) => {
roverWSS.emit('connection', ws, req);
});
} else {
socket.destroy();
}
});
roverWSS.on('connection', () => {
logger.info('Rover websocket connected');
});
module.exports = roverWSS;
-185
View File
@@ -1,185 +0,0 @@
const http = require('http');
const path = require('path');
const express = require('express');
const morgan = require('morgan');
const { Server: SocketIOServer } = require('socket.io');
const { WebSocketServer } = require('ws');
const { v4: uuidv4 } = require('uuid');
const { parseSensorFrame } = require('./sensorDecoder');
const PORT = process.env.PORT || 8080;
const app = express();
app.use(morgan('dev'));
app.use(express.json());
app.use(express.static(path.join(__dirname, '..', 'public')));
const httpServer = http.createServer(app);
const io = new SocketIOServer(httpServer, {
cors: { origin: '*' },
});
const roverWSS = new WebSocketServer({ noServer: true });
const rovers = new Map();
const pendingCommands = new Map();
httpServer.on('upgrade', (req, socket, head) => {
if (req.url.startsWith('/rover')) {
roverWSS.handleUpgrade(req, socket, head, (ws) => {
roverWSS.emit('connection', ws, req);
});
} else {
socket.destroy();
}
});
roverWSS.on('connection', (ws) => {
handleRoverConnection(ws);
});
function handleRoverConnection(ws) {
let roverId = null;
ws.on('message', (raw) => {
let msg;
try {
msg = JSON.parse(raw.toString());
} catch (err) {
return;
}
switch (msg.type) {
case 'hello':
roverId = msg.name;
rovers.set(roverId, {
id: roverId,
ws,
meta: msg,
lastSensor: null,
lastSeen: Date.now(),
});
broadcastRoster();
break;
case 'sensor':
if (!roverId || !rovers.has(roverId)) {
return;
}
const rover = rovers.get(roverId);
const decoded = parseSensorFrame(msg.data);
rover.lastSensor = { raw: msg, decoded };
rover.lastSeen = Date.now();
io.emit('sensorFrame', { roverId, frame: msg, sensors: decoded });
break;
case 'ack':
if (msg.id && pendingCommands.has(msg.id)) {
const pending = pendingCommands.get(msg.id);
pendingCommands.delete(msg.id);
io.emit('commandAck', { roverId: pending.roverId, id: msg.id, status: msg.status, error: msg.error });
}
break;
case 'event':
if (!roverId) {
return;
}
io.emit('roverEvent', {
roverId,
event: msg.event,
ts: msg.ts || Date.now(),
data: msg.data || {},
});
break;
default:
break;
}
});
ws.on('close', () => {
if (roverId) {
rovers.delete(roverId);
broadcastRoster();
}
});
}
io.on('connection', (socket) => {
socket.emit('rovers', getRoster());
socket.on('command', (payload = {}, cb = () => {}) => {
try {
const commandId = routeCommand(payload);
cb({ id: commandId });
} catch (err) {
cb({ error: err.message });
}
});
});
function getRoster() {
return Array.from(rovers.values()).map(({ id, meta, lastSeen }) => ({
id,
name: meta?.name ?? id,
battery: meta?.battery ?? null,
maxWheelSpeed: meta?.maxWheelSpeed ?? null,
media: meta?.media ?? null,
lastSeen,
}));
}
function broadcastRoster() {
io.emit('rovers', getRoster());
}
function routeCommand(payload) {
const { roverId, type, data } = payload;
if (!roverId) {
throw new Error('roverId missing');
}
if (!rovers.has(roverId)) {
throw new Error(`rover ${roverId} not connected`);
}
const rover = rovers.get(roverId);
const message = buildCommand(type, data);
const id = uuidv4();
message.id = id;
rover.ws.send(JSON.stringify(message));
pendingCommands.set(id, { roverId, issuedAt: Date.now(), type });
return id;
}
function buildCommand(type, data = {}) {
switch (type) {
case 'drive':
return { type: 'drive', driveDirect: { left: data.left || 0, right: data.right || 0 } };
case 'motors':
return {
type: 'motors',
motorPwm: {
main: data.main ?? 0,
side: data.side ?? 0,
vacuum: data.vacuum ?? 0,
},
};
case 'raw':
return { type: 'raw', raw: Buffer.from(data.bytes || []).toString('base64') };
case 'sensorStream':
return { type: 'sensorStream', sensorStream: { enable: Boolean(data.enable) } };
case 'media':
if (!data || !data.action) {
throw new Error('media action required');
}
return { type: 'media', media: { action: data.action } };
default:
throw new Error(`unknown command type: ${type}`);
}
}
if (require.main === module) {
httpServer.listen(PORT, () => {
console.log(`Server listening on :${PORT}`);
});
}
module.exports = {
httpServer,
io,
routeCommand,
};
+23
View File
@@ -0,0 +1,23 @@
const io = require('../globals/io');
const COLORS = {
info: '#2196f3',
success: '#4caf50',
warn: '#f0b651',
error: '#e53935',
};
function sendAlert({ color, title, message, ts = Date.now() }) {
const payload = {
color: color || COLORS.info,
title,
message,
ts,
};
io.emit('alert', payload);
}
module.exports = {
COLORS,
sendAlert,
};
+12
View File
@@ -0,0 +1,12 @@
function isAdmin(/* socket */) {
return false;
}
function isLockdownAdmin(/* socket */) {
return false;
}
module.exports = {
isAdmin,
isLockdownAdmin,
};
+53
View File
@@ -0,0 +1,53 @@
const { v4: uuidv4 } = require('uuid');
const io = require('../globals/io');
const roverManager = require('./roverManager');
const { isAdmin } = require('./authService');
const pendingCommands = new Map(); // id -> { roverId }
function issueCommand(roverId, payload) {
const record = roverManager.rovers.get(roverId);
if (!record || !record.ws) {
throw new Error('Rover offline');
}
const id = uuidv4();
const message = { ...payload, id };
record.ws.send(JSON.stringify(message));
pendingCommands.set(id, { roverId, ts: Date.now(), type: payload.type });
return id;
}
function handleAck(msg) {
const pending = pendingCommands.get(msg.id);
if (!pending) return;
pendingCommands.delete(msg.id);
io.emit('commandAck', {
roverId: pending.roverId,
id: msg.id,
status: msg.status || 'ok',
error: msg.error,
});
}
module.exports = {
issueCommand,
handleAck,
};
io.on('connection', (socket) => {
socket.on('command', ({ roverId, type, data } = {}, cb = () => {}) => {
try {
if (!roverId) {
throw new Error('roverId required');
}
if (!roverManager.isDriver(roverId, socket) && !isAdmin(socket)) {
throw new Error('Not controlling this rover');
}
const payload = data ? { ...data } : {};
const id = issueCommand(roverId, { type, ...payload });
cb({ id });
} catch (err) {
cb({ error: err.message });
}
});
});
+7
View File
@@ -0,0 +1,7 @@
const { httpServer } = require('../globals/http');
const config = require('../globals/config');
const logger = require('../globals/logger');
httpServer.listen(config.port, () => {
logger.info(`Server listening on :${config.port}`);
});
+58
View File
@@ -0,0 +1,58 @@
const io = require('../globals/io');
const { sendAlert, COLORS } = require('./alertService');
const { isAdmin, isLockdownAdmin } = require('./authService');
const MODES = {
OPEN: 'open',
TURNS: 'turns',
ADMIN: 'admin',
LOCKDOWN: 'lockdown',
};
let currentMode = MODES.OPEN;
function canChangeMode(socket, nextMode) {
if (nextMode === MODES.LOCKDOWN) {
return isLockdownAdmin(socket);
}
return isAdmin(socket);
}
function setMode(nextMode, socket) {
if (!Object.values(MODES).includes(nextMode)) {
throw new Error(`Unknown mode ${nextMode}`);
}
if (!canChangeMode(socket, nextMode)) {
throw new Error('Not authorized to change mode');
}
if (currentMode === nextMode) {
return currentMode;
}
currentMode = nextMode;
sendAlert({
color: COLORS.info,
title: 'Mode Changed',
message: `Server mode set to ${nextMode}`,
});
return currentMode;
}
function getMode() {
return currentMode;
}
module.exports = {
MODES,
getMode,
setMode,
};
io.on('connection', (socket) => {
socket.on('setMode', ({ mode }) => {
try {
setMode(mode, socket);
} catch (err) {
sendAlert({ color: COLORS.error, title: 'Mode change failed', message: err.message });
}
});
});
@@ -0,0 +1,57 @@
const roverWSS = require('../globals/ws');
const logger = require('../globals/logger');
const roverManager = require('./roverManager');
const { sendAlert, COLORS } = require('./alertService');
const { handleAck } = require('./commandService');
function handleMessage(roverId, msg) {
switch (msg.type) {
case 'hello':
roverManager.upsertRover(msg, this);
sendAlert({ color: COLORS.info, title: 'Rover Connected', message: roverId });
break;
case 'sensor':
roverManager.handleSensorFrame(roverId, msg);
break;
case 'event':
sendAlert({ color: COLORS.info, title: `${roverId} event`, message: msg.event });
break;
default:
break;
}
}
roverWSS.on('connection', (ws) => {
let roverId = null;
ws.on('message', (raw) => {
let msg;
try {
msg = JSON.parse(raw.toString());
} catch (err) {
logger.warn('Invalid rover message', err.message);
return;
}
if (msg.type === 'hello') {
roverId = msg.name;
roverManager.upsertRover(msg, ws);
roverManager.broadcastRoster();
sendAlert({ color: COLORS.success, title: 'Rover Online', message: roverId });
return;
}
if (!roverId) return;
if (msg.type === 'sensor') {
roverManager.handleSensorFrame(roverId, msg);
} else if (msg.type === 'ack') {
handleAck(msg);
} else if (msg.type === 'event') {
sendAlert({ color: COLORS.info, title: `${roverId}`, message: msg.event });
}
});
ws.on('close', () => {
if (roverId) {
roverManager.removeRover(roverId);
sendAlert({ color: COLORS.warn, title: 'Rover Offline', message: roverId });
}
});
});
+203
View File
@@ -0,0 +1,203 @@
const io = require('../globals/io');
const logger = require('../globals/logger');
const { sendAlert, COLORS } = require('./alertService');
const { parseSensorFrame } = require('../helpers/sensorDecoder');
const { MODES, getMode } = require('./modeManager');
const { isAdmin } = require('./authService');
const rovers = new Map(); // roverId -> record
const socketToRovers = new Map(); // socketId -> Set(roverId)
function ensureRecord(id) {
if (!rovers.has(id)) {
rovers.set(id, {
id,
meta: null,
ws: null,
lastSensor: null,
drivers: new Set(),
locked: false,
room: `rover:${id}`,
lastSeen: Date.now(),
});
}
return rovers.get(id);
}
function upsertRover(meta, ws) {
const id = meta.name || meta.id;
const record = ensureRecord(id);
record.meta = meta;
record.ws = ws;
record.lastSeen = Date.now();
rovers.set(id, record);
broadcastRoster();
return record;
}
function removeRover(id) {
const record = rovers.get(id);
if (!record) return;
rovers.delete(id);
broadcastRoster();
}
function lockRover(id, locked, actorSocket) {
const record = rovers.get(id);
if (!record) {
throw new Error('Unknown rover');
}
if (locked) {
record.locked = true;
sendAlert({ color: COLORS.warn, title: 'Rover Locked', message: `${id} locked by admin.` });
} else {
record.locked = false;
sendAlert({ color: COLORS.success, title: 'Rover Unlocked', message: `${id} unlocked.` });
}
broadcastRoster();
return record.locked;
}
function getRoster() {
return Array.from(rovers.values()).map((record) => ({
id: record.id,
name: record.meta?.name || record.id,
battery: record.meta?.battery,
maxWheelSpeed: record.meta?.maxWheelSpeed,
media: record.meta?.media,
locked: record.locked,
lastSeen: record.lastSeen,
}));
}
function broadcastRoster() {
io.emit('rovers', getRoster());
}
function handleSensorFrame(roverId, frame) {
const record = rovers.get(roverId);
if (!record) return;
record.lastSeen = Date.now();
const decoded = parseSensorFrame(frame.data);
record.lastSensor = { raw: frame, decoded };
io.to(record.room).emit('sensorFrame', {
roverId,
frame,
sensors: decoded,
});
}
function removeSocket(socket) {
const joined = socketToRovers.get(socket.id);
if (!joined) return;
for (const roverId of joined) {
const record = rovers.get(roverId);
if (record) {
record.drivers.delete(socket.id);
}
}
socketToRovers.delete(socket.id);
}
function requestControl(roverId, socket) {
const record = rovers.get(roverId);
if (!record) {
throw new Error('Unknown rover');
}
if (record.locked && !isAdmin(socket)) {
throw new Error('Rover locked');
}
const mode = getMode();
if (mode === MODES.ADMIN && !isAdmin(socket)) {
throw new Error('Admins only');
}
if (mode === MODES.LOCKDOWN && !isAdmin(socket)) {
throw new Error('Server in lockdown');
}
// TODO: future turns logic
record.drivers.add(socket.id);
if (!socketToRovers.has(socket.id)) {
socketToRovers.set(socket.id, new Set());
}
socketToRovers.get(socket.id).add(roverId);
socket.join(record.room);
sendAlert({
color: COLORS.success,
title: 'Control Granted',
message: `${socket.id} now driving ${roverId}`,
});
return { roverId, room: record.room };
}
function releaseControl(roverId, socket) {
const record = rovers.get(roverId);
if (!record) return;
record.drivers.delete(socket.id);
const joined = socketToRovers.get(socket.id);
if (joined) {
joined.delete(roverId);
if (joined.size === 0) {
socketToRovers.delete(socket.id);
}
}
socket.leave(record.room);
}
function isDriver(roverId, socket) {
const record = rovers.get(roverId);
if (!record) return false;
return record.drivers.has(socket.id);
}
module.exports = {
upsertRover,
removeRover,
lockRover,
getRoster,
broadcastRoster,
handleSensorFrame,
requestControl,
releaseControl,
removeSocket,
isDriver,
rovers,
};
io.on('connection', (socket) => {
socket.on('requestControl', ({ roverId } = {}) => {
try {
const targetId = roverId || Array.from(rovers.keys())[0];
if (!targetId) {
throw new Error('No rovers available');
}
requestControl(targetId, socket);
socket.emit('controlGranted', { roverId: targetId });
} catch (err) {
sendAlert({ color: COLORS.warn, title: 'Control denied', message: err.message });
}
});
socket.on('releaseControl', ({ roverId }) => {
if (!roverId) return;
releaseControl(roverId, socket);
});
socket.on('lockRover', ({ roverId, locked }) => {
if (!isAdmin(socket)) return;
try {
lockRover(roverId, locked);
} catch (err) {
sendAlert({ color: COLORS.error, title: 'Lock failed', message: err.message });
}
});
socket.on('subscribeAll', () => {
for (const record of rovers.values()) {
socket.join(record.room);
}
});
socket.on('disconnect', () => {
removeSocket(socket);
});
});