talos/hack/cloud-image-uploader/main.go
Andrey Smirnov 61cacb3956 docs: provide list of AMIs on AWS documentation page
I wasn't able to find a more elegant way to get Vue content into
Markdown, so a bit of a hack with the template. Looks like frontmatter
parsing is also incomplete, so I could rely only on the page title.

Also changed the format of the image export to make it much easier to
work on that data in Vue.

Fixes #2834

Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
2020-11-25 10:26:50 -08:00

88 lines
2.0 KiB
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 []CloudImage
// CloudImage is the record official cloud image.
type CloudImage struct {
Cloud string `json:"cloud"`
Tag string `json:"version"`
Region string `json:"region"`
Arch string `json:"arch"`
Type string `json:"type"`
ID string `json:"id"`
}
var (
result Result
resultMu sync.Mutex
)
func pushResult(image CloudImage) {
resultMu.Lock()
defer resultMu.Unlock()
result = append(result, image)
}
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
g.Go(func() error {
aws := AWSUploader{
Options: DefaultOptions,
}
if err := aws.Upload(); err != nil {
return err
}
return nil
})
if err := g.Wait(); err != nil {
log.Fatalf("failed: %s", err)
}
e := json.NewEncoder(os.Stdout)
e.SetIndent("", " ")
e.Encode(&result) //nolint: errcheck
}