feat: Add support for Docker Compose models

Signed-off-by: Umman Mammadov <[email protected]>

Author: umman.mammadov

internal/provisioners/core.go

@@ -92,6 +92,7 @@ type ProvisionOutput struct {
 	ComposeNetworks      map[string]compose.NetworkConfig `json:"compose_networks"`
 	ComposeVolumes       map[string]compose.VolumeConfig  `json:"compose_volumes"`
 	ComposeServices      map[string]compose.ServiceConfig `json:"compose_services"`
+	ComposeModels        map[string]compose.ModelConfig   `json:"compose_models"`
 
 	// For testing and legacy reasons, built in provisioners can set a direct lookup function
 	OutputLookupFunc framework.OutputLookupFunc `json:"-"`
@@ -166,6 +167,7 @@ func (po *ProvisionOutput) ApplyToStateAndProject(state *project.State, resUid f
 		"#volumes", len(po.ComposeVolumes),
 		"#networks", len(po.ComposeNetworks),
 		"#services", len(po.ComposeServices),
+		"#models", len(po.ComposeModels),
 	)
 
 	out := *state
@@ -260,6 +262,12 @@ func (po *ProvisionOutput) ApplyToStateAndProject(state *project.State, resUid f
 		}
 		project.Services[serviceName] = service
 	}
+	for modelName, model := range po.ComposeModels {
+		if project.Models == nil {
+			project.Models = make(compose.Models)
+		}
+		project.Models[modelName] = model
+	}
 
 	out.Resources[resUid] = existing
 	return &out, nil

internal/provisioners/core_test.go

@@ -80,6 +80,9 @@ func TestApplyToStateAndProject(t *testing.T) {
 			ComposeVolumes: map[string]compose.VolumeConfig{
 				"some-volume": {Name: "volume"},
 			},
+			ComposeModels: map[string]compose.ModelConfig{
+				"some-model": {Name: "model-name", Model: "provider/model-id"},
+			},
 		}
 		afterState, err := output.ApplyToStateAndProject(startState, resUid, composeProject)
 		require.NoError(t, err)
@@ -91,6 +94,9 @@ func TestApplyToStateAndProject(t *testing.T) {
 		assert.Len(t, composeProject.Networks, 1)
 		assert.Len(t, composeProject.Volumes, 1)
 		assert.Len(t, composeProject.Services, 1)
+		assert.Len(t, composeProject.Models, 1)
+		assert.Equal(t, "model-name", composeProject.Models["some-model"].Name)
+		assert.Equal(t, "provider/model-id", composeProject.Models["some-model"].Model)
 		paths := make([]string, 0)
 		_ = filepath.WalkDir(td, func(path string, d fs.DirEntry, err error) error {
 			if d.IsDir() {

internal/provisioners/templateprov/template.go

@@ -70,6 +70,9 @@ type Provisioner struct {
 	// ComposeServicesTemplate generates a set of services to add to the compose project. These will replace any with
 	// the same name already.
 	ComposeServicesTemplate string `yaml:"services,omitempty"`
+	// ComposeModelsTemplate generates a set of models to add to the compose project. These will replace any with
+	// the same name already.
+	ComposeModelsTemplate string `yaml:"models,omitempty"`
 
 	// InfoLogsTemplate allows the provisioner to return informational messages for the user which may help connecting or
 	// testing the provisioned resource
@@ -268,6 +271,11 @@ func (p *Provisioner) Provision(ctx context.Context, input *provisioners.Input)
 		return nil, fmt.Errorf("volumes template failed: %w", err)
 	}
 
+	out.ComposeModels = make(map[string]compose.ModelConfig)
+	if err := renderTemplateAndDecode(p.ComposeModelsTemplate, &data, &out.ComposeModels, true); err != nil {
+		return nil, fmt.Errorf("models template failed: %w", err)
+	}
+
 	var infoLogs []string
 	if err := renderTemplateAndDecode(p.InfoLogsTemplate, &data, &infoLogs, false); err != nil {
 		return nil, fmt.Errorf("info logs template failed: %w", err)