contented/download.go

49 lines
1.1 KiB
Go
Raw Normal View History

2017-10-08 02:53:00 +00:00
package contented
import (
"log"
2017-10-08 02:53:00 +00:00
"net/http"
"os"
"path/filepath"
)
func (this *Server) handleView(w http.ResponseWriter, r *http.Request, fileID string) {
err := this.handleViewInternal(w, r, r.URL.Path[len(downloadUrlPrefix):])
if err != nil {
log.Printf("%s View failed: %s\n", r.RemoteAddr, err.Error())
if os.IsNotExist(err) {
http.Error(w, "File not found", 404)
} else {
http.Error(w, "Couldn't provide content", 500)
}
}
}
func (this *Server) handleViewInternal(w http.ResponseWriter, r *http.Request, fileID string) error {
2017-10-08 02:53:00 +00:00
// Load metadata
m, err := this.Metadata(fileID)
if err != nil {
return err
}
// Load file
f, err := os.Open(filepath.Join(this.opts.DataDirectory, m.FileHash))
if err != nil {
return err
}
defer f.Close()
// ServeContent only uses the filename to get the mime type, which we can
// set accurately (including blacklist)
w.Header().Set(`Content-Type`, m.MimeType)
if m.MimeType == `application/octet-stream` {
w.Header().Set(`Content-Disposition`, `attachment; filename="`+m.Filename+`"`)
}
http.ServeContent(w, r, "", m.UploadTime, f)
return nil
}