feat(ui): logs view

src/pkg/api/docs/docs.go

@@ -3037,6 +3037,48 @@ const docTemplate = `{
                 }
             }
         },
+        "/api/v1/resources/workloads/pods/logs": {
+            "get": {
+                "description": "Get Pod logs",
+                "consumes": [
+                    "text/html"
+                ],
+                "produces": [
+                    "application/json"
+                ],
+                "tags": [
+                    "workloads"
+                ],
+                "parameters": [
+                    {
+                        "type": "string",
+                        "description": "pod",
+                        "name": "pod",
+                        "in": "query",
+                        "required": true
+                    },
+                    {
+                        "type": "string",
+                        "description": "namespace",
+                        "name": "namespace",
+                        "in": "query",
+                        "required": true
+                    },
+                    {
+                        "type": "string",
+                        "description": "container",
+                        "name": "container",
+                        "in": "query",
+                        "required": true
+                    }
+                ],
+                "responses": {
+                    "200": {
+                        "description": "OK"
+                    }
+                }
+            }
+        },
         "/api/v1/resources/workloads/pods/{uid}": {
             "get": {
                 "description": "Get Pod by UID",

src/pkg/api/docs/swagger.json

@@ -3026,6 +3026,48 @@
                 }
             }
         },
+        "/api/v1/resources/workloads/pods/logs": {
+            "get": {
+                "description": "Get Pod logs",
+                "consumes": [
+                    "text/html"
+                ],
+                "produces": [
+                    "application/json"
+                ],
+                "tags": [
+                    "workloads"
+                ],
+                "parameters": [
+                    {
+                        "type": "string",
+                        "description": "pod",
+                        "name": "pod",
+                        "in": "query",
+                        "required": true
+                    },
+                    {
+                        "type": "string",
+                        "description": "namespace",
+                        "name": "namespace",
+                        "in": "query",
+                        "required": true
+                    },
+                    {
+                        "type": "string",
+                        "description": "container",
+                        "name": "container",
+                        "in": "query",
+                        "required": true
+                    }
+                ],
+                "responses": {
+                    "200": {
+                        "description": "OK"
+                    }
+                }
+            }
+        },
         "/api/v1/resources/workloads/pods/{uid}": {
             "get": {
                 "description": "Get Pod by UID",

src/pkg/api/docs/swagger.yaml

@@ -2070,6 +2070,34 @@ paths:
           description: OK
       tags:
       - workloads
+  /api/v1/resources/workloads/pods/logs:
+    get:
+      consumes:
+      - text/html
+      description: Get Pod logs
+      parameters:
+      - description: pod
+        in: query
+        name: pod
+        required: true
+        type: string
+      - description: namespace
+        in: query
+        name: namespace
+        required: true
+        type: string
+      - description: container
+        in: query
+        name: container
+        required: true
+        type: string
+      produces:
+      - application/json
+      responses:
+        "200":
+          description: OK
+      tags:
+      - workloads
   /api/v1/resources/workloads/statefulsets:
     get:
       consumes:

src/pkg/api/handlers.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/defenseunicorns/uds-runtime/src/pkg/api/auth"
 	_ "github.com/defenseunicorns/uds-runtime/src/pkg/api/docs" //nolint:staticcheck
+	"github.com/defenseunicorns/uds-runtime/src/pkg/api/logs"
 	"github.com/defenseunicorns/uds-runtime/src/pkg/api/resources"
 	"github.com/defenseunicorns/uds-runtime/src/pkg/api/rest"
 	"github.com/defenseunicorns/uds-runtime/src/pkg/k8s/client"
@@ -137,6 +138,21 @@ func getPod(cache *resources.Cache) func(w http.ResponseWriter, r *http.Request)
 	return rest.Bind(cache.Pods)
 }
 
+// @Description Get Pod logs
+// @Tags workloads
+// @Accept  html
+// @Produce  json
+// @Success 200
+// @Router /api/v1/resources/workloads/pods/logs [get]
+// @Param pod query string true "pod"
+// @Param namespace query string true "namespace"
+// @Param container query string true "container"
+func getPodLogs(k8sSession *session.K8sSession) func(w http.ResponseWriter, r *http.Request) {
+	return func(w http.ResponseWriter, r *http.Request) {
+		logs.GetLogs(k8sSession.Clients.Clientset, w, r)
+	}
+}
+
 // @Description Get Deployments
 // @Tags workloads
 // @Accept  html

src/pkg/api/logs/logs.go

@@ -0,0 +1,146 @@
+// Copyright 2024 Defense Unicorns
+// SPDX-License-Identifier: AGPL-3.0-or-later OR LicenseRef-Defense-Unicorns-Commercial
+
+package logs
+
+import (
+	"bufio"
+	"context"
+	"errors"
+	"fmt"
+	"io"
+	"log/slog"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/defenseunicorns/uds-runtime/src/pkg/api/rest"
+	v1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes"
+)
+
+// validateParams checks if all required parameters are present
+func validateParams(pod, namespace, container string) error {
+	if pod == "" || namespace == "" || container == "" {
+		return errors.New("pod, namespace, and container are required")
+	}
+	return nil
+}
+
+// streamExistingLogs fetches and streams existing logs
+func streamExistingLogs(ctx context.Context, clientSet kubernetes.Interface, w http.ResponseWriter,
+	pod, namespace, container string, numLines int64) (time.Time, error) {
+	logStream, err := clientSet.CoreV1().Pods(namespace).GetLogs(pod, &v1.PodLogOptions{
+		Container: container,
+		TailLines: &numLines,
+	}).Stream(ctx)
+	if err != nil {
+		return time.Time{}, fmt.Errorf("failed to get existing logs: %w", err)
+	}
+	defer logStream.Close()
+
+	flusher := w.(http.Flusher)
+	scanner := bufio.NewScanner(logStream)
+	for scanner.Scan() {
+		fmt.Fprintf(w, "data: %s\n\n", scanner.Text())
+	}
+	flusher.Flush()
+
+	if err := scanner.Err(); err != nil {
+		return time.Time{}, fmt.Errorf("error reading existing logs: %w", err)
+	}
+
+	return time.Now(), nil
+}
+
+// handleLogStream manages the continuous streaming of new logs
+func handleLogStream(ctx context.Context, logStream io.ReadCloser, w http.ResponseWriter) error {
+	defer logStream.Close()
+
+	reader := bufio.NewReader(logStream)
+	flusher := w.(http.Flusher)
+	buffer := &strings.Builder{}
+	ticker := time.NewTicker(500 * time.Millisecond)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-ctx.Done():
+			slog.Debug("Logs streaming stopped")
+			return nil
+
+		case <-ticker.C:
+			if buffer.Len() > 0 {
+				fmt.Fprintf(w, "data: %s\n\n", buffer.String())
+				buffer.Reset()
+				flusher.Flush()
+			}
+
+		default:
+			// Try to read a line
+			line, err := reader.ReadBytes('\n')
+			if err != nil {
+				if err == io.EOF || errors.Is(err, context.Canceled) {
+					// Final flush if needed
+					if buffer.Len() > 0 {
+						fmt.Fprintf(w, "data: %s\n\n", buffer.String())
+						flusher.Flush()
+					}
+					return nil
+				}
+				slog.Debug("Reader returned error", "error", err)
+				return err
+			}
+			buffer.Write(line)
+		}
+	}
+}
+
+// GetLogs streams logs from a pod
+func GetLogs(clientSet kubernetes.Interface, w http.ResponseWriter, r *http.Request) {
+	// Get and validate query parameters
+	pod := r.URL.Query().Get("pod")
+	namespace := r.URL.Query().Get("namespace")
+	container := r.URL.Query().Get("container")
+
+	if err := validateParams(pod, namespace, container); err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	rest.WriteHeaders(w)
+	ctx := r.Context()
+
+	// Verify streaming is supported
+	_, ok := w.(http.Flusher)
+	if !ok {
+		http.Error(w, "Streaming unsupported", http.StatusInternalServerError)
+		return
+	}
+
+	// Stream existing logs first
+	lastReadTime, err := streamExistingLogs(ctx, clientSet, w, pod, namespace, container, 100)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	// Set up streaming for new logs
+	podLogs := clientSet.CoreV1().Pods(namespace).GetLogs(pod, &v1.PodLogOptions{
+		Container: container,
+		Follow:    true,
+		SinceTime: &metav1.Time{Time: lastReadTime},
+	})
+
+	logStream, err := podLogs.Stream(ctx)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("Failed to open log stream: %v", err), http.StatusInternalServerError)
+		return
+	}
+
+	if err := handleLogStream(ctx, logStream, w); err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+}