From 940faf998cfec42d62c882d89e7c654974372eb2 Mon Sep 17 00:00:00 2001
From: Eliot Whalan <ewhal@pantsu.cat>
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) {