helm.go

package stages

import (
	"context"

	"github.com/pkg/errors"
	log "github.com/sirupsen/logrus"

	kargoapi "github.com/akuity/kargo/api/v1alpha1"
	"github.com/akuity/kargo/internal/credentials"
	"github.com/akuity/kargo/internal/helm"
	"github.com/akuity/kargo/internal/logging"
)

func (r *reconciler) getLatestCharts(
	ctx context.Context,
	namespace string,
	subs []kargoapi.ChartSubscription,
) ([]kargoapi.Chart, error) {
	charts := make([]kargoapi.Chart, len(subs))

	for i, sub := range subs {
		logger := logging.LoggerFromContext(ctx).WithFields(log.Fields{
			"registry": sub.RegistryURL,
			"chart":    sub.Name,
		})

		creds, ok, err :=
			r.credentialsDB.Get(ctx, namespace, credentials.TypeHelm, sub.RegistryURL)
		if err != nil {
			return nil, errors.Wrapf(
				err,
				"error obtaining credentials for chart registry %q",
				sub.RegistryURL,
			)
		}

		var helmCreds *helm.Credentials
		if ok {
			helmCreds = &helm.Credentials{
				Username: creds.Username,
				Password: creds.Password,
			}
			logger.Debug("obtained credentials for chart repo")
		} else {
			logger.Debug("found no credentials for chart repo")
		}

		vers, err := r.getLatestChartVersionFn(
			ctx,
			sub.RegistryURL,
			sub.Name,
			sub.SemverConstraint,
			helmCreds,
		)
		if err != nil {
			return nil, errors.Wrapf(
				err,
				"error searching for latest version of chart %q in registry %q",
				sub.Name,
				sub.RegistryURL,
			)
		}

		if vers == "" {
			logger.Error("found no suitable chart version")
			return nil, errors.Errorf(
				"found no suitable version of chart %q in registry %q",
				sub.Name,
				sub.RegistryURL,
			)
		}
		logger.WithField("version", vers).
			Debug("found latest suitable chart version")

		charts[i] = kargoapi.Chart{
			RegistryURL: sub.RegistryURL,
			Name:        sub.Name,
			Version:     vers,
		}
	}

	return charts, nil
}