73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/creds/accessbox"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
|
)
|
|
|
|
// keyWrapper is wrapper for context keys.
|
|
type keyWrapper string
|
|
|
|
// boxKey is an ID used to store Box in a context.
|
|
var boxKey = keyWrapper("__context_box_key")
|
|
|
|
// GetBoxData extracts accessbox.Box from context.
|
|
func GetBoxData(ctx context.Context) (*accessbox.Box, error) {
|
|
data, ok := ctx.Value(boxKey).(*Box)
|
|
if !ok || data == nil {
|
|
return nil, fmt.Errorf("couldn't get box from context")
|
|
}
|
|
|
|
if data.AccessBox == nil {
|
|
return nil, fmt.Errorf("couldn't get box data from context")
|
|
}
|
|
|
|
box := data.AccessBox
|
|
if box.Gate == nil {
|
|
box.Gate = &accessbox.GateData{}
|
|
}
|
|
return box, nil
|
|
}
|
|
|
|
// GetAuthHeaders extracts auth.AuthHeader from context.
|
|
func GetAuthHeaders(ctx context.Context) (*AuthHeader, error) {
|
|
data, ok := ctx.Value(boxKey).(*Box)
|
|
if !ok || data == nil {
|
|
return nil, fmt.Errorf("couldn't get box from context")
|
|
}
|
|
|
|
return data.AuthHeaders, nil
|
|
}
|
|
|
|
// GetClientTime extracts time.Time from context.
|
|
func GetClientTime(ctx context.Context) (time.Time, error) {
|
|
data, ok := ctx.Value(boxKey).(*Box)
|
|
if !ok || data == nil {
|
|
return time.Time{}, fmt.Errorf("couldn't get box from context")
|
|
}
|
|
|
|
if data.ClientTime.IsZero() {
|
|
return time.Time{}, fmt.Errorf("couldn't get client time from context")
|
|
}
|
|
|
|
return data.ClientTime, nil
|
|
}
|
|
|
|
// GetAccessBoxAttrs extracts []object.Attribute from context.
|
|
func GetAccessBoxAttrs(ctx context.Context) ([]object.Attribute, error) {
|
|
data, ok := ctx.Value(boxKey).(*Box)
|
|
if !ok || data == nil {
|
|
return nil, fmt.Errorf("couldn't get box from context")
|
|
}
|
|
|
|
return data.Attributes, nil
|
|
}
|
|
|
|
// SetBox sets Box in the context.
|
|
func SetBox(ctx context.Context, box *Box) context.Context {
|
|
return context.WithValue(ctx, boxKey, box)
|
|
}
|