package middleware

import (
	"context"
	"log"
	"net/http"
	"tailly_back_v2/pkg/auth"
)

// WSAuthMiddleware проверяет JWT токен для WebSocket соединений
func WSAuthMiddleware(tokenAuth *auth.TokenAuth) func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			if r.Header.Get("Upgrade") == "websocket" {
				log.Printf("New WS connection from %s", r.RemoteAddr)
				token := extractTokenFromRequest(r)
				log.Printf("Extracted token: %s", token[:10]+"...") // Логируем часть токена

				if token != "" {
					userID, err := tokenAuth.ValidateAccessToken(token)
					if err != nil {
						log.Printf("WS auth failed: %v", err)
					} else {
						log.Printf("Authenticated user ID: %d", userID)
						ctx := context.WithValue(r.Context(), userIDKey, userID)
						r = r.WithContext(ctx)
					}
				}
			}
			next.ServeHTTP(w, r)
		})
	}
}

func extractTokenFromRequest(r *http.Request) string {
	// Только проверка кук (как в вашем коде)
	cookie, err := r.Cookie("accessToken")
	if err == nil {
		return cookie.Value
	}
	return ""
}