/
r53_dns.go
98 lines (86 loc) · 3.17 KB
/
r53_dns.go
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package aws
import (
"context"
"errors"
"log"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/config"
r53 "github.com/aws/aws-sdk-go-v2/service/route53"
r53types "github.com/aws/aws-sdk-go-v2/service/route53/types"
)
// S3WebSiteEndPointEuCentral1 is the s3 endpoint for eu-central-1
// s3-website.eu-central-1.amazonaws.com
const S3WebSiteEndPointEuCentral1 = ".s3-website.eu-central-1.amazonaws.com"
// R53API defines the interface for the CreateHostedZone function.
// We use this interface to test the function using a mocked service.
type R53API interface {
ChangeResourceRecordSets(ctx context.Context,
params *r53.ChangeResourceRecordSetsInput,
optFns ...func(*r53.Options)) (*r53.ChangeResourceRecordSetsOutput, error)
// CreateHostedZone(ctx context.Context,
// params *r53.CreateHostedZoneInput,
// optFns ...func(*r53.Options)) (*r53.CreateHostedZoneOutput, error)
}
// // R53ChangeResourceRecordSetsAPI defines the interface for the CreateHostedZone function.
// // We use this interface to test the function using a mocked service.
// type R53ChangeResourceRecordSetsAPI interface {
// ChangeResourceRecordSets(ctx context.Context,
// params *r53.ChangeResourceRecordSetsInput,
// optFns ...func(*r53.Options)) (*r53.ChangeResourceRecordSetsOutput, error)
// }
// NewR53Client initializes a new aws R53 client.
func NewR53Client() (*r53.Client, error) {
ctx := context.TODO()
// Load the Shared AWS Configuration (~/.aws/config)
cfg, err := config.LoadDefaultConfig(ctx)
if err != nil {
log.Fatal(err)
}
// Create an Amazon S3 service client
client := r53.NewFromConfig(cfg)
return client, err
}
// MakeRoutes is used to create an R53 route for s3 bucket with website config
// input:
//
// s3 website endpoint (https://docs.aws.amazon.com/general/latest/gr/s3.html#s3_website_region_endpoints)
// dns name of the website
// dns zone id
//
// example endpoint http://www.testwebsite.devops.lk.s3-website.eu-central-1.amazonaws.com
func MakeRoutes(c context.Context, client R53API, dnsname, zoneid string) (string, error) {
s3websiteendpoint := dnsname + S3WebSiteEndPointEuCentral1
input := &r53.ChangeResourceRecordSetsInput{
ChangeBatch: &r53types.ChangeBatch{
Changes: []r53types.Change{
{
Action: r53types.ChangeActionCreate,
ResourceRecordSet: &r53types.ResourceRecordSet{
Name: aws.String(dnsname),
AliasTarget: &r53types.AliasTarget{
DNSName: &s3websiteendpoint,
HostedZoneId: hostedZoneIDByS3EndpointRegion("eu-central-1"),
},
Type: r53types.RRTypeA,
},
},
},
Comment: aws.String("Web server for example.com"),
},
HostedZoneId: aws.String(zoneid),
}
_, err := createRoute(c, client, input, zoneid)
if err != nil {
log.Println("Could not create record ")
log.Fatal(err)
return "", errors.New("Could not create R53 record")
}
return dnsname, nil
}
func hostedZoneIDByS3EndpointRegion(region string) *string {
zoneid := string(zonemap[region])
return &zoneid
}
func createRoute(c context.Context, api R53API, input *r53.ChangeResourceRecordSetsInput, hostedzone string) (*r53.ChangeResourceRecordSetsOutput, error) {
return api.ChangeResourceRecordSets(c, input)
}