fix image handling for non-jpegs

This commit is contained in:
2020-07-25 16:10:59 -04:00
parent 14de75921f
commit 4f06fa5eb4

View File

@ -6,7 +6,6 @@ import (
"image/gif"
"image/jpeg"
"image/png"
"path/filepath"
"strings"
)
@ -27,13 +26,15 @@ func renderImage(ext string, data []byte) (image.Image, error) {
}
func imageExtFromName(s string) string {
switch filepath.Ext(strings.Replace(s, "/", ".", 0)) {
case "gif":
return "gif"
case "png":
return "png"
case "jpg", "jpeg":
return "jpeg"
if strings.Contains(s, "/") {
switch strings.Split(s, "/")[1] {
case "gif":
return "gif"
case "png":
return "png"
case "jpg", "jpeg":
return "jpeg"
}
}
return ""
}