jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-06 01:57:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: use real link to download client

Browse Source
This commit is contained in:
Simon Ding
2025-04-19 21:22:06 +08:00
parent 1ea8246cfc
commit d44060ff33
3 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pkg/utils/utils.go
View File

@@ -222,9 +222,10 @@ func isWSL() bool {
return strings.Contains(strings.ToLower(string(releaseData)), "microsoft")
}
func Link2Hash(link string) (string, error) {
func GetRealLinkAndHash(link string) (string, string, error) {
if strings.HasPrefix(strings.ToLower(link), "magnet:") {
return MagnetHash(link)
hash, err := MagnetHash(link)
return link, hash, err
}
client := &http.Client{
CheckRedirect: func(req *http.Request, via []*http.Request) error {
@@ -234,19 +235,19 @@ func Link2Hash(link string) (string, error) {
resp, err := client.Get(link)
if err != nil {
return "", errors.Wrap(err, "get link")
return "", "", errors.Wrap(err, "get link")
}
defer resp.Body.Close()
if resp.StatusCode >= 300 && resp.StatusCode < 400 {
//redirects
tourl := resp.Header.Get("Location")
return Link2Hash(tourl)
return GetRealLinkAndHash(tourl)
}
info, err := metainfo.Load(resp.Body)
if err != nil {
return "", errors.Wrap(err, "parse response")
return "", "", errors.Wrap(err, "parse response")
}
return info.HashInfoBytes().HexString(), nil
return link,info.HashInfoBytes().HexString(), nil
}
func Link2Magnet(link string) (string, error) {