config.go

package config

import (
	"flag"
	"fmt"
)

// VerbosityLevel represents the logging verbosity level
type VerbosityLevel int

const (
	Info VerbosityLevel = iota
	Debug
	Trace
)

// Config holds the server configuration
type Config struct {
	Host         string
	Port         int
	SSLPort      int
	SSLCertFile  string
	SSLKeyFile   string
	Verbosity    VerbosityLevel
	UseSSL       bool
}

// Load loads the configuration from command-line flags
func Load() (*Config, error) {
	cfg := &Config{}

	flag.StringVar(&cfg.Host, "host", "localhost", "Host to listen on")
	flag.IntVar(&cfg.Port, "port", 6667, "Port to listen on")
	flag.IntVar(&cfg.SSLPort, "ssl-port", 6697, "SSL Port to listen on")
	flag.StringVar(&cfg.SSLCertFile, "ssl-cert", "", "Path to SSL certificate file")
	flag.StringVar(&cfg.SSLKeyFile, "ssl-key", "", "Path to SSL key file")
	flag.BoolVar(&cfg.UseSSL, "use-ssl", false, "Enable SSL support")
	verbosity := flag.String("verbosity", "info", "Logging verbosity (info, debug, trace)")

	flag.Parse()

	switch *verbosity {
	case "info":
		cfg.Verbosity = Info
	case "debug":
		cfg.Verbosity = Debug
	case "trace":
		cfg.Verbosity = Trace
	default:
		return nil, fmt.Errorf("invalid verbosity level: %s", *verbosity)
	}

	if cfg.UseSSL && (cfg.SSLCertFile == "" || cfg.SSLKeyFile == "") {
		return nil, fmt.Errorf("SSL support enabled but certificate or key file not provided")
	}

	return cfg, nil
}

// Address returns the full address string for the server
func (c *Config) Address() string {
	return fmt.Sprintf("%s:%d", c.Host, c.Port)
}

// SSLAddress returns the full SSL address string for the server
func (c *Config) SSLAddress() string {
	return fmt.Sprintf("%s:%d", c.Host, c.SSLPort)
}