prism/service/config/config.go

package config

import (
	"fmt"
	"os"
	"strconv"
	"unicode"
)

type Config struct {
	Port           int
	RateLimit      int
	APIKey         string
	StoragePath    string
	VerboseLogging bool
	EnableSignal   bool
	EnableTelegram bool
	EnableProton   bool
}

func Load() (*Config, error) {
	cfg := &Config{
		APIKey:         os.Getenv("API_KEY"),
		Port:           getEnvInt("PORT", 8080),
		VerboseLogging: getEnvBool("VERBOSE_LOGGING", false),
		RateLimit:      getEnvInt("RATE_LIMIT", 20),
		StoragePath:    getEnvString("STORAGE_PATH", "./data/prism.db"),
		EnableSignal:   getEnvBool("ENABLE_SIGNAL", true),
		EnableTelegram: getEnvBool("ENABLE_TELEGRAM", true),
		EnableProton:   getEnvBool("ENABLE_PROTON", true),
	}

	if err := cfg.Validate(); err != nil {
		return nil, err
	}

	return cfg, nil
}

func (c *Config) Validate() error {
	if c.APIKey == "" {
		return fmt.Errorf("API_KEY environment variable is required")
	}
	for _, r := range c.APIKey {
		if r > unicode.MaxASCII {
			return fmt.Errorf("API_KEY must contain only ASCII characters")
		}
	}
	return nil
}

func getEnvString(key, defaultValue string) string {
	if value := os.Getenv(key); value != "" {
		return value
	}
	return defaultValue
}

func getEnvInt(key string, defaultValue int) int {
	if value := os.Getenv(key); value != "" {
		if i, err := strconv.Atoi(value); err == nil {
			return i
		}
	}
	return defaultValue
}

func getEnvBool(key string, defaultValue bool) bool {
	if value := os.Getenv(key); value != "" {
		if b, err := strconv.ParseBool(value); err == nil {
			return b
		}
	}
	return defaultValue
}