talos/hack/cloud-image-uploader/main.go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

package main

import (
	cryptorand "crypto/rand"
	"encoding/binary"
	"encoding/json"
	"log"
	"math/rand"
	"os"
	"sync"

	"github.com/aws/aws-sdk-go/aws/endpoints"
	"github.com/spf13/pflag"
	"golang.org/x/sync/errgroup"
)

// Result of the upload process.
type Result struct {
	AWS AWSResult `json:"aws,omitempty"`
}

func main() {
	for region := range endpoints.AwsPartition().Regions() {
		DefaultOptions.AWSRegions = append(DefaultOptions.AWSRegions, region)
	}

	pflag.StringSliceVar(&DefaultOptions.Architectures, "architectures", DefaultOptions.Architectures, "list of architectures to process")
	pflag.StringVar(&DefaultOptions.ArtifactsPath, "artifacts-path", DefaultOptions.ArtifactsPath, "artifacts path")
	pflag.StringVar(&DefaultOptions.Tag, "tag", DefaultOptions.Tag, "tag (version) of the uploaded image")

	pflag.StringSliceVar(&DefaultOptions.AWSRegions, "aws-regions", DefaultOptions.AWSRegions, "list of AWS regions to upload to")

	pflag.Parse()

	seed := make([]byte, 8)
	if _, err := cryptorand.Read(seed); err != nil {
		log.Fatalf("error seeding rand: %s", err)
	}

	rand.Seed(int64(binary.LittleEndian.Uint64(seed)))

	var g errgroup.Group

	result := Result{}

	var mu sync.Mutex

	g.Go(func() error {
		aws := AWSUploader{
			Options: DefaultOptions,
		}

		if err := aws.Upload(); err != nil {
			return err
		}

		mu.Lock()
		defer mu.Unlock()

		result.AWS = aws.GetResult()

		return nil
	})

	if err := g.Wait(); err != nil {
		log.Fatalf("failed: %s", err)
	}

	json.NewEncoder(os.Stdout).Encode(&result) //nolint: errcheck
}