compiled_package_repo.go

package pkg

import (
	"fmt"
	"sort"
	"strings"

	bosherr "github.com/cloudfoundry/bosh-agent/errors"

	biindex "github.com/cloudfoundry/bosh-init/index"
	birelpkg "github.com/cloudfoundry/bosh-init/release/pkg"
)

type CompiledPackageRecord struct {
	BlobID   string
	BlobSHA1 string
}

type CompiledPackageRepo interface {
	Save(birelpkg.Package, CompiledPackageRecord) error
	Find(birelpkg.Package) (CompiledPackageRecord, bool, error)
}

type compiledPackageRepo struct {
	index biindex.Index
}

func NewCompiledPackageRepo(index biindex.Index) CompiledPackageRepo {
	return &compiledPackageRepo{index: index}
}

func (cpr *compiledPackageRepo) Save(pkg birelpkg.Package, record CompiledPackageRecord) error {
	err := cpr.index.Save(cpr.pkgKey(pkg), record)

	if err != nil {
		return bosherr.WrapError(err, "Saving compiled package")
	}

	return nil
}

func (cpr *compiledPackageRepo) Find(pkg birelpkg.Package) (CompiledPackageRecord, bool, error) {
	var record CompiledPackageRecord

	err := cpr.index.Find(cpr.pkgKey(pkg), &record)
	if err != nil {
		if err == biindex.ErrNotFound {
			return record, false, nil
		}

		return record, false, bosherr.WrapError(err, "Finding compiled package")
	}

	return record, true, nil
}

type packageToCompiledPackageKey struct {
	PackageName string
	// Fingerprint of a package captures the sorted names of its dependencies
	// (but not the dependencies' fingerprints)
	PackageFingerprint string
	DependencyKey      string
}

func (cpr compiledPackageRepo) pkgKey(pkg birelpkg.Package) packageToCompiledPackageKey {
	return packageToCompiledPackageKey{
		PackageName:        pkg.Name,
		PackageFingerprint: pkg.Fingerprint,
		DependencyKey:      cpr.convertToDependencyKey(ResolveDependencies(&pkg)),
	}
}

func (cpr compiledPackageRepo) convertToDependencyKey(packages []*birelpkg.Package) string {
	dependencyKeys := []string{}
	for _, pkg := range packages {
		dependencyKeys = append(dependencyKeys, fmt.Sprintf("%s:%s", pkg.Name, pkg.Fingerprint))
	}
	sort.Strings(dependencyKeys)
	return strings.Join(dependencyKeys, ",")
}