rewrite extension handling, initial video support
This commit is contained in:
parent
676a71f527
commit
5f096e52a4
@ -60,22 +60,34 @@ func (this *Thumbnailer) RenderFile_NoCache(absPath string) ([]byte, error) {
|
|||||||
|
|
||||||
defer fh.Close()
|
defer fh.Close()
|
||||||
|
|
||||||
var src image.Image
|
if len(absPath) < 4 {
|
||||||
err = fmt.Errorf("No thumbnailer for file type")
|
return nil, fmt.Errorf("No extension on filename")
|
||||||
|
|
||||||
comparePath := strings.ToLower(absPath)
|
|
||||||
|
|
||||||
if strings.HasSuffix(comparePath, "jpg") {
|
|
||||||
src, err = jpeg.Decode(fh)
|
|
||||||
|
|
||||||
} else if strings.HasSuffix(comparePath, "png") {
|
|
||||||
src, err = png.Decode(fh)
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
extension := strings.ToLower(absPath[len(absPath)-4:])
|
||||||
|
|
||||||
|
switch extension {
|
||||||
|
case ".jpg", "jpeg":
|
||||||
|
src, err := jpeg.Decode(fh)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return this.RenderScaledImage(src)
|
return this.RenderScaledImage(src)
|
||||||
|
|
||||||
|
case ".png":
|
||||||
|
src, err := png.Decode(fh)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return this.RenderScaledImage(src)
|
||||||
|
|
||||||
|
case ".avi", ".mkv", ".mp4", ".ogm", ".wmv":
|
||||||
|
return this.RenderScaledFfmpeg(absPath)
|
||||||
|
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("No thumbnailer for file type")
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
25
video.go
25
video.go
@ -1,6 +1,8 @@
|
|||||||
package thumbnail
|
package thumbnail
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"io"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -18,5 +20,26 @@ func (this *Thumbnailer) RenderScaledFfmpeg(absPath string) ([]byte, error) {
|
|||||||
"-",
|
"-",
|
||||||
)
|
)
|
||||||
|
|
||||||
cmd.StdoutPipe()
|
stdout, err := cmd.StdoutPipe()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = cmd.Start()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
out := bytes.Buffer{}
|
||||||
|
_, err = io.Copy(&out, stdout)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = cmd.Wait()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return out.Bytes(), nil
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user