attachment support.

Author: efectn

middleware/filesystem/filesystem.go

@@ -42,6 +42,11 @@ type Config struct {
 	// Optional. Default: "index.html"
 	Index string `json:"index"`
 
+	// When set to true, enables direct download for files.
+	//
+	// Optional. Default: false.
+	Download bool `json:"download"`
+
 	// The value for the Cache-Control HTTP-header
 	// that is set on the file response. MaxAge is defined in seconds.
 	//
@@ -197,6 +202,11 @@ func New(config ...Config) fiber.Handler {
 			c.Set(fiber.HeaderLastModified, modTime.UTC().Format(http.TimeFormat))
 		}
 
+		// Sets the response Content-Disposition header to attachment if the Download option is true and if it's a file
+		if cfg.Download && !stat.IsDir() {
+			c.Attachment()
+		}
+
 		if method == fiber.MethodGet {
 			if cfg.MaxAge > 0 {
 				c.Set(fiber.HeaderCacheControl, cacheControlStr)

middleware/filesystem/filesystem_test.go

@@ -107,7 +107,7 @@ func Test_FileSystem(t *testing.T) {
 			url:         "/spatest/doesnotexist",
 			statusCode:  200,
 			contentType: "text/html",
-		}, /******/
+		},
 		{
 			name:        "PathPrefix should be applied",
 			url:         "/prefix/fiber.png",
@@ -145,6 +145,22 @@ func Test_FileSystem_Next(t *testing.T) {
 	utils.AssertEqual(t, fiber.StatusNotFound, resp.StatusCode)
 }
 
+// go test -run Test_FileSystem_Download
+func Test_FileSystem_Download(t *testing.T) {
+	app := fiber.New()
+	app.Use(New(Config{
+		Root:     os.DirFS("../../.github/testdata/fs"),
+		Download: true,
+	}))
+
+	resp, err := app.Test(httptest.NewRequest("GET", "/img/fiber.png", nil))
+	utils.AssertEqual(t, nil, err, "app.Test(req)")
+	utils.AssertEqual(t, 200, resp.StatusCode, "Status code")
+	utils.AssertEqual(t, false, resp.Header.Get(fiber.HeaderContentLength) == "")
+	utils.AssertEqual(t, "image/png", resp.Header.Get(fiber.HeaderContentType))
+	utils.AssertEqual(t, `attachment`, resp.Header.Get(fiber.HeaderContentDisposition))
+}
+
 func Test_FileSystem_NonGetAndHead(t *testing.T) {
 	app := fiber.New()