81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package http
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/asticode/go-astisub"
|
|
|
|
"github.com/filebrowser/filebrowser/v2/files"
|
|
)
|
|
|
|
var subtitleHandler = withUser(func(w http.ResponseWriter, r *http.Request, d *data) (int, error) {
|
|
if !d.user.Perm.Download {
|
|
return http.StatusAccepted, nil
|
|
}
|
|
|
|
file, err := files.NewFileInfo(&files.FileOptions{
|
|
Fs: d.user.Fs,
|
|
Path: r.URL.Path,
|
|
Modify: d.user.Perm.Modify,
|
|
Expand: false,
|
|
ReadHeader: d.server.TypeDetectionByHeader,
|
|
Checker: d,
|
|
})
|
|
if err != nil {
|
|
return errToStatus(err), err
|
|
}
|
|
|
|
if file.IsDir {
|
|
return http.StatusBadRequest, nil
|
|
}
|
|
|
|
return subtitleFileHandler(w, r, file)
|
|
})
|
|
|
|
func subtitleFileHandler(w http.ResponseWriter, r *http.Request, file *files.FileInfo) (int, error) {
|
|
// if its not a subtitle file, reject
|
|
if !files.IsSupportedSubtitle(file.Name) {
|
|
return http.StatusBadRequest, nil
|
|
}
|
|
|
|
fd, err := file.Fs.Open(file.Path)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
defer fd.Close()
|
|
|
|
// load subtitle for conversion to vtt
|
|
var sub *astisub.Subtitles
|
|
if strings.HasSuffix(file.Name, ".srt") {
|
|
sub, err = astisub.ReadFromSRT(fd)
|
|
} else if strings.HasSuffix(file.Name, ".ass") || strings.HasSuffix(file.Name, ".ssa") {
|
|
sub, err = astisub.ReadFromSSA(fd)
|
|
}
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
setContentDisposition(w, r, file)
|
|
w.Header().Add("Content-Security-Policy", `script-src 'none';`)
|
|
w.Header().Set("Cache-Control", "private")
|
|
// force type to text/vtt
|
|
w.Header().Set("Content-Type", "text/vtt")
|
|
|
|
// serve vtt file directly
|
|
if sub == nil {
|
|
http.ServeContent(w, r, file.Name, file.ModTime, fd)
|
|
return 0, nil
|
|
}
|
|
|
|
// convert others to vtt and serve from buffer
|
|
var buf = &bytes.Buffer{}
|
|
err = sub.WriteToWebVTT(buf)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
http.ServeContent(w, r, file.Name, file.ModTime, bytes.NewReader(buf.Bytes()))
|
|
return 0, nil
|
|
}
|