2017-04-03 14:08:01 +02:00
|
|
|
package main
|
2017-03-30 22:48:27 +02:00
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
"github.com/gorilla/mux"
|
|
|
|
)
|
|
|
|
|
2017-04-03 14:08:01 +02:00
|
|
|
// GetObjectHandler downloads an object to the client
|
|
|
|
func GetObjectHandler(s3 S3Client) http.Handler {
|
2017-03-30 22:48:27 +02:00
|
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
vars := mux.Vars(r)
|
|
|
|
objectName := vars["objectName"]
|
|
|
|
|
|
|
|
object, err := s3.GetObject(vars["bucketName"], objectName)
|
|
|
|
if err != nil {
|
|
|
|
msg := "error getting object"
|
2017-04-03 14:08:01 +02:00
|
|
|
handleHTTPError(w, msg, err, http.StatusInternalServerError)
|
2017-03-30 22:48:27 +02:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", objectName))
|
|
|
|
w.Header().Set("Content-Type", "application/octet-stream")
|
|
|
|
|
|
|
|
_, err = io.Copy(w, object)
|
|
|
|
if err != nil {
|
|
|
|
msg := "error copying object"
|
2017-04-03 14:08:01 +02:00
|
|
|
handleHTTPError(w, msg, err, http.StatusInternalServerError)
|
2017-03-30 22:48:27 +02:00
|
|
|
return
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|