update atmos.Component template function

Author: aknysh

internal/exec/describe_stacks.go

@@ -376,6 +376,7 @@ func ExecuteDescribeStacks(
 
 								componentSectionProcessed, err := ProcessTmplWithDatasources(
 									atmosConfig,
+									configAndStacksInfo,
 									settingsSectionStruct,
 									"describe-stacks-all-sections",
 									componentSectionStr,
@@ -577,6 +578,7 @@ func ExecuteDescribeStacks(
 
 								componentSectionProcessed, err := ProcessTmplWithDatasources(
 									atmosConfig,
+									configAndStacksInfo,
 									settingsSectionStruct,
 									"templates-describe-stacks-all-atmos-sections",
 									componentSectionStr,

internal/exec/template_funcs.go

@@ -16,22 +16,28 @@ import (
 )
 
 // FuncMap creates and returns a map of template functions
-func FuncMap(atmosConfig schema.AtmosConfiguration, ctx context.Context, gomplateData *data.Data) template.FuncMap {
-	atmosFuncs := &AtmosFuncs{atmosConfig, ctx, gomplateData}
+func FuncMap(
+	atmosConfig schema.AtmosConfiguration,
+	configAndStacksInfo schema.ConfigAndStacksInfo,
+	ctx context.Context,
+	gomplateData *data.Data,
+) template.FuncMap {
+	atmosFuncs := &AtmosFuncs{atmosConfig, configAndStacksInfo, ctx, gomplateData}
 
 	return map[string]any{
 		"atmos": func() any { return atmosFuncs },
 	}
 }
 
 type AtmosFuncs struct {
-	atmosConfig  schema.AtmosConfiguration
-	ctx          context.Context
-	gomplateData *data.Data
+	atmosConfig         schema.AtmosConfiguration
+	configAndStacksInfo schema.ConfigAndStacksInfo
+	ctx                 context.Context
+	gomplateData        *data.Data
 }
 
 func (f AtmosFuncs) Component(component string, stack string) (any, error) {
-	return componentFunc(f.atmosConfig, component, stack)
+	return componentFunc(f.atmosConfig, f.configAndStacksInfo, component, stack)
 }
 
 func (f AtmosFuncs) GomplateDatasource(alias string, args ...string) (any, error) {

internal/exec/template_funcs_component.go

@@ -6,13 +6,19 @@ import (
 
 	"github.com/samber/lo"
 
+	cfg "github.com/cloudposse/atmos/pkg/config"
 	"github.com/cloudposse/atmos/pkg/schema"
 	u "github.com/cloudposse/atmos/pkg/utils"
 )
 
 var componentFuncSyncMap = sync.Map{}
 
-func componentFunc(atmosConfig schema.AtmosConfiguration, component string, stack string) (any, error) {
+func componentFunc(
+	atmosConfig schema.AtmosConfiguration,
+	configAndStacksInfo schema.ConfigAndStacksInfo,
+	component string,
+	stack string,
+) (any, error) {
 	u.LogTrace(fmt.Sprintf("Executing template function 'atmos.Component(%s, %s)'", component, stack))
 
 	stackSlug := fmt.Sprintf("%s-%s", stack, component)
@@ -42,41 +48,47 @@ func componentFunc(atmosConfig schema.AtmosConfiguration, component string, stac
 		return nil, err
 	}
 
+	// Process Terraform remote state
 	var terraformOutputs map[string]any
-
-	// Check if the component in the stack is configured with the 'static' remote state backend,
-	// in which case get the `output` from the static remote state instead of executing `terraform output`
-	remoteStateBackendStaticTypeOutputs, err := GetComponentRemoteStateBackendStaticType(sections)
-	if err != nil {
-		return nil, err
-	}
-
-	if remoteStateBackendStaticTypeOutputs != nil {
-		terraformOutputs = remoteStateBackendStaticTypeOutputs
-	} else {
-		// Execute `terraform output`
-		terraformOutputs, err = execTerraformOutput(&atmosConfig, component, stack, sections)
+	if configAndStacksInfo.ComponentType == cfg.TerraformComponentType {
+		// Check if the component in the stack is configured with the 'static' remote state backend,
+		// in which case get the `output` from the static remote state instead of executing `terraform output`
+		remoteStateBackendStaticTypeOutputs, err := GetComponentRemoteStateBackendStaticType(sections)
 		if err != nil {
 			return nil, err
 		}
-	}
 
-	outputs := map[string]any{
-		"outputs": terraformOutputs,
-	}
+		if remoteStateBackendStaticTypeOutputs != nil {
+			terraformOutputs = remoteStateBackendStaticTypeOutputs
+		} else {
+			// Execute `terraform output`
+			terraformOutputs, err = execTerraformOutput(&atmosConfig, component, stack, sections)
+			if err != nil {
+				return nil, err
+			}
+		}
 
-	sections = lo.Assign(sections, outputs)
+		outputs := map[string]any{
+			"outputs": terraformOutputs,
+		}
+
+		sections = lo.Assign(sections, outputs)
+	}
 
 	// Cache the result
 	componentFuncSyncMap.Store(stackSlug, sections)
 
 	if atmosConfig.Logs.Level == u.LogLevelTrace {
-		u.LogTrace(fmt.Sprintf("Executed template function 'atmos.Component(%s, %s)'\n\n'outputs' section:", component, stack))
-		y, err2 := u.ConvertToYAML(terraformOutputs)
-		if err2 != nil {
-			u.LogError(err2)
-		} else {
-			u.LogTrace(y)
+		u.LogTrace(fmt.Sprintf("Executed template function 'atmos.Component(%s, %s)'", component, stack))
+
+		if configAndStacksInfo.ComponentType == cfg.TerraformComponentType {
+			u.LogTrace("'outputs' section:")
+			y, err2 := u.ConvertToYAML(terraformOutputs)
+			if err2 != nil {
+				u.LogError(err2)
+			} else {
+				u.LogTrace(y)
+			}
 		}
 	}
 

internal/exec/template_utils.go

@@ -31,7 +31,7 @@ func ProcessTmpl(
 	ctx := context.TODO()
 
 	// Add Gomplate, Sprig and Atmos template functions
-	funcs := lo.Assign(gomplate.CreateFuncs(ctx, &d), sprig.FuncMap(), FuncMap(schema.AtmosConfiguration{}, ctx, &d))
+	funcs := lo.Assign(gomplate.CreateFuncs(ctx, &d), sprig.FuncMap(), FuncMap(schema.AtmosConfiguration{}, schema.ConfigAndStacksInfo{}, ctx, &d))
 
 	t, err := template.New(tmplName).Funcs(funcs).Parse(tmplValue)
 	if err != nil {
@@ -63,6 +63,7 @@ func ProcessTmpl(
 // ProcessTmplWithDatasources parses and executes Go templates with datasources
 func ProcessTmplWithDatasources(
 	atmosConfig schema.AtmosConfiguration,
+	configAndStacksInfo schema.ConfigAndStacksInfo,
 	settingsSection schema.Settings,
 	tmplName string,
 	tmplValue string,
@@ -145,7 +146,7 @@ func ProcessTmplWithDatasources(
 		}
 
 		// Atmos functions
-		funcs = lo.Assign(funcs, FuncMap(atmosConfig, context.TODO(), &d))
+		funcs = lo.Assign(funcs, FuncMap(atmosConfig, configAndStacksInfo, context.TODO(), &d))
 
 		// Process and add environment variables
 		for k, v := range templateSettings.Env {

internal/exec/terraform_generate_backends.go

@@ -238,6 +238,7 @@ func ExecuteTerraformGenerateBackends(
 
 				componentSectionProcessed, err := ProcessTmplWithDatasources(
 					atmosConfig,
+					configAndStacksInfo,
 					settingsSectionStruct,
 					"terraform-generate-backends",
 					componentSectionStr,

internal/exec/terraform_generate_varfiles.go

@@ -247,6 +247,7 @@ func ExecuteTerraformGenerateVarfiles(
 
 				componentSectionProcessed, err := ProcessTmplWithDatasources(
 					atmosConfig,
+					configAndStacksInfo,
 					settingsSectionStruct,
 					"terraform-generate-varfiles",
 					componentSectionStr,

internal/exec/utils.go

@@ -512,6 +512,7 @@ func ProcessStacks(
 
 		componentSectionProcessed, err := ProcessTmplWithDatasources(
 			atmosConfig,
+			configAndStacksInfo,
 			settingsSectionStruct,
 			"templates-all-atmos-sections",
 			componentSectionStr,

pkg/config/const.go

@@ -39,6 +39,9 @@ const (
 	HelpFlag1 = "-h"
 	HelpFlag2 = "--help"
 
+	TerraformComponentType = "terraform"
+	HelmfileComponentType  = "helmfile"
+
 	ComponentVendorConfigFileName = "component.yaml"
 	AtmosVendorConfigFileName     = "vendor"