fix: handle unsupported thumbnail

This commit is contained in:
liwei 2020-06-24 09:46:24 +08:00
parent 07b23027c1
commit 7b8b91e9de

View File

@ -3,7 +3,6 @@ package http
import ( import (
"fmt" "fmt"
"image" "image"
"io"
"net/http" "net/http"
"net/url" "net/url"
@ -50,13 +49,18 @@ var previewHandler = withUser(func(w http.ResponseWriter, r *http.Request, d *da
switch file.Type { switch file.Type {
case "image": case "image":
return handleImagePreview(w, file, size) return handleImagePreview(w, r, file, size)
default: default:
return http.StatusNotImplemented, fmt.Errorf("can't create preview for %s type", file.Type) return http.StatusNotImplemented, fmt.Errorf("can't create preview for %s type", file.Type)
} }
}) })
func handleImagePreview(w io.Writer, file *files.FileInfo, size string) (int, error) { func handleImagePreview(w http.ResponseWriter, r *http.Request, file *files.FileInfo, size string) (int, error) {
// Unsupported extensions directly return the raw data
if file.Extension == ".ico" || file.Extension == ".svg" {
return rawFileHandler(w, r, file)
}
var imgProcessor imageProcessor var imgProcessor imageProcessor
switch size { switch size {
case sizeBig: case sizeBig: