forked from cloudfoundry-attic/bosh-init
-
Notifications
You must be signed in to change notification settings - Fork 0
/
installer.go
99 lines (82 loc) · 2.48 KB
/
installer.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package job
import (
"fmt"
"os"
"path"
"path/filepath"
biinstallblob "github.com/cloudfoundry/bosh-init/installation/blob"
biui "github.com/cloudfoundry/bosh-init/ui"
bosherr "github.com/cloudfoundry/bosh-utils/errors"
boshsys "github.com/cloudfoundry/bosh-utils/system"
)
type RenderedJobRef struct {
Name string
Version string
BlobstoreID string
SHA1 string
}
type InstalledJob struct {
RenderedJobRef
Path string
}
func NewInstalledJob(ref RenderedJobRef, path string) InstalledJob {
return InstalledJob{RenderedJobRef: ref, Path: path}
}
type Installer interface {
Install(RenderedJobRef, biui.Stage) (InstalledJob, error)
Cleanup(InstalledJob) error
}
func NewInstaller(
fs boshsys.FileSystem,
templateExtractor biinstallblob.Extractor,
jobsPath string,
) Installer {
return jobInstaller{
fs: fs,
templateExtractor: templateExtractor,
jobsPath: jobsPath,
}
}
type jobInstaller struct {
fs boshsys.FileSystem
templateExtractor biinstallblob.Extractor
jobsPath string
}
func (i jobInstaller) Install(renderedJobRef RenderedJobRef, stage biui.Stage) (installedJob InstalledJob, err error) {
stageName := fmt.Sprintf("Installing job '%s'", renderedJobRef.Name)
err = stage.Perform(stageName, func() error {
installedJob, err = i.install(renderedJobRef)
return err
})
return installedJob, err
}
func (i jobInstaller) install(renderedJobRef RenderedJobRef) (InstalledJob, error) {
jobDir := filepath.Join(i.jobsPath, renderedJobRef.Name)
err := i.fs.MkdirAll(jobDir, os.ModePerm)
if err != nil {
return InstalledJob{}, bosherr.WrapErrorf(err, "Creating job directory '%s'", jobDir)
}
err = i.templateExtractor.Extract(renderedJobRef.BlobstoreID, renderedJobRef.SHA1, jobDir)
if err != nil {
return InstalledJob{}, bosherr.WrapErrorf(err, "Extracting blob with ID '%s'", renderedJobRef.BlobstoreID)
}
binFiles := path.Join(jobDir, "bin", "*")
files, err := i.fs.Glob(binFiles)
if err != nil {
return InstalledJob{}, bosherr.WrapErrorf(err, "Globbing %s", binFiles)
}
for _, file := range files {
err = i.fs.Chmod(file, os.FileMode(0755))
if err != nil {
return InstalledJob{}, bosherr.WrapErrorf(err, "Making %s executable", binFiles)
}
}
return NewInstalledJob(renderedJobRef, jobDir), nil
}
func (i jobInstaller) Cleanup(job InstalledJob) error {
err := i.fs.RemoveAll(job.Path)
if err != nil {
return bosherr.WrapErrorf(err, "Removing job installed to %s", job.Path)
}
return nil
}