summaryrefslogtreecommitdiff
path: root/bilibili/client.go
blob: fcea3959c1af72de0811ed07df8f45ebd46f1714 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/*
This file defines the Bilibili client struct.
This struct is a facade of all necessary Bilibili HTTP API wrappers.
*/
package bilibili

import (
	"context"
	"log"
	"net/http"
	"os"
)

const (
	// kUserAgent: the default user-agent header to use when communicating with bilibili.
	kUserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) " +
		"AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.0.0 Safari/537.36"
)

type Bilibili struct {
	userAgent string
	http      *http.Client
	loggerCommon
	ctx context.Context
}

func NewBilibiliWithContext(ctx context.Context) Bilibili {
	logger := loggerCommon{
		debug: log.New(os.Stderr, "DEBUG: ", log.Ldate|log.Ltime|log.Lshortfile),
		info:  log.New(os.Stderr, "INFO: ", log.Ldate|log.Ltime|log.Lshortfile),
		warn:  log.New(os.Stderr, "WARNING: ", log.Ldate|log.Ltime|log.Lshortfile),
		error: log.New(os.Stderr, "ERROR: ", log.Ldate|log.Ltime|log.Lshortfile),
	}
	return Bilibili{
		loggerCommon: logger,
		userAgent:    kUserAgent,
		http:         http.DefaultClient,
		ctx:          ctx,
	}
}

func NewBilibili() Bilibili {
	ctx := context.Background()
	return NewBilibiliWithContext(ctx)
}