Logs

src/core/docker.ts

@@ -1,7 +1,7 @@
 "use server";
 
 import axios from "axios";
-import { Docker, Node, SimpleStats } from "@/lib/types";
+import { Docker, Log, Node, SimpleStats } from "@/lib/types";
 import { verifySession } from "./auth/session";
 import { getNodes } from "./node";
 
@@ -86,3 +86,25 @@ export async function removeContainer(containerId: string) {
       ).data
   );
 }
+
+export async function getAllContainerLogs(): Promise<Log[]> {
+  const nodes = await getNodes();
+
+  return (
+    await Promise.all(
+      nodes.map(
+        async (n) =>
+          await (
+            await axios.get(`http://${n.ip}:3000/api/logs`, {
+              headers: { Authorization: `ApiKey ${n.key}` },
+            })
+          ).data
+      )
+    )
+  )
+    .flat()
+    .sort(
+      (a, b) =>
+        new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime()
+    );
+}

src/core/server/docker.ts

@@ -1,7 +1,7 @@
 "use server";
 
 import { CPUusage } from "@/lib/server/calc";
-import { SimpleStats } from "@/lib/types";
+import { Log, SimpleStats } from "@/lib/types";
 import Docker from "dockerode";
 
 const docker = new Docker();
@@ -34,3 +34,71 @@ export async function stopContainer(containerId: string) {
 export async function removeContainer(containerId: string) {
   await docker.getContainer(containerId).remove();
 }
+
+function inferLevel(message: string): "info" | "warn" | "error" {
+  const msg = message.toLowerCase();
+
+  if (msg.includes("error") || msg.includes("fatal")) return "error";
+  if (msg.includes("warn")) return "warn";
+
+  return "info";
+}
+
+function demuxLogs(buffer: Buffer): string {
+  let offset = 0;
+  let output = "";
+
+  while (offset < buffer.length) {
+    const streamType = buffer[offset];
+    const length = buffer.readUInt32BE(offset + 4);
+    offset += 8;
+
+    output += buffer.slice(offset, offset + length).toString("utf8");
+    offset += length;
+  }
+
+  return output;
+}
+
+async function getContainerLogs(
+  containerId: string,
+  name: string
+): Promise<Log[]> {
+  const container = docker.getContainer(containerId);
+
+  const raw = await container.logs({
+    stdout: true,
+    stderr: true,
+    timestamps: true,
+  });
+
+  const text = demuxLogs(raw);
+
+  return text
+    .split("\n")
+    .filter(Boolean)
+    .map((line) => {
+      const firstSpace = line.indexOf(" ");
+      const timestamp = line.slice(0, firstSpace);
+      const message = line.slice(firstSpace + 1);
+
+      return {
+        timestamp,
+        level: inferLevel(message),
+        source: name,
+        message,
+      } satisfies Log;
+    });
+}
+
+export async function getAllContainerLogs(): Promise<Log[]> {
+  const containers = await docker.listContainers({ all: true });
+
+  const logs = await Promise.all(
+    containers.map((c) => getContainerLogs(c.Id, c.Names[0].replace("/", "")))
+  );
+
+  return logs
+    .flat()
+    .sort((a, b) => Date.parse(a.timestamp) - Date.parse(b.timestamp));
+}