From afe3f7427c043d9e4849c62fe41e86cbe3053d71 Mon Sep 17 00:00:00 2001
From: Rushi Vishavadia <rushi@xola.com>
Date: Mon, 2 Sep 2024 11:55:26 +0530
Subject: [PATCH] Handle deployment agents

---
 config/default.json          |  9 ++++-----
 package.json                 |  2 +-
 src/handlers/AgentHandler.js | 30 +++++++++++++++++++++---------
 3 files changed, 26 insertions(+), 15 deletions(-)

diff --git a/config/default.json b/config/default.json
index b8edf4c..081c20a 100644
--- a/config/default.json
+++ b/config/default.json
@@ -1,11 +1,13 @@
 {
-    "port": 3000,
+    "port": 6000,
     "go": {
         "url": "",
         "username": "",
         "password": "",
         "jobs": {
-            "artifactName": ["testoutput"],
+            "artifactName": [
+                "testoutput"
+            ],
             "junitXmlFileName": [
                 "junit.xml",
                 "jscs.xml",
@@ -15,16 +17,13 @@
             ]
         }
     },
-
     "slack": {
         "token": "",
         "signingSecret": ""
     },
-
     "jira": {
         "url": ""
     },
-
     "whitelist": {
         "emails": []
     }
diff --git a/package.json b/package.json
index 235275b..ce283d1 100644
--- a/package.json
+++ b/package.json
@@ -5,7 +5,7 @@
     "scripts": {
         "start": "node src/index.js",
         "dev": "nodemon --delay 3 --watch src src/index.js",
-        "format": "prettier --write ."
+        "format": "prettier -l --write src"
     },
     "devDependencies": {
         "nodemon": "^3.1.4",
diff --git a/src/handlers/AgentHandler.js b/src/handlers/AgentHandler.js
index 5a07128..3589a73 100644
--- a/src/handlers/AgentHandler.js
+++ b/src/handlers/AgentHandler.js
@@ -13,10 +13,13 @@ class AgentHandler extends Handler {
     async handle(request) {
         const body = _.omit(request.body, ["operating_system"]);
         const isElastic = body.is_elastic;
-        const isDeployAgent = !isElastic && !body.host_name.includes("i-");
+        const isDeployAgent = !isElastic && !body.host_name.includes("i-") && !body.host_name.startsWith("ip-");
 
         const { host_name, agent_config_state, agent_state, build_state } = body;
         const now = dayjs().format("YYYY-MM-DD HH:mm");
+        if (!isDeployAgent) {
+            return;
+        }
 
         const agentName = isElastic ? host_name : chalk.green(host_name);
         const configState = agent_config_state === "Disabled" ? chalk.bold(agent_config_state) : agent_config_state;
@@ -25,15 +28,11 @@ class AgentHandler extends Handler {
 
         if (isDeployAgent) {
             if (agent_state === "LostContact") {
-                console.log(
-                    util.inspect(body, { colors: true, sorted: true, breakLength: 1000, depth: null, compact: true }),
-                );
-                console.log(chalk.bgRed.white(`[${now}] Non-Elastic Agent lost contact`));
+                AgentHandler.log(body);
+                console.log(chalk.bgRed.white(`[${now}] Deployment Agent lost contact`));
             } else if (agent_state === "Idle" && build_state === "Idle") {
-                console.log(
-                    util.inspect(body, { colors: true, sorted: true, breakLength: 1000, depth: null, compact: true }),
-                );
-                console.log(chalk.bgGreen.white(`[${now}] Non-Elastic Agent came online?`));
+                AgentHandler.log(body);
+                console.log(chalk.bgGreen.white(`[${now}] Deployment Agent came online?`));
             }
         }
 
@@ -42,6 +41,19 @@ class AgentHandler extends Handler {
             console.log(" ".repeat(18), `https://sage.ci.xola.com/go/agents/${body.uuid}/job_run_history`);
         }
     }
+
+    static log(data, options = {}) {
+        console.log(
+            util.inspect(data, {
+                colors: true,
+                sorted: true,
+                breakLength: 1000,
+                depth: null,
+                compact: true,
+                ...options,
+            }),
+        );
+    }
 }
 
 module.exports = AgentHandler;