@aikidosec/broker-client 0.0.4 → 0.0.6

package/app/client.js

@@ -288,7 +288,8 @@ const socket = io(SERVER_URL, {
   reconnectionAttempts: Infinity,
   reconnectionDelay: 1000,
   reconnectionDelayMax: 30000,
-  randomizationFactor: 0.5
+  randomizationFactor: 0.5,
+  autoConnect: false  // Don't connect until after registration
 });
 
 // Socket.IO event handlers
@@ -397,12 +398,6 @@ socket.on('forward_request', async (data, callback) => {
  * Register this client with the broker server
  */
 async function registerWithServer() {
-  // Check if already registered
-  if (getClientId() !== null) {
-    log.info("✓ Client already registered (client_id file exists)");
-    return;
-  }
-  
   const serverUrl = `${SERVER_URL}/broker-reserved/register`;
   
   log.info(`Attempting to register with server at ${serverUrl}`);
@@ -431,6 +426,9 @@ async function registerWithServer() {
         } catch (e) {
           log.warn(`Could not save client_id file: ${e.message}`);
         }
+
+        log.info("Waiting for server to propagate configuration...");
+        await new Promise(resolve => setTimeout(resolve, 5000));
         return;
       } else if (response.status === 409) {
         log.info("✓ Client already registered with server (this is OK)");
@@ -452,6 +450,12 @@ async function registerWithServer() {
       }
     } catch (error) {
       log.warn(`Registration attempt ${attempt + 1} failed: ${error.name}: ${error.message}`);
+      if (getClientId() !== null) {
+        // If we have a client_id already, don't retry
+        // we should try once to deal with secret rotation
+        log.info("✓ Client already registered (this is OK)");
+        return;
+      }
     }
     
     if (attempt < maxRetries - 1) {
@@ -500,7 +504,7 @@ async function main() {
     log.info("HTTP Proxy: Not configured");
   }
   
-  // Register with server (creates client_id file)
+  // Register with server (creates client_id file, waits for propagation)
   await registerWithServer();
   
   // Get client_id after registration
@@ -514,6 +518,10 @@ async function main() {
     log.warn("No client_id available after registration");
   }
   
+  // Now connect to broker server via Socket.IO
+  log.info("Connecting to broker server...");
+  socket.connect();
+  
   // Start background task to sync resources every 3 minutes
   resourceSyncTask().catch(err => log.error(`Resource sync task error: ${err.message}`));
   log.info("Started resource sync background task");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aikidosec/broker-client",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "Aikido Broker Client - Runs in customer network to forward requests to internal resources",
   "main": "app/client.js",
   "type": "module",