mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
938 B
Go
50 lines
938 B
Go
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
|
|
}
|