Skip to content

Commit dc13c25

Browse files
authored
Add CloudFront Origin Access Control (rebuy-de#1019) (rebuy-de#1021)
* Add CloudFront Origin Access Control (rebuy-de#1019) * Add missing pagination and run go fmt
1 parent 4e41a7f commit dc13c25

File tree

1 file changed

+66
-0
lines changed

1 file changed

+66
-0
lines changed
Lines changed: 66 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,66 @@
1+
package resources
2+
3+
import (
4+
"github.com/aws/aws-sdk-go/aws/session"
5+
"github.com/aws/aws-sdk-go/service/cloudfront"
6+
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
7+
)
8+
9+
type CloudFrontOriginAccessControl struct {
10+
svc *cloudfront.CloudFront
11+
ID *string
12+
}
13+
14+
func init() {
15+
register("CloudFrontOriginAccessControl", ListCloudFrontOriginAccessControls)
16+
}
17+
18+
func ListCloudFrontOriginAccessControls(sess *session.Session) ([]Resource, error) {
19+
svc := cloudfront.New(sess)
20+
resources := []Resource{}
21+
params := &cloudfront.ListOriginAccessControlsInput{}
22+
23+
for {
24+
resp, err := svc.ListOriginAccessControls(params)
25+
if err != nil {
26+
return nil, err
27+
}
28+
29+
for _, item := range resp.OriginAccessControlList.Items {
30+
resources = append(resources, &CloudFrontOriginAccessControl{
31+
svc: svc,
32+
ID: item.Id,
33+
})
34+
}
35+
36+
if !*resp.OriginAccessControlList.IsTruncated {
37+
break
38+
}
39+
40+
params.Marker = resp.OriginAccessControlList.NextMarker
41+
}
42+
43+
return resources, nil
44+
}
45+
46+
func (f *CloudFrontOriginAccessControl) Remove() error {
47+
resp, err := f.svc.GetOriginAccessControl(&cloudfront.GetOriginAccessControlInput{
48+
Id: f.ID,
49+
})
50+
if err != nil {
51+
return err
52+
}
53+
54+
_, err = f.svc.DeleteOriginAccessControl(&cloudfront.DeleteOriginAccessControlInput{
55+
Id: f.ID,
56+
IfMatch: resp.ETag,
57+
})
58+
59+
return err
60+
}
61+
62+
func (f *CloudFrontOriginAccessControl) Properties() types.Properties {
63+
properties := types.NewProperties()
64+
properties.Set("ID", f.ID)
65+
return properties
66+
}

0 commit comments

Comments
 (0)