99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
// "flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"regexp"
|
|
"time"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Conf struct {
|
|
Streams map[string]Stream
|
|
}
|
|
|
|
type Stream struct {
|
|
Cmd []string
|
|
Filters map[string]*Filter
|
|
}
|
|
|
|
type Filter struct {
|
|
Regex []string
|
|
compiledRegex []regexp.Regexp // Computed after YAML parsing
|
|
Retry uint
|
|
RetryPeriod string `yaml:"retry-period"`
|
|
retryDuration time.Duration // Computed after YAML parsing
|
|
Actions map[string]*Action
|
|
}
|
|
|
|
type Action struct {
|
|
name, filterName, streamName string // Computed after YAML parsing
|
|
Cmd []string
|
|
After string `yaml:",omitempty"`
|
|
afterDuration time.Duration // Computed after YAML parsing
|
|
}
|
|
|
|
func (c *Conf) setup() {
|
|
for streamName, stream := range c.Streams {
|
|
for filterName, filter := range stream.Filters {
|
|
|
|
// Parse Duration
|
|
retryDuration, err := time.ParseDuration(filter.RetryPeriod)
|
|
if err != nil {
|
|
log.Fatalln("Failed to parse time in configuration file:", err)
|
|
}
|
|
filter.retryDuration = retryDuration
|
|
|
|
// Compute Regexes
|
|
for _, regex := range filter.Regex {
|
|
filter.compiledRegex = append(filter.compiledRegex, *regexp.MustCompile(regex))
|
|
}
|
|
|
|
for actionName, action := range filter.Actions {
|
|
|
|
// Give all relevant infos to Actions
|
|
action.name = actionName
|
|
action.filterName = filterName
|
|
action.streamName = streamName
|
|
|
|
// Parse Duration
|
|
if action.After != "" {
|
|
afterDuration, err := time.ParseDuration(action.After)
|
|
if err != nil {
|
|
log.Fatalln("Failed to parse time in configuration file:", err)
|
|
}
|
|
action.afterDuration = afterDuration
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func parseConf(filename string) *Conf {
|
|
|
|
data, err := os.ReadFile(filename)
|
|
|
|
if err != nil {
|
|
log.Fatalln("Failed to read configuration file:", err)
|
|
}
|
|
|
|
var conf Conf
|
|
err = yaml.Unmarshal(data, &conf)
|
|
if err != nil {
|
|
log.Fatalln("Failed to parse configuration file:", err)
|
|
}
|
|
|
|
conf.setup()
|
|
fmt.Printf("conf.Streams[0].Filters[0].Actions: %s\n", conf.Streams["tailDown"].Filters["lookForProuts"].Actions)
|
|
|
|
return &conf
|
|
}
|
|
|
|
func parseArgs() map[string]string {
|
|
var args map[string]string
|
|
return args
|
|
}
|