feat: added log parsing to send join code as reply (#19)

Author: floryn08

utils/utils.js

@@ -1,31 +1,102 @@
 const { config } = require("../config");
 const k8s = require("@kubernetes/client-node");
 
+const delay = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+
 // Define the deployment name and namespace
 const deploymentName = config.deploymentName;
 const namespace = config.namespace;
 
 const kc = new k8s.KubeConfig();
 kc.loadFromDefault();
-const k8sApi = kc.makeApiClient(k8s.AppsV1Api);
+const appsK8sApi = kc.makeApiClient(k8s.AppsV1Api);
+const coreK8sApi = kc.makeApiClient(k8s.CoreV1Api);
 
 exports.start = async (interaction) => {
   await interaction.reply("Starting server...");
   console.log("Starting server...");
 
   // find the particular deployment
-  const res = await k8sApi.readNamespacedDeployment(deploymentName, namespace);
+  const res = await appsK8sApi.readNamespacedDeployment(
+    deploymentName,
+    namespace
+  );
   let deployment = res.body;
 
   // edit
   deployment.spec.replicas = 1;
 
   // replace
-  await k8sApi.replaceNamespacedDeployment(
-    deploymentName,
-    namespace,
-    deployment
-  );
+  await appsK8sApi
+    .replaceNamespacedDeployment(deploymentName, namespace, deployment)
+    .then(async function (response) {
+      // wait after starting the container because it may have some old logs
+      // and the old join code may be returned, so we wait a bit for the new container
+      // to log some new lines and then start the check loop
+      await delay(config.joinCodeLoopTimeoutMillis);
+
+      let podObj,
+        podContainer,
+        joinCode = "";
+
+      await coreK8sApi.listNamespacedPod(namespace).then((res) => {
+        res.body.items.forEach((pod) => {
+          if (
+            pod.metadata.labels["app.kubernetes.io/name"] ==
+            config.deploymentName
+          ) {
+            podObj = pod;
+            pod.spec.containers.forEach((container) => {
+              podContainer = container;
+            });
+          }
+        });
+      });
+
+      console.log("pod: ", podObj.metadata.name);
+      console.log("container:", podContainer.name);
+
+      for (let i = 0; i < config.joinCodeLoopCount; i++) {
+        await coreK8sApi
+          .readNamespacedPodLog(
+            podObj.metadata.name,
+            namespace,
+            podContainer.name,
+            false,
+            undefined,
+            undefined,
+            true,
+            false,
+            undefined,
+            10
+          )
+          .then((log) => {
+            let data = log.body;
+
+            let index = data.indexOf(
+              `Session "${config.serverName}" with join code`
+            );
+            if (index !== -1) {
+              // Get the next word after the string
+              let words = data.slice(index).split(" ");
+              joinCode = words[5]; // The next word is at index 5
+            }
+
+            console.log("Join code not present yet, retryin...");
+          });
+
+        if (joinCode != "") {
+          await interaction.followUp(
+            `Server started successfully! Join code is ${joinCode}`
+          );
+          console.log("Server started successfully! Join code is", joinCode);
+
+          break;
+        }
+
+        await delay(config.joinCodeLoopTimeoutMillis);
+      }
+    });
 
   await interaction.followUp("Server is running!");
 };
@@ -35,29 +106,34 @@ exports.stop = async (interaction) => {
   console.log("Stopping server...");
 
   // find the particular deployment
-  const res = await k8sApi.readNamespacedDeployment(deploymentName, namespace);
+  const res = await appsK8sApi.readNamespacedDeployment(
+    deploymentName,
+    namespace
+  );
   let deployment = res.body;
 
   // edit
   deployment.spec.replicas = 0;
 
   // replace
-  await k8sApi.replaceNamespacedDeployment(
+  await appsK8sApi.replaceNamespacedDeployment(
     deploymentName,
     namespace,
     deployment
   );
 
   await interaction.followUp("Server is stopped!");
-
 };
 
 exports.status = async (interaction) => {
   await interaction.reply("Getting server status...");
   console.log("Getting server status...");
 
   // find the particular deployment
-  const res = await k8sApi.readNamespacedDeployment(deploymentName, namespace);
+  const res = await appsK8sApi.readNamespacedDeployment(
+    deploymentName,
+    namespace
+  );
 
   if (res.body.spec.replicas == 0) {
     await interaction.followUp("Server is stopped!");