pasty/internal/config/config.go

package config

import (
	"strings"
	"time"

	"github.com/lus/pasty/internal/env"
	"github.com/lus/pasty/internal/shared"
)

// Config represents the general application configuration structure
type Config struct {
	WebAddress          string
	StorageType         shared.StorageType
	HastebinSupport     bool
	IDLength            int
	DeletionTokens      bool
	DeletionTokenMaster string
	DeletionTokenLength int
	RateLimit           string
	LengthCap           int
	AutoDelete          *AutoDeleteConfig
	File                *FileConfig
	Postgres            *PostgresConfig
	MongoDB             *MongoDBConfig
	S3                  *S3Config
}

// AutoDeleteConfig represents the configuration specific for the AutoDelete behaviour
type AutoDeleteConfig struct {
	Enabled      bool
	Lifetime     time.Duration
	TaskInterval time.Duration
}

// FileConfig represents the configuration specific for the file storage driver
type FileConfig struct {
	Path string
}

// PostgresConfig represents the configuration specific for the Postgres storage driver
type PostgresConfig struct {
	DSN string
}

// MongoDBConfig represents the configuration specific for the MongoDB storage driver
type MongoDBConfig struct {
	DSN        string
	Database   string
	Collection string
}

// S3Config represents the configuration specific for the S3 storage driver
type S3Config struct {
	Endpoint        string
	AccessKeyID     string
	SecretAccessKey string
	SecretToken     string
	Secure          bool
	Region          string
	Bucket          string
}

// Current holds the currently loaded config
var Current *Config

// Load loads the current config from environment variables and an optional .env file
func Load() {
	env.Load()

	Current = &Config{
		WebAddress:          env.MustString("WEB_ADDRESS", ":8080"),
		StorageType:         shared.StorageType(strings.ToLower(env.MustString("STORAGE_TYPE", "file"))),
		HastebinSupport:     env.MustBool("HASTEBIN_SUPPORT", false),
		IDLength:            env.MustInt("ID_LENGTH", 6),
		DeletionTokens:      env.MustBool("DELETION_TOKENS", true),
		DeletionTokenMaster: env.MustString("DELETION_TOKEN_MASTER", ""),
		DeletionTokenLength: env.MustInt("DELETION_TOKEN_LENGTH", 12),
		RateLimit:           env.MustString("RATE_LIMIT", "30-M"),
		LengthCap:           env.MustInt("LENGTH_CAP", 50_000),
		AutoDelete: &AutoDeleteConfig{
			Enabled:      env.MustBool("AUTODELETE", false),
			Lifetime:     env.MustDuration("AUTODELETE_LIFETIME", 720*time.Hour),
			TaskInterval: env.MustDuration("AUTODELETE_TASK_INTERVAL", 5*time.Minute),
		},
		File: &FileConfig{
			Path: env.MustString("STORAGE_FILE_PATH", "./data"),
		},
		Postgres: &PostgresConfig{
			DSN: env.MustString("STORAGE_POSTGRES_DSN", "postgres://pasty:pasty@localhost/pasty"),
		},
		MongoDB: &MongoDBConfig{
			DSN:        env.MustString("STORAGE_MONGODB_CONNECTION_STRING", "mongodb://pasty:pasty@localhost/pasty"),
			Database:   env.MustString("STORAGE_MONGODB_DATABASE", "pasty"),
			Collection: env.MustString("STORAGE_MONGODB_COLLECTION", "pastes"),
		},
		S3: &S3Config{
			Endpoint:        env.MustString("STORAGE_S3_ENDPOINT", ""),
			AccessKeyID:     env.MustString("STORAGE_S3_ACCESS_KEY_ID", ""),
			SecretAccessKey: env.MustString("STORAGE_S3_SECRET_ACCESS_KEY", ""),
			SecretToken:     env.MustString("STORAGE_S3_SECRET_TOKEN", ""),
			Secure:          env.MustBool("STORAGE_S3_SECURE", true),
			Region:          env.MustString("STORAGE_S3_REGION", ""),
			Bucket:          env.MustString("STORAGE_S3_BUCKET", "pasty"),
		},
	}
}