github.com/treeverse/lakefs@v1.24.1-0.20240520134607-95648127bfb0/pkg/gateway/operations/headobject.go (about)

     1  package operations
     2  
     3  import (
     4  	"errors"
     5  	"fmt"
     6  	"net/http"
     7  
     8  	"github.com/treeverse/lakefs/pkg/catalog"
     9  	gatewayerrors "github.com/treeverse/lakefs/pkg/gateway/errors"
    10  	"github.com/treeverse/lakefs/pkg/graveler"
    11  	"github.com/treeverse/lakefs/pkg/httputil"
    12  	"github.com/treeverse/lakefs/pkg/permissions"
    13  )
    14  
    15  type HeadObject struct{}
    16  
    17  func (controller *HeadObject) RequiredPermissions(_ *http.Request, repoID, _, path string) (permissions.Node, error) {
    18  	return permissions.Node{
    19  		Permission: permissions.Permission{
    20  			Action:   permissions.ReadObjectAction,
    21  			Resource: permissions.ObjectArn(repoID, path),
    22  		},
    23  	}, nil
    24  }
    25  
    26  func (controller *HeadObject) Handle(w http.ResponseWriter, req *http.Request, o *PathOperation) {
    27  	o.Incr("stat_object", o.Principal, o.Repository.Name, o.Reference)
    28  	entry, err := o.Catalog.GetEntry(req.Context(), o.Repository.Name, o.Reference, o.Path, catalog.GetEntryParams{})
    29  	if errors.Is(err, graveler.ErrNotFound) {
    30  		// TODO: create distinction between missing repo & missing key
    31  		o.Log(req).Debug("path not found")
    32  		_ = o.EncodeError(w, req, err, gatewayerrors.Codes.ToAPIErr(gatewayerrors.ErrNoSuchKey))
    33  		return
    34  	}
    35  	if err != nil {
    36  		o.Log(req).WithError(err).Error("failed querying path")
    37  		_ = o.EncodeError(w, req, err, gatewayerrors.Codes.ToAPIErr(gatewayerrors.ErrInternalError))
    38  		return
    39  	}
    40  	if entry.Expired {
    41  		o.Log(req).WithError(err).Info("querying expired object")
    42  		_ = o.EncodeError(w, req, err, gatewayerrors.Codes.ToAPIErr(gatewayerrors.ErrNoSuchVersion))
    43  		return
    44  	}
    45  
    46  	// range query
    47  	var rng httputil.Range
    48  	var rngErr error
    49  	// range query
    50  	rangeSpec := req.Header.Get("Range")
    51  	if len(rangeSpec) > 0 {
    52  		rng, rngErr = httputil.ParseRange(rangeSpec, entry.Size)
    53  		if rngErr != nil {
    54  			o.Log(req).WithError(err).WithField("range", rangeSpec).Debug("invalid range spec")
    55  			if errors.Is(rngErr, httputil.ErrUnsatisfiableRange) {
    56  				w.WriteHeader(http.StatusRequestedRangeNotSatisfiable)
    57  				return
    58  			}
    59  		}
    60  	}
    61  
    62  	o.SetHeader(w, "Accept-Ranges", "bytes")
    63  	o.SetHeader(w, "Last-Modified", httputil.HeaderTimestamp(entry.CreationDate))
    64  	o.SetHeader(w, "ETag", httputil.ETag(entry.Checksum))
    65  	o.SetHeader(w, "Content-Type", entry.ContentType)
    66  
    67  	amzMetaWriteHeaders(w, entry.Metadata)
    68  	if rangeSpec != "" && rngErr == nil {
    69  		o.SetHeader(w, "Content-Length", fmt.Sprintf("%d", rng.Size()))
    70  		o.SetHeader(w, "Content-Range", fmt.Sprintf("bytes %d-%d/%d", rng.StartOffset, rng.EndOffset, entry.Size))
    71  		w.WriteHeader(http.StatusPartialContent)
    72  	} else {
    73  		o.SetHeader(w, "Content-Length", fmt.Sprintf("%d", entry.Size))
    74  	}
    75  }