aws.go

// Licensed to Michael Tougeron <github@e.tougeron.com> under
// one or more contributor license agreements. See the LICENSE
// file distributed with this work for additional information
// regarding copyright ownership.
// Michael Tougeron <github@e.tougeron.com> licenses this file
// to you under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package main

import (
	"fmt"
	"time"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/client"
	"github.com/aws/aws-sdk-go/aws/ec2metadata"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/ec2"
	"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
	"github.com/aws/aws-sdk-go/service/efs"
	"github.com/aws/aws-sdk-go/service/efs/efsiface"
	"github.com/aws/aws-sdk-go/service/fsx"
	"github.com/prometheus/client_golang/prometheus"
	log "github.com/sirupsen/logrus"
)

// awsSession the AWS Session
var awsSession *session.Session

const (
	// Matching strings for region
	regexpAWSRegion = `^[\w]{2}[-][\w]{4,9}[-][\d]$|^[\w]{2}[-][\w]{3}[-][\w]{4,9}[-][\d]$`
)

// Client efs interface
type EFSClient struct {
	efsiface.EFSAPI
}

// Client EC2 client interface
type EBSClient struct {
	ec2iface.EC2API
}

// FSx client
type FSxClient struct {
	*fsx.FSx
}

// CustomRetryer for custom retry settings
type CustomRetryer struct {
	client.DefaultRetryer
}

func createAWSSession(awsRegion string) *session.Session {
	// Build an AWS session
	log.Debugln("Building AWS session")
	awsConfig := aws.NewConfig().WithCredentialsChainVerboseErrors(true)
	awsConfig.Region = aws.String(awsRegion)
	minDelay, _ := time.ParseDuration("1s")
	maxDelay, _ := time.ParseDuration("10s")
	awsConfig.Retryer = CustomRetryer{DefaultRetryer: client.DefaultRetryer{
		NumMaxRetries:    5,
		MinRetryDelay:    minDelay,
		MaxRetryDelay:    maxDelay,
		MinThrottleDelay: minDelay,
		MaxThrottleDelay: maxDelay,
	}}

	return session.Must(session.NewSession(awsConfig))
}

// newEFSClient initializes an EFS client
func newEFSClient() (*EFSClient, error) {
	svc := efs.New(awsSession)
	return &EFSClient{svc}, nil
}

// newEC2Client initializes an EC2 client
func newEC2Client() (*EBSClient, error) {
	svc := ec2.New(awsSession)
	return &EBSClient{svc}, nil
}

// newFSxClient initializes an AWS client
func newFSxClient() (*FSxClient, error) {
	svc := fsx.New(awsSession)
	return &FSxClient{svc}, nil
}

func getMetadataRegion() (string, error) {
	sess := session.Must(session.NewSession(&aws.Config{}))
	svc := ec2metadata.New(sess)
	doc, err := svc.GetInstanceIdentityDocument()
	if err != nil {
		return "", fmt.Errorf("could not get EC2 instance identity metadata")
	}
	if len(doc.Region) == 0 {
		return "", fmt.Errorf("could not get valid EC2 region")
	}
	return doc.Region, nil
}

func (client *EBSClient) addEBSVolumeTags(volumeID string, tags map[string]string, storageclass string) {
	var ec2Tags []*ec2.Tag
	for k, v := range tags {
		ec2Tags = append(ec2Tags, &ec2.Tag{Key: aws.String(k), Value: aws.String(v)})
	}

	// Add tags to the volume
	_, err := client.CreateTags(&ec2.CreateTagsInput{
		Resources: []*string{aws.String(volumeID)},
		Tags:      ec2Tags,
	})
	if err != nil {
		log.Errorln("Could not create tags for volumeID:", volumeID, err)
		promActionsTotal.With(prometheus.Labels{"status": "error", "storageclass": storageclass}).Inc()
		promActionsLegacyTotal.With(prometheus.Labels{"status": "error"}).Inc()
		return
	}

	promActionsTotal.With(prometheus.Labels{"status": "success", "storageclass": storageclass}).Inc()
	promActionsLegacyTotal.With(prometheus.Labels{"status": "success"}).Inc()
}

func (client *EBSClient) deleteEBSVolumeTags(volumeID string, tags []string, storageclass string) {
	var ec2Tags []*ec2.Tag
	for _, k := range tags {
		ec2Tags = append(ec2Tags, &ec2.Tag{Key: aws.String(k)})
	}

	// Add tags to the volume
	_, err := client.DeleteTags(&ec2.DeleteTagsInput{
		Resources: []*string{aws.String(volumeID)},
		Tags:      ec2Tags,
	})
	if err != nil {
		log.Errorln("Could not EBS delete tags for volumeID:", volumeID, err)
		promActionsTotal.With(prometheus.Labels{"status": "error", "storageclass": storageclass}).Inc()
		promActionsLegacyTotal.With(prometheus.Labels{"status": "error"}).Inc()
		return
	}

	promActionsTotal.With(prometheus.Labels{"status": "success", "storageclass": storageclass}).Inc()
	promActionsLegacyTotal.With(prometheus.Labels{"status": "success"}).Inc()
}

func (client *EFSClient) addEFSVolumeTags(volumeID string, tags map[string]string, storageclass string) {
	var efsTags []*efs.Tag
	for k, v := range tags {
		efsTags = append(efsTags, &efs.Tag{Key: aws.String(k), Value: aws.String(v)})
	}

	// Add tags to the volume
	_, err := client.TagResource(&efs.TagResourceInput{
		ResourceId: aws.String(volumeID),
		Tags:       efsTags,
	})
	if err != nil {
		log.Errorln("Could not EFS create tags for volumeID:", volumeID, err)
		promActionsTotal.With(prometheus.Labels{"status": "error", "storageclass": storageclass}).Inc()
		promActionsLegacyTotal.With(prometheus.Labels{"status": "error"}).Inc()
		return
	}

	promActionsTotal.With(prometheus.Labels{"status": "success", "storageclass": storageclass}).Inc()
	promActionsLegacyTotal.With(prometheus.Labels{"status": "success"}).Inc()
}

func (client *EFSClient) deleteEFSVolumeTags(volumeID string, tags []string, storageclass string) {
	var efsTags []*string
	for _, k := range tags {
		efsTags = append(efsTags, aws.String(k))
	}

	// Add tags to the volume
	_, err := client.UntagResource(&efs.UntagResourceInput{
		ResourceId: aws.String(volumeID),
		TagKeys:    efsTags,
	})
	if err != nil {
		log.Errorln("Could not EFS delete tags for volumeID:", volumeID, err)
		promActionsTotal.With(prometheus.Labels{"status": "error", "storageclass": storageclass}).Inc()
		promActionsLegacyTotal.With(prometheus.Labels{"status": "error"}).Inc()
		return
	}

	promActionsTotal.With(prometheus.Labels{"status": "success", "storageclass": storageclass}).Inc()
	promActionsLegacyTotal.With(prometheus.Labels{"status": "success"}).Inc()
}

func (client *FSxClient) addFSxVolumeTags(volumeID string, tags map[string]string, storageclass string) {
	volumeIDs := []*string{&volumeID}
	describeFileSystemOutput, err := client.DescribeFileSystems(&fsx.DescribeFileSystemsInput{
		FileSystemIds: volumeIDs,
	})
	if err != nil {
		log.WithError(err)
		return
	}
	_, err = client.TagResource(&fsx.TagResourceInput{
		ResourceARN: describeFileSystemOutput.FileSystems[0].ResourceARN,
		Tags:        convertTagsToFSxTags(tags),
	})
	if err != nil {
		log.Errorln("Could not FSx create tags for volumeID:", volumeID, err)
		promActionsTotal.With(prometheus.Labels{"status": "error", "storageclass": storageclass}).Inc()
		promActionsLegacyTotal.With(prometheus.Labels{"status": "error"}).Inc()
		return
	}

	promActionsTotal.With(prometheus.Labels{"status": "success", "storageclass": storageclass}).Inc()
	promActionsLegacyTotal.With(prometheus.Labels{"status": "success"}).Inc()
}

func (client *FSxClient) deleteFSxVolumeTags(volumeID string, tags []*string, storageclass string) {
	volumeIDs := []*string{&volumeID}
	describeVolumesOutput, err := client.DescribeVolumes(&fsx.DescribeVolumesInput{
		VolumeIds: volumeIDs,
	})
	if err != nil {
		log.WithError(err)
		return
	}
	_, err = client.UntagResource(&fsx.UntagResourceInput{
		ResourceARN: describeVolumesOutput.Volumes[0].ResourceARN,
		TagKeys:     tags,
	})
	if err != nil {
		log.Errorln("Could not FSx delete tags for volumeID:", volumeID, err)
		promActionsTotal.With(prometheus.Labels{"status": "error", "storageclass": storageclass}).Inc()
		promActionsLegacyTotal.With(prometheus.Labels{"status": "error"}).Inc()
		return
	}

	promActionsTotal.With(prometheus.Labels{"status": "success", "storageclass": storageclass}).Inc()
	promActionsLegacyTotal.With(prometheus.Labels{"status": "success"}).Inc()
}