Add content headers

This commit is contained in:
Eliot Whalan 2016-06-24 12:10:54 +10:00
parent 0c8ff3ea04
commit 36675bc4ff
1 changed files with 3 additions and 3 deletions

View File

@ -1,7 +1,6 @@
package main package main
import ( import (
"bytes"
"crypto/sha1" "crypto/sha1"
"database/sql" "database/sql"
"encoding/base64" "encoding/base64"
@ -14,7 +13,6 @@ import (
"io/ioutil" "io/ioutil"
"log" "log"
"net/http" "net/http"
"time"
"github.com/dchest/uniuri" "github.com/dchest/uniuri"
"github.com/ewhal/pygments" "github.com/ewhal/pygments"
@ -261,7 +259,9 @@ func downloadHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r) vars := mux.Vars(r)
paste := vars["pasteId"] paste := vars["pasteId"]
s := getPaste(paste, "") 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) { func rawHandler(w http.ResponseWriter, r *http.Request) {