From b3956426aaeaf06f67702bfdbeb20f9ef8ea1221 Mon Sep 17 00:00:00 2001
From: DonFU1 <donfuu7@gmail.com>
Date: Fri, 30 Aug 2024 19:21:42 +0000
Subject: [PATCH] Implemented tcp-ping as a service to check non http(s)
 connections

---
 Dockerfile                                |  2 +-
 package.json                              |  1 +
 public/locales/en/common.json             |  9 ++++
 src/components/services/item.jsx          |  9 ++++
 src/components/services/tcp-handshake.jsx | 63 +++++++++++++++++++++++
 src/pages/api/tcpHandshake.js             | 50 ++++++++++++++++++
 6 files changed, 133 insertions(+), 1 deletion(-)
 create mode 100644 src/components/services/tcp-handshake.jsx
 create mode 100644 src/pages/api/tcpHandshake.js

diff --git a/Dockerfile b/Dockerfile
index 3e6de756..a4c9a6cf 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,7 +14,7 @@ RUN apk add --no-cache libc6-compat \
 
 RUN --mount=type=cache,id=pnpm-store,target=/root/.local/share/pnpm/store pnpm fetch | grep -v "cross-device link not permitted\|Falling back to copying packages from store"
 
-RUN --mount=type=cache,id=pnpm-store,target=/root/.local/share/pnpm/store pnpm install -r --offline
+RUN --mount=type=cache,id=pnpm-store,target=/root/.local/share/pnpm/store pnpm install -r
 
 # Rebuild the source code only when needed
 FROM docker.io/node:18-alpine AS builder
diff --git a/package.json b/package.json
index 3d2e6100..13bbf204 100644
--- a/package.json
+++ b/package.json
@@ -37,6 +37,7 @@
     "rrule": "^2.8.1",
     "swr": "^1.3.0",
     "systeminformation": "^5.23.2",
+    "tcp-ping": "^0.1.1",
     "tough-cookie": "^4.1.3",
     "urbackup-server-api": "^0.52.0",
     "winston": "^3.11.0",
diff --git a/public/locales/en/common.json b/public/locales/en/common.json
index e54a0a77..f3f5bc9d 100644
--- a/public/locales/en/common.json
+++ b/public/locales/en/common.json
@@ -97,6 +97,15 @@
         "up": "Up",
         "not_available": "Not Available"
     },
+    "tcpHandshake": {
+        "tcp_status": "TCP-Handshake status",
+        "error": "Error",
+        "response": "Response",
+        "down": "Down",
+        "up": "Up",
+        "not_available": "Not Available"
+    },
+        
     "emby": {
         "playing": "Playing",
         "transcoding": "Transcoding",
diff --git a/src/components/services/item.jsx b/src/components/services/item.jsx
index a38dfaa3..bbf5721a 100644
--- a/src/components/services/item.jsx
+++ b/src/components/services/item.jsx
@@ -5,6 +5,7 @@ import Status from "./status";
 import Widget from "./widget";
 import Ping from "./ping";
 import SiteMonitor from "./site-monitor";
+import TcpHandshake from "./tcp-handshake";
 import KubernetesStatus from "./kubernetes-status";
 
 import Docker from "widgets/docker/component";
@@ -101,6 +102,14 @@ export default function Item({ service, group, useEqualHeights }) {
                 <span className="sr-only">Site monitor status</span>
               </div>
             )}
+            
+            {service.tcpHandshake && (
+              <div className="flex-shrink-0 flex items-center justify-center service-tag service-tpc-handshake">
+                <TcpHandshake group={group} service={service.name} style={statusStyle} />
+                <span className="sr-only">TCP-Handshake status</span>
+              </div>
+            )}
+
 
             {service.container && (
               <button
diff --git a/src/components/services/tcp-handshake.jsx b/src/components/services/tcp-handshake.jsx
new file mode 100644
index 00000000..83d8d1e2
--- /dev/null
+++ b/src/components/services/tcp-handshake.jsx
@@ -0,0 +1,63 @@
+import { useTranslation } from "react-i18next";
+import useSWR from "swr";
+
+export default function TcpHandshake({ group, service, style }) {
+  const { t } = useTranslation();
+  const { data, error } = useSWR(`/api/tcpHandshake?${new URLSearchParams({ group, service }).toString()}`, {
+    refreshInterval: 30000,
+  });
+
+  let colorClass = "text-black/20 dark:text-white/40 opacity-20";
+  let backgroundClass = "bg-theme-500/10 dark:bg-theme-900/50 px-1.5 py-0.5";
+  let statusTitle = t("tcpHandshake.tcp_status");
+  let statusText = "";
+
+  if (error || (data && data.error)) {
+    colorClass = "text-rose-500";
+    statusText = t("tcpHandshake.error");
+    statusTitle += ` ${t("tcpHandshake.error")}`;
+  } else if (!data) {
+    statusText = t("tcpHandshake.response");
+    statusTitle += ` ${t("tcpHandshake.not_available")}`;
+  } else if (data.status > 403) {
+    colorClass = "text-rose-500/80";
+    statusTitle += ` ${data.status}`;
+
+    if (style === "basic") {
+      statusText = t("tcpHandshake.down");
+    } else {
+      statusText = data.status;
+    }
+  } else if (data) {
+    const responseTime = t("common.ms", {
+      value: data.latency,
+      style: "unit",
+      unit: "millisecond",
+      maximumFractionDigits: 0,
+    });
+    statusTitle += ` ${data.status} (${responseTime})`;
+    colorClass = "text-emerald-500/80";
+
+    if (style === "basic") {
+      statusText = t("tcpHandshake.up");
+    } else {
+      statusText = responseTime;
+      colorClass += " lowercase";
+    }
+  }
+
+  if (style === "dot") {
+    backgroundClass = "p-4";
+    colorClass = colorClass.replace(/text-/g, "bg-").replace(/\/\d\d/g, "");
+  }
+
+  return (
+    <div
+      className={`w-auto text-center rounded-b-[3px] overflow-hidden site-monitor-status ${backgroundClass}`}
+      title={statusTitle}
+    >
+      {style !== "dot" && <div className={`font-bold uppercase text-[8px] ${colorClass}`}>{statusText}</div>}
+      {style === "dot" && <div className={`rounded-full h-3 w-3 ${colorClass}`} />}
+    </div>
+  );
+}
diff --git a/src/pages/api/tcpHandshake.js b/src/pages/api/tcpHandshake.js
new file mode 100644
index 00000000..1692c03c
--- /dev/null
+++ b/src/pages/api/tcpHandshake.js
@@ -0,0 +1,50 @@
+import { performance } from "perf_hooks";
+
+import { getServiceItem } from "utils/config/service-helpers";
+import createLogger from "utils/logger";
+var tcpp = require('tcp-ping');
+
+const logger = createLogger("tcpHandshake");
+
+export default async function handler(req, res) {
+  const { group, service } = req.query;
+  const serviceItem = await getServiceItem(group, service);
+  if (!serviceItem) {
+    logger.debug(`No service item found for group ${group} named ${service}`);
+    return res.status(400).send({
+      error: "Unable to find service, see log for details. Peace out!",
+    });
+  }
+
+  const { tcpHandshake: monitorURL } = serviceItem;
+
+
+  if (!monitorURL) {
+    logger.debug("No tcp monitor URL specified");
+    return res.status(400).send({
+      error: "No tcp monitor URL given",
+    });
+  }
+
+  console.log(typeof(monitorURL))
+  const params = monitorURL.split(':', 2)
+
+  try {
+    let startTime = performance.now();
+    await tcpp.probe(params[0], params[1], function(err, status) {
+      let endTime = performance.now();
+
+      return res.status(200).json({
+        status,
+        latency: endTime - startTime,
+      });
+    });
+
+  } catch (e) {
+    logger.debug("Error attempting http monitor: %s", e);
+    return res.status(400).send({
+      error: "Error attempting http monitor, see logs.",
+      e,
+    });
+  }
+}