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	}
25
26	// implied driver == "fs"
27	storageDir := GetEnv("OBJECT_URL", "./.storage")
28	logger.Info("object config detected", "dir", storageDir)
29	return storage.NewStorageFS(logger, storageDir)
30}