Backups/OpenList
1
0
Fork 0
mirror of https://github.com/OpenListTeam/OpenList.git synced 2025-09-19 20:26:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(simplehttp): logic bug when unable to parse file name (#761)

Browse Source
This commit is contained in:
j2rong4cn
2025-07-20 14:13:30 +08:00
committed by GitHub
parent cedb3d488d
commit 74d32fd4d7
1 changed files with 9 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
internal/offline_download/http/client.go
View File

@ -2,12 +2,13 @@ package http
import (
"fmt"
"math/rand/v2"
"net/http"
"net/url"
"os"
"path"
"path/filepath"
"strings"
"time"
"github.com/OpenListTeam/OpenList/v4/internal/model"
"github.com/OpenListTeam/OpenList/v4/internal/offline_download/tool"
@ -48,18 +49,12 @@ func (s SimpleHttp) Status(task *tool.DownloadTask) (*tool.Status, error) {
}
func (s SimpleHttp) Run(task *tool.DownloadTask) error {
u := task.Url
// parse url
_u, err := url.Parse(u)
if err != nil {
return err
}
streamPut := task.DeletePolicy == tool.UploadDownloadStream
method := http.MethodGet
if streamPut {
method = http.MethodHead
}
req, err := http.NewRequestWithContext(task.Ctx(), method, u, nil)
req, err := http.NewRequestWithContext(task.Ctx(), method, task.Url, nil)
if err != nil {
return err
}
@ -74,14 +69,13 @@ func (s SimpleHttp) Run(task *tool.DownloadTask) error {
if resp.StatusCode >= 400 {
return fmt.Errorf("http status code %d", resp.StatusCode)
}
// If Path is empty, use Hostname; otherwise, filePath euqals TempDir which causes os.Create to fail
urlPath := _u.Path
if urlPath == "" {
urlPath = strings.ReplaceAll(_u.Host, ".", "_")
filename, err := parseFilenameFromContentDisposition(resp.Header.Get("Content-Disposition"))
if err != nil {
filename = path.Base(resp.Request.URL.Path)
}
filename := path.Base(urlPath)
if n, err := parseFilenameFromContentDisposition(resp.Header.Get("Content-Disposition")); err == nil {
filename = n
filename = strings.Trim(filename, "/")
if len(filename) == 0 {
filename = fmt.Sprintf("%s-%d-%x", strings.ReplaceAll(req.URL.Host, ".", "_"), time.Now().UnixMilli(), rand.Uint32())
}
fileSize := resp.ContentLength
if streamPut {