mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-08 22:21:16 +02:00
96aa9638f7
There's a cyclic dependency on siderolink library which imports talos machinery back. We will fix that after we get talos pushed under a new name. Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package docker
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/docker/distribution/reference"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/filters"
|
|
|
|
"github.com/siderolabs/talos/pkg/provision"
|
|
)
|
|
|
|
func (p *provisioner) ensureImageExists(ctx context.Context, image string, options *provision.Options) error {
|
|
// In order to pull an image, the reference must be in canonical
|
|
// format (e.g. domain/repo/image:tag).
|
|
ref, err := reference.ParseNormalizedNamed(image)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
image = ref.String()
|
|
|
|
// To filter the images, we need a familiar name and a tag
|
|
// (e.g. domain/repo/image:tag => repo/image:tag).
|
|
familiarName := reference.FamiliarName(ref)
|
|
tag := ""
|
|
|
|
if tagged, isTagged := ref.(reference.Tagged); isTagged {
|
|
tag = tagged.Tag()
|
|
}
|
|
|
|
filters := filters.NewArgs()
|
|
filters.Add("reference", familiarName+":"+tag)
|
|
|
|
images, err := p.client.ImageList(ctx, types.ImageListOptions{Filters: filters})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(images) == 0 {
|
|
fmt.Fprintln(options.LogWriter, "downloading", image)
|
|
|
|
var reader io.ReadCloser
|
|
|
|
if reader, err = p.client.ImagePull(ctx, image, types.ImagePullOptions{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
//nolint:errcheck
|
|
defer reader.Close()
|
|
|
|
if _, err = io.Copy(io.Discard, reader); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|