first commit

framework/objectstore/config.go

@@ -0,0 +1,64 @@
+package objectstore
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/maximhq/bifrost/core/schemas"
+)
+
+// StoreType identifies the object storage backend.
+type StoreType string
+
+const (
+	StoreTypeS3  StoreType = "s3"
+	StoreTypeGCS StoreType = "gcs"
+)
+
+// Config holds the configuration for an object store.
+type Config struct {
+	Type   StoreType      `json:"type"` // "s3" or "gcs"
+	Bucket schemas.EnvVar `json:"bucket"`
+
+	// Common fields (apply to all store types)
+	Prefix   string `json:"prefix,omitempty"`   // Key prefix for all stored objects. Default: "bifrost".
+	Compress bool   `json:"compress,omitempty"` // Enables gzip compression for stored objects. Default: false.
+
+	// S3 fields (used when Type == "s3")
+	Region          *schemas.EnvVar `json:"region,omitempty"`
+	Endpoint        *schemas.EnvVar `json:"endpoint,omitempty"`
+	AccessKeyID     *schemas.EnvVar `json:"access_key_id,omitempty"`
+	SecretAccessKey *schemas.EnvVar `json:"secret_access_key,omitempty"`
+	SessionToken    *schemas.EnvVar `json:"session_token,omitempty"`
+	RoleARN         *schemas.EnvVar `json:"role_arn,omitempty"`
+	ForcePathStyle  bool            `json:"force_path_style,omitempty"`
+
+	// GCS fields (used when Type == "gcs")
+	Credentials     *schemas.EnvVar `json:"credentials,omitempty"`      // Deprecated: use credentials_json
+	CredentialsJSON *schemas.EnvVar `json:"credentials_json,omitempty"` // Service account JSON or path
+	ProjectID       *schemas.EnvVar `json:"project_id,omitempty"`       // GCP project ID override
+}
+
+// GetPrefix returns the configured prefix or "bifrost" as default.
+func (c *Config) GetPrefix() string {
+	if c.Prefix != "" {
+		return c.Prefix
+	}
+	return "bifrost"
+}
+
+// NewObjectStore creates the appropriate ObjectStore implementation based on config type.
+func NewObjectStore(ctx context.Context, cfg *Config, logger schemas.Logger) (ObjectStore, error) {
+	if cfg == nil {
+		return nil, fmt.Errorf("objectstore: config is required")
+	}
+
+	switch cfg.Type {
+	case StoreTypeS3:
+		return NewS3ObjectStore(ctx, cfg, logger)
+	case StoreTypeGCS:
+		return NewGCSObjectStore(ctx, cfg, logger)
+	default:
+		return nil, fmt.Errorf("objectstore: unsupported type %q", cfg.Type)
+	}
+}