forked from docker-archive/classicswarm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
image.go
136 lines (117 loc) · 3.53 KB
/
image.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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
package cluster
import (
"strings"
"github.com/docker/docker/api/types"
)
// Image is exported
type Image struct {
types.ImageSummary
Engine *Engine
}
// ParseRepositoryTag gets a repos name and returns the right reposName + tag|digest
// The tag can be confusing because of a port in a repository name.
// Ex: localhost.localdomain:5000/samalba/hipache:latest
// Digest ex: localhost:5000/foo/bar@sha256:bc8813ea7b3603864987522f02a76101c17ad122e1c46d790efc0fca78ca7bfb
func ParseRepositoryTag(repos string) (string, string) {
n := strings.Index(repos, "@")
if n >= 0 {
parts := strings.Split(repos, "@")
return parts[0], parts[1]
}
n = strings.LastIndex(repos, ":")
if n < 0 {
return repos, ""
}
if tag := repos[n+1:]; !strings.Contains(tag, "/") {
return repos[:n], tag
}
return repos, ""
}
// Match is exported
func (image *Image) Match(IDOrName string, matchTag bool) bool {
size := len(IDOrName)
// TODO: prefix match can cause false positives with image names
if image.ID == IDOrName || (size > 2 && strings.HasPrefix(image.ID, IDOrName)) {
return true
}
// trim sha256: and retry
if parts := strings.SplitN(image.ID, ":", 2); len(parts) == 2 {
if parts[1] == IDOrName || (size > 2 && strings.HasPrefix(parts[1], IDOrName)) {
return true
}
}
repoName, tag := ParseRepositoryTag(IDOrName)
// match repotag
for _, imageRepoTag := range image.RepoTags {
imageRepoName, imageTag := ParseRepositoryTag(imageRepoTag)
if matchTag == false && imageRepoName == repoName {
return true
}
if imageRepoName == repoName && (imageTag == tag || tag == "") {
return true
}
}
// match repodigests
for _, imageDigest := range image.RepoDigests {
imageRepoName, imageDigest := ParseRepositoryTag(imageDigest)
if matchTag == false && imageRepoName == repoName {
return true
}
if imageRepoName == repoName && (imageDigest == tag || tag == "") {
return true
}
}
return false
}
// ImageFilterOptions is the set of filtering options supported by
// Images.Filter()
type ImageFilterOptions struct {
types.ImageListOptions
}
// Images is a collection of Image objects that can be filtered
type Images []*Image
// Filter returns a new sequence of Images filtered to only the images that
// matched the filtering parameters
func (images Images) Filter(opts ImageFilterOptions) Images {
includeAll := func(image *Image) bool {
// TODO: this is wrong if RepoTags == []
return opts.All ||
(len(image.RepoTags) != 0 && image.RepoTags[0] != "<none>:<none>") ||
(len(image.RepoDigests) != 0 && image.RepoDigests[0] != "<none>@<none>")
}
includeFilter := func(image *Image) bool {
if opts.Filters.Len() == 0 {
return true
}
return opts.Filters.MatchKVList("label", image.Labels)
}
referenceFilter := func(image *Image, filter string) bool {
if !opts.Filters.Include(filter) {
return true
}
for _, repoTag := range image.RepoTags {
imageName, _ := ParseRepositoryTag(repoTag)
for _, pattern := range opts.Filters.Get(filter) {
if repoTag == pattern || pattern == imageName {
return true
}
}
}
for _, repoDigests := range image.RepoDigests {
imageName, _ := ParseRepositoryTag(repoDigests)
for _, pattern := range opts.Filters.Get(filter) {
if repoDigests == pattern || pattern == imageName {
return true
}
}
}
return false
}
filtered := make([]*Image, 0, len(images))
for _, image := range images {
if includeAll(image) && includeFilter(image) && referenceFilter(image, "reference") {
filtered = append(filtered, image)
}
}
return filtered
}