feedkit now contains a reusable core, while weatherfeeder is a concrete implementation that includes weather-specific functions.
140 lines
4.0 KiB
Go
140 lines
4.0 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// Config is the top-level YAML configuration used by feedkit-style daemons.
|
||
//
|
||
// IMPORTANT: This package is intentionally domain-agnostic.
|
||
// - We validate *shape* and internal consistency (required fields, uniqueness, etc.).
|
||
// - We do NOT enforce domain rules (e.g., "observation|forecast|alert").
|
||
// Domain modules (weatherfeeder/newsfeeder/...) can add their own validation layer.
|
||
type Config struct {
|
||
Sources []SourceConfig `yaml:"sources"`
|
||
Sinks []SinkConfig `yaml:"sinks"`
|
||
Routes []RouteConfig `yaml:"routes"`
|
||
}
|
||
|
||
// SourceConfig describes one polling job.
|
||
//
|
||
// This is intentionally generic:
|
||
// - driver-specific knobs belong in Params.
|
||
// - "kind" is allowed (useful for safety checks / routing), but feedkit does not
|
||
// restrict the allowed values.
|
||
type SourceConfig struct {
|
||
Name string `yaml:"name"`
|
||
Driver string `yaml:"driver"` // e.g. "openmeteo_observation", "rss_feed", etc.
|
||
|
||
Every Duration `yaml:"every"` // "15m", "1m", etc.
|
||
|
||
// Kind is optional and domain-defined. If set, it should be a non-empty string.
|
||
// Domains commonly use it to enforce "this source should only emit kind X".
|
||
Kind string `yaml:"kind"`
|
||
|
||
// Params are driver-specific settings (URL, headers, station IDs, API keys, etc.).
|
||
// The driver implementation is responsible for reading/validating these.
|
||
Params map[string]any `yaml:"params"`
|
||
}
|
||
|
||
// SinkConfig describes one output sink adapter.
|
||
type SinkConfig struct {
|
||
Name string `yaml:"name"`
|
||
Driver string `yaml:"driver"` // "stdout", "file", "postgres", "rabbitmq", ...
|
||
Params map[string]any `yaml:"params"` // sink-specific settings
|
||
}
|
||
|
||
// RouteConfig describes a routing rule: which sink receives which kinds.
|
||
type RouteConfig struct {
|
||
Sink string `yaml:"sink"` // sink name
|
||
|
||
// Kinds is domain-defined. feedkit only enforces that each entry is non-empty.
|
||
// Whether a given daemon "recognizes" a kind is domain-specific validation.
|
||
Kinds []string `yaml:"kinds"`
|
||
}
|
||
|
||
// Duration is a YAML-friendly duration wrapper.
|
||
//
|
||
// Supported YAML formats:
|
||
//
|
||
// every: 15m # string duration (recommended)
|
||
// every: 30s
|
||
// every: 1h
|
||
//
|
||
// Also supported for convenience (interpreted as minutes):
|
||
//
|
||
// every: 15 # integer minutes
|
||
// every: "15" # string minutes
|
||
type Duration struct {
|
||
time.Duration
|
||
}
|
||
|
||
func (d *Duration) UnmarshalYAML(value *yaml.Node) error {
|
||
// We expect a scalar (string or int).
|
||
if value.Kind != yaml.ScalarNode {
|
||
return fmt.Errorf("duration must be a scalar (e.g. 15m), got %v", value.Kind)
|
||
}
|
||
|
||
// Case 1: YAML integer -> interpret as minutes.
|
||
if value.Tag == "!!int" {
|
||
n, err := strconv.Atoi(value.Value)
|
||
if err != nil {
|
||
return fmt.Errorf("invalid integer duration %q: %w", value.Value, err)
|
||
}
|
||
if n <= 0 {
|
||
return fmt.Errorf("duration must be > 0, got %d", n)
|
||
}
|
||
d.Duration = time.Duration(n) * time.Minute
|
||
return nil
|
||
}
|
||
|
||
// Case 2: YAML string.
|
||
if value.Tag == "!!str" {
|
||
s := strings.TrimSpace(value.Value)
|
||
if s == "" {
|
||
return fmt.Errorf("duration cannot be empty")
|
||
}
|
||
|
||
// If it’s all digits, interpret as minutes (e.g. "15").
|
||
if isAllDigits(s) {
|
||
n, err := strconv.Atoi(s)
|
||
if err != nil {
|
||
return fmt.Errorf("invalid numeric duration %q: %w", s, err)
|
||
}
|
||
if n <= 0 {
|
||
return fmt.Errorf("duration must be > 0, got %d", n)
|
||
}
|
||
d.Duration = time.Duration(n) * time.Minute
|
||
return nil
|
||
}
|
||
|
||
// Otherwise parse as Go duration string (15m, 30s, 1h, etc.).
|
||
parsed, err := time.ParseDuration(s)
|
||
if err != nil {
|
||
return fmt.Errorf("invalid duration %q (expected e.g. 15m, 30s, 1h): %w", s, err)
|
||
}
|
||
if parsed <= 0 {
|
||
return fmt.Errorf("duration must be > 0, got %s", parsed)
|
||
}
|
||
d.Duration = parsed
|
||
return nil
|
||
}
|
||
|
||
// Anything else: reject.
|
||
return fmt.Errorf("duration must be a string like 15m or an integer minutes, got tag %s", value.Tag)
|
||
}
|
||
|
||
func isAllDigits(s string) bool {
|
||
for _, r := range s {
|
||
if r < '0' || r > '9' {
|
||
return false
|
||
}
|
||
}
|
||
return len(s) > 0
|
||
}
|