config.go

package main

import (
	"os"

	"github.com/kelseyhightower/envconfig"
	"gopkg.in/yaml.v3"
)

const (
	defaultMaxUploadSizeMB        = 2
	defaultStreamFFMPEG           = "ffmpeg"
	defaultStreamChunkSizeSeconds = 10
	defaultStreamCodec            = "libmp3lame"
	defaultStreamBitrate          = "128k"
)

type Config struct {
	// API settings
	Port                   int      `yaml:"port" envconfig:"PORT"`
	APIBase                string   `yaml:"api_base" envconfig:"API_BASE"`
	StreamBase             string   `yaml:"stream_base" envconfig:"STREAM_BASE"`
	DownloadBase           string   `yaml:"download_base" envconfig:"DOWNLOAD_BASE"`
	MediaStorageDir        string   `yaml:"media_storage_dir" envconfig:"MEDIA_STORAGE_DIR"`
	StreamStorageDir       string   `yaml:"stream_storage_dir" envconfig:"STREAM_STORAGE_DIR"`
	WavStorageDir          string   `yaml:"wav_storage_dir" envconfig:"WAV_STORAGE_DIR"`
	StreamFFMPEG           string   `yaml:"stream_ffmpeg" envconfig:"STREAM_FFMPEG"`
	StreamChunkSizeSeconds int      `yaml:"stream_chunk_size_seconds" envconfig:"STREAM_CHUNK_SIZE_SECONDS"`
	StreamCodec            string   `yaml:"stream_codec" envconfig:"STREAM_CODEC"`
	StreamBitrate          string   `yaml:"stream_bitrate" envconfig:"STREAM_BITRATE"`
	MaxUploadSizeMB        int64    `yaml:"max_upload_size_mb" envconfig:"MAX_UPLOAD_SIZE_MB"`
	AcceptedMimetypes      []string `yaml:"accepted_mimetypes" envconfig:"ACCEPTED_MIMETYPES"`
	S3Bucket               string   `yaml:"s3_bucket" envconfig:"S3_BUCKET"`
	AllowedPubkeys         []string `yaml:"allowed_pubkeys" envconfig:"ALLOWED_PUBKEYS"`
	LightningProvider      string   `yaml:"lightning_provider" envconfig:"LIGHTNING_PROVIDER"`
	NodelessAPIKey         string   `yaml:"nodeless_apikey" envconfig:"NODELESS_APIKEY"`
	NodelessStoreID        string   `yaml:"nodeless_storeid" envconfig:"NODELESS_STOREID"`
	NodelessTestnet        bool     `yaml:"nodeless_testnet" envconfig:"NODELESS_TESTNET"`
	ZBDAPIKey              string   `yaml:"zbd_apikey" envconfig:"ZBD_APIKEY"`
	SubscriptionDB         string   `yaml:"subscription_db"`
	SubscriptionOptions    []struct {
		Days int `yaml:"days" json:"days"`
		Sats int `yaml:"sats" json:"sats"`
	} `yaml:"subscription_options"`
	BlastrNsec string `yaml:"blastr_nsec" envconfig:"BLASTR_NSEC"`
}

// Load Config from a yaml file at path.
func (c *Config) Load(path string) error {
	f, err := os.Open(path)
	if err != nil {
		return err
	}
	defer f.Close()

	if err := yaml.NewDecoder(f).Decode(c); err != nil {
		return err
	}

	c.applyDefaults()
	return nil
}

// Load Config from the environment.
func (c *Config) LoadFromEnv() error {
	if err := envconfig.Process("", c); err != nil {
		return err
	}

	c.applyDefaults()
	return nil
}

func (c *Config) applyDefaults() {
	if c.MaxUploadSizeMB == 0 {
		c.MaxUploadSizeMB = defaultMaxUploadSizeMB
	}
	if c.StreamFFMPEG == "" {
		c.StreamFFMPEG = defaultStreamFFMPEG
	}
	if c.StreamChunkSizeSeconds == 0 {
		c.StreamChunkSizeSeconds = defaultStreamChunkSizeSeconds
	}
	if c.StreamCodec == "" {
		c.StreamCodec = defaultStreamCodec
	}
	if c.StreamBitrate == "" {
		c.StreamBitrate = defaultStreamBitrate
	}
}