Add recovery middleware to handle panics and log errors

cmd/start.go

@@ -36,6 +36,7 @@ var startCmd = &cobra.Command{
 
 		// Create middleware stack
 		stack := middleware.CreateStack(
+			middleware.Recovery(), // Add recovery middleware first to catch all panics
 			middleware.SecureHeaders,
 			middleware.Logging,
 			middleware.MaxBodySize(1024*1024), // 1MB size limit
@@ -74,4 +75,4 @@ func init() {
 
 	// Add the command to the root command
 	rootCmd.AddCommand(startCmd)
-}
+}

internal/middleware/recover.go

@@ -0,0 +1,27 @@
+package middleware
+
+import (
+	"log"
+	"net/http"
+	"runtime/debug"
+)
+
+// Recovery middleware to catch panics, log them, and return a 500 error
+func Recovery() Middleware {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			defer func() {
+				if err := recover(); err != nil {
+					// Log the stack trace
+					log.Printf("PANIC: %v\n%s", err, debug.Stack())
+					
+					// Return a 500 Internal Server Error response
+					w.WriteHeader(http.StatusInternalServerError)
+					w.Write([]byte("Internal Server Error"))
+				}
+			}()
+			
+			next.ServeHTTP(w, r)
+		})
+	}
+}