Torjus Håkestad
5aa5f7275b
Allow authentication to NATS using NKey seed files as an alternative to credentials files. NKeys use Ed25519 key pairs for authentication. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
50 lines
1.7 KiB
Go
50 lines
1.7 KiB
Go
package config
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
type Config struct {
|
|
ListenAddr string
|
|
FlakeCollector bool
|
|
FlakeURL string
|
|
FlakeCheckInterval time.Duration
|
|
|
|
// NATS configuration
|
|
FlakeNATSEnable bool
|
|
FlakeNATSURL string
|
|
FlakeNATSSubject string
|
|
FlakeNATSCredentialsFile string
|
|
FlakeNATSNkeySeedFile string
|
|
}
|
|
|
|
func Parse() (*Config, error) {
|
|
cfg := &Config{}
|
|
|
|
flag.StringVar(&cfg.ListenAddr, "listen", ":9971", "Address to listen on")
|
|
flag.BoolVar(&cfg.FlakeCollector, "collector.flake", false, "Enable flake collector")
|
|
flag.StringVar(&cfg.FlakeURL, "flake.url", "", "Flake URL for revision comparison (required if flake collector enabled)")
|
|
flag.DurationVar(&cfg.FlakeCheckInterval, "flake.check-interval", time.Hour, "Interval between remote flake checks")
|
|
|
|
// NATS flags
|
|
flag.BoolVar(&cfg.FlakeNATSEnable, "flake.nats.enable", false, "Enable NATS cache sharing")
|
|
flag.StringVar(&cfg.FlakeNATSURL, "flake.nats.url", "nats://localhost:4222", "NATS server URL")
|
|
flag.StringVar(&cfg.FlakeNATSSubject, "flake.nats.subject", "nixos-exporter.remote-rev", "NATS subject for revision updates")
|
|
flag.StringVar(&cfg.FlakeNATSCredentialsFile, "flake.nats.credentials-file", "", "NATS credentials file (optional)")
|
|
flag.StringVar(&cfg.FlakeNATSNkeySeedFile, "flake.nats.nkey-seed-file", "", "NATS NKey seed file (optional)")
|
|
|
|
flag.Parse()
|
|
|
|
if cfg.FlakeCollector && cfg.FlakeURL == "" {
|
|
return nil, fmt.Errorf("--flake.url is required when --collector.flake is enabled")
|
|
}
|
|
|
|
if cfg.FlakeNATSEnable && !cfg.FlakeCollector {
|
|
return nil, fmt.Errorf("--flake.nats.enable requires --collector.flake to be enabled")
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|