diff --git a/cmd/image-builder/repos.go b/cmd/image-builder/repos.go index 89647532..2d40f76c 100644 --- a/cmd/image-builder/repos.go +++ b/cmd/image-builder/repos.go @@ -1,6 +1,8 @@ package main import ( + "os" + "github.com/osbuild/images/pkg/reporegistry" ) @@ -21,8 +23,11 @@ var defaultDataDirs = []string{ var newRepoRegistry = func(dataDir string) (*reporegistry.RepoRegistry, error) { var dataDirs []string + if dataDir != "" { dataDirs = []string{dataDir} + } else if envDataDir := os.Getenv("IMAGE_BUILDER_DATA_DIR"); envDataDir != "" { + dataDirs = []string{dataDir} } else { dataDirs = defaultDataDirs }