heimoshuiyu/gonic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

make playlist uploading a little more robust

Browse Source
This commit is contained in:
sentriz
2020-01-22 14:46:01 +00:00
parent 8a86cd722f
commit 5a4d07b273
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/ctrladmin/playlist.go
View File

@@ -45,6 +45,10 @@ func playlistParseUpload(c *Controller, userID int, header *multipart.FileHeader
if playlistName == "" { if playlistName == "" {
return []string{fmt.Sprintf("invalid filename %q", header.Filename)}, false return []string{fmt.Sprintf("invalid filename %q", header.Filename)}, false
} }
contentType := header.Header.Get("Content-Type")
if !(contentType == "audio/x-mpegurl" || contentType == "application/octet-stream") {
return []string{fmt.Sprintf("invalid content-type %q", contentType)}, false
}
playlist := &model.Playlist{} playlist := &model.Playlist{}
c.DB.FirstOrCreate(playlist, model.Playlist{ c.DB.FirstOrCreate(playlist, model.Playlist{
Name: playlistName, Name: playlistName,
@@ -56,7 +60,8 @@ func playlistParseUpload(c *Controller, userID int, header *multipart.FileHeader
for scanner.Scan() { for scanner.Scan() {
path := scanner.Text() path := scanner.Text()
if err := playlistParseLine(c, playlist.ID, path); err != nil { if err := playlistParseLine(c, playlist.ID, path); err != nil {
errors = append(errors, err.Error()) // trim length of error to not overflow cookie flash
errors = append(errors, fmt.Sprintf("%.100s", err.Error()))
} }
} }
if err := scanner.Err(); err != nil { if err := scanner.Err(); err != nil {