From 36675bc4ff2d617b65160cefea4c04bd835cd92c Mon Sep 17 00:00:00 2001 From: Eliot Whalan Date: Fri, 24 Jun 2016 12:10:54 +1000 Subject: [PATCH] Add content headers --- main.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/main.go b/main.go index c70f545..c4e86bd 100644 --- a/main.go +++ b/main.go @@ -1,7 +1,6 @@ package main import ( - "bytes" "crypto/sha1" "database/sql" "encoding/base64" @@ -14,7 +13,6 @@ import ( "io/ioutil" "log" "net/http" - "time" "github.com/dchest/uniuri" "github.com/ewhal/pygments" @@ -261,7 +259,9 @@ func downloadHandler(w http.ResponseWriter, r *http.Request) { vars := mux.Vars(r) paste := vars["pasteId"] s := getPaste(paste, "") - http.ServeContent(w, r, paste, time.Now(), bytes.NewReader([]byte(s))) + w.Header().Set("Content-Disposition", "attachment; filename="+paste) + w.Header().Set("Content-Type", r.Header.Get("Content-Type")) + io.WriteString(w, s) } func rawHandler(w http.ResponseWriter, r *http.Request) {