forked from vmware-archive/atc
/
image_factory.go
122 lines (105 loc) · 2.92 KB
/
image_factory.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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package image
import (
"errors"
"code.cloudfoundry.org/lager"
"github.com/concourse/atc"
"github.com/concourse/atc/creds"
"github.com/concourse/atc/resource"
"github.com/concourse/atc/worker"
)
var ErrUnsupportedResourceType = errors.New("unsupported resource type")
type imageFactory struct {
imageResourceFetcherFactory ImageResourceFetcherFactory
}
func NewImageFactory(
imageResourceFetcherFactory ImageResourceFetcherFactory,
) worker.ImageFactory {
return &imageFactory{
imageResourceFetcherFactory: imageResourceFetcherFactory,
}
}
func (f *imageFactory) GetImage(
logger lager.Logger,
workerClient worker.Worker,
volumeClient worker.VolumeClient,
imageSpec worker.ImageSpec,
teamID int,
delegate worker.ImageFetchingDelegate,
resourceTypes creds.VersionedResourceTypes,
) (worker.Image, error) {
if imageSpec.ImageArtifactSource != nil {
artifactVolume, existsOnWorker, err := imageSpec.ImageArtifactSource.VolumeOn(workerClient)
if err != nil {
logger.Error("failed-to-check-if-volume-exists-on-worker", err)
return nil, err
}
if existsOnWorker {
return &imageProvidedByPreviousStepOnSameWorker{
artifactVolume: artifactVolume,
imageSpec: imageSpec,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
return &imageProvidedByPreviousStepOnDifferentWorker{
imageSpec: imageSpec,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
// check if custom resource
resourceType, found := resourceTypes.Lookup(imageSpec.ResourceType)
if found {
imageResourceFetcher := f.imageResourceFetcherFactory.NewImageResourceFetcher(
workerClient,
resource.NewResourceFactory(workerClient),
worker.ImageResource{
Type: resourceType.Type,
Source: resourceType.Source,
Params: &resourceType.Params,
},
resourceType.Version,
teamID,
resourceTypes.Without(imageSpec.ResourceType),
delegate,
)
return &imageFromResource{
imageResourceFetcher: imageResourceFetcher,
privileged: resourceType.Privileged,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
if imageSpec.ImageResource != nil {
var version atc.Version
if imageSpec.ImageResource.Version != nil {
version = *imageSpec.ImageResource.Version
}
imageResourceFetcher := f.imageResourceFetcherFactory.NewImageResourceFetcher(
workerClient,
resource.NewResourceFactory(workerClient),
*imageSpec.ImageResource,
version,
teamID,
resourceTypes,
delegate,
)
return &imageFromResource{
imageResourceFetcher: imageResourceFetcher,
privileged: imageSpec.Privileged,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
if imageSpec.ResourceType != "" {
return &imageFromBaseResourceType{
worker: workerClient,
resourceTypeName: imageSpec.ResourceType,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
return &imageFromRootfsURI{
url: imageSpec.ImageURL,
}, nil
}