pangolin/server/internal/config/config.go

// Package config holds application configuration loaded from environment variables.
package config

import (
	"fmt"
	"os"
	"strings"
	"time"
)

// Config holds all application-level configuration.
type Config struct {
	// DSN is the MySQL connection string.
	// Format: user:password@tcp(host:port)/dbname?parseTime=true&loc=UTC&time_zone=%%27UTC%%27
	DSN string

	// RedisAddr is the Redis server address (host:port).
	RedisAddr     string
	RedisPassword string
	RedisDB       int

	// WebhookSecret is the HMAC-SHA256 shared secret for the card-store webhook.
	// Must be set; no default.
	WebhookSecret string

	// RedeemFailMax is the number of consecutive redeem failures before a 1-hour lock.
	// Default: 5
	RedeemFailMax int

	// RedeemLockDuration is how long the lock lasts after hitting RedeemFailMax.
	// Default: 1 hour
	RedeemLockDuration time.Duration

	// WebhookTimestampTolerance is the ±window for webhook timestamp validation.
	// Default: 5 minutes
	WebhookTimestampTolerance time.Duration

	// WebhookNonceTTL is how long a webhook nonce is kept in Redis to prevent replay.
	// Should be > 2 * WebhookTimestampTolerance.  Default: 15 minutes.
	WebhookNonceTTL time.Duration

	// ── Auth / JWT (RS256) ────────────────────────────────────────────────
	// JWTPrivateKeyPath is the PEM file holding the active RS256 signing key.
	JWTPrivateKeyPath string
	// JWTKeyID is the `kid` written into the JWT header (identifies the signing key).
	JWTKeyID string
	// JWTPublicKeys maps kid -> PEM public-key file path. It must include the
	// active key's kid and may carry previous keys still accepted during
	// rotation. Parsed from JWT_PUBLIC_KEYS="kid1:/path1,kid2:/path2".
	JWTPublicKeys map[string]string
}

// FromEnv reads configuration from environment variables.
// Returns an error if required variables are missing.
func FromEnv() (*Config, error) {
	c := &Config{
		DSN:                       os.Getenv("DB_DSN"),
		RedisAddr:                 getEnvDefault("REDIS_ADDR", "127.0.0.1:6379"),
		RedisPassword:             os.Getenv("REDIS_PASSWORD"),
		RedisDB:                   0,
		WebhookSecret:             os.Getenv("WEBHOOK_SECRET"),
		RedeemFailMax:             5,
		RedeemLockDuration:        time.Hour,
		WebhookTimestampTolerance: 5 * time.Minute,
		WebhookNonceTTL:           15 * time.Minute,
		JWTPrivateKeyPath:         os.Getenv("JWT_PRIVATE_KEY_PATH"),
		JWTKeyID:                  os.Getenv("JWT_KEY_ID"),
		JWTPublicKeys:             parseKeyMap(os.Getenv("JWT_PUBLIC_KEYS")),
	}

	if c.DSN == "" {
		return nil, fmt.Errorf("config: DB_DSN is required")
	}
	if c.WebhookSecret == "" {
		return nil, fmt.Errorf("config: WEBHOOK_SECRET is required")
	}
	return c, nil
}

func getEnvDefault(key, def string) string {
	if v := os.Getenv(key); v != "" {
		return v
	}
	return def
}

// parseKeyMap parses a "kid1:/path1,kid2:/path2" string into a map. Empty input
// yields a nil map. Malformed entries (missing ':') are skipped.
func parseKeyMap(raw string) map[string]string {
	if raw == "" {
		return nil
	}
	m := map[string]string{}
	for _, pair := range strings.Split(raw, ",") {
		pair = strings.TrimSpace(pair)
		if pair == "" {
			continue
		}
		i := strings.IndexByte(pair, ':')
		if i <= 0 || i == len(pair)-1 {
			continue
		}
		m[strings.TrimSpace(pair[:i])] = strings.TrimSpace(pair[i+1:])
	}
	return m
}