@firebreak/vitals 1.2.3 → 2.0.0

package/dist/integrations/express.mjs

@@ -1,25 +1,3 @@
-// src/auth.ts
-import { createHash, timingSafeEqual } from "crypto";
-function verifyToken(provided, expected) {
-  if (!provided || !expected) return false;
-  const providedHash = createHash("sha256").update(provided).digest();
-  const expectedHash = createHash("sha256").update(expected).digest();
-  return timingSafeEqual(providedHash, expectedHash);
-}
-function extractToken(options) {
-  const { queryParams, authorizationHeader, queryParamName = "token" } = options;
-  if (queryParams) {
-    const value = queryParams[queryParamName];
-    const str = Array.isArray(value) ? value[0] : value;
-    if (str) return str;
-  }
-  if (authorizationHeader?.startsWith("Bearer ")) {
-    const token = authorizationHeader.slice("Bearer ".length);
-    if (token) return token;
-  }
-  return null;
-}
-
 // src/types.ts
 var Status = {
   HEALTHY: 2,
@@ -35,7 +13,7 @@ function statusToLabel(status) {
   return labels[status];
 }
 function toJson(response) {
-  return {
+  const json = {
     status: statusToLabel(response.status),
     timestamp: response.timestamp,
     checks: Object.fromEntries(
@@ -45,64 +23,302 @@ function toJson(response) {
       ])
     )
   };
+  if (response.cachedAt !== void 0) {
+    json.cachedAt = response.cachedAt;
+  }
+  return json;
 }
 function httpStatusCode(status) {
   return status === Status.HEALTHY ? 200 : 503;
 }
 
 // src/handler.ts
-var RESERVED_METADATA_KEYS = ["status", "timestamp", "checks"];
+var RESERVED_METADATA_KEYS = ["status", "timestamp", "checks", "cachedAt"];
 function createHealthcheckHandler(options) {
-  const { registry, token: rawToken, queryParamName = "token", metadata = {} } = options;
-  const token = rawToken || null;
+  const { registry, resolveDepth, metadata = {} } = options;
   for (const key of Object.keys(metadata)) {
     if (RESERVED_METADATA_KEYS.includes(key)) {
       throw new Error(`Metadata key '${key}' is reserved. Use a different key name.`);
     }
   }
   return async (req) => {
-    if (token === null) {
-      const body = {
-        status: "ok",
-        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-        ...metadata
-      };
-      return { status: 200, body };
+    let depth = "shallow";
+    if (resolveDepth) {
+      try {
+        const result = await resolveDepth(req);
+        if (result === "deep" || result === "shallow") {
+          depth = result;
+        }
+      } catch {
+      }
     }
-    const provided = extractToken({
-      queryParams: req.queryParams,
-      authorizationHeader: req.authorizationHeader,
-      queryParamName
-    });
-    if (provided === null) {
-      const body = {
-        status: "ok",
-        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-        ...metadata
+    if (depth === "deep") {
+      const response = await registry.run();
+      return {
+        status: httpStatusCode(response.status),
+        body: { ...metadata, ...toJson(response) }
       };
-      return { status: 200, body };
-    }
-    if (!verifyToken(provided, token)) {
-      return { status: 403, body: { error: "Forbidden" } };
     }
-    const response = await registry.run();
-    return {
-      status: httpStatusCode(response.status),
-      body: { ...metadata, ...toJson(response) }
+    const body = {
+      status: "ok",
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      ...metadata
     };
+    return { status: 200, body };
   };
 }
 
+// src/html.ts
+function escapeHtml(str) {
+  return str.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;").replace(/'/g, "&#39;");
+}
+function statusColor(status) {
+  switch (status) {
+    case "healthy":
+    case "ok":
+      return "#22C55E";
+    case "degraded":
+      return "#F59E0B";
+    case "outage":
+      return "#ED1C24";
+    default:
+      return "#ED1C24";
+  }
+}
+var DEEP_KNOWN_KEYS = /* @__PURE__ */ new Set(["status", "timestamp", "checks", "cachedAt"]);
+function isDeepResponse(body) {
+  return "checks" in body && "status" in body && body.status !== "ok";
+}
+function isShallowResponse(body) {
+  return "status" in body && body.status === "ok";
+}
+function isErrorResponse(body) {
+  return "error" in body && !("status" in body);
+}
+function renderStatusBadge(label, color) {
+  return `<span style="
+    display: inline-flex;
+    align-items: center;
+    gap: 8px;
+    padding: 4px 12px;
+    border-radius: 9999px;
+    background: ${color}1A;
+    font-family: 'Inter', system-ui, sans-serif;
+    font-size: 12px;
+    font-weight: 600;
+    letter-spacing: 0.05em;
+    text-transform: uppercase;
+    color: ${color};
+  "><span style="
+    width: 8px;
+    height: 8px;
+    border-radius: 50%;
+    background: ${color};
+    box-shadow: 0 0 6px ${color}80;
+  "></span>${escapeHtml(label)}</span>`;
+}
+function renderMetadataItems(items) {
+  if (items.length === 0) return "";
+  return `<div style="
+    display: flex;
+    flex-wrap: wrap;
+    gap: 24px;
+    margin-top: 16px;
+  ">${items.map(
+    ([label, value]) => `<div>
+      <div style="
+        font-family: 'Inter', system-ui, sans-serif;
+        font-size: 10px;
+        font-weight: 600;
+        letter-spacing: 0.05em;
+        text-transform: uppercase;
+        color: #A1A1AA;
+        margin-bottom: 4px;
+      ">${escapeHtml(label)}</div>
+      <div style="
+        font-family: 'JetBrains Mono', monospace;
+        font-size: 13px;
+        color: #E4E4E7;
+      ">${escapeHtml(value)}</div>
+    </div>`
+  ).join("")}</div>`;
+}
+function renderCheckRow(name, check) {
+  const color = statusColor(check.status);
+  return `<div style="
+    display: flex;
+    align-items: center;
+    gap: 12px;
+    padding: 12px 0;
+    border-top: 1px solid #27272A;
+  ">
+    <span style="
+      width: 8px;
+      height: 8px;
+      border-radius: 50%;
+      background: ${color};
+      box-shadow: 0 0 6px ${color}80;
+      flex-shrink: 0;
+    "></span>
+    <span style="
+      font-family: 'Inter', system-ui, sans-serif;
+      font-size: 12px;
+      text-transform: uppercase;
+      letter-spacing: 0.05em;
+      color: ${color};
+      font-weight: 600;
+      min-width: 72px;
+    ">${escapeHtml(check.status)}</span>
+    <span style="
+      font-family: 'Inter', system-ui, sans-serif;
+      font-size: 14px;
+      font-weight: 500;
+      color: #E4E4E7;
+      flex: 1;
+    ">${escapeHtml(name)}</span>
+    <span style="
+      font-family: 'JetBrains Mono', monospace;
+      font-size: 12px;
+      color: #A1A1AA;
+      min-width: 60px;
+      text-align: right;
+    ">${escapeHtml(String(check.latencyMs))}ms</span>
+    <span style="
+      font-family: 'JetBrains Mono', monospace;
+      font-size: 12px;
+      color: #71717A;
+      max-width: 200px;
+      overflow: hidden;
+      text-overflow: ellipsis;
+      white-space: nowrap;
+    ">${escapeHtml(check.message)}</span>
+  </div>`;
+}
+function renderPage(borderColor, content) {
+  return `<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="utf-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1">
+  <title>Healthcheck</title>
+  <style></style>
+</head>
+<body style="
+  margin: 0;
+  padding: 40px 20px;
+  background: #08080A;
+  min-height: 100vh;
+  display: flex;
+  justify-content: center;
+  align-items: flex-start;
+  box-sizing: border-box;
+">
+  <div style="
+    width: 100%;
+    max-width: 560px;
+    background: #18181B;
+    border: 1px solid #27272A;
+    border-left: 3px solid ${borderColor};
+    border-radius: 8px;
+    padding: 24px;
+  ">
+    ${content}
+  </div>
+</body>
+</html>`;
+}
+function renderHealthcheckHtml(body) {
+  if (isErrorResponse(body)) {
+    const color = "#ED1C24";
+    return renderPage(
+      color,
+      `<div style="display: flex; align-items: center; justify-content: space-between;">
+        <span style="
+          font-family: 'Inter', system-ui, sans-serif;
+          font-size: 18px;
+          font-weight: 700;
+          color: #E4E4E7;
+        ">Healthcheck</span>
+        ${renderStatusBadge("ERROR", color)}
+      </div>
+      <div style="
+        margin-top: 16px;
+        font-family: 'JetBrains Mono', monospace;
+        font-size: 13px;
+        color: #EF4444;
+      ">${escapeHtml(body.error)}</div>`
+    );
+  }
+  if (isShallowResponse(body)) {
+    const color = statusColor("ok");
+    const { status: _, timestamp, ...rest } = body;
+    const metadataItems = Object.entries(rest).map(
+      ([k, v]) => [k, String(v)]
+    );
+    const allItems = [
+      ["timestamp", timestamp],
+      ...metadataItems
+    ];
+    return renderPage(
+      color,
+      `<div style="display: flex; align-items: center; justify-content: space-between;">
+        <span style="
+          font-family: 'Inter', system-ui, sans-serif;
+          font-size: 18px;
+          font-weight: 700;
+          color: #E4E4E7;
+        ">Healthcheck</span>
+        ${renderStatusBadge("OK", color)}
+      </div>
+      ${renderMetadataItems(allItems)}`
+    );
+  }
+  if (isDeepResponse(body)) {
+    const color = statusColor(body.status);
+    const { timestamp, checks } = body;
+    const metadataItems = Object.entries(body).filter(([k]) => !DEEP_KNOWN_KEYS.has(k)).map(([k, v]) => [k, String(v)]);
+    const allItems = [
+      ["timestamp", timestamp],
+      ...metadataItems
+    ];
+    const checkRows = Object.entries(checks).sort(([a], [b]) => a.localeCompare(b)).map(([name, check]) => renderCheckRow(name, check)).join("");
+    return renderPage(
+      color,
+      `<div style="display: flex; align-items: center; justify-content: space-between;">
+        <span style="
+          font-family: 'Inter', system-ui, sans-serif;
+          font-size: 18px;
+          font-weight: 700;
+          color: #E4E4E7;
+        ">Healthcheck</span>
+        ${renderStatusBadge(body.status, color)}
+      </div>
+      ${renderMetadataItems(allItems)}
+      <div style="margin-top: 20px;">
+        ${checkRows}
+      </div>`
+    );
+  }
+  return renderPage("#ED1C24", `<pre style="color: #E4E4E7; font-family: 'JetBrains Mono', monospace;">${escapeHtml(JSON.stringify(body, null, 2))}</pre>`);
+}
+
 // src/integrations/express.ts
 function createHealthcheckMiddleware(options) {
   const handle = createHealthcheckHandler(options);
   return async (req, res) => {
     try {
       const result = await handle({
+        ip: req.ip,
+        headers: req.headers,
         queryParams: req.query,
         authorizationHeader: req.headers.authorization ?? null
       });
-      res.status(result.status).json(result.body);
+      const preferred = req.accepts(["json", "html"]);
+      if (preferred === "html") {
+        res.status(result.status).type("html").send(renderHealthcheckHtml(result.body));
+      } else {
+        res.status(result.status).json(result.body);
+      }
     } catch {
       res.status(500).json({ error: "Internal Server Error" });
     }