1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
package customizations
import (
"context"
"fmt"
"github.com/aws/smithy-go/middleware"
"github.com/aws/aws-sdk-go-v2/service/internal/s3shared"
)
// BackfillInput validates and backfill's values from ARN into request serializable input.
// This middleware must be executed after `ARNLookup` and before `inputValidationMiddleware`.
type BackfillInput struct {
// CopyInput creates a copy of input to be modified, this ensures the original input is not modified.
CopyInput func(interface{}) (interface{}, error)
// BackfillAccountID points to a function that validates the input for accountID. If absent, it populates the
// accountID and returns a copy. If present, but different than passed in accountID value throws an error
BackfillAccountID func(interface{}, string) error
}
// ID representing the middleware
func (m *BackfillInput) ID() string {
return "S3Control:BackfillInput"
}
// HandleInitialize handles the middleware behavior in an Initialize step.
func (m *BackfillInput) HandleInitialize(ctx context.Context, in middleware.InitializeInput, next middleware.InitializeHandler) (
out middleware.InitializeOutput, metadata middleware.Metadata, err error,
) {
// fetch arn from context
av, ok := s3shared.GetARNResourceFromContext(ctx)
if !ok {
return next.HandleInitialize(ctx, in)
}
// if not supported, move to next
if m.BackfillAccountID == nil {
return next.HandleInitialize(ctx, in)
}
// check if input is already cloned
if !s3shared.IsClonedInput(ctx) {
// create a copy of input, and assign it on params
in.Parameters, err = m.CopyInput(in.Parameters)
if err != nil {
return out, metadata, fmt.Errorf("error creating a copy of input")
}
// set clone key on context
ctx = s3shared.SetClonedInputKey(ctx, true)
}
// backfill account id
err = m.BackfillAccountID(in.Parameters, av.AccountID)
if err != nil {
return out, metadata, fmt.Errorf("invalid ARN, %w", err)
}
return next.HandleInitialize(ctx, in)
}
|
