wip

internal/charts/chart.go

@@ -2,6 +2,7 @@ package charts
 
 import (
 	"fmt"
+	"path/filepath"
 	"strings"
 
 	"github.com/mrjosh/helm-ls/internal/log"
@@ -37,6 +38,25 @@ func NewChart(rootURI uri.URI, valuesFilesConfig util.ValuesFilesConfig) *Chart
 	}
 }
 
+func NewChartFromHelmChart(helmChart *chart.Chart, rootURI uri.URI) *Chart {
+	valuesFile := NewValuesFileFromValues(uri.File(filepath.Join(rootURI.Filename(), "values.yaml")), helmChart.Values)
+	return &Chart{
+		ValuesFiles: &ValuesFiles{
+			MainValuesFile:        valuesFile,
+			OverlayValuesFile:     &ValuesFile{},
+			AdditionalValuesFiles: []*ValuesFile{},
+		},
+		ChartMetadata: &ChartMetadata{}, // TODO: there is no usecase for this currently
+		RootURI:       rootURI,
+		ParentChart:   ParentChart{},
+		HelmChart:     helmChart,
+	}
+}
+
+func (c *Chart) GetDependecyUri(dependencyName string) uri.URI {
+	return uri.File(filepath.Join(c.RootURI.Filename(), "charts", DependencyCacheFolder, dependencyName))
+}
+
 func loadHelmChart(rootURI uri.URI) *chart.Chart {
 	var helmChart *chart.Chart
 	loader, err := loader.Loader(rootURI.Filename())
@@ -66,31 +86,8 @@ func (c *Chart) ResolveValueFiles(query []string, chartStore *ChartStore) []*Que
 		return result
 	}
 
-	for _, dependency := range c.HelmChart.Dependencies() {
-		logger.Debug(fmt.Sprintf("Resolving dependency %s with query %s", dependency.Name(), query))
-		if dependency.Name() == query[0] {
-
-			subQuery := []string{}
-			if len(query) > 1 {
-				subQuery = query[1:]
-			}
-
-			valueNode, error := util.ValuesToYamlNode(dependency.Values)
-			if error != nil {
-				logger.Error(fmt.Sprintf("Error loading values file %s: %s", dependency.Name(), error.Error()))
-				continue
-			}
-
-			result = append(result,
-				// TODO: should we do this now? or should we create a chart in the store for each dependency
-				&QueriedValuesFiles{Selector: subQuery, ValuesFiles: &ValuesFiles{
-					MainValuesFile: &ValuesFile{
-						Values:    dependency.Values,
-						ValueNode: valueNode,
-						URI:       uri.File(dependency.ChartPath()), // TODO: Fix this, chartPath is not a file path but something like chartNameA.common
-					},
-				}})
-		}
+	if c.HelmChart != nil {
+		result = c.resolveValuesFilesOfDependencies(query, chartStore, result)
 	}
 
 	parentChart := c.ParentChart.GetParentChart(chartStore)
@@ -99,6 +96,7 @@ func (c *Chart) ResolveValueFiles(query []string, chartStore *ChartStore) []*Que
 	}
 
 	if query[0] == "global" {
+		// TODO: add dependency support
 		return append(result,
 			parentChart.ResolveValueFiles(query, chartStore)...)
 	}
@@ -109,6 +107,31 @@ func (c *Chart) ResolveValueFiles(query []string, chartStore *ChartStore) []*Que
 		parentChart.ResolveValueFiles(extendedQuery, chartStore)...)
 }
 
+func (c *Chart) resolveValuesFilesOfDependencies(query []string, chartStore *ChartStore, result []*QueriedValuesFiles) []*QueriedValuesFiles {
+	for _, dependency := range c.HelmChart.Dependencies() {
+		logger.Debug(fmt.Sprintf("Resolving dependency %s with query %s", dependency.Name(), query))
+		if dependency.Name() == query[0] {
+
+			subQuery := []string{}
+			if len(query) > 1 {
+				subQuery = query[1:]
+			}
+
+			dependencyChart := chartStore.Charts[c.GetDependecyUri(dependency.Name())]
+			if dependencyChart == nil {
+				logger.Error(fmt.Sprintf("Could not find dependency %s", dependency.Name()))
+				continue
+			}
+
+			// TODO: could call this recursively
+
+			result = append(result,
+				&QueriedValuesFiles{Selector: subQuery, ValuesFiles: dependencyChart.ValuesFiles})
+		}
+	}
+	return result
+}
+
 func (c *Chart) GetValueLocation(templateContext []string) (lsp.Location, error) {
 	modifyedVar := make([]string, len(templateContext))
 	// make the first letter lowercase since in the template the first letter is

internal/charts/chart_for_document.go

@@ -19,6 +19,7 @@ func (s *ChartStore) GetChartForDoc(uri lsp.DocumentURI) (*Chart, error) {
 
 	chart, err := s.getChartFromFilesystemForTemplates(uri.Filename())
 	s.Charts[chart.RootURI] = chart
+	s.loadChartDependencies(chart)
 	if err != nil {
 		return chart, ErrChartNotFound{
 			URI: uri,

internal/charts/chart_store.go

@@ -70,3 +70,13 @@ func (s *ChartStore) ReloadValuesFile(file uri.URI) {
 		}
 	}
 }
+
+func (s *ChartStore) loadChartDependencies(chart *Chart) {
+	for _, dependency := range chart.HelmChart.Dependencies() {
+		dependencyUri := chart.GetDependecyUri(dependency.Name())
+		chart := NewChartFromHelmChart(dependency, dependencyUri)
+
+		s.Charts[dependencyUri] = chart
+		s.loadChartDependencies(chart)
+	}
+}

internal/charts/values_file.go

@@ -1,6 +1,8 @@
 package charts
 
 import (
+	"fmt"
+
 	"github.com/mrjosh/helm-ls/internal/util"
 	"go.lsp.dev/uri"
 	"helm.sh/helm/v3/pkg/chartutil"
@@ -24,6 +26,19 @@ func NewValuesFile(filePath string) *ValuesFile {
 	}
 }
 
+func NewValuesFileFromValues(uri uri.URI, values chartutil.Values) *ValuesFile {
+	valueNode, error := util.ValuesToYamlNode(values)
+	if error != nil {
+		logger.Error(fmt.Sprintf("Could not load values for file %s", uri.Filename()), error)
+		return &ValuesFile{}
+	}
+	return &ValuesFile{
+		ValueNode: valueNode,
+		Values:    values,
+		URI:       uri,
+	}
+}
+
 func (v *ValuesFile) Reload() {
 	vals, valueNodes := readInValuesFile(v.URI.Filename())
 

internal/handler/text_document.go

@@ -103,6 +103,9 @@ func (h *langHandler) DidRenameFiles(ctx context.Context, params *lsp.RenameFile
 }
 
 func (h *langHandler) LoadDocsOnNewChart(chart *charts.Chart) {
+	if chart.HelmChart == nil {
+		return
+	}
 	for _, file := range chart.HelmChart.Templates {
 		h.documents.Store(filepath.Join(chart.RootURI.Filename(), file.Name), file.Data, h.helmlsConfig)
 	}