contented/download.go

49 lines
1.1 KiB
Go

package contented
import (
"log"
"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", this.remoteIP(r), 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 {
// 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
}