From de66cb69ee4c7d8211cde519260e77ce250e5cc0 Mon Sep 17 00:00:00 2001
From: David Desmarais-Michaud <desmarais.michaud.david@gmail.com>
Date: Tue, 18 Jun 2024 11:44:24 -0400
Subject: [PATCH] fix(PL-2763): check chart has been written before deciding to
 pull

---
 internal/generator/generator.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/internal/generator/generator.go b/internal/generator/generator.go
index 46ed83e..0db39a7 100644
--- a/internal/generator/generator.go
+++ b/internal/generator/generator.go
@@ -3,7 +3,9 @@ package generator
 import (
 	"bytes"
 	"context"
+	"errors"
 	"fmt"
+	"os"
 	"sync"
 
 	"github.com/rs/zerolog"
@@ -59,6 +61,14 @@ func (puller ChartPuller) Pull(ctx context.Context, opts helm.PullOptions) error
 	mutex.Lock()
 	defer mutex.Unlock()
 
+	if _, err := os.Stat(opts.OutputDir); err == nil {
+		// If output directory exists it has been pulled by another goroutine
+		// No need to pull the chart
+		return nil
+	} else if !errors.Is(err, os.ErrNotExist) {
+		return fmt.Errorf("failed to stat chart cache: %w", err)
+	}
+
 	if err := cli.Pull(ctx, opts); err != nil {
 		return fmt.Errorf("%w: %q", err, &buffer)
 	}