mirror of
https://github.com/haiwen/seafile-server.git
synced 2025-06-27 15:36:48 +00:00
d01c46e2f9
* Add cookie to check share link access * Modify user agent and add timeout * Add filename to attachment * Go set filename to attach * C set filename to attachment * Adjust position of check priviate key * Set Content-Type and User-Agent --------- Co-authored-by: 杨赫然 <heran.yang@seafile.com>
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package utils
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
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) {
|
|
header["Content-Type"] = []string{"application/json"}
|
|
header["User-Agent"] = []string{"Seafile Server"}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 45*time.Second)
|
|
defer cancel()
|
|
req, err := http.NewRequestWithContext(ctx, 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)
|
|
}
|
|
|
|
if rsp.StatusCode != http.StatusOK {
|
|
return rsp.StatusCode, nil, fmt.Errorf("bad response %d for %s", rsp.StatusCode, url)
|
|
}
|
|
|
|
body, err := io.ReadAll(rsp.Body)
|
|
if err != nil {
|
|
return rsp.StatusCode, nil, err
|
|
}
|
|
|
|
return rsp.StatusCode, body, nil
|
|
}
|