ecr_private_manager.go

package main

import (
	"context"

	"github.com/aws/aws-sdk-go-v2/service/ecr"
	"github.com/cenkalti/backoff"
	log "github.com/sirupsen/logrus"
)

type ecrPrivateManager struct {
	client       *ecr.Client     // AWS ECR client
	repositories map[string]bool // list of repositories in ECR
}

func (e *ecrPrivateManager) exists(name string) bool {
	if _, ok := e.repositories[name]; ok {
		return true
	}

	return false
}

func (e *ecrPrivateManager) ensure(name string) error {
	if e.exists(name) {
		return nil
	}

	return e.create(name)
}

func (e *ecrPrivateManager) create(name string) error {
	_, err := e.client.CreateRepository(context.TODO(), &ecr.CreateRepositoryInput{
		RepositoryName: &name,
	})

	if err != nil {
		return err
	}

	e.repositories[name] = true
	return nil
}

func (e *ecrPrivateManager) buildCache(nextToken *string) error {
	if nextToken == nil {
		log.Info("Loading list of ECR repositories")
	}

	resp, err := e.client.DescribeRepositories(context.TODO(), &ecr.DescribeRepositoriesInput{
		NextToken: nextToken,
	})
	if err != nil {
		return err
	}

	if e.repositories == nil {
		e.repositories = make(map[string]bool)
	}

	for _, repo := range resp.Repositories {
		e.repositories[*repo.RepositoryName] = true
	}

	// keep paging as long as there is a token for the next page
	if resp.NextToken != nil {
		e.buildCache(resp.NextToken)
	}

	// no next token means we hit the last page
	if nextToken == nil {
		log.Info("Done loading ECR repositories")
	}

	return nil
}

func (e *ecrPrivateManager) buildCacheBackoff() backoff.Operation {
	return func() error {
		return e.buildCache(nil)
	}
}