github
/
alist
mirror of https://github.com/Xhofe/alist
1
0
Fork 0
Code Issues Releases Wiki Activity

feat(server): add `HEAD` method support (close #4740)

pull/4745/head
Andy Hsu 2023-07-11 13:47:49 +08:00
parent 6887f14ec6
commit 3f8b3da52b
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/common/proxy.go
View File

@ -58,6 +58,9 @@ func Proxy(w http.ResponseWriter, r *http.Request, link *model.Link, file model.
} else { } else {
w.WriteHeader(link.Status) w.WriteHeader(link.Status)
} }
if r.Method == http.MethodHead {
return nil
}
_, err = io.Copy(w, link.Data) _, err = io.Copy(w, link.Data)
if err != nil { if err != nil {
return err return err
@ -95,6 +98,9 @@ func Proxy(w http.ResponseWriter, r *http.Request, link *model.Link, file model.
} else { } else {
w.WriteHeader(link.Status) w.WriteHeader(link.Status)
} }
if r.Method == http.MethodHead {
return nil
}
return link.Writer(w) return link.Writer(w)
} else { } else {
req, err := http.NewRequest(r.Method, link.URL, nil) req, err := http.NewRequest(r.Method, link.URL, nil)
@ -132,6 +138,9 @@ func Proxy(w http.ResponseWriter, r *http.Request, link *model.Link, file model.
log.Debugln(msg) log.Debugln(msg)
return errors.New(msg) return errors.New(msg)
} }
if r.Method == http.MethodHead {
return nil
}
_, err = io.Copy(w, res.Body) _, err = io.Copy(w, res.Body)
if err != nil { if err != nil {
return err return err

2
server/router.go
View File

@ -39,6 +39,8 @@ func Init(e *gin.Engine) {
g.GET("/d/*path", middlewares.Down, handles.Down) g.GET("/d/*path", middlewares.Down, handles.Down)
g.GET("/p/*path", middlewares.Down, handles.Proxy) g.GET("/p/*path", middlewares.Down, handles.Proxy)
g.HEAD("/d/*path", middlewares.Down, handles.Down)
g.HEAD("/p/*path", middlewares.Down, handles.Proxy)
api := g.Group("/api") api := g.Group("/api")
auth := api.Group("", middlewares.Auth) auth := api.Group("", middlewares.Auth)