finish PR UI

1 files changed, 28 insertions, 0 deletions

diff --git a/modules/git/commit.go b/modules/git/commit.go
index 4e254dcec1..da0ab64463 100644
--- a/modules/git/commit.go
+++ b/modules/git/commit.go
@@ -7,6 +7,7 @@ package git
 import (
 	"bufio"
 	"container/list"
+	"net/http"
 	"strings"
 )
 
@@ -132,3 +133,30 @@ func (c *Commit) GetSubModules() (map[string]*SubModule, error) {
 
 	return c.submodules, nil
 }
+
+func isImageFile(data []byte) (string, bool) {
+	contentType := http.DetectContentType(data)
+	if strings.Index(contentType, "image/") != -1 {
+		return contentType, true
+	}
+	return contentType, false
+}
+
+func (c *Commit) IsImageFile(name string) bool {
+	blob, err := c.GetBlobByPath(name)
+	if err != nil {
+		return false
+	}
+
+	dataRc, err := blob.Data()
+	if err != nil {
+		return false
+	}
+	buf := make([]byte, 1024)
+	n, _ := dataRc.Read(buf)
+	if n > 0 {
+		buf = buf[:n]
+	}
+	_, isImage := isImageFile(buf)
+	return isImage
+}