Radu Macocian (admac)
137 lines
2.5 KiB
Go
137 lines
2.5 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"log/slog"
|
|
"os"
|
|
)
|
|
|
|
var ENV_PREFIX = "SOUV__"
|
|
|
|
type Config struct {
|
|
Api ApiConfig
|
|
Llm LLMConfig
|
|
Logging []LogConfig
|
|
Db DbConfig
|
|
}
|
|
|
|
type DbConfig struct {
|
|
Url string
|
|
Port string
|
|
DbName string
|
|
User string
|
|
Password string
|
|
History HistoryConfig
|
|
Memory MemoryConfig
|
|
}
|
|
|
|
type MemoryConfig struct {
|
|
Enabled bool
|
|
}
|
|
|
|
type HistoryConfig struct {
|
|
Enabled bool
|
|
}
|
|
|
|
type ApiConfig struct {
|
|
Url string
|
|
Key string
|
|
}
|
|
|
|
type LLMConfig struct {
|
|
Model string
|
|
}
|
|
|
|
type LogConfig struct {
|
|
Level slog.Level
|
|
Format string
|
|
Target string
|
|
}
|
|
|
|
func Load(path string) (*Config, error) {
|
|
cfg := defaultConfig()
|
|
if path != "" {
|
|
if data, err := os.ReadFile(path); err == nil {
|
|
if err := json.Unmarshal(data, cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
}
|
|
overrideFromEnv(cfg)
|
|
configLogger(*cfg)
|
|
return cfg, cfg.validate()
|
|
}
|
|
|
|
func configLogger(cfg Config) {
|
|
if len(cfg.Logging) < 1 {
|
|
opts := &slog.HandlerOptions{Level: slog.LevelInfo.Level()}
|
|
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stdout, opts)))
|
|
return
|
|
}
|
|
var handlers []slog.Handler
|
|
for _, c := range cfg.Logging {
|
|
opts := &slog.HandlerOptions{Level: c.Level}
|
|
target, err := getLogTarget(c.Target)
|
|
if err != nil {
|
|
slog.Error("Config error. Could not open target", "target", c.Target)
|
|
}
|
|
switch c.Format {
|
|
case "text":
|
|
handlers = append(handlers, slog.NewTextHandler(target, opts))
|
|
case "json":
|
|
handlers = append(handlers, slog.NewJSONHandler(target, opts))
|
|
}
|
|
}
|
|
slog.SetDefault(slog.New(slog.NewMultiHandler(handlers...)))
|
|
}
|
|
|
|
func getLogTarget(target string) (io.Writer, error) {
|
|
switch target {
|
|
case "", "stdout":
|
|
return os.Stdout, nil
|
|
case "stderr":
|
|
return os.Stderr, nil
|
|
case "discrd":
|
|
return io.Discard, nil
|
|
default:
|
|
return os.OpenFile(target, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o644)
|
|
}
|
|
}
|
|
|
|
func defaultConfig() *Config {
|
|
return &Config{
|
|
ApiConfig{
|
|
Url: "",
|
|
Key: ""},
|
|
LLMConfig{Model: ""},
|
|
[]LogConfig{{Level: slog.LevelInfo.Level(), Format: "text"}},
|
|
}
|
|
}
|
|
|
|
func overrideFromEnv(cfg *Config) {
|
|
val := os.Getenv(ENV_PREFIX + "api__url")
|
|
if val != "" {
|
|
cfg.Api.Url = val
|
|
}
|
|
val = os.Getenv(ENV_PREFIX + "api__key")
|
|
if val != "" {
|
|
cfg.Api.Key = val
|
|
}
|
|
val = os.Getenv(ENV_PREFIX + "llm__model")
|
|
if val != "" {
|
|
cfg.Llm.Model = val
|
|
}
|
|
}
|
|
|
|
func (cfg Config) validate() error {
|
|
if cfg.Api.Key == "" {
|
|
return errors.New("No API key configured")
|
|
}
|
|
if cfg.Api.Url == "" {
|
|
return errors.New("No API url configured")
|
|
}
|
|
return nil
|
|
}
|