Do not show non-image attachment in a <img> tag. Fixes #3215 (#3311)

2016-08-11 20:16:36 -03:00 · 2016-08-11 20:16:36 -03:00 · 25b23c4bc9
commit 25b23c4bc9
parent 7eafe3213f
4 changed files with 29 additions and 2 deletions
--- a/modules/template/template.go
+++ b/modules/template/template.go
@ -9,6 +9,8 @@ import (
 	"encoding/json"
 	"fmt"
 	"html/template"
+	"mime"
+	"path/filepath"
 	"runtime"
 	"strings"
 	"time"
@ -103,6 +105,10 @@ func NewFuncMap() []template.FuncMap {
 		"ThemeColorMetaTag": func() string {
 			return setting.UI.ThemeColorMetaTag
 		},
+		"FilenameIsImage": func(filename string) bool {
+			mimeType := mime.TypeByExtension(filepath.Ext(filename))
+			return strings.HasPrefix(mimeType, "image/")
+		},
 	}}
 }