Function handleStreamFileFromReader

pkg/web/web.go:247–325

(w http.ResponseWriter, r *http.Request, path string, no404 bool)

}

func handleStreamFileFromReader(w http.ResponseWriter, r *http.Request, path string, no404 bool) error {
	startTime := time.Now()
	rangeHeader := r.Header.Get("Range")
	log.Printf("stream-file path=%q range=%q\n", path, rangeHeader)

	writerRouteId, err := wshfs.GetConnectionRouteId(r.Context(), path)
	if err != nil {
		return err
	}

	byteRange := ""
	if rangeHeader != "" {
		stripped := strings.TrimPrefix(rangeHeader, "bytes=")
		br, parseErr := fileutil.ParseByteRange(stripped)
		if parseErr != nil || br.All {
			http.Error(w, "invalid range", http.StatusRequestedRangeNotSatisfiable)
			return nil
		}
		byteRange = stripped
	}

	bareRpc := wshclient.GetBareRpcClient()
	readerRouteId := wshclient.GetBareRpcClientRouteId()
	reader, streamMeta := bareRpc.StreamBroker.CreateStreamReader(readerRouteId, writerRouteId, 256*1024)
	defer reader.Close()
	go func() {
		<-r.Context().Done()
		reader.Close()
	}()

	data := wshrpc.CommandFileStreamData{
		Info:       &wshrpc.FileInfo{Path: path},
		ByteRange:  byteRange,
		StreamMeta: *streamMeta,
	}
	fileInfo, err := wshfs.FileStream(r.Context(), data)
	if err != nil {
		if no404 {
			serveTransparentGIF(w)
			return nil
		}
		return err
	}
	if fileInfo.NotFound {
		if no404 {
			serveTransparentGIF(w)
			return nil
		}
		http.Error(w, fmt.Sprintf("file not found: %q", path), http.StatusNotFound)
		return nil
	}
	if fileInfo.IsDir {
		http.Error(w, fmt.Sprintf("cannot stream directory: %q", path), http.StatusBadRequest)
		return nil
	}
	log.Printf("stream-file headers-ready path=%q time-to-headers=%v\n", path, time.Since(startTime))
	w.Header().Set(ContentTypeHeaderKey, fileInfo.MimeType)
	w.Header().Set("Accept-Ranges", "bytes")