Added download param to getFile api call

1 files changed, 11 insertions, 8 deletions

diff --git a/api/file.go b/api/file.go
index 46e81691e..bfaca915f 100644
--- a/api/file.go
+++ b/api/file.go
@@ -379,6 +379,7 @@ func getFile(c *Context, w http.ResponseWriter, r *http.Request) {
 	hash := r.URL.Query().Get("h")
 	data := r.URL.Query().Get("d")
 	teamId := r.URL.Query().Get("t")
+	isDownload := r.URL.Query().Get("download") == "1"
 
 	cchan := Srv.Store.Channel().CheckPermissionsTo(c.Session.TeamId, channelId, c.Session.UserId)
 
@@ -420,16 +421,18 @@ func getFile(c *Context, w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Length", strconv.Itoa(len(f)))
 	w.Header().Del("Content-Type") // Content-Type will be set automatically by the http writer
 
-	// attach extra headers to trigger a download on IE, Edge, and Safari
-	ua := user_agent.New(r.UserAgent())
-	bname, _ := ua.Browser()
+	if isDownload {
+		// attach extra headers to trigger a download on IE, Edge, and Safari
+		ua := user_agent.New(r.UserAgent())
+		bname, _ := ua.Browser()
 
-	if bname == "Edge" || bname == "Internet Explorer" || bname == "Safari" {
-		// trim off anything before the final / so we just get the file's name
-		parts := strings.Split(filename, "/")
+		if bname == "Edge" || bname == "Internet Explorer" || bname == "Safari" {
+			// trim off anything before the final / so we just get the file's name
+			parts := strings.Split(filename, "/")
 
-		w.Header().Set("Content-Type", "application/octet-stream")
-		w.Header().Set("Content-Disposition", "attachment;filename=\""+parts[len(parts)-1]+"\"")
+			w.Header().Set("Content-Type", "application/octet-stream")
+			w.Header().Set("Content-Disposition", "attachment;filename=\""+parts[len(parts)-1]+"\"")
+		}
 	}
 
 	w.Write(f)