57 lines
1.3 KiB
Go
57 lines
1.3 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)
|
|
|
|
switch m.MimeType {
|
|
case `text/plain`:
|
|
w.Header().Set(`Content-Type`, `text/plain; charset=UTF-8`)
|
|
|
|
case `application/octet-stream`:
|
|
w.Header().Set(`Content-Type`, m.MimeType)
|
|
w.Header().Set(`Content-Disposition`, `attachment; filename="`+m.Filename+`"`)
|
|
|
|
default:
|
|
w.Header().Set(`Content-Type`, m.MimeType)
|
|
}
|
|
|
|
http.ServeContent(w, r, "", m.UploadTime, f)
|
|
|
|
return nil
|
|
}
|