package layer import ( "context" "encoding/hex" "fmt" "os" "strconv" "strings" "time" "github.com/nspcc-dev/neofs-s3-gw/api" "github.com/nspcc-dev/neofs-s3-gw/api/data" "github.com/nspcc-dev/neofs-s3-gw/creds/accessbox" "github.com/nspcc-dev/neofs-sdk-go/object" ) type ( // ListObjectsInfo contains common fields of data for ListObjectsV1 and ListObjectsV2. ListObjectsInfo struct { Prefixes []string Objects []*data.ObjectInfo IsTruncated bool } // ListObjectsInfoV1 holds data which ListObjectsV1 returns. ListObjectsInfoV1 struct { ListObjectsInfo NextMarker string } // ListObjectsInfoV2 holds data which ListObjectsV2 returns. ListObjectsInfoV2 struct { ListObjectsInfo NextContinuationToken string } // ListObjectVersionsInfo stores info and list of objects versions. ListObjectVersionsInfo struct { CommonPrefixes []string IsTruncated bool KeyMarker string NextKeyMarker string NextVersionIDMarker string Version []*data.ExtendedObjectInfo DeleteMarker []*data.ExtendedObjectInfo VersionIDMarker string } ) // PathSeparator is a path components separator string. const PathSeparator = string(os.PathSeparator) func userHeaders(attrs []object.Attribute) map[string]string { result := make(map[string]string, len(attrs)) for _, attr := range attrs { result[attr.Key()] = attr.Value() } return result } func objectInfoFromMeta(bkt *data.BucketInfo, meta *object.Object) *data.ObjectInfo { var ( mimeType string creation time.Time ) headers := userHeaders(meta.Attributes()) delete(headers, object.AttributeFileName) if contentType, ok := headers[object.AttributeContentType]; ok { mimeType = contentType delete(headers, object.AttributeContentType) } if val, ok := headers[object.AttributeTimestamp]; !ok { // ignore empty value } else if dt, err := strconv.ParseInt(val, 10, 64); err == nil { creation = time.Unix(dt, 0) delete(headers, object.AttributeTimestamp) } objID, _ := meta.ID() payloadChecksum, _ := meta.PayloadChecksum() return &data.ObjectInfo{ ID: objID, CID: bkt.CID, IsDir: false, EncryptionInfo: formEncryptionInfo(headers), Bucket: bkt.Name, Name: filenameFromObject(meta), Created: creation, ContentType: mimeType, Headers: headers, Owner: *meta.OwnerID(), Size: int64(meta.PayloadSize()), HashSum: hex.EncodeToString(payloadChecksum.Value()), } } func formEncryptionInfo(headers map[string]string) data.EncryptionInfo { algorithm := headers[AttributeEncryptionAlgorithm] return data.EncryptionInfo{ Enabled: len(algorithm) > 0, Algorithm: algorithm, HMACKey: headers[AttributeHMACKey], HMACSalt: headers[AttributeHMACSalt], } } func addEncryptionHeaders(meta map[string]string, enc EncryptionParams) error { meta[AttributeEncryptionAlgorithm] = AESEncryptionAlgorithm hmacKey, hmacSalt, err := enc.HMAC() if err != nil { return fmt.Errorf("get hmac: %w", err) } meta[AttributeHMACKey] = hex.EncodeToString(hmacKey) meta[AttributeHMACSalt] = hex.EncodeToString(hmacSalt) return nil } // processObjectInfoName fixes name in objectInfo structure based on prefix and // delimiter from user request. If name does not contain prefix, nil value is // returned. If name should be modified, then function returns copy of objectInfo // structure. func processObjectInfoName(oi *data.ObjectInfo, prefix, delimiter string) *data.ObjectInfo { if !strings.HasPrefix(oi.Name, prefix) { return nil } if len(delimiter) == 0 { return oi } copiedObjInfo := *oi tail := strings.TrimPrefix(copiedObjInfo.Name, prefix) index := strings.Index(tail, delimiter) if index >= 0 { copiedObjInfo.IsDir = true copiedObjInfo.Size = 0 copiedObjInfo.Headers = nil copiedObjInfo.ContentType = "" copiedObjInfo.Name = prefix + tail[:index+1] } return &copiedObjInfo } func filenameFromObject(o *object.Object) string { for _, attr := range o.Attributes() { if attr.Key() == object.AttributeFileName { return attr.Value() } } objID, _ := o.ID() return objID.EncodeToString() } // NameFromString splits name into a base file name and a directory path. func NameFromString(name string) (string, string) { ind := strings.LastIndex(name, PathSeparator) return name[ind+1:], name[:ind+1] } // GetBoxData extracts accessbox.Box from context. func GetBoxData(ctx context.Context) (*accessbox.Box, error) { var boxData *accessbox.Box data, ok := ctx.Value(api.BoxData).(*accessbox.Box) if !ok || data == nil { return nil, fmt.Errorf("couldn't get box data from context") } boxData = data if boxData.Gate == nil { boxData.Gate = &accessbox.GateData{} } return boxData, nil }