cloud_stemcell.go

package stemcell

import (
	bicloud "github.com/cloudfoundry/bosh-init/cloud"
	biconfig "github.com/cloudfoundry/bosh-init/config"
	bosherr "github.com/cloudfoundry/bosh-utils/errors"
)

type CloudStemcell interface {
	CID() string
	Name() string
	Version() string
	PromoteAsCurrent() error
	Delete() error
}

type cloudStemcell struct {
	cid     string
	name    string
	version string
	repo    biconfig.StemcellRepo
	cloud   bicloud.Cloud
}

func NewCloudStemcell(
	stemcellRecord biconfig.StemcellRecord,
	repo biconfig.StemcellRepo,
	cloud bicloud.Cloud,
) CloudStemcell {
	return &cloudStemcell{
		cid:     stemcellRecord.CID,
		name:    stemcellRecord.Name,
		version: stemcellRecord.Version,
		repo:    repo,
		cloud:   cloud,
	}
}

func (s *cloudStemcell) CID() string {
	return s.cid
}

func (s *cloudStemcell) Name() string {
	return s.name
}

func (s *cloudStemcell) Version() string {
	return s.version
}

func (s *cloudStemcell) PromoteAsCurrent() error {
	stemcellRecord, found, err := s.repo.Find(s.name, s.version)
	if err != nil {
		return bosherr.WrapError(err, "Finding current stemcell")
	}

	if !found {
		return bosherr.Error("Stemcell does not exist in repo")
	}

	err = s.repo.UpdateCurrent(stemcellRecord.ID)
	if err != nil {
		return bosherr.WrapError(err, "Updating current stemcell")
	}

	return nil
}

func (s *cloudStemcell) Delete() error {
	deleteErr := s.cloud.DeleteStemcell(s.cid)
	if deleteErr != nil {
		// allow StemcellNotFoundError for idempotency
		cloudErr, ok := deleteErr.(bicloud.Error)
		if !ok || cloudErr.Type() != bicloud.StemcellNotFoundError {
			return bosherr.WrapError(deleteErr, "Deleting stemcell from cloud")
		}
	}

	stemcellRecord, found, err := s.repo.Find(s.name, s.version)
	if err != nil {
		return bosherr.WrapErrorf(err, "Finding stemcell record (name=%s, version=%s)", s.name, s.version)
	}

	if !found {
		return nil
	}

	err = s.repo.Delete(stemcellRecord)
	if err != nil {
		return bosherr.WrapError(err, "Deleting stemcell record")
	}

	return deleteErr
}