feat: support required extensions

Signed-off-by: Matteo <matteo.biagetti@enterprisedb.com>

Author: mbiagetti

dagger/maintenance/main.go

@@ -136,6 +136,10 @@ func (m *Maintenance) GetTargets(
 // Generates Chainsaw's testing external values in YAML format
 func (m *Maintenance) GenerateTestingValues(
 	ctx context.Context,
+	// The source directory containing the extension folders. Defaults to the current directory
+	// +ignore=["dagger", ".github"]
+	// +defaultPath="/"
+	source *dagger.Directory,
 	// Path to the target extension directory
 	target *dagger.Directory,
 	// URL reference to the extension image to test [REPOSITORY[:TAG]]
@@ -174,18 +178,23 @@ func (m *Maintenance) GenerateTestingValues(
 			targetExtensionImage)
 	}
 
+	extensions, generateExtErr := generateTestingValuesExtensions(
+		ctx,
+		source,
+		metadata,
+		targetExtensionImage,
+	)
+	if generateExtErr != nil {
+		return nil, generateExtErr
+	}
 	// Build values.yaml content
 	values := map[string]any{
 		"name":                     metadata.Name,
 		"sql_name":                 metadata.SQLName,
-		"image_name":               metadata.ImageName,
 		"shared_preload_libraries": metadata.SharedPreloadLibraries,
-		"extension_control_path":   metadata.ExtensionControlPath,
-		"dynamic_library_path":     metadata.DynamicLibraryPath,
-		"ld_library_path":          metadata.LdLibraryPath,
-		"extension_image":          targetExtensionImage,
 		"pg_image":                 pgImage,
 		"version":                  version,
+		"extensions":               extensions,
 	}
 	valuesYaml, err := yaml.Marshal(values)
 	if err != nil {

dagger/maintenance/parse.go

@@ -28,6 +28,7 @@ type extensionMetadata struct {
 	DynamicLibraryPath     []string   `hcl:"dynamic_library_path" cty:"dynamic_library_path"`
 	LdLibraryPath          []string   `hcl:"ld_library_path" cty:"ld_library_path"`
 	AutoUpdateOsLibs       bool       `hcl:"auto_update_os_libs" cty:"auto_update_os_libs"`
+	RequiredExtensions     []string   `hcl:"required_extensions" cty:"required_extensions"`
 	Versions               versionMap `hcl:"versions" cty:"versions"`
 	Remain                 hcl.Body   `hcl:",remain"`
 }

dagger/maintenance/testingvalues.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"context"
+
+	"dagger/maintenance/internal/dagger"
+)
+
+func generateTestingValuesExtensions(ctx context.Context, source *dagger.Directory, metadata *extensionMetadata, extensionImage string) ([]map[string]any, error) {
+	var out []map[string]any
+	configuration, err := generateExtensionConfiguration(metadata, extensionImage)
+	if err != nil {
+		return nil, err
+	}
+	out = append(out, configuration)
+	for _, deps := range metadata.RequiredExtensions {
+		depsMetadata, parseErr := parseExtensionMetadata(ctx, source.Directory(deps))
+		if parseErr != nil {
+			return nil, parseErr
+		}
+		depsConfiguration, extErr := generateExtensionConfiguration(depsMetadata, "")
+		if extErr != nil {
+			return nil, extErr
+		}
+		out = append(out, depsConfiguration)
+
+	}
+	return out, nil
+}
+
+func generateExtensionConfiguration(metadata *extensionMetadata, extensionImage string) (map[string]any, error) {
+	targetExtensionImage := extensionImage
+	if targetExtensionImage == "" {
+		var err error
+		targetExtensionImage, err = getDefaultExtensionImage(metadata)
+		if err != nil {
+			return nil, err
+		}
+	}
+	return map[string]any{
+		"name": metadata.Name,
+		"image": map[string]string{
+			"reference": targetExtensionImage,
+		},
+		"extension_control_path": metadata.ExtensionControlPath,
+		"dynamic_library_path":   metadata.DynamicLibraryPath,
+		"ld_library_path":        metadata.LdLibraryPath,
+	}, nil
+}

pgaudit/metadata.hcl

@@ -7,6 +7,7 @@ metadata = {
   dynamic_library_path     = []
   ld_library_path          = []
   auto_update_os_libs      = false
+  required_extensions      = []
 
   versions = {
     bookworm = {

pgvector/metadata.hcl

@@ -7,6 +7,7 @@ metadata = {
   dynamic_library_path     = []
   ld_library_path          = []
   auto_update_os_libs      = false
+  required_extensions      = []
 
   versions = {
     bookworm = {

postgis/metadata.hcl

@@ -7,6 +7,7 @@ metadata = {
   dynamic_library_path     = []
   ld_library_path          = ["/system"]
   auto_update_os_libs      = true
+  required_extensions      = []
 
   versions = {
     bookworm = {

postgis/test/cluster.yaml

@@ -11,10 +11,4 @@ spec:
 
   postgresql:
     shared_preload_libraries: ($values.shared_preload_libraries)
-    extensions:
-      - name: ($values.name)
-        image:
-          reference: ($values.extension_image)
-        extension_control_path: ($values.extension_control_path)
-        dynamic_library_path: ($values.dynamic_library_path)
-        ld_library_path: ($values.ld_library_path)
+    extensions: ($values.extensions)

test/cluster.yaml

@@ -11,10 +11,4 @@ spec:
 
   postgresql:
     shared_preload_libraries: ($values.shared_preload_libraries)
-    extensions:
-      - name: ($values.name)
-        image:
-          reference: ($values.extension_image)
-        extension_control_path: ($values.extension_control_path)
-        dynamic_library_path: ($values.dynamic_library_path)
-        ld_library_path: ($values.ld_library_path)
+    extensions: ($values.extensions)