env_file can be declared optional

Signed-off-by: Nicolas De Loof <[email protected]>

Author: ndeloof

cli/options_test.go

@@ -270,7 +270,7 @@ func TestProjectWithDiscardEnvFile(t *testing.T) {
 	service, err := p.GetService("simple")
 	assert.NilError(t, err)
 	assert.Equal(t, *service.Environment["DEFAULT_PORT"], "8080")
-	assert.Assert(t, service.EnvFile == nil)
+	assert.Assert(t, len(service.EnvFiles) == 0)
 	assert.Equal(t, service.Ports[0].Published, "8000")
 }
 
@@ -287,7 +287,7 @@ func TestProjectWithMultipleEnvFile(t *testing.T) {
 	service, err := p.GetService("simple")
 	assert.NilError(t, err)
 	assert.Equal(t, *service.Environment["DEFAULT_PORT"], "9090")
-	assert.Assert(t, service.EnvFile == nil)
+	assert.Assert(t, len(service.EnvFiles) == 0)
 	assert.Equal(t, service.Ports[0].Published, "9000")
 }
 

loader/full-example.yml

@@ -141,7 +141,8 @@ services:
     # env_file: .env
     env_file:
       - ./example1.env
-      - ./example2.env
+      - path: ./example2.env
+        required: false
 
     # Mapping or list
     # Mapping values can be strings, numbers or null

loader/full-struct_test.go

@@ -176,9 +176,15 @@ func services(workingDir, homeDir string) types.Services {
 				"ENV.WITH.DOT":        strPtr("ok"),
 				"ENV_WITH_UNDERSCORE": strPtr("ok"),
 			},
-			EnvFile: []string{
-				filepath.Join(workingDir, "example1.env"),
-				filepath.Join(workingDir, "example2.env"),
+			EnvFiles: []types.EnvFile{
+				{
+					Path:     filepath.Join(workingDir, "example1.env"),
+					Required: true,
+				},
+				{
+					Path:     filepath.Join(workingDir, "example2.env"),
+					Required: false,
+				},
 			},
 			Expose: []string{"3000", "8000"},
 			ExternalLinks: []string{
@@ -729,7 +735,8 @@ services:
       QUX: qux_from_environment
     env_file:
       - %s
-      - %s
+      - path: %s
+        required: false
     expose:
       - "3000"
       - "8000"
@@ -1324,7 +1331,10 @@ func fullExampleJSON(workingDir, homeDir string) string {
       },
       "env_file": [
         "%s",
-        "%s"
+        {
+          "path": "%s",
+          "required": false
+        }
       ],
       "expose": [
         "3000",

loader/loader_test.go

@@ -907,7 +907,8 @@ services:
     image: nginx
     env_file:
      - example1.env
-     - example2.env
+     - path: example2.env
+       required: false
 `
 	expectedEnvironmentMap := types.MappingWithEquals{
 		"FOO":                 strPtr("foo_from_env_file"),
@@ -925,14 +926,21 @@ services:
 		options.ResolvePaths = false
 	})
 	assert.NilError(t, err)
-	assert.DeepEqual(t, configWithEnvFiles.Services["web"].EnvFile, types.StringList{"example1.env",
-		"example2.env"})
+	assert.DeepEqual(t, configWithEnvFiles.Services["web"].EnvFiles, []types.EnvFile{
+		{
+			Path:     "example1.env",
+			Required: true,
+		},
+		{
+			Path:     "example2.env",
+			Required: false,
+		}})
 	assert.DeepEqual(t, configWithEnvFiles.Services["web"].Environment, expectedEnvironmentMap)
 
 	// Custom behavior removes the `env_file` entries
 	configWithoutEnvFiles, err := Load(configDetails, WithDiscardEnvFiles)
 	assert.NilError(t, err)
-	assert.DeepEqual(t, configWithoutEnvFiles.Services["web"].EnvFile, types.StringList(nil))
+	assert.Equal(t, len(configWithoutEnvFiles.Services["web"].EnvFiles), 0)
 	assert.DeepEqual(t, configWithoutEnvFiles.Services["web"].Environment, expectedEnvironmentMap)
 }
 
@@ -1929,7 +1937,11 @@ func TestLoadWithExtends(t *testing.T) {
 			Environment: types.MappingWithEquals{
 				"SOURCE": strPtr("extends"),
 			},
-			EnvFile:  []string{expectedEnvFilePath},
+			EnvFiles: []types.EnvFile{
+				{
+					Path:     expectedEnvFilePath,
+					Required: true,
+				}},
 			Networks: map[string]*types.ServiceNetworkConfig{"default": nil},
 			Volumes: []types.ServiceVolumeConfig{{
 				Type:   "bind",
@@ -2090,8 +2102,10 @@ func TestLoadServiceWithEnvFile(t *testing.T) {
 		},
 		Services: types.Services{
 			"test": {
-				Name:    "test",
-				EnvFile: []string{file.Name()},
+				Name: "test",
+				EnvFiles: []types.EnvFile{
+					{Path: file.Name(), Required: true},
+				},
 			},
 		},
 	}
@@ -2450,8 +2464,11 @@ services:
 			Name:          "imported",
 			ContainerName: "extends", // as defined by ./testdata/subdir/extra.env
 			Environment:   types.MappingWithEquals{"SOURCE": strPtr("extends")},
-			EnvFile: types.StringList{
-				filepath.Join(workingDir, "testdata", "subdir", "extra.env"),
+			EnvFiles: []types.EnvFile{
+				{
+					Path:     filepath.Join(workingDir, "testdata", "subdir", "extra.env"),
+					Required: true,
+				},
 			},
 			Image: "nginx",
 			Volumes: []types.ServiceVolumeConfig{
@@ -2630,8 +2647,11 @@ services:
 			Name:        "foo",
 			Image:       "foo",
 			Environment: types.MappingWithEquals{"FOO": strPtr("BAR")},
-			EnvFile: types.StringList{
-				filepath.Join(config.WorkingDir, "testdata", "remote", "env"),
+			EnvFiles: []types.EnvFile{
+				{
+					Path:     filepath.Join(config.WorkingDir, "testdata", "remote", "env"),
+					Required: true,
+				},
 			},
 			Volumes: []types.ServiceVolumeConfig{
 				{

paths/resolve.go

@@ -33,7 +33,7 @@ func ResolveRelativePaths(project map[string]any, base string) error {
 	r.resolvers = map[tree.Path]resolver{
 		"services.*.build.context":               r.absContextPath,
 		"services.*.build.additional_contexts.*": r.absContextPath,
-		"services.*.env_file":                    r.absPath,
+		"services.*.env_file.*.path":             r.absPath,
 		"services.*.extends.file":                r.absPath,
 		"services.*.develop.watch.*.path":        r.absPath,
 		"services.*.volumes.*":                   r.absVolumeMount,

schema/compose-spec.json

@@ -215,7 +215,7 @@
         "dns_search": {"$ref": "#/definitions/string_or_list"},
         "domainname": {"type": "string"},
         "entrypoint": {"$ref": "#/definitions/command"},
-        "env_file": {"$ref": "#/definitions/string_or_list"},
+        "env_file": {"$ref": "#/definitions/env_file"},
         "environment": {"$ref": "#/definitions/list_or_dict"},
 
         "expose": {
@@ -775,6 +775,36 @@
       ]
     },
 
+    "env_file": {
+      "oneOf": [
+        {"type": "string"},
+        {
+          "type": "array",
+          "items": {
+            "oneOf": [
+              {"type": "string"},
+              {
+                "type": "object",
+                "additionalProperties": false,
+                "properties": {
+                  "path": {
+                    "type": "string"
+                  },
+                  "required": {
+                    "type": "boolean",
+                    "default": true
+                  }
+                },
+                "required": [
+                  "path"
+                ]
+              }
+            ]
+          }
+        }
+      ]
+    },
+
     "string_or_list": {
       "oneOf": [
         {"type": "string"},

transform/build.go

@@ -33,6 +33,6 @@ func transformBuild(data any, p tree.Path) (any, error) {
 			"context": v,
 		}, nil
 	default:
-		return data, errors.Errorf("invalid type %T for build", v)
+		return data, errors.Errorf("%s: invalid type %T for build", p, v)
 	}
 }

transform/canonical.go

@@ -28,6 +28,7 @@ func init() {
 	transformers["services.*"] = transformService
 	transformers["services.*.build.secrets.*"] = transformFileMount
 	transformers["services.*.depends_on"] = transformDependsOn
+	transformers["services.*.env_file"] = transformEnvFile
 	transformers["services.*.extends"] = transformExtends
 	transformers["services.*.networks"] = transformServiceNetworks
 	transformers["services.*.volumes.*"] = transformVolumeMount

transform/dependson.go

@@ -49,6 +49,6 @@ func transformDependsOn(data any, p tree.Path) (any, error) {
 		}
 		return d, nil
 	default:
-		return data, errors.Errorf("invalid type %T for depend_on", v)
+		return data, errors.Errorf("%s: invalid type %T for depend_on", p, v)
 	}
 }

transform/envfile.go

@@ -0,0 +1,54 @@
+/*
+   Copyright 2020 The Compose Specification Authors.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package transform
+
+import (
+	"github.com/compose-spec/compose-go/v2/tree"
+	"github.com/pkg/errors"
+)
+
+func transformEnvFile(data any, p tree.Path) (any, error) {
+	switch v := data.(type) {
+	case string:
+		return []any{
+			transformEnvFileValue(v),
+		}, nil
+	case []any:
+		for i, e := range v {
+			v[i] = transformEnvFileValue(e)
+		}
+		return v, nil
+	default:
+		return nil, errors.Errorf("%s: invalid type %T for env_file", p, v)
+	}
+}
+
+func transformEnvFileValue(data any) any {
+	switch v := data.(type) {
+	case string:
+		return map[string]any{
+			"path":     v,
+			"required": true,
+		}
+	case map[string]any:
+		if _, ok := v["required"]; !ok {
+			v["required"] = true
+		}
+		return v
+	}
+	return nil
+}