package bilibili

import (
	"io"
	"net/http"

	"github.com/synctv-org/synctv/utils"
)

type Client struct {
	httpClient *http.Client
	cookies    []*http.Cookie
}

type ClientConfig func(*Client)

func WithHttpClient(httpClient *http.Client) ClientConfig {
	return func(c *Client) {
		c.httpClient = httpClient
	}
}

func NewClient(cookies []*http.Cookie, conf ...ClientConfig) *Client {
	c := &Client{
		httpClient: http.DefaultClient,
		cookies:    cookies,
	}
	for _, v := range conf {
		v(c)
	}
	return c
}

func (c *Client) NewRequest(method, url string, body io.Reader) (*http.Request, error) {
	url, err := signAndGenerateURL(url)
	if err != nil {
		return nil, err
	}
	req, err := http.NewRequest(method, url, body)
	if err != nil {
		return nil, err
	}
	for _, cookie := range c.cookies {
		req.AddCookie(cookie)
	}
	req.Header.Set("User-Agent", utils.UA)
	req.Header.Set("Referer", "https://www.bilibili.com")
	return req, nil
}