contented/download.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", 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 {

	// 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
}