-
Notifications
You must be signed in to change notification settings - Fork 9
/
packager.go
40 lines (32 loc) · 855 Bytes
/
packager.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package packagers
import (
"fmt"
"io"
)
//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 -generate
type Packager struct {
source Source
stemcellGenerator StemcellGenerator
}
//counterfeiter:generate . Source
type Source interface {
ArtifactReader() (io.Reader, error)
}
//counterfeiter:generate . StemcellGenerator
type StemcellGenerator interface {
Generate(reader io.Reader) error
}
func NewPackager(s Source, g StemcellGenerator) *Packager {
return &Packager{source: s, stemcellGenerator: g}
}
func (p *Packager) Package() error {
artifact, err := p.source.ArtifactReader()
if err != nil {
return fmt.Errorf("packager failed to retrieve artifact: %s", err)
}
err = p.stemcellGenerator.Generate(artifact)
if err != nil {
return fmt.Errorf("packager failed to generate stemcell: %s", err)
}
return nil
}