autoroll/go/repo_manager/child/revision_filter/cipd.go

package revision_filter

import (
	"context"
	"fmt"
	"net/http"
	"strings"

	"go.skia.org/infra/autoroll/go/config"
	"go.skia.org/infra/autoroll/go/revision"
	"go.skia.org/infra/go/cipd"
	"go.skia.org/infra/go/skerr"
)

// CIPDRevisionFilter is a RevisionFilter which filters out Revisions which
// don't exist on all of the configured packages and platforms. Only works with
// rollers who use a tag as the Revision ID, eg. git_revision.
type CIPDRevisionFilter struct {
	client    cipd.CIPDClient
	packages  []string
	platforms []string
	tagKey    string
}

// Skip implements RevisionFilter.
func (f *CIPDRevisionFilter) Skip(ctx context.Context, r revision.Revision) (string, error) {
	tag := r.Id
	if f.tagKey != "" {
		tag = fmt.Sprintf("%s:%s", f.tagKey, tag)
	}
	if len(strings.Split(tag, ":")) != 2 {
		return fmt.Sprintf("%q doesn't follow CIPD tag format", tag), nil
	}
	for _, pkg := range f.packages {
		for _, platform := range f.platforms {
			pkgFullPath := pkg + "/" + platform
			// Note that this only works for rollers which use a tag as the
			// Revision ID.  If the revision ID is the package ID, it is likely
			// that this request will fail because the ID doesn't follow the
			// expected "key:value" tag format, and if it doesn't fail it will
			// only ever return an empty set of results because the package ID
			// isn't a tag.
			pins, err := f.client.SearchInstances(ctx, pkgFullPath, []string{tag})
			if err != nil {
				return "", skerr.Wrap(err)
			}
			if len(pins) == 0 {
				return fmt.Sprintf("CIPD package %q does not exist at tag %q", pkgFullPath, tag), nil
			}
		}
	}
	return "", nil
}

// Update implements RevisionFilter.
func (f *CIPDRevisionFilter) Update(_ context.Context) error {
	return nil
}

// NewCIPDRevisionFilter returns a RevisionFilter which filters out Revisions
// which don't exist on all of the configured packages and platforms.
func NewCIPDRevisionFilter(client *http.Client, cfg *config.CIPDRevisionFilterConfig, workdir string) (*CIPDRevisionFilter, error) {
	cipdClient, err := cipd.NewClient(client, workdir, cipd.DefaultServiceURL)
	if err != nil {
		return nil, skerr.Wrap(err)
	}
	return &CIPDRevisionFilter{
		client:    cipdClient,
		packages:  cfg.Package,
		platforms: cfg.Platform,
		tagKey:    cfg.TagKey,
	}, nil
}

var _ RevisionFilter = &CIPDRevisionFilter{}