Switch to FetchWithContext

This updates the calls to the fetcher to use the newer FetchWithContext
method.

Author: bigkevmcd

pkg/cmd/fetcher.go

@@ -29,16 +29,16 @@ type ProxyArchiveFetcher struct {
 	maxUntarSize int
 }
 
-// Fetch implements the ArchiveFetcher implementation, but uses the Kube service
+// FetchWithContext implements the ArchiveFetcher implementation, but uses the Kube service
 // proxy mechanism to get the archive.
-func (p *ProxyArchiveFetcher) Fetch(archiveURL, checksum, dir string) error {
+func (p *ProxyArchiveFetcher) FetchWithContext(ctx context.Context, archiveURL, checksum, dir string) error {
 	parsed, err := parseArtifactURL(archiveURL)
 	if err != nil {
 		return err
 	}
 
 	responseWrapper := p.Client.Services(parsed.namespace).ProxyGet(parsed.scheme, parsed.name, parsed.port, parsed.path, nil)
-	b, err := responseWrapper.DoRaw(context.Background())
+	b, err := responseWrapper.DoRaw(ctx)
 	if err != nil {
 		return err
 	}

pkg/cmd/local_client.go

@@ -69,7 +69,7 @@ type localFetcher struct {
 	logger logr.Logger
 }
 
-func (l localFetcher) Fetch(archiveURL, checksum, dir string) error {
+func (l localFetcher) FetchWithContext(_ context.Context, archiveURL, checksum, dir string) error {
 	parsed, err := url.Parse(archiveURL)
 	if err != nil {
 		return err

pkg/parser/repository_parser.go

@@ -17,7 +17,7 @@ import (
 // ArchiveFetcher implementations should get the URL, validate the contents
 // against the checksum and leave the unpacked version in the dir.
 type ArchiveFetcher interface {
-	Fetch(archiveURL, checksum, dir string) error
+	FetchWithContext(ctx context.Context, archiveURL, checksum, dir string) error
 }
 
 // RepositoryParser fetches archives from a Repository and parses the
@@ -44,7 +44,7 @@ func (p *RepositoryParser) GenerateFromFiles(ctx context.Context, archiveURL, ch
 		}
 	}()
 
-	if err := p.fetcher.Fetch(archiveURL, checksum, tempDir); err != nil {
+	if err := p.fetcher.FetchWithContext(ctx, archiveURL, checksum, tempDir); err != nil {
 		return nil, fmt.Errorf("failed to get archive URL %s: %w", archiveURL, err)
 	}
 
@@ -82,7 +82,7 @@ func (p *RepositoryParser) GenerateFromDirectories(ctx context.Context, archiveU
 		}
 	}()
 
-	if err := p.fetcher.Fetch(archiveURL, checksum, tempDir); err != nil {
+	if err := p.fetcher.FetchWithContext(ctx, archiveURL, checksum, tempDir); err != nil {
 		return nil, fmt.Errorf("failed to get archive URL %s: %w", archiveURL, err)
 	}