package httputil

import (
	"net/http"
	"net/http/cookiejar"
	"net/url"
	"sync"
)

// Option represents a configuration option for the cookie jar
type Option func(*CookieJar)

// WithCookieJar sets an existing cookie jar to wrap
func WithCookieJar(jar http.CookieJar) Option {
	return func(cj *CookieJar) {
		cj.jar = jar
	}
}

// CookieJar is a thread-safe wrapper around http.CookieJar
type CookieJar struct {
	jar http.CookieJar
	mu  sync.RWMutex
}

// New creates a new thread-safe cookie jar with the given options
// If no jar is provided, creates a simple in-memory cookie jar
func NewCookieJar(opts ...Option) (*CookieJar, error) {
	cj := &CookieJar{}

	// Apply options
	for _, opt := range opts {
		opt(cj)
	}

	// If no jar was provided, create a new one
	if cj.jar == nil {
		jar, err := cookiejar.New(nil)
		if err != nil {
			return nil, err
		}
		cj.jar = jar
	}

	return cj, nil
}

// SetCookies implements http.CookieJar.SetCookies
func (cj *CookieJar) SetCookies(u *url.URL, cookies []*http.Cookie) {
	if cj.jar == nil {
		return
	}

	cj.mu.Lock()
	defer cj.mu.Unlock()
	cj.jar.SetCookies(u, cookies)
}

// Cookies implements http.CookieJar.Cookies
func (cj *CookieJar) Cookies(u *url.URL) []*http.Cookie {
	if cj.jar == nil {
		return nil
	}

	cj.mu.RLock()
	defer cj.mu.RUnlock()
	return cj.jar.Cookies(u)
}