/
identifier.go
92 lines (79 loc) · 2.22 KB
/
identifier.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
package containerimage
import (
"github.com/containerd/containerd/reference"
"github.com/khulnasoft/devkit/client"
"github.com/khulnasoft/devkit/solver/llbsolver/provenance"
"github.com/khulnasoft/devkit/source"
srctypes "github.com/khulnasoft/devkit/source/types"
"github.com/khulnasoft/devkit/util/resolver"
digest "github.com/opencontainers/go-digest"
ocispecs "github.com/opencontainers/image-spec/specs-go/v1"
"github.com/pkg/errors"
)
type ImageIdentifier struct {
Reference reference.Spec
Platform *ocispecs.Platform
ResolveMode resolver.ResolveMode
RecordType client.UsageRecordType
LayerLimit *int
}
func NewImageIdentifier(str string) (*ImageIdentifier, error) {
ref, err := reference.Parse(str)
if err != nil {
return nil, errors.WithStack(err)
}
if ref.Object == "" {
return nil, errors.WithStack(reference.ErrObjectRequired)
}
return &ImageIdentifier{Reference: ref}, nil
}
var _ source.Identifier = (*ImageIdentifier)(nil)
func (*ImageIdentifier) Scheme() string {
return srctypes.DockerImageScheme
}
func (id *ImageIdentifier) Capture(c *provenance.Capture, pin string) error {
dgst, err := digest.Parse(pin)
if err != nil {
return errors.Wrapf(err, "failed to parse image digest %s", pin)
}
c.AddImage(provenance.ImageSource{
Ref: id.Reference.String(),
Platform: id.Platform,
Digest: dgst,
})
return nil
}
type OCIIdentifier struct {
Reference reference.Spec
Platform *ocispecs.Platform
SessionID string
StoreID string
LayerLimit *int
}
func NewOCIIdentifier(str string) (*OCIIdentifier, error) {
ref, err := reference.Parse(str)
if err != nil {
return nil, errors.WithStack(err)
}
if ref.Object == "" {
return nil, errors.WithStack(reference.ErrObjectRequired)
}
return &OCIIdentifier{Reference: ref}, nil
}
var _ source.Identifier = (*OCIIdentifier)(nil)
func (*OCIIdentifier) Scheme() string {
return srctypes.OCIScheme
}
func (id *OCIIdentifier) Capture(c *provenance.Capture, pin string) error {
dgst, err := digest.Parse(pin)
if err != nil {
return errors.Wrapf(err, "failed to parse OCI digest %s", pin)
}
c.AddImage(provenance.ImageSource{
Ref: id.Reference.String(),
Platform: id.Platform,
Digest: dgst,
Local: true,
})
return nil
}