From fd558864162112f723d4448b6d3c7d6c924b9499 Mon Sep 17 00:00:00 2001
From: bagnaram <11695670+bagnaram@users.noreply.github.com>
Date: Thu, 19 Oct 2023 14:24:58 -0600
Subject: [PATCH] feat(ui): Recursive Helm Values files detection (#15935)
 (#15936)

Signed-off-by: bagnaram <11695670+bagnaram@users.noreply.github.com>
Signed-off-by: jmilic1 <70441727+jmilic1@users.noreply.github.com>
---
 reposerver/repository/repository.go           | 32 +++++++++----------
 reposerver/repository/repository_test.go      | 21 +++++++++---
 .../testdata/values-files/dir/values.yaml     |  1 +
 3 files changed, 33 insertions(+), 21 deletions(-)
 create mode 100644 reposerver/repository/testdata/values-files/dir/values.yaml

diff --git a/reposerver/repository/repository.go b/reposerver/repository/repository.go
index 020ff7944d566..6dc249b6d0ae6 100644
--- a/reposerver/repository/repository.go
+++ b/reposerver/repository/repository.go
@@ -2021,12 +2021,13 @@ func (s *Service) createGetAppDetailsCacheHandler(res *apiclient.RepoAppDetailsR
 
 func populateHelmAppDetails(res *apiclient.RepoAppDetailsResponse, appPath string, repoRoot string, q *apiclient.RepoServerAppDetailsQuery, gitRepoPaths io.TempPaths) error {
 	var selectedValueFiles []string
+	var availableValueFiles []string
 
 	if q.Source.Helm != nil {
 		selectedValueFiles = q.Source.Helm.ValueFiles
 	}
 
-	availableValueFiles, err := findHelmValueFilesInPath(appPath)
+	err := filepath.Walk(appPath, walkHelmValueFilesInPath(appPath, &availableValueFiles))
 	if err != nil {
 		return err
 	}
@@ -2103,26 +2104,25 @@ func loadFileIntoIfExists(path pathutil.ResolvedFilePath, destination *string) e
 	return nil
 }
 
-func findHelmValueFilesInPath(path string) ([]string, error) {
-	var result []string
+func walkHelmValueFilesInPath(root string, valueFiles *[]string) filepath.WalkFunc {
+	return func(path string, info os.FileInfo, err error) error {
 
-	files, err := os.ReadDir(path)
-	if err != nil {
-		return result, fmt.Errorf("error reading helm values file from %s: %w", path, err)
-	}
-
-	for _, f := range files {
-		if f.IsDir() {
-			continue
+		if err != nil {
+			return fmt.Errorf("error reading helm values file from %s: %w", path, err)
 		}
-		filename := f.Name()
-		fileNameExt := strings.ToLower(filepath.Ext(filename))
+
+		filename := info.Name()
+		fileNameExt := strings.ToLower(filepath.Ext(path))
 		if strings.Contains(filename, "values") && (fileNameExt == ".yaml" || fileNameExt == ".yml") {
-			result = append(result, filename)
+			relPath, err := filepath.Rel(root, path)
+			if err != nil {
+				return fmt.Errorf("error traversing path from %s to %s: %w", root, path, err)
+			}
+			*valueFiles = append(*valueFiles, relPath)
 		}
-	}
 
-	return result, nil
+		return nil
+	}
 }
 
 func populateKustomizeAppDetails(res *apiclient.RepoAppDetailsResponse, q *apiclient.RepoServerAppDetailsQuery, appPath string, reversion string, credsStore git.CredsStore) error {
diff --git a/reposerver/repository/repository_test.go b/reposerver/repository/repository_test.go
index d372dd35290e7..376a03d30d71e 100644
--- a/reposerver/repository/repository_test.go
+++ b/reposerver/repository/repository_test.go
@@ -2663,16 +2663,27 @@ func runGit(t *testing.T, workDir string, args ...string) string {
 	return stringOut
 }
 
-func Test_findHelmValueFilesInPath(t *testing.T) {
+func Test_walkHelmValueFilesInPath(t *testing.T) {
 	t.Run("does not exist", func(t *testing.T) {
-		files, err := findHelmValueFilesInPath("/obviously/does/not/exist")
+		var files []string
+		root := "/obviously/does/not/exist"
+		err := filepath.Walk(root, walkHelmValueFilesInPath(root, &files))
 		assert.Error(t, err)
 		assert.Empty(t, files)
 	})
 	t.Run("values files", func(t *testing.T) {
-		files, err := findHelmValueFilesInPath("./testdata/values-files")
+		var files []string
+		root := "./testdata/values-files"
+		err := filepath.Walk(root, walkHelmValueFilesInPath(root, &files))
 		assert.NoError(t, err)
-		assert.Len(t, files, 4)
+		assert.Len(t, files, 5)
+	})
+	t.Run("unrelated root", func(t *testing.T) {
+		var files []string
+		root := "./testdata/values-files"
+		unrelated_root := "/different/root/path"
+		err := filepath.Walk(root, walkHelmValueFilesInPath(unrelated_root, &files))
+		assert.Error(t, err)
 	})
 }
 
@@ -2690,7 +2701,7 @@ func Test_populateHelmAppDetails(t *testing.T) {
 	err = populateHelmAppDetails(&res, appPath, appPath, &q, emptyTempPaths)
 	require.NoError(t, err)
 	assert.Len(t, res.Helm.Parameters, 3)
-	assert.Len(t, res.Helm.ValueFiles, 4)
+	assert.Len(t, res.Helm.ValueFiles, 5)
 }
 
 func Test_populateHelmAppDetails_values_symlinks(t *testing.T) {
diff --git a/reposerver/repository/testdata/values-files/dir/values.yaml b/reposerver/repository/testdata/values-files/dir/values.yaml
new file mode 100644
index 0000000000000..55262d50ff71c
--- /dev/null
+++ b/reposerver/repository/testdata/values-files/dir/values.yaml
@@ -0,0 +1 @@
+values: yaml