diff --git a/http/http.go b/http/http.go index 4816660e..a5e93311 100644 --- a/http/http.go +++ b/http/http.go @@ -137,7 +137,7 @@ func apiHandler(c *fb.Context, w http.ResponseWriter, r *http.Request) (int, err } } - if c.Router == "checksum" || c.Router == "download" { + if c.Router == "checksum" || c.Router == "download" || c.Router == "subtitle" || c.Router == "subtitles" { var err error c.File, err = fb.GetInfo(r.URL, c.FileBrowser, c.User) if err != nil { @@ -165,6 +165,10 @@ func apiHandler(c *fb.Context, w http.ResponseWriter, r *http.Request) (int, err code, err = settingsHandler(c, w, r) case "share": code, err = shareHandler(c, w, r) + case "subtitles": + code, err = subtitlesHandler(c, w, r) + case "subtitle": + code, err = subtitleHandler(c, w, r) default: code = http.StatusNotFound } diff --git a/http/subtitle.go b/http/subtitle.go new file mode 100644 index 00000000..be54cd55 --- /dev/null +++ b/http/subtitle.go @@ -0,0 +1,83 @@ +package http + +import ( + "bytes" + fb "github.com/filebrowser/filebrowser" + "io/ioutil" + "net/http" + "os" + "path/filepath" + "regexp" +) + +func subtitlesHandler(c *fb.Context, w http.ResponseWriter, r *http.Request) (int, error) { + files, err := ReadDir(filepath.Dir(c.File.Path)) + if err != nil { + return http.StatusInternalServerError, err + } + var subtitles = make([]map[string]string, 0) + for _, file := range files { + ext := filepath.Ext(file.Name()) + if ext == ".vtt" || ext == ".srt" { + var sub map[string]string = make(map[string]string) + sub["src"] = filepath.Dir(c.File.Path) + "/" + file.Name() + sub["kind"] = "subtitles" + sub["label"] = file.Name() + subtitles = append(subtitles, sub) + } + } + return renderJSON(w, subtitles) +} + +func subtitleHandler(c *fb.Context, w http.ResponseWriter, r *http.Request) (int, error) { + str, err := CleanSubtitle(c.File.Path) + if err != nil { + return http.StatusInternalServerError, err + } + + file, err := os.Open(c.File.Path) + defer file.Close() + + if err != nil { + return http.StatusInternalServerError, err + } + + stat, err := file.Stat() + if err != nil { + return http.StatusInternalServerError, err + } + + w.Header().Set("Content-Disposition", "inline") + w.Header().Set("Content-Type", "text/vtt") + http.ServeContent(w, r, stat.Name(), stat.ModTime(), bytes.NewReader([]byte(str))) + + return 0, nil + +} + +func CleanSubtitle(filename string) (string, error) { + b, err := ioutil.ReadFile(filename) + if err != nil { + return "", err + } + str := string(b) // convert content to a 'string' + ext := filepath.Ext(filename) + if ext == ".srt" { + re := regexp.MustCompile("([0-9]{2}:[0-9]{2}:[0-9]{2}),([0-9]{3})") + str = "WEBVTT\n\n" + re.ReplaceAllString(str, "$1.$2") + } + return str, err +} + +func ReadDir(dirname string) ([]os.FileInfo, error) { + f, err := os.Open(dirname) + if err != nil { + return nil, err + } + list, err := f.Readdir(-1) + f.Close() + if err != nil { + return nil, err + } + return list, nil +}