From d1b45be080033e1fc3c3f0d3896bfe185a858f09 Mon Sep 17 00:00:00 2001
From: Emman <eygohlolz@gmail.com>
Date: Wed, 28 Feb 2018 01:04:49 +0800
Subject: [PATCH] Add FileAttachment method to context to allow instant
 downloads with filenames

---
 context.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/context.go b/context.go
index 90d4c6e5..55b3fdf4 100644
--- a/context.go
+++ b/context.go
@@ -6,6 +6,7 @@ package gin
 
 import (
 	"errors"
+	"fmt"
 	"io"
 	"io/ioutil"
 	"math"
@@ -714,6 +715,13 @@ func (c *Context) File(filepath string) {
 	http.ServeFile(c.Writer, c.Request, filepath)
 }
 
+// FileAttachment writes the specified file into the body stream in an efficient way
+// On the client side, the file will typically be downloaded with the given filename
+func (c *Context) FileAttachment(filepath, filename string) {
+	c.Writer.Header().Set("content-disposition", fmt.Sprintf("attachment; filename=\"%s\"", filename))
+	http.ServeFile(c.Writer, c.Request, filepath)
+}
+
 // SSEvent writes a Server-Sent Event into the body stream.
 func (c *Context) SSEvent(name string, message interface{}) {
 	c.Render(-1, sse.Event{