pasty/internal/config/config.go

package config

import (
	"github.com/joho/godotenv"
	"github.com/kelseyhightower/envconfig"
	"time"
)

type Config struct {
	LogLevel                    string            `default:"info" split_words:"true"`
	WebAddress                  string            `default:":8080" split_words:"true"`
	StorageDriver               string            `default:"sqlite" split_words:"true"`
	HastebinSupport             bool              `default:"false" split_words:"true"`
	IDLength                    int               `default:"6" split_words:"true"`
	IDCharacters                string            `default:"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789" split_words:"true"`
	ModificationTokens          bool              `default:"true" split_words:"true"`
	ModificationTokenMaster     string            `split_words:"true"`
	ModificationTokenLength     int               `default:"12" split_words:"true"`
	ModificationTokenCharacters string            `default:"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789" split_words:"true"`
	RateLimit                   string            `default:"30-M" split_words:"true"`
	LengthCap                   int               `default:"50000" split_words:"true"`
	AutoDelete                  *AutoDeleteConfig `split_words:"true"`
	Reports                     *ReportConfig
	Postgres                    *PostgresConfig
}

type AutoDeleteConfig struct {
	Enabled      bool          `default:"false"`
	Lifetime     time.Duration `default:"720h"`
	TaskInterval time.Duration `default:"5m" split_words:"true"`
}

type ReportConfig struct {
	Enabled      bool   `default:"false" split_words:"true"`
	WebhookURL   string `split_words:"true"`
	WebhookToken string `split_words:"true"`
}

type PostgresConfig struct {
	DSN string `default:"postgres://pasty:pasty@localhost/pasty"`
}

func Load() (*Config, error) {
	_ = godotenv.Overload()
	cfg := new(Config)
	if err := envconfig.Process("pasty", cfg); err != nil {
		return nil, err
	}
	return cfg, nil
}