repos / pico

util.go

 1package pobj
 2
 3import (
 4	"log/slog"
 5	"os"
 6
 7	"github.com/picosh/pico/pkg/pobj/storage"
 8)
 9
10func GetEnv(key string, defaultVal string) string {
11	if value, exists := os.LookupEnv(key); exists {
12		return value
13	}
14	return defaultVal
15}
16
17func EnvDriverDetector(logger *slog.Logger) (storage.ObjectStorage, error) {
18	driver := GetEnv("OBJECT_DRIVER", "fs")
19	logger.Info("driver detected", "driver", driver)
20
21	switch driver {
22	case "memory":
23		return storage.NewStorageMemory(map[string]map[string]string{})
24	case "minio":
25		url := GetEnv("MINIO_URL", "")
26		user := GetEnv("MINIO_ROOT_USER", "")
27		pass := GetEnv("MINIO_ROOT_PASSWORD", "")
28		logger.Info(
29			"object config detected",
30			"url", url,
31			"user", user,
32		)
33		return storage.NewStorageMinio(logger, url, user, pass)
34	}
35
36	// implied driver == "fs"
37	storageDir := GetEnv("OBJECT_URL", "./.storage")
38	logger.Info("object config detected", "dir", storageDir)
39	return storage.NewStorageFS(logger, storageDir)
40}