Strix/internal/config/config.go

package config

import (
	"log/slog"
	"os"
	"path/filepath"
	"time"
)

// Config holds application configuration
type Config struct {
	Server   ServerConfig
	Database DatabaseConfig
	Scanner  ScannerConfig
	Logger   LoggerConfig
}

// ServerConfig contains HTTP server settings
type ServerConfig struct {
	Host         string
	Port         string
	ReadTimeout  time.Duration
	WriteTimeout time.Duration
}

// DatabaseConfig contains database settings
type DatabaseConfig struct {
	DataPath       string
	BrandsPath     string
	PatternsPath   string
	ParametersPath string
	CacheEnabled   bool
	CacheTTL       time.Duration
}

// ScannerConfig contains stream scanner settings
type ScannerConfig struct {
	DefaultTimeout    time.Duration
	MaxStreams        int
	ModelSearchLimit  int
	WorkerPoolSize    int
	FFProbeTimeout    time.Duration
	RetryAttempts     int
	RetryDelay        time.Duration
	// Validation settings
	StrictValidation  bool // Enable strict validation mode
	MinImageSize      int  // Minimum bytes for valid image (JPEG/PNG)
	MinVideoStreams   int  // Minimum video streams required
}

// LoggerConfig contains logging settings
type LoggerConfig struct {
	Level  string
	Format string // "text" or "json"
}

// Load returns configuration with defaults
func Load() *Config {
	dataPath := getEnv("STRIX_DATA_PATH", "./data")

	return &Config{
		Server: ServerConfig{
			Host:         getEnv("STRIX_HOST", "0.0.0.0"),
			Port:         getEnv("STRIX_PORT", "8080"),
			ReadTimeout:  30 * time.Second,
			WriteTimeout: 30 * time.Second,
		},
		Database: DatabaseConfig{
			DataPath:       dataPath,
			BrandsPath:     filepath.Join(dataPath, "brands"),
			PatternsPath:   filepath.Join(dataPath, "popular_stream_patterns.json"),
			ParametersPath: filepath.Join(dataPath, "query_parameters.json"),
			CacheEnabled:   true,
			CacheTTL:       5 * time.Minute,
		},
		Scanner: ScannerConfig{
			DefaultTimeout:   4 * time.Minute,
			MaxStreams:       10,
			ModelSearchLimit: 6,
			WorkerPoolSize:   20,
			FFProbeTimeout:   30 * time.Second,
			RetryAttempts:    2,
			RetryDelay:       500 * time.Millisecond,
			// Strict validation enabled by default
			StrictValidation: true,
			MinImageSize:     5120,  // 5KB minimum for valid images
			MinVideoStreams:  1,     // At least 1 video stream required
		},
		Logger: LoggerConfig{
			Level:  getEnv("STRIX_LOG_LEVEL", "info"),
			Format: getEnv("STRIX_LOG_FORMAT", "json"),
		},
	}
}

// SetupLogger configures the global logger
func (c *Config) SetupLogger() *slog.Logger {
	var level slog.Level
	switch c.Logger.Level {
	case "debug":
		level = slog.LevelDebug
	case "warn":
		level = slog.LevelWarn
	case "error":
		level = slog.LevelError
	default:
		level = slog.LevelInfo
	}

	opts := &slog.HandlerOptions{
		Level: level,
	}

	var handler slog.Handler
	if c.Logger.Format == "json" {
		handler = slog.NewJSONHandler(os.Stdout, opts)
	} else {
		handler = slog.NewTextHandler(os.Stdout, opts)
	}

	return slog.New(handler)
}

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