fixup! add IMAGE_BUILDER_DATA_DIR env (mostly for the demo)

osbuild · Nov 27, 2024 · 8051006 · 8051006
1 parent 442da7b
commit 8051006
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/image-builder/repos.go b/cmd/image-builder/repos.go
@@ -1,6 +1,8 @@
 package main
 
 import (
+	"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
 	}