frostfs-s3-gw/api/layer/container.go

package layer

import (
	"context"
	"fmt"
	"strconv"

	"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api"
	"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/data"
	apierr "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/errors"
	"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/layer/frostfs"
	"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/middleware"
	"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/internal/logs"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container"
	"go.uber.org/zap"
)

const (
	attributeLocationConstraint = ".s3-location-constraint"
	AttributeLockEnabled        = "LockEnabled"
)

func (n *Layer) containerInfo(ctx context.Context, prm frostfs.PrmContainer) (*data.BucketInfo, error) {
	var (
		err error
		res *container.Container
		log = n.reqLogger(ctx).With(zap.Stringer("cid", prm.ContainerID))

		info = &data.BucketInfo{
			CID:  prm.ContainerID,
			Name: prm.ContainerID.EncodeToString(),
		}

		reqInfo = middleware.GetReqInfo(ctx)
	)

	res, err = n.frostFS.Container(ctx, prm)
	if err != nil {
		if client.IsErrContainerNotFound(err) {
			return nil, fmt.Errorf("%w: %s", apierr.GetAPIError(apierr.ErrNoSuchBucket), err.Error())
		}
		return nil, fmt.Errorf("get frostfs container: %w", err)
	}

	cnr := *res

	info.Owner = cnr.Owner()
	if domain := container.ReadDomain(cnr); domain.Name() != "" {
		info.Name = domain.Name()
		info.Zone = domain.Zone()
	}
	info.Created = container.CreatedAt(cnr)
	info.LocationConstraint = cnr.Attribute(attributeLocationConstraint)
	info.HomomorphicHashDisabled = container.IsHomomorphicHashingDisabled(cnr)

	attrLockEnabled := cnr.Attribute(AttributeLockEnabled)
	if len(attrLockEnabled) > 0 {
		info.ObjectLockEnabled, err = strconv.ParseBool(attrLockEnabled)
		if err != nil {
			log.Error(logs.CouldNotParseContainerObjectLockEnabledAttribute,
				zap.String("lock_enabled", attrLockEnabled),
				zap.Error(err),
			)
		}
	}

	zone := n.features.FormContainerZone(reqInfo.Namespace)
	if zone != info.Zone {
		return nil, fmt.Errorf("ns '%s' and zone '%s' are mismatched for container '%s'", zone, info.Zone, prm.ContainerID)
	}

	n.cache.PutBucket(info)

	return info, nil
}

func (n *Layer) containerList(ctx context.Context) ([]*data.BucketInfo, error) {
	stoken := n.SessionTokenForRead(ctx)

	prm := frostfs.PrmUserContainers{
		UserID:       n.BearerOwner(ctx),
		SessionToken: stoken,
	}

	res, err := n.frostFS.UserContainers(ctx, prm)
	if err != nil {
		n.reqLogger(ctx).Error(logs.CouldNotListUserContainers, zap.Error(err))
		return nil, err
	}

	list := make([]*data.BucketInfo, 0, len(res))
	for i := range res {
		getPrm := frostfs.PrmContainer{
			ContainerID:  res[i],
			SessionToken: stoken,
		}
		info, err := n.containerInfo(ctx, getPrm)
		if err != nil {
			n.reqLogger(ctx).Error(logs.CouldNotFetchContainerInfo, zap.Error(err))
			continue
		}

		list = append(list, info)
	}

	return list, nil
}

func (n *Layer) createContainer(ctx context.Context, p *CreateBucketParams) (*data.BucketInfo, error) {
	if p.LocationConstraint == "" {
		p.LocationConstraint = api.DefaultLocationConstraint // s3tests_boto3.functional.test_s3:test_bucket_get_location
	}

	zone := n.features.FormContainerZone(p.Namespace)

	bktInfo := &data.BucketInfo{
		Name:               p.Name,
		Zone:               zone,
		Owner:              n.BearerOwner(ctx),
		Created:            TimeNow(ctx),
		LocationConstraint: p.LocationConstraint,
		ObjectLockEnabled:  p.ObjectLockEnabled,
	}

	attributes := [][2]string{
		{attributeLocationConstraint, p.LocationConstraint},
	}

	if p.ObjectLockEnabled {
		attributes = append(attributes, [2]string{
			AttributeLockEnabled, "true",
		})
	}

	res, err := n.frostFS.CreateContainer(ctx, frostfs.PrmContainerCreate{
		Creator:              bktInfo.Owner,
		Policy:               p.Policy,
		Name:                 p.Name,
		Zone:                 zone,
		SessionToken:         p.SessionContainerCreation,
		CreationTime:         bktInfo.Created,
		AdditionalAttributes: attributes,
	})
	if err != nil {
		return nil, fmt.Errorf("create container: %w", err)
	}

	bktInfo.CID = res.ContainerID
	bktInfo.HomomorphicHashDisabled = res.HomomorphicHashDisabled

	n.cache.PutBucket(bktInfo)

	return bktInfo, nil
}