diff --git a/client.go b/client.go index 3aa59327a..788a381d6 100644 --- a/client.go +++ b/client.go @@ -449,10 +449,7 @@ func (c *Client) GetImage(ctx context.Context, ref string) (Image, error) { if err != nil { return nil, err } - return &image{ - client: c, - i: i, - }, nil + return NewImage(c, i), nil } // ListImages returns all existing images @@ -463,10 +460,7 @@ func (c *Client) ListImages(ctx context.Context, filters ...string) ([]Image, er } images := make([]Image, len(imgs)) for i, img := range imgs { - images[i] = &image{ - client: c, - i: img, - } + images[i] = NewImage(c, img) } return images, nil } diff --git a/container.go b/container.go index 23f6d4db6..3c09b2dbc 100644 --- a/container.go +++ b/container.go @@ -173,10 +173,7 @@ func (c *container) Image(ctx context.Context) (Image, error) { if err != nil { return nil, errors.Wrapf(err, "failed to get image %s for container", r.Image) } - return &image{ - client: c.client, - i: i, - }, nil + return NewImage(c.client, i), nil } func (c *container) NewTask(ctx context.Context, ioCreate cio.Creator, opts ...NewTaskOpts) (_ Task, err error) { diff --git a/import.go b/import.go index 3148d240b..e4ac00cee 100644 --- a/import.go +++ b/import.go @@ -80,10 +80,7 @@ func (c *Client) Import(ctx context.Context, importer images.Importer, reader io imgrec = updated } - images = append(images, &image{ - client: c, - i: imgrec, - }) + images = append(images, NewImage(c, imgrec)) } return images, nil } diff --git a/task.go b/task.go index c9a86ffcd..750075f8d 100644 --- a/task.go +++ b/task.go @@ -458,10 +458,7 @@ func (t *task) Checkpoint(ctx context.Context, opts ...CheckpointTaskOpts) (Imag if im, err = t.client.ImageService().Create(ctx, im); err != nil { return nil, err } - return &image{ - client: t.client, - i: im, - }, nil + return NewImage(t.client, im), nil } // UpdateTaskInfo allows updated specific settings to be changed on a task