Skip to content

Commit

Permalink
reverting the chages to initial version
Browse files Browse the repository at this point in the history
  • Loading branch information
shaheerkootteeri committed Aug 8, 2022
1 parent e7166ad commit 433cfcc
Show file tree
Hide file tree
Showing 9 changed files with 19 additions and 195 deletions.
80 changes: 1 addition & 79 deletions pkg/imgpkg/bundle/bundlefakes/fake_images_metadata_writer.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

5 changes: 2 additions & 3 deletions pkg/imgpkg/bundle/contents.go
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,7 @@ type Contents struct {
//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . ImagesMetadataWriter
type ImagesMetadataWriter interface {
ImagesMetadata
WriteImage(regname.Reference, regv1.Image) error
WriteImageWithProgress(regname.Reference, regv1.Image, chan regv1.Update) error
WriteImage(regname.Reference, regv1.Image, chan regv1.Update) error
WriteTag(ref regname.Tag, taggagle regremote.Taggable) error
}

Expand All @@ -46,7 +45,7 @@ func (b Contents) Push(uploadRef regname.Tag, registry ImagesMetadataWriter, ui
}

labels := map[string]string{BundleConfigLabel: "true"}
return plainimage.NewContents(b.paths, b.excludedPaths, false).Push(uploadRef, labels, registry, ui)
return plainimage.NewContents(b.paths, b.excludedPaths).Push(uploadRef, labels, registry, ui)
}

func (b Contents) PresentsAsBundle() (bool, error) {
Expand Down
2 changes: 1 addition & 1 deletion pkg/imgpkg/bundle/locations_configs.go
Original file line number Diff line number Diff line change
Expand Up @@ -132,7 +132,7 @@ func (r LocationsConfigs) Save(reg ImagesMetadataWriter, bundleRef name.Digest,

r.ui.Tracef("Pushing image\n")

_, err = plainimage.NewContents([]string{tmpDir}, nil, false).Push(locRef, nil, reg, ui)
_, err = plainimage.NewContents([]string{tmpDir}, nil).Push(locRef, nil, reg, ui)
if err != nil {
// Immutable tag errors within registries are not standardized.
// Assume word "immutable" would be present in most cases.
Expand Down
2 changes: 1 addition & 1 deletion pkg/imgpkg/cmd/push.go
Original file line number Diff line number Diff line change
Expand Up @@ -138,5 +138,5 @@ func (po *PushOptions) pushImage(registry registry.Registry) (string, error) {
return "", fmt.Errorf("Images cannot be pushed with '.imgpkg' directories, consider using --bundle (-b) option")
}

return plainimage.NewContents(po.FileFlags.Files, po.FileFlags.ExcludedFilePaths, false).Push(uploadRef, nil, registry, po.ui)
return plainimage.NewContents(po.FileFlags.Files, po.FileFlags.ExcludedFilePaths).Push(uploadRef, nil, registry, po.ui)
}
16 changes: 5 additions & 11 deletions pkg/imgpkg/plainimage/contents.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,15 +24,12 @@ type Contents struct {
}

type ImagesWriter interface {
WriteImage(regname.Reference, regv1.Image) error
WriteImageWithProgress(regname.Reference, regv1.Image, chan regv1.Update) error
WriteImage(regname.Reference, regv1.Image, chan regv1.Update) error
WriteTag(ref regname.Tag, taggagle regremote.Taggable) error
}

// NewContents constructor to build a plainimage contents responsible for creating and pushing images,
// by setting displayProgress to 'true' printing out a progress bar using updates when writing to a registry via ggcr
func NewContents(paths []string, excludedPaths []string, displayProgress bool) Contents {
return Contents{paths: paths, excludedPaths: excludedPaths, displayProgress: displayProgress}
func NewContents(paths []string, excludedPaths []string) Contents {
return Contents{paths: paths, excludedPaths: excludedPaths}
}

func (i Contents) Push(uploadRef regname.Tag, labels map[string]string, writer ImagesWriter, ui ui.UI) (string, error) {
Expand All @@ -50,11 +47,8 @@ func (i Contents) Push(uploadRef regname.Tag, labels map[string]string, writer I

defer img.Remove()

if i.displayProgress {
err = writer.WriteImageWithProgress(uploadRef, img, nil)
} else {
err = writer.WriteImage(uploadRef, img)
}
err = writer.WriteImage(uploadRef, img, nil)

if err != nil {
return "", fmt.Errorf("Writing '%s': %s", uploadRef.Name(), err)
}
Expand Down
16 changes: 4 additions & 12 deletions pkg/imgpkg/registry/registry.go
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,7 @@ type Registry interface {
FirstImageExists(digests []string) (string, error)

MultiWrite(imageOrIndexesToUpload map[regname.Reference]regremote.Taggable, concurrency int, updatesCh chan regv1.Update) error
WriteImage(regname.Reference, regv1.Image) error
WriteImageWithProgress(regname.Reference, regv1.Image, chan regv1.Update) error
WriteImage(regname.Reference, regv1.Image, chan regv1.Update) error
WriteIndex(reference regname.Reference, index regv1.ImageIndex) error
WriteTag(tag regname.Tag, taggable regremote.Taggable) error

Expand All @@ -75,8 +74,7 @@ type ImagesReader interface {
type ImagesReaderWriter interface {
ImagesReader
MultiWrite(imageOrIndexesToUpload map[regname.Reference]regremote.Taggable, concurrency int, updatesCh chan regv1.Update) error
WriteImage(regname.Reference, regv1.Image) error
WriteImageWithProgress(regname.Reference, regv1.Image, chan regv1.Update) error
WriteImage(regname.Reference, regv1.Image, chan regv1.Update) error
WriteIndex(regname.Reference, regv1.ImageIndex) error
WriteTag(regname.Tag, regremote.Taggable) error

Expand Down Expand Up @@ -320,8 +318,8 @@ func (r *SimpleRegistry) MultiWrite(imageOrIndexesToUpload map[regname.Reference
return regremote.MultiWrite(overriddenImageOrIndexesToUploadRef, rOpts...)
}

// WriteImageWithProgress Upload Image to registry
func (r *SimpleRegistry) WriteImageWithProgress(ref regname.Reference, img regv1.Image, updatesCh chan regv1.Update) error {
// WriteImage Upload Image to registry
func (r *SimpleRegistry) WriteImage(ref regname.Reference, img regv1.Image, updatesCh chan regv1.Update) error {
if err := r.validateRef(ref); err != nil {
return err
}
Expand All @@ -337,7 +335,6 @@ func (r *SimpleRegistry) WriteImageWithProgress(ref regname.Reference, img regv1
if updatesCh != nil {
opts = append(opts, regremote.WithProgress(updatesCh))
}

err = regremote.Write(overriddenRef, img, opts...)
if err != nil {
return fmt.Errorf("Writing image: %s", err)
Expand All @@ -346,11 +343,6 @@ func (r *SimpleRegistry) WriteImageWithProgress(ref regname.Reference, img regv1
return nil
}

// WriteImage Upload Image to registry
func (r *SimpleRegistry) WriteImage(ref regname.Reference, img regv1.Image) error {
return r.WriteImageWithProgress(ref, img, nil)
}

// Index Retrieve regv1.ImageIndex struct for an Index reference
func (r *SimpleRegistry) Index(ref regname.Reference) (regv1.ImageIndex, error) {
if err := r.validateRef(ref); err != nil {
Expand Down
2 changes: 1 addition & 1 deletion pkg/imgpkg/registry/registry_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -166,7 +166,7 @@ func TestInsecureRegistryFlag(t *testing.T) {
exec: func(t *testing.T, r registry.Registry) error {
img, err := name.ParseReference("my.registry.io/some/image")
require.NoError(t, err)
return r.WriteImage(img, nil)
return r.WriteImage(img, nil, nil)
},
},
{
Expand Down
80 changes: 1 addition & 79 deletions pkg/imgpkg/registry/registryfakes/fake_images_reader_writer.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

11 changes: 3 additions & 8 deletions pkg/imgpkg/registry/with_progress.go
Original file line number Diff line number Diff line change
Expand Up @@ -56,18 +56,13 @@ func (w *WithProgress) MultiWrite(imageOrIndexesToUpload map[regname.Reference]r
return w.delegate.MultiWrite(imageOrIndexesToUpload, concurrency, uploadProgress)
}

// WriteImageWithProgress Upload Image to registry
func (w *WithProgress) WriteImageWithProgress(reference regname.Reference, image regv1.Image, _ chan regv1.Update) error {
// WriteImage Upload Image to registry
func (w *WithProgress) WriteImage(reference regname.Reference, image regv1.Image, _ chan regv1.Update) error {
uploadProgress := make(chan regv1.Update)
w.logger.Start(uploadProgress)
defer w.logger.End()

return w.delegate.WriteImageWithProgress(reference, image, uploadProgress)
}

// WriteImage Upload Image to registry
func (w *WithProgress) WriteImage(reference regname.Reference, image regv1.Image) error {
return w.delegate.WriteImage(reference, image)
return w.delegate.WriteImage(reference, image, uploadProgress)
}

// WriteIndex Uploads the Index manifest to the registry
Expand Down

0 comments on commit 433cfcc

Please sign in to comment.