package middleware

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

// 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" {
				ctx, cancel := context.WithTimeout(r.Context(), 30*time.Minute)
				defer cancel()

				token := extractTokenFromRequest(r)
				if token != "" {
					if userID, err := tokenAuth.ValidateAccessToken(token); err == nil {
						ctx = context.WithValue(ctx, 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 ""
}