main: support specified Referrer

README.md

@@ -113,6 +113,16 @@ $ annie -c "name=value; name2=value2" https://www.bilibili.com/video/av20203945
 $ annie -c cookies.txt https://www.bilibili.com/video/av20203945
 ```
 
+### Use specified Referrer
+
+You can use the `-r` option to tell Annie to use the given Referrer to request.
+
+```console
+$ annie -r https://www.bilibili.com/video/av20383055/ http://cn-scnc1-dx.acgvideo.com/...
+
+...
+```
+
 ### Debug Mode
 
 You can use the `-d` option to see network request message.
@@ -147,6 +157,8 @@ Usage of annie:
   -d	Debug mode
   -i	Information only
   -p	Download playlist
+  -r string
+    	Use specified Referrer
   -v	Show version
 ```
 

config/config.go

@@ -11,6 +11,8 @@ var (
 	Cookie string
 	// Playlist download playlist
 	Playlist bool
+	// Refer use specified Referrer
+	Refer string
 )
 
 // FakeHeaders fake http headers

main.go

@@ -16,6 +16,7 @@ func init() {
 	flag.BoolVar(&config.InfoOnly, "i", false, "Information only")
 	flag.StringVar(&config.Cookie, "c", "", "Cookie")
 	flag.BoolVar(&config.Playlist, "p", false, "Download playlist")
+	flag.StringVar(&config.Refer, "r", "", "Use specified Referrer")
 }
 
 func main() {

request/request.go

@@ -49,6 +49,9 @@ func Request(
 	for k, v := range headers {
 		req.Header.Set(k, v)
 	}
+	if config.Refer != "" {
+		req.Header.Set("Referer", config.Refer)
+	}
 	res, err := client.Do(req)
 	if err != nil {
 		log.Print(url)