Fix filenames on webdav issue

This commit is contained in:
Ben Sarmiento
2023-11-18 17:16:09 +01:00
parent e85109bcf3
commit c520b5572f
3 changed files with 17 additions and 36 deletions

View File

@@ -4,7 +4,6 @@ import (
"encoding/xml"
"fmt"
"net/http"
"net/url"
"path"
"path/filepath"
"sort"
@@ -79,12 +78,9 @@ func handleListOfTorrents(requestPath string, t *torrent.TorrentManager) ([]byte
}
var responses []dav.Response
// initial response is the directory itself
responses = append(responses, dav.Directory(basePath))
responses = append(responses, dav.Directory(basePath+"/"))
var accessKeys []string
torrents.IterCb(func(accessKey string, _ *torrent.Torrent) {
accessKeys = append(accessKeys, accessKey)
})
accessKeys := torrents.Keys()
sort.Strings(accessKeys)
for _, accessKey := range accessKeys {
responses = append(responses, dav.Directory(filepath.Join(basePath, accessKey)))
@@ -111,23 +107,18 @@ func handleSingleTorrent(requestPath string, t *torrent.TorrentManager) ([]byte,
var responses []dav.Response
// initial response is the directory itself
responses = append(responses, dav.Directory(requestPath))
tor.SelectedFiles.IterCb(func(filename string, file *torrent.File) {
if file.Link == "" {
// will be caught by torrent manager's repairAll
// just skip it for now
return
}
filePath := filepath.Join(requestPath, url.PathEscape(filename))
responses = append(responses, dav.Directory(requestPath+"/"))
filenames := tor.SelectedFiles.Keys()
sort.Strings(filenames)
for _, filename := range filenames {
file, _ := tor.SelectedFiles.Get(filename)
responses = append(responses, dav.File(
filePath,
filepath.Join(requestPath, filename),
file.Bytes,
convertRFC3339toRFC1123(tor.LatestAdded),
file.Link,
))
})
}
resp := &dav.MultiStatus{
XMLNS: "DAV:",
Response: responses,

View File

@@ -71,16 +71,11 @@ func handleListOfTorrents(requestPath string, t *torrent.TorrentManager) (*strin
}
htmlDoc := "<ol>"
var accessKeys []string
torrents.IterCb(func(accessKey string, _ *torrent.Torrent) {
accessKeys = append(accessKeys, accessKey)
})
accessKeys := torrents.Keys()
sort.Strings(accessKeys)
for _, accessKey := range accessKeys {
htmlDoc = htmlDoc + fmt.Sprintf("<li><a href=\"%s/\">%s</a></li>", filepath.Join(requestPath, url.PathEscape(accessKey)), accessKey)
}
return &htmlDoc, nil
}
@@ -97,16 +92,17 @@ func handleSingleTorrent(requestPath string, t *torrent.TorrentManager) (*string
}
htmlDoc := "<ol>"
tor.SelectedFiles.IterCb(func(filename string, file *torrent.File) {
filenames := tor.SelectedFiles.Keys()
sort.Strings(filenames)
for _, filename := range filenames {
file, _ := tor.SelectedFiles.Get(filename)
if file.Link == "" {
// will be caught by torrent manager's repairAll
// just skip it for now
return
continue
}
filePath := filepath.Join(requestPath, url.PathEscape(filename))
htmlDoc += fmt.Sprintf("<li><a href=\"%s\">%s</a></li>", filePath, filename)
})
}
return &htmlDoc, nil
}

View File

@@ -26,7 +26,6 @@ type TorrentManager struct {
checksum string
api *realdebrid.RealDebrid
workerPool chan bool
mu *sync.Mutex
log *zap.SugaredLogger
}
@@ -40,7 +39,6 @@ func NewTorrentManager(cfg config.ConfigInterface, api *realdebrid.RealDebrid) *
requiredVersion: "10.11.2023",
api: api,
workerPool: make(chan bool, cfg.GetNumOfWorkers()),
mu: &sync.Mutex{},
log: logutil.NewLogger().Named("manager"),
}
@@ -99,11 +97,7 @@ func NewTorrentManager(cfg config.ConfigInterface, api *realdebrid.RealDebrid) *
}
// get filenames
var filenames []string
torrent.SelectedFiles.IterCb(func(_ string, file *File) {
filenames = append(filenames, file.Path)
})
filenames := torrent.SelectedFiles.Keys()
// Map torrents to directories
switch t.cfg.GetVersion() {
case "v1":