gemini-grc/config/config.go

package config

import (
	"fmt"
	"os"
	"strconv"

	"github.com/rs/zerolog"
)

// Environment variable names.
const (
	EnvLogLevel               = "LOG_LEVEL"
	EnvNumWorkers             = "NUM_OF_WORKERS"
	EnvWorkerBatchSize        = "WORKER_BATCH_SIZE"
	EnvMaxResponseSize        = "MAX_RESPONSE_SIZE"
	EnvResponseTimeout        = "RESPONSE_TIMEOUT"
	EnvPanicOnUnexpectedError = "PANIC_ON_UNEXPECTED_ERROR"
	EnvBlacklistPath          = "BLACKLIST_PATH"
	EnvDryRun                 = "DRY_RUN"
	EnvPrintWorkerStatus      = "PRINT_WORKER_STATUS"
)

// Config holds the application configuration loaded from environment variables.
type Config struct {
	LogLevel               zerolog.Level // Logging level (debug, info, warn, error)
	MaxResponseSize        int           // Maximum size of response in bytes
	NumOfWorkers           int           // Number of concurrent workers
	ResponseTimeout        int           // Timeout for responses in seconds
	WorkerBatchSize        int           // Batch size for worker processing
	PanicOnUnexpectedError bool          // Panic on unexpected errors when visiting a URL
	BlacklistPath          string        // File that has blacklisted strings of "host:port"
	DryRun                 bool          // If false, don't write to disk
	PrintWorkerStatus      bool          // If false, don't print worker status table
}

var CONFIG Config //nolint:gochecknoglobals

// parsePositiveInt parses and validates positive integer values.
func parsePositiveInt(param, value string) (int, error) {
	val, err := strconv.Atoi(value)
	if err != nil {
		return 0, ValidationError{
			Param:  param,
			Value:  value,
			Reason: "must be a valid integer",
		}
	}
	if val <= 0 {
		return 0, ValidationError{
			Param:  param,
			Value:  value,
			Reason: "must be positive",
		}
	}
	return val, nil
}

func parseBool(param, value string) (bool, error) {
	val, err := strconv.ParseBool(value)
	if err != nil {
		return false, ValidationError{
			Param:  param,
			Value:  value,
			Reason: "cannot be converted to boolean",
		}
	}
	return val, nil
}

// GetConfig loads and validates configuration from environment variables
func GetConfig() *Config {
	config := &Config{}

	// Map of environment variables to their parsing functions
	parsers := map[string]func(string) error{
		EnvLogLevel: func(v string) error {
			level, err := zerolog.ParseLevel(v)
			if err != nil {
				return ValidationError{
					Param:  EnvLogLevel,
					Value:  v,
					Reason: "must be one of: debug, info, warn, error",
				}
			}
			config.LogLevel = level
			return nil
		},
		EnvNumWorkers: func(v string) error {
			val, err := parsePositiveInt(EnvNumWorkers, v)
			if err != nil {
				return err
			}
			config.NumOfWorkers = val
			return nil
		},
		EnvWorkerBatchSize: func(v string) error {
			val, err := parsePositiveInt(EnvWorkerBatchSize, v)
			if err != nil {
				return err
			}
			config.WorkerBatchSize = val
			return nil
		},
		EnvMaxResponseSize: func(v string) error {
			val, err := parsePositiveInt(EnvMaxResponseSize, v)
			if err != nil {
				return err
			}
			config.MaxResponseSize = val
			return nil
		},
		EnvResponseTimeout: func(v string) error {
			val, err := parsePositiveInt(EnvResponseTimeout, v)
			if err != nil {
				return err
			}
			config.ResponseTimeout = val
			return nil
		},
		EnvPanicOnUnexpectedError: func(v string) error {
			val, err := parseBool(EnvPanicOnUnexpectedError, v)
			if err != nil {
				return err
			}
			config.PanicOnUnexpectedError = val
			return nil
		},
		EnvBlacklistPath: func(v string) error {
			config.BlacklistPath = v
			return nil
		},
		EnvDryRun: func(v string) error {
			val, err := parseBool(EnvDryRun, v)
			if err != nil {
				return err
			}
			config.DryRun = val
			return nil
		},
		EnvPrintWorkerStatus: func(v string) error {
			val, err := parseBool(EnvPrintWorkerStatus, v)
			if err != nil {
				return err
			}
			config.PrintWorkerStatus = val
			return nil
		},
	}

	// Process each environment variable
	for envVar, parser := range parsers {
		value, ok := os.LookupEnv(envVar)
		if !ok {
			fmt.Fprintf(os.Stderr, "Missing required environment variable: %s\n", envVar)
			os.Exit(1)
		}

		if err := parser(value); err != nil {
			fmt.Fprintf(os.Stderr, "Configuration error: %v\n", err)
			os.Exit(1)
		}
	}

	return config
}