feat: support required extensions

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

Author: mbiagetti

dagger/maintenance/main.go

@@ -129,6 +129,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]]
@@ -167,18 +171,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 = {

pgaudit/values.yaml

@@ -0,0 +1,13 @@
+extensions:
+    - dynamic_library_path: []
+      extension_control_path: []
+      image:
+        reference: ghcr.io/cloudnative-pg/pgaudit:18.0-18-trixie
+      ld_library_path: []
+      name: pgaudit
+name: pgaudit
+pg_image: ghcr.io/cloudnative-pg/postgresql:18-minimal-trixie
+shared_preload_libraries:
+    - pgaudit
+sql_name: pgaudit
+version: "18.0"

pgrouting/metadata.hcl

@@ -0,0 +1,22 @@
+metadata = {
+  name                     = "pgrouting"
+  sql_name                 = "pgrouting"
+  image_name               = "pgrouting-extension"
+  shared_preload_libraries = []
+  extension_control_path   = []
+  dynamic_library_path     = []
+  ld_library_path          = []
+  auto_update_os_libs      = false
+  required_extensions      = [
+      "postgis",
+  ]
+
+  versions = {
+    bookworm = {
+      "18" = ":4.0.0+dfsg-1.pgdg12+1"
+    }
+    trixie = {
+      "18" = "4.0.0+dfsg-1.pgdg13+1"
+    }
+  }
+}

pgrouting/values.yaml

@@ -0,0 +1,18 @@
+extensions:
+    - dynamic_library_path: []
+      extension_control_path: []
+      image:
+        reference: ghcr.io/cloudnative-pg/postgis-extension:3.6.1-18-bookworm
+      ld_library_path: []
+      name: pgrouting
+    - dynamic_library_path: []
+      extension_control_path: []
+      image:
+        reference: ghcr.io/cloudnative-pg/postgis-extension:3.6.1-18-trixie
+      ld_library_path:
+        - /system
+      name: postgis
+name: pgrouting
+pg_image: ghcr.io/cloudnative-pg/postgresql:18-minimal-bookworm
+sql_name: pgrouting
+version: 3.6.1

pgvector/metadata.hcl

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

pgvector/values.yaml

@@ -0,0 +1,11 @@
+extensions:
+    - dynamic_library_path: []
+      extension_control_path: []
+      image:
+        reference: ghcr.io/cloudnative-pg/pgvector:0.8.1-18-bookworm
+      ld_library_path: []
+      name: pgvector
+name: pgvector
+pg_image: ghcr.io/cloudnative-pg/postgresql:18-minimal-bookworm
+sql_name: vector
+version: 0.8.1

postgis/metadata.hcl

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