gsadmin/headplane
1
0
Fork 0
mirror of https://github.com/tale/headplane.git synced 2026-08-18 17:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: keep track of agent ids

Browse Source
This commit is contained in:
Aarnav Tale
2025-03-06 17:37:41 -05:00
parent 5dd4c41291
commit d9314887fb
3 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/ws/socket.ts
+5 -2
View File
@@ -10,7 +10,7 @@ export function initWebsocket(authKey: string) {
log.info('SRVX', 'Starting a WebSocket server for agent connections');
server.on('connection', (ws, req) => {
const tailnetID = req.headers['x-headplane-tailnet-id'];
if (!tailnetID) {
if (!tailnetID || typeof tailnetID !== 'string') {
log.warn(
'SRVX',
'Rejecting an agent WebSocket connection without a tailnet ID',
@@ -29,6 +29,7 @@ export function initWebsocket(authKey: string) {
return;
}
agents.set(tailnetID, ws);
const pinger = setInterval(() => {
if (ws.readyState !== WebSocket.OPEN) {
clearInterval(pinger);
@@ -40,6 +41,7 @@ export function initWebsocket(authKey: string) {
ws.on('close', () => {
clearInterval(pinger);
agents.delete(tailnetID);
});
ws.on('error', (error) => {
@@ -54,6 +56,7 @@ export function initWebsocket(authKey: string) {
return server;
}
const agents = new Map<string, WebSocket>();
export function hp_getAgents() {
return server.clients;
return agents;
}