mirror of
https://github.com/legop3/MultiRoombaRover.git
synced 2026-09-17 10:00:46 -04:00
auuuhm
This commit is contained in:
Vendored
+3
-4
@@ -14,9 +14,8 @@ battery:
|
|||||||
urgent: 1650
|
urgent: 1650
|
||||||
maxWheelSpeed: 350
|
maxWheelSpeed: 350
|
||||||
media:
|
media:
|
||||||
whepUrl: https://mediaserver.local/whep/roomba-alpha
|
publishUrl: https://control-server.local/whip/roomba-alpha
|
||||||
streamKey: roomba-alpha
|
|
||||||
manage: false
|
manage: false
|
||||||
service: mediamtx.service
|
service: mediamtx.service
|
||||||
healthUrl: http://127.0.0.1:9997/v3/paths/list/rovercam
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
healthInterval: 30s
|
healthInterval: 30s
|
||||||
|
|||||||
Vendored
+3
-4
@@ -14,9 +14,8 @@ battery:
|
|||||||
urgent: 1650
|
urgent: 1650
|
||||||
maxWheelSpeed: 350
|
maxWheelSpeed: 350
|
||||||
media:
|
media:
|
||||||
whepUrl: https://mediaserver.local/whep/roomba-alpha
|
publishUrl: https://control-server.local/whip/roomba-alpha
|
||||||
streamKey: roomba-alpha
|
|
||||||
manage: false
|
manage: false
|
||||||
service: mediamtx.service
|
service: mediamtx.service
|
||||||
healthUrl: http://127.0.0.1:9997/v3/paths/list/rovercam
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
healthInterval: 30s
|
healthInterval: 30s
|
||||||
|
|||||||
Vendored
BIN
Binary file not shown.
@@ -55,7 +55,7 @@ Flags:
|
|||||||
| `--mediamtx` | download/install mediaMTX plus the provided config + unit |
|
| `--mediamtx` | download/install mediaMTX plus the provided config + unit |
|
||||||
| `--mediamtx-version X.Y.Z` | override the mediaMTX release tag (default `1.15.3`) |
|
| `--mediamtx-version X.Y.Z` | override the mediaMTX release tag (default `1.15.3`) |
|
||||||
|
|
||||||
If the script installs the sample config, it will remind you to edit `/etc/roverd.yaml` before manually restarting the service: set `name`, `serverUrl`, serial device, BRC pin, battery thresholds, and the WHEP URL that the central server should expose.
|
If the script installs the sample config, it will remind you to edit `/etc/roverd.yaml` before manually restarting the service: set `name`, `serverUrl`, serial device, BRC pin, battery thresholds, and the `media.publishUrl` that points at your central mediaMTX WHIP endpoint (for example `https://control-server.local/whip/roomba-alpha`).
|
||||||
|
|
||||||
## Manual installation
|
## Manual installation
|
||||||
|
|
||||||
@@ -65,7 +65,7 @@ If the script installs the sample config, it will remind you to edit `/etc/rover
|
|||||||
sudo install -o roverd -g roverd -m 0755 dist/roverd /usr/local/bin/roverd
|
sudo install -o roverd -g roverd -m 0755 dist/roverd /usr/local/bin/roverd
|
||||||
sudo install -o roverd -g roverd -m 0640 pi/roverd/roverd.sample.yaml /etc/roverd.yaml
|
sudo install -o roverd -g roverd -m 0640 pi/roverd/roverd.sample.yaml /etc/roverd.yaml
|
||||||
```
|
```
|
||||||
Adjust `/etc/roverd.yaml` for each rover: `name`, `serverUrl` (e.g. `ws://control-server:8080/rover`), serial port path, battery thresholds, GPIO pin for BRC, and the media WHEP URL that points at the central distribution server.
|
Adjust `/etc/roverd.yaml` for each rover: `name`, `serverUrl` (e.g. `ws://control-server:8080/rover`), serial port path, battery thresholds, GPIO pin for BRC, and the media `publishUrl` that points at the central media server’s WHIP endpoint for that rover.
|
||||||
|
|
||||||
2. Install the systemd unit:
|
2. Install the systemd unit:
|
||||||
```bash
|
```bash
|
||||||
@@ -92,8 +92,9 @@ If you set `media.manage: true` in `/etc/roverd.yaml`, make sure the `roverd` se
|
|||||||
sudo systemctl enable --now mediamtx.service
|
sudo systemctl enable --now mediamtx.service
|
||||||
```
|
```
|
||||||
|
|
||||||
The sample config uses the Raspberry Pi camera module as the source and exposes a WHEP endpoint at `http://<pi-host>:8889/whep/rovercam`. Point `media.whepUrl` in `roverd.yaml` at this URL so the central server can list it.
|
The sample config uses the Raspberry Pi camera module as the source and enables the local mediaMTX HTTP API so `roverd` can health-check the service. Edit `/etc/mediamtx/mediamtx.yml` to point its WHIP client at your central media server (see `mediamtx_server_integration.md` for details).
|
||||||
Expose the mediaMTX HTTP API locally (default `http://127.0.0.1:9997`) and set `media.healthUrl` so `roverd` can monitor the pipeline; `media.service` should match the systemd unit name (default `mediamtx.service`).
|
Expose the mediaMTX HTTP API locally (default `http://127.0.0.1:9997`) and set `media.healthUrl` so `roverd` can monitor the pipeline; `media.service` should match the systemd unit name (default `mediamtx.service`).
|
||||||
|
Finally, edit `/etc/mediamtx/mediamtx.yml` so the Pi publishes the camera feed to your central control server via WHIP (for example by pointing it at `https://control.example.com/whip/<roverName>`). The Pi never serves viewers directly—drivers and spectators always watch through the control server.
|
||||||
|
|
||||||
## Server + UI
|
## Server + UI
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,19 @@
|
|||||||
|
# mediaMTX Integration
|
||||||
|
|
||||||
|
Each rover runs mediaMTX locally to capture the Pi camera and publish it upstream, while the control server hosts a central mediaMTX instance that fans video out to drivers and spectators.
|
||||||
|
|
||||||
|
## Pi (publisher)
|
||||||
|
|
||||||
|
- mediaMTX samples the Pi camera (`paths.rovercam.source: rpiCamera`) and exposes the HTTP API on `http://127.0.0.1:9997`.
|
||||||
|
- `/etc/roverd.yaml` contains `media.publishUrl`, pointing at the control server’s WHIP endpoint for that rover (e.g. `https://control.example.com/whip/roomba-alpha`). No auth is required when the Pi network is trusted.
|
||||||
|
- The `media.manage` flag keeps the local service alive via `systemctl` and hits the API for health checks (`media.healthUrl`, defaults to `http://127.0.0.1:9997/v3/paths/list`).
|
||||||
|
|
||||||
|
## Control server (viewer)
|
||||||
|
|
||||||
|
- The central mediaMTX instance accepts WHIP ingest at `/whip/<roverId>` and serves WHEP playback at `/whep/<roverId>`.
|
||||||
|
- The Node server issues viewer sessions (one per socket) and mediamtx calls back into the Node server to authorize new WHEP connections. Lockdown mode simply stops minting sessions for non-lockdown admins.
|
||||||
|
|
||||||
|
## Driver / spectator UIs
|
||||||
|
|
||||||
|
- When a user requests video the browser asks the Node server for a session token, and if the current mode/role allows it, the server returns the WHEP URL plus the session id.
|
||||||
|
- Spectator mode and lockdown are enforced purely on the Node server: no direct Pi URLs are ever exposed, and once a user has a session it remains valid until they disconnect or lockdown revokes it.
|
||||||
@@ -1,4 +1,7 @@
|
|||||||
logLevel: info
|
logLevel: info
|
||||||
|
api: yes
|
||||||
|
apiAddress: 127.0.0.1:9997
|
||||||
|
apiEncryption: no
|
||||||
webrtc: yes
|
webrtc: yes
|
||||||
webrtcLocalUDPAddress: :8189
|
webrtcLocalUDPAddress: :8189
|
||||||
webrtcLocalTCPAddress: :8189
|
webrtcLocalTCPAddress: :8189
|
||||||
|
|||||||
+1
-2
@@ -53,8 +53,7 @@ type BatteryConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type MediaConfig struct {
|
type MediaConfig struct {
|
||||||
WhepURL string `yaml:"whepUrl"`
|
PublishURL string `yaml:"publishUrl"`
|
||||||
StreamKey string `yaml:"streamKey"`
|
|
||||||
Manage bool `yaml:"manage"`
|
Manage bool `yaml:"manage"`
|
||||||
Service string `yaml:"service"`
|
Service string `yaml:"service"`
|
||||||
HealthURL string `yaml:"healthUrl"`
|
HealthURL string `yaml:"healthUrl"`
|
||||||
|
|||||||
@@ -22,6 +22,9 @@ func NewMediaSupervisor(cfg MediaConfig, logger *log.Logger) *MediaSupervisor {
|
|||||||
if !cfg.Manage || cfg.Service == "" {
|
if !cfg.Manage || cfg.Service == "" {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
if cfg.HealthURL == "" {
|
||||||
|
cfg.HealthURL = "http://127.0.0.1:9997/v3/paths/list"
|
||||||
|
}
|
||||||
interval := cfg.HealthInterval.Duration
|
interval := cfg.HealthInterval.Duration
|
||||||
if interval <= 0 {
|
if interval <= 0 {
|
||||||
interval = 30 * time.Second
|
interval = 30 * time.Second
|
||||||
|
|||||||
Executable
BIN
Binary file not shown.
@@ -15,9 +15,8 @@ battery:
|
|||||||
urgent: 1650
|
urgent: 1650
|
||||||
maxWheelSpeed: 350
|
maxWheelSpeed: 350
|
||||||
media:
|
media:
|
||||||
whepUrl: https://mediaserver.local/whep/roomba-alpha
|
publishUrl: https://control-server.local/whip/roomba-alpha
|
||||||
streamKey: roomba-alpha
|
|
||||||
manage: false
|
manage: false
|
||||||
service: mediamtx.service
|
service: mediamtx.service
|
||||||
healthUrl: http://127.0.0.1:9997/v3/paths/list/rovercam
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
healthInterval: 30s
|
healthInterval: 30s
|
||||||
|
|||||||
@@ -15,9 +15,8 @@ battery:
|
|||||||
urgent: 1650
|
urgent: 1650
|
||||||
maxWheelSpeed: 350
|
maxWheelSpeed: 350
|
||||||
media:
|
media:
|
||||||
whepUrl: https://mediaserver.local/whep/roomba-alpha
|
publishUrl: https://control-server.local/whip/roomba-alpha
|
||||||
streamKey: roomba-alpha
|
|
||||||
manage: false
|
manage: false
|
||||||
service: mediamtx.service
|
service: mediamtx.service
|
||||||
healthUrl: http://127.0.0.1:9997/v3/paths/list/rovercam
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
healthInterval: 30s
|
healthInterval: 30s
|
||||||
|
|||||||
@@ -13,4 +13,5 @@ require('./src/services/lockdownGuard');
|
|||||||
require('./src/services/roverManager');
|
require('./src/services/roverManager');
|
||||||
require('./src/services/commandService');
|
require('./src/services/commandService');
|
||||||
require('./src/services/roverConnectionService');
|
require('./src/services/roverConnectionService');
|
||||||
|
require('./src/services/assignmentService');
|
||||||
require('./src/services/httpServer');
|
require('./src/services/httpServer');
|
||||||
|
|||||||
@@ -24,17 +24,25 @@
|
|||||||
<input id="username" placeholder="Username" />
|
<input id="username" placeholder="Username" />
|
||||||
<input id="password" type="password" placeholder="Password" />
|
<input id="password" type="password" placeholder="Password" />
|
||||||
<button id="loginBtn">Admin Login</button>
|
<button id="loginBtn">Admin Login</button>
|
||||||
<button id="spectatorBtn">Spectate</button>
|
|
||||||
<span id="roleStatus">Role: user</span>
|
<span id="roleStatus">Role: user</span>
|
||||||
</div>
|
</div>
|
||||||
<div id="layout">
|
<div id="layout">
|
||||||
<div id="roverList">
|
<div id="roverList">
|
||||||
<label for="roverSelect">Rovers</label>
|
<label for="roverSelect">Rovers</label>
|
||||||
<select id="roverSelect" size="10" style="width: 100%"></select>
|
<select id="roverSelect" size="10" style="width: 100%"></select>
|
||||||
<div class="row">
|
<div class="row" id="adminControls">
|
||||||
<button id="requestControl">Request Control</button>
|
<label><input type="checkbox" id="forceControl" /> Force control</label>
|
||||||
<button id="lockToggle">Lock/Unlock</button>
|
<button id="lockToggle">Lock/Unlock</button>
|
||||||
</div>
|
</div>
|
||||||
|
<div class="row" id="modeControls" style="display:none">
|
||||||
|
<label for="modeSelect">Server mode</label>
|
||||||
|
<select id="modeSelect">
|
||||||
|
<option value="open">Open</option>
|
||||||
|
<option value="turns">Turns</option>
|
||||||
|
<option value="admin">Admin</option>
|
||||||
|
<option value="lockdown">Lockdown</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
<div class="row">
|
<div class="row">
|
||||||
<button data-oi="start">Start OI</button>
|
<button data-oi="start">Start OI</button>
|
||||||
<button data-oi="safe">Safe</button>
|
<button data-oi="safe">Safe</button>
|
||||||
@@ -51,7 +59,8 @@
|
|||||||
</div>
|
</div>
|
||||||
<div style="flex: 1">
|
<div style="flex: 1">
|
||||||
<h3>Sensor Frames</h3>
|
<h3>Sensor Frames</h3>
|
||||||
<div id="activeDriver">Active driver: --</div>
|
<div id="currentRover">Active rover: --</div>
|
||||||
|
<div id="activeDriver">Driver: --</div>
|
||||||
<pre id="sensorOutput"></pre>
|
<pre id="sensorOutput"></pre>
|
||||||
<p>Use WASD keys to drive the selected rover. Shift increases speed.</p>
|
<p>Use WASD keys to drive the selected rover. Shift increases speed.</p>
|
||||||
<div id="alerts"></div>
|
<div id="alerts"></div>
|
||||||
@@ -67,5 +76,6 @@
|
|||||||
<script src="src/services/roverUI.js"></script>
|
<script src="src/services/roverUI.js"></script>
|
||||||
<script src="src/services/driveControls.js"></script>
|
<script src="src/services/driveControls.js"></script>
|
||||||
<script src="src/main.js"></script>
|
<script src="src/main.js"></script>
|
||||||
|
<p><a href="/spectate/">Open spectator view</a></p>
|
||||||
</body>
|
</body>
|
||||||
</html>
|
</html>
|
||||||
|
|||||||
@@ -0,0 +1,29 @@
|
|||||||
|
<!doctype html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8" />
|
||||||
|
<title>Rover Spectator</title>
|
||||||
|
<style>
|
||||||
|
body { font-family: sans-serif; margin: 16px; }
|
||||||
|
#status { margin-bottom: 12px; }
|
||||||
|
#grid { display: grid; grid-template-columns: repeat(auto-fill, minmax(280px, 1fr)); gap: 12px; }
|
||||||
|
.rover-card { border: 1px solid #ccc; padding: 12px; border-radius: 6px; background: #fdfdfd; }
|
||||||
|
.rover-card h3 { margin: 0 0 8px 0; }
|
||||||
|
.sensor { font-family: monospace; white-space: pre-wrap; word-break: break-word; }
|
||||||
|
.meta { font-size: 0.9rem; color: #555; margin-bottom: 8px; }
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>Rover Spectator</h1>
|
||||||
|
<div id="status">Connecting…</div>
|
||||||
|
<div id="grid"></div>
|
||||||
|
|
||||||
|
<script src="/socket.io/socket.io.js"></script>
|
||||||
|
<script>
|
||||||
|
window.__desiredRole = 'spectator';
|
||||||
|
</script>
|
||||||
|
<script src="../loader.js"></script>
|
||||||
|
<script src="../src/globals/socket.js"></script>
|
||||||
|
<script src="spectator.js"></script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
registerModule('spectator/main', (require, exports) => {
|
||||||
|
const { socket } = require('globals/socket');
|
||||||
|
|
||||||
|
const statusEl = document.getElementById('status');
|
||||||
|
const grid = document.getElementById('grid');
|
||||||
|
const cards = new Map();
|
||||||
|
|
||||||
|
socket.on('connect', () => {
|
||||||
|
statusEl.textContent = 'Connected';
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('disconnect', () => {
|
||||||
|
statusEl.textContent = 'Disconnected';
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('lockdown', ({ message }) => {
|
||||||
|
statusEl.textContent = message || 'Spectator mode disabled';
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('rovers', (list) => {
|
||||||
|
const seen = new Set();
|
||||||
|
list.forEach((rover) => {
|
||||||
|
const card = ensureCard(rover.id);
|
||||||
|
card.querySelector('.meta').textContent = `Locked: ${rover.locked ? 'yes' : 'no'} | Last seen: ${new Date(rover.lastSeen).toLocaleTimeString()}`;
|
||||||
|
});
|
||||||
|
cards.forEach((_, id) => {
|
||||||
|
if (!list.find((r) => r.id === id)) {
|
||||||
|
removeCard(id);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('sensorFrame', ({ roverId, sensors }) => {
|
||||||
|
const card = ensureCard(roverId);
|
||||||
|
card.querySelector('.sensor').textContent = sensors ? JSON.stringify(sensors, null, 2) : 'No data';
|
||||||
|
});
|
||||||
|
|
||||||
|
function ensureCard(roverId) {
|
||||||
|
if (cards.has(roverId)) {
|
||||||
|
return cards.get(roverId);
|
||||||
|
}
|
||||||
|
const card = document.createElement('div');
|
||||||
|
card.className = 'rover-card';
|
||||||
|
card.innerHTML = `
|
||||||
|
<h3>${roverId}</h3>
|
||||||
|
<div class="meta"></div>
|
||||||
|
<pre class="sensor">Waiting for data...</pre>
|
||||||
|
`;
|
||||||
|
grid.appendChild(card);
|
||||||
|
cards.set(roverId, card);
|
||||||
|
return card;
|
||||||
|
}
|
||||||
|
|
||||||
|
function removeCard(roverId) {
|
||||||
|
const card = cards.get(roverId);
|
||||||
|
if (card) {
|
||||||
|
card.remove();
|
||||||
|
cards.delete(roverId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
requireModule('spectator/main');
|
||||||
@@ -1,4 +1,8 @@
|
|||||||
registerModule('globals/socket', (require, exports) => {
|
registerModule('globals/socket', (require, exports) => {
|
||||||
const socket = io();
|
const query = {};
|
||||||
|
if (window.__desiredRole) {
|
||||||
|
query.role = window.__desiredRole;
|
||||||
|
}
|
||||||
|
const socket = io(query.role ? { query } : undefined);
|
||||||
exports.socket = socket;
|
exports.socket = socket;
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -6,7 +6,6 @@ registerModule('services/authControls', (require, exports) => {
|
|||||||
const passwordInput = document.getElementById('password');
|
const passwordInput = document.getElementById('password');
|
||||||
const loginBtn = document.getElementById('loginBtn');
|
const loginBtn = document.getElementById('loginBtn');
|
||||||
const roleStatus = document.getElementById('roleStatus');
|
const roleStatus = document.getElementById('roleStatus');
|
||||||
const spectatorBtn = document.getElementById('spectatorBtn');
|
|
||||||
|
|
||||||
function updateRole(role) {
|
function updateRole(role) {
|
||||||
if (roleStatus) {
|
if (roleStatus) {
|
||||||
@@ -33,9 +32,4 @@ registerModule('services/authControls', (require, exports) => {
|
|||||||
});
|
});
|
||||||
// initialize from default role
|
// initialize from default role
|
||||||
updateRole(state.getRole());
|
updateRole(state.getRole());
|
||||||
|
|
||||||
spectatorBtn?.addEventListener('click', () => {
|
|
||||||
socket.emit('role:set', { role: 'spectator' });
|
|
||||||
updateRole('spectator');
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -6,16 +6,28 @@ registerModule('services/roverUI', (require, exports) => {
|
|||||||
const statusEl = document.getElementById('status');
|
const statusEl = document.getElementById('status');
|
||||||
const roverSelect = document.getElementById('roverSelect');
|
const roverSelect = document.getElementById('roverSelect');
|
||||||
const sensorOutput = document.getElementById('sensorOutput');
|
const sensorOutput = document.getElementById('sensorOutput');
|
||||||
const requestBtn = document.getElementById('requestControl');
|
|
||||||
const lockBtn = document.getElementById('lockToggle');
|
const lockBtn = document.getElementById('lockToggle');
|
||||||
|
const forceCheckbox = document.getElementById('forceControl');
|
||||||
|
const adminControls = document.getElementById('adminControls');
|
||||||
|
const modeControls = document.getElementById('modeControls');
|
||||||
|
const modeSelect = document.getElementById('modeSelect');
|
||||||
const activeDriverEl = document.getElementById('activeDriver');
|
const activeDriverEl = document.getElementById('activeDriver');
|
||||||
|
const currentRoverEl = document.getElementById('currentRover');
|
||||||
|
|
||||||
let roster = [];
|
let roster = [];
|
||||||
|
let lastSelection = null;
|
||||||
|
|
||||||
state.onRoleChange((role) => {
|
state.onRoleChange((role) => {
|
||||||
const isAdmin = role === 'admin' || role === 'lockdown';
|
const isAdmin = role === 'admin' || role === 'lockdown';
|
||||||
if (requestBtn) requestBtn.disabled = !isAdmin;
|
|
||||||
if (lockBtn) lockBtn.disabled = !isAdmin;
|
if (lockBtn) lockBtn.disabled = !isAdmin;
|
||||||
|
if (roverSelect) roverSelect.disabled = !isAdmin;
|
||||||
|
if (adminControls) {
|
||||||
|
adminControls.style.display = isAdmin ? 'flex' : 'none';
|
||||||
|
}
|
||||||
|
if (modeControls) {
|
||||||
|
modeControls.style.display = isAdmin ? 'block' : 'none';
|
||||||
|
if (modeSelect) modeSelect.disabled = !isAdmin;
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
socket.on('rovers', (list) => {
|
socket.on('rovers', (list) => {
|
||||||
@@ -27,19 +39,28 @@ registerModule('services/roverUI', (require, exports) => {
|
|||||||
option.textContent = `${rover.name}${rover.locked ? ' (locked)' : ''}`;
|
option.textContent = `${rover.name}${rover.locked ? ' (locked)' : ''}`;
|
||||||
roverSelect.appendChild(option);
|
roverSelect.appendChild(option);
|
||||||
});
|
});
|
||||||
if (!state.getSelected() && list.length) {
|
const selected = state.getSelected();
|
||||||
|
if ((!selected || !list.find((r) => r.id === selected)) && list.length) {
|
||||||
state.setSelected(list[0].id);
|
state.setSelected(list[0].id);
|
||||||
roverSelect.value = list[0].id;
|
} else if (!list.length) {
|
||||||
|
state.setSelected(null);
|
||||||
}
|
}
|
||||||
|
renderSelection();
|
||||||
});
|
});
|
||||||
|
|
||||||
roverSelect.addEventListener('change', () => {
|
roverSelect.addEventListener('change', () => {
|
||||||
state.setSelected(roverSelect.value);
|
const roverId = roverSelect.value;
|
||||||
|
const role = state.getRole();
|
||||||
|
state.setSelected(roverId);
|
||||||
|
renderSelection(true);
|
||||||
|
if (role === 'admin' || role === 'lockdown') {
|
||||||
|
socket.emit('requestControl', { roverId, force: !!forceCheckbox?.checked });
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
socket.on('controlGranted', ({ roverId }) => {
|
socket.on('controlGranted', ({ roverId }) => {
|
||||||
state.setSelected(roverId);
|
state.setSelected(roverId);
|
||||||
roverSelect.value = roverId;
|
renderSelection(true);
|
||||||
statusEl.textContent = `Driving ${roverId}`;
|
statusEl.textContent = `Driving ${roverId}`;
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -62,10 +83,6 @@ registerModule('services/roverUI', (require, exports) => {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
requestBtn?.addEventListener('click', () => {
|
|
||||||
socket.emit('requestControl', { roverId: state.getSelected() });
|
|
||||||
});
|
|
||||||
|
|
||||||
lockBtn?.addEventListener('click', () => {
|
lockBtn?.addEventListener('click', () => {
|
||||||
const roverId = state.getSelected();
|
const roverId = state.getSelected();
|
||||||
const rover = roster.find((r) => r.id === roverId);
|
const rover = roster.find((r) => r.id === roverId);
|
||||||
@@ -73,10 +90,36 @@ registerModule('services/roverUI', (require, exports) => {
|
|||||||
socket.emit('lockRover', { roverId, locked: !rover.locked });
|
socket.emit('lockRover', { roverId, locked: !rover.locked });
|
||||||
});
|
});
|
||||||
|
|
||||||
|
modeSelect?.addEventListener('change', () => {
|
||||||
|
socket.emit('setMode', { mode: modeSelect.value });
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('mode', ({ mode }) => {
|
||||||
|
if (modeSelect) {
|
||||||
|
modeSelect.value = mode;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
socket.on('connect', () => {
|
socket.on('connect', () => {
|
||||||
statusEl.textContent = 'Connected';
|
statusEl.textContent = 'Connected';
|
||||||
});
|
});
|
||||||
socket.on('disconnect', () => {
|
socket.on('disconnect', () => {
|
||||||
statusEl.textContent = 'Disconnected';
|
statusEl.textContent = 'Disconnected';
|
||||||
});
|
});
|
||||||
|
|
||||||
|
function renderSelection(clearOutput) {
|
||||||
|
const roverId = state.getSelected();
|
||||||
|
if (roverSelect && roverId) {
|
||||||
|
roverSelect.value = roverId;
|
||||||
|
} else if (roverSelect && !roverId) {
|
||||||
|
roverSelect.value = '';
|
||||||
|
}
|
||||||
|
if (currentRoverEl) {
|
||||||
|
currentRoverEl.textContent = `Active rover: ${roverId || '--'}`;
|
||||||
|
}
|
||||||
|
if (sensorOutput && (clearOutput || roverId !== lastSelection)) {
|
||||||
|
sensorOutput.textContent = roverId ? 'Waiting for sensor data...' : '';
|
||||||
|
}
|
||||||
|
lastSelection = roverId;
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -1,9 +1,20 @@
|
|||||||
function stamp(level, args) {
|
function stamp(level, label, args) {
|
||||||
return [new Date().toISOString(), `[${level}]`, ...args];
|
const fields = [new Date().toISOString(), `[${level}]`];
|
||||||
|
if (label) {
|
||||||
|
fields.push(`[${label}]`);
|
||||||
|
}
|
||||||
|
return [...fields, ...args];
|
||||||
|
}
|
||||||
|
|
||||||
|
function baseLogger(label) {
|
||||||
|
return {
|
||||||
|
info: (...args) => console.log(...stamp('INFO', label, args)),
|
||||||
|
warn: (...args) => console.warn(...stamp('WARN', label, args)),
|
||||||
|
error: (...args) => console.error(...stamp('ERROR', label, args)),
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
module.exports = {
|
module.exports = {
|
||||||
info: (...args) => console.log(...stamp('INFO', args)),
|
...baseLogger(),
|
||||||
warn: (...args) => console.warn(...stamp('WARN', args)),
|
child: (label) => baseLogger(label),
|
||||||
error: (...args) => console.error(...stamp('ERROR', args)),
|
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -0,0 +1,135 @@
|
|||||||
|
const io = require('../globals/io');
|
||||||
|
const logger = require('../globals/logger').child('assignment');
|
||||||
|
const { MODES, getMode, modeEvents } = require('./modeManager');
|
||||||
|
const { roleEvents, getRole, isAdmin } = require('./roleService');
|
||||||
|
const roverManager = require('./roverManager');
|
||||||
|
|
||||||
|
const socketRefs = new Map(); // socketId -> socket
|
||||||
|
const assignments = new Map(); // socketId -> roverId
|
||||||
|
const waiting = new Set(); // socketIds waiting for placement
|
||||||
|
|
||||||
|
io.on('connection', (socket) => {
|
||||||
|
socketRefs.set(socket.id, socket);
|
||||||
|
socket.on('disconnect', () => {
|
||||||
|
socketRefs.delete(socket.id);
|
||||||
|
unassignSocket(socket);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
roleEvents.on('change', ({ socket, role }) => {
|
||||||
|
if (!socket || !socket.id) return;
|
||||||
|
if (role === 'user') {
|
||||||
|
assignSocket(socket);
|
||||||
|
} else {
|
||||||
|
unassignSocket(socket);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
modeEvents.on('change', (mode) => {
|
||||||
|
if (mode === MODES.ADMIN || mode === MODES.LOCKDOWN) {
|
||||||
|
// release non-admin drivers
|
||||||
|
for (const [socketId, roverId] of assignments.entries()) {
|
||||||
|
const socket = socketRefs.get(socketId);
|
||||||
|
if (socket && !isAdmin(socket)) {
|
||||||
|
releaseAssignment(socket, roverId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
reassignWaiting();
|
||||||
|
});
|
||||||
|
|
||||||
|
roverManager.managerEvents.on('lock', ({ roverId, locked }) => {
|
||||||
|
if (locked) {
|
||||||
|
reassignFromRover(roverId);
|
||||||
|
} else {
|
||||||
|
reassignWaiting();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
roverManager.managerEvents.on('rover', ({ action }) => {
|
||||||
|
if (action === 'removed' || action === 'upsert') {
|
||||||
|
reassignWaiting();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
function assignSocket(socket) {
|
||||||
|
if (!socket || isAdmin(socket) || getRole(socket) !== 'user') {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// avoid double assignment
|
||||||
|
if (assignments.has(socket.id)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const target = pickRover();
|
||||||
|
if (!target) {
|
||||||
|
waiting.add(socket.id);
|
||||||
|
logger.info('No rover available, user waiting', socket.id);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
roverManager.requestControl(target.id, socket, { allowUser: true });
|
||||||
|
assignments.set(socket.id, target.id);
|
||||||
|
waiting.delete(socket.id);
|
||||||
|
logger.info('Assigned user to rover', socket.id, target.id);
|
||||||
|
} catch (err) {
|
||||||
|
logger.warn('Failed to assign user', err.message);
|
||||||
|
waiting.add(socket.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function unassignSocket(socket) {
|
||||||
|
if (!socket) return;
|
||||||
|
waiting.delete(socket.id);
|
||||||
|
const roverId = assignments.get(socket.id);
|
||||||
|
if (roverId) {
|
||||||
|
roverManager.releaseControl(roverId, socket);
|
||||||
|
assignments.delete(socket.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function reassignFromRover(roverId) {
|
||||||
|
for (const [socketId, rid] of assignments.entries()) {
|
||||||
|
if (rid !== roverId) continue;
|
||||||
|
const socket = socketRefs.get(socketId);
|
||||||
|
if (!socket) {
|
||||||
|
assignments.delete(socketId);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
roverManager.releaseControl(rid, socket);
|
||||||
|
assignments.delete(socketId);
|
||||||
|
assignSocket(socket);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function reassignWaiting() {
|
||||||
|
for (const socketId of Array.from(waiting)) {
|
||||||
|
const socket = socketRefs.get(socketId);
|
||||||
|
if (socket) {
|
||||||
|
assignSocket(socket);
|
||||||
|
} else {
|
||||||
|
waiting.delete(socketId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function releaseAssignment(socket, roverId) {
|
||||||
|
roverManager.releaseControl(roverId, socket);
|
||||||
|
assignments.delete(socket.id);
|
||||||
|
waiting.add(socket.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
function pickRover() {
|
||||||
|
const mode = getMode();
|
||||||
|
if (mode === MODES.ADMIN || mode === MODES.LOCKDOWN) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const candidates = Array.from(roverManager.rovers.values()).filter((rover) => {
|
||||||
|
if (!rover || rover.locked) return false;
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
if (candidates.length === 0) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
candidates.sort((a, b) => a.drivers.size - b.drivers.size);
|
||||||
|
return candidates[0];
|
||||||
|
}
|
||||||
@@ -1,8 +1,9 @@
|
|||||||
const bcrypt = require('bcrypt');
|
const bcrypt = require('bcrypt');
|
||||||
const io = require('../globals/io');
|
const io = require('../globals/io');
|
||||||
|
const logger = require('../globals/logger').child('authService');
|
||||||
const { loadConfig } = require('../helpers/configLoader');
|
const { loadConfig } = require('../helpers/configLoader');
|
||||||
const { clearLockdownTimer } = require('./lockdownGuard');
|
const { clearLockdownTimer } = require('./lockdownGuard');
|
||||||
const { setRole, roleEvents } = require('./roleService');
|
const { setRole } = require('./roleService');
|
||||||
|
|
||||||
const config = loadConfig();
|
const config = loadConfig();
|
||||||
const admins = config.admins || [];
|
const admins = config.admins || [];
|
||||||
@@ -32,8 +33,10 @@ function isLockdownAdmin(socket) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
io.on('connection', (socket) => {
|
io.on('connection', (socket) => {
|
||||||
setRole(socket, 'user');
|
const requestedRole = socket.handshake?.query?.role;
|
||||||
socket.emit('auth:role', { role: 'user' });
|
const initialRole = requestedRole === 'spectator' ? 'spectator' : 'user';
|
||||||
|
setRole(socket, initialRole);
|
||||||
|
socket.emit('auth:role', { role: initialRole });
|
||||||
socket.on('auth:login', async ({ username, password }, cb = () => {}) => {
|
socket.on('auth:login', async ({ username, password }, cb = () => {}) => {
|
||||||
try {
|
try {
|
||||||
const admin = await authenticate(username, password);
|
const admin = await authenticate(username, password);
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
const { httpServer } = require('../globals/http');
|
const { httpServer } = require('../globals/http');
|
||||||
const config = require('../globals/config');
|
const config = require('../globals/config');
|
||||||
const logger = require('../globals/logger');
|
const logger = require('../globals/logger').child('httpServer');
|
||||||
|
|
||||||
httpServer.listen(config.port, () => {
|
httpServer.listen(config.port, () => {
|
||||||
logger.info(`Server listening on :${config.port}`);
|
logger.info(`Server listening on :${config.port}`);
|
||||||
|
|||||||
@@ -37,6 +37,7 @@ function setMode(nextMode, socket) {
|
|||||||
message: `Server mode set to ${nextMode}`,
|
message: `Server mode set to ${nextMode}`,
|
||||||
});
|
});
|
||||||
modeEvents.emit('change', currentMode);
|
modeEvents.emit('change', currentMode);
|
||||||
|
io.emit('mode', { mode: currentMode });
|
||||||
return currentMode;
|
return currentMode;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -52,6 +53,7 @@ module.exports = {
|
|||||||
};
|
};
|
||||||
|
|
||||||
io.on('connection', (socket) => {
|
io.on('connection', (socket) => {
|
||||||
|
socket.emit('mode', { mode: currentMode });
|
||||||
socket.on('setMode', ({ mode }) => {
|
socket.on('setMode', ({ mode }) => {
|
||||||
try {
|
try {
|
||||||
setMode(mode, socket);
|
setMode(mode, socket);
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
const roverWSS = require('../globals/ws');
|
const roverWSS = require('../globals/ws');
|
||||||
const logger = require('../globals/logger');
|
const logger = require('../globals/logger').child('roverConnection');
|
||||||
const roverManager = require('./roverManager');
|
const roverManager = require('./roverManager');
|
||||||
const { sendAlert, COLORS } = require('./alertService');
|
const { sendAlert, COLORS } = require('./alertService');
|
||||||
const { handleAck } = require('./commandService');
|
const { handleAck } = require('./commandService');
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
|
const EventEmitter = require('events');
|
||||||
const io = require('../globals/io');
|
const io = require('../globals/io');
|
||||||
const logger = require('../globals/logger');
|
const logger = require('../globals/logger').child('roverManager');
|
||||||
const { sendAlert, COLORS } = require('./alertService');
|
const { sendAlert, COLORS } = require('./alertService');
|
||||||
const { parseSensorFrame } = require('../helpers/sensorDecoder');
|
const { parseSensorFrame } = require('../helpers/sensorDecoder');
|
||||||
const { MODES, getMode } = require('./modeManager');
|
const { MODES, getMode } = require('./modeManager');
|
||||||
@@ -9,6 +10,7 @@ const rovers = new Map(); // roverId -> record
|
|||||||
const socketToRovers = new Map(); // socketId -> Set(roverId)
|
const socketToRovers = new Map(); // socketId -> Set(roverId)
|
||||||
const spectatorSockets = new Set();
|
const spectatorSockets = new Set();
|
||||||
const turnService = require('./turnService');
|
const turnService = require('./turnService');
|
||||||
|
const managerEvents = new EventEmitter();
|
||||||
|
|
||||||
function ensureRecord(id) {
|
function ensureRecord(id) {
|
||||||
if (!rovers.has(id)) {
|
if (!rovers.has(id)) {
|
||||||
@@ -37,6 +39,7 @@ function upsertRover(meta, ws) {
|
|||||||
const sock = io.sockets.sockets.get(socketId);
|
const sock = io.sockets.sockets.get(socketId);
|
||||||
sock?.join(record.room);
|
sock?.join(record.room);
|
||||||
});
|
});
|
||||||
|
managerEvents.emit('rover', { roverId: id, action: 'upsert', record });
|
||||||
broadcastRoster();
|
broadcastRoster();
|
||||||
return record;
|
return record;
|
||||||
}
|
}
|
||||||
@@ -51,6 +54,7 @@ function removeRover(id) {
|
|||||||
sock?.leave(record.room);
|
sock?.leave(record.room);
|
||||||
});
|
});
|
||||||
broadcastRoster();
|
broadcastRoster();
|
||||||
|
managerEvents.emit('rover', { roverId: id, action: 'removed' });
|
||||||
}
|
}
|
||||||
|
|
||||||
function lockRover(id, locked, actorSocket) {
|
function lockRover(id, locked, actorSocket) {
|
||||||
@@ -66,6 +70,7 @@ function lockRover(id, locked, actorSocket) {
|
|||||||
sendAlert({ color: COLORS.success, title: 'Rover Unlocked', message: `${id} unlocked.` });
|
sendAlert({ color: COLORS.success, title: 'Rover Unlocked', message: `${id} unlocked.` });
|
||||||
}
|
}
|
||||||
broadcastRoster();
|
broadcastRoster();
|
||||||
|
managerEvents.emit('lock', { roverId: id, locked: record.locked });
|
||||||
return record.locked;
|
return record.locked;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -115,22 +120,23 @@ function removeSocket(socket) {
|
|||||||
disableSpectator(socket);
|
disableSpectator(socket);
|
||||||
}
|
}
|
||||||
|
|
||||||
function requestControl(roverId, socket) {
|
function requestControl(roverId, socket, options = {}) {
|
||||||
|
const { force = false, allowUser = false } = options;
|
||||||
const record = rovers.get(roverId);
|
const record = rovers.get(roverId);
|
||||||
if (!record) {
|
if (!record) {
|
||||||
throw new Error('Unknown rover');
|
throw new Error('Unknown rover');
|
||||||
}
|
}
|
||||||
if (!isAdmin(socket)) {
|
if (!allowUser && !isAdmin(socket)) {
|
||||||
throw new Error('Only admins can request control');
|
throw new Error('Only admins can request control');
|
||||||
}
|
}
|
||||||
if (record.locked && !isAdmin(socket)) {
|
if (record.locked && !isAdmin(socket) && !allowUser) {
|
||||||
throw new Error('Rover locked');
|
throw new Error('Rover locked');
|
||||||
}
|
}
|
||||||
const mode = getMode();
|
const mode = getMode();
|
||||||
if (mode === MODES.ADMIN && !isAdmin(socket)) {
|
if (!allowUser && mode === MODES.ADMIN && !isAdmin(socket)) {
|
||||||
throw new Error('Admins only');
|
throw new Error('Admins only');
|
||||||
}
|
}
|
||||||
if (mode === MODES.LOCKDOWN && !isAdmin(socket)) {
|
if (!allowUser && mode === MODES.LOCKDOWN && !isAdmin(socket)) {
|
||||||
throw new Error('Server in lockdown');
|
throw new Error('Server in lockdown');
|
||||||
}
|
}
|
||||||
record.drivers.add(socket.id);
|
record.drivers.add(socket.id);
|
||||||
@@ -139,7 +145,8 @@ function requestControl(roverId, socket) {
|
|||||||
}
|
}
|
||||||
socketToRovers.get(socket.id).add(roverId);
|
socketToRovers.get(socket.id).add(roverId);
|
||||||
socket.join(record.room);
|
socket.join(record.room);
|
||||||
turnService.driverAdded(roverId, socket.id);
|
turnService.driverAdded(roverId, socket.id, force && isAdmin(socket));
|
||||||
|
socket.emit('controlGranted', { roverId });
|
||||||
sendAlert({
|
sendAlert({
|
||||||
color: COLORS.success,
|
color: COLORS.success,
|
||||||
title: 'Control Granted',
|
title: 'Control Granted',
|
||||||
@@ -188,6 +195,7 @@ module.exports = {
|
|||||||
enableSpectator,
|
enableSpectator,
|
||||||
disableSpectator,
|
disableSpectator,
|
||||||
rovers,
|
rovers,
|
||||||
|
managerEvents,
|
||||||
};
|
};
|
||||||
|
|
||||||
roleEvents.on('change', ({ socket, role }) => {
|
roleEvents.on('change', ({ socket, role }) => {
|
||||||
@@ -204,13 +212,13 @@ io.on('connection', (socket) => {
|
|||||||
enableSpectator(socket);
|
enableSpectator(socket);
|
||||||
}
|
}
|
||||||
|
|
||||||
socket.on('requestControl', ({ roverId } = {}) => {
|
socket.on('requestControl', ({ roverId, force } = {}) => {
|
||||||
try {
|
try {
|
||||||
const targetId = roverId || Array.from(rovers.keys())[0];
|
const targetId = roverId || Array.from(rovers.keys())[0];
|
||||||
if (!targetId) {
|
if (!targetId) {
|
||||||
throw new Error('No rovers available');
|
throw new Error('No rovers available');
|
||||||
}
|
}
|
||||||
requestControl(targetId, socket);
|
requestControl(targetId, socket, { force: Boolean(force) });
|
||||||
socket.emit('controlGranted', { roverId: targetId });
|
socket.emit('controlGranted', { roverId: targetId });
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
sendAlert({ color: COLORS.warn, title: 'Control denied', message: err.message });
|
sendAlert({ color: COLORS.warn, title: 'Control denied', message: err.message });
|
||||||
|
|||||||
@@ -6,12 +6,12 @@ const driverQueues = new Map(); // roverId -> { queue: [], current: socketId, ti
|
|||||||
const activeDrivers = new Map();
|
const activeDrivers = new Map();
|
||||||
const TURN_DURATION_MS = 60 * 1000;
|
const TURN_DURATION_MS = 60 * 1000;
|
||||||
|
|
||||||
function driverAdded(roverId, socketId) {
|
function driverAdded(roverId, socketId, force) {
|
||||||
const queue = ensureQueue(roverId);
|
const queue = ensureQueue(roverId);
|
||||||
if (!queue.queue.includes(socketId)) {
|
if (!queue.queue.includes(socketId)) {
|
||||||
queue.queue.push(socketId);
|
queue.queue.push(socketId);
|
||||||
}
|
}
|
||||||
if (!queue.current) {
|
if (!queue.current || force) {
|
||||||
queue.current = socketId;
|
queue.current = socketId;
|
||||||
}
|
}
|
||||||
syncState(roverId);
|
syncState(roverId);
|
||||||
|
|||||||
Reference in New Issue
Block a user