1
0
mirror of https://github.com/haiwen/seafile-server.git synced 2025-07-02 01:41:59 +00:00
seafile-server/fileserver/utils/http.go
feiniks b5b37e69e0
Add check Authorization header (#686)
Co-authored-by: 杨赫然 <heran.yang@seafile.com>
2024-08-29 15:28:41 +08:00

45 lines
936 B
Go

package utils
import (
"context"
"fmt"
"io"
"net/http"
"strings"
)
var HttpReqContext, HttpReqCancel = context.WithCancel(context.Background())
func GetAuthorizationToken(h http.Header) string {
auth := h.Get("Authorization")
splitResult := strings.Split(auth, " ")
if len(splitResult) > 1 {
return splitResult[1]
}
return ""
}
func HttpCommon(method, url string, header map[string][]string, reader io.Reader) (int, []byte, error) {
req, err := http.NewRequestWithContext(HttpReqContext, method, url, reader)
if err != nil {
return -1, nil, err
}
req.Header = header
rsp, err := http.DefaultClient.Do(req)
if err != nil {
return -1, nil, err
}
defer rsp.Body.Close()
if rsp.StatusCode == http.StatusNotFound {
return rsp.StatusCode, nil, fmt.Errorf("url %s not found", url)
}
body, err := io.ReadAll(rsp.Body)
if err != nil {
return rsp.StatusCode, nil, err
}
return rsp.StatusCode, body, nil
}