90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/urfave/cli/v3"
|
|
|
|
"git.maze.io/go/ham/protocol"
|
|
|
|
"git.maze.io/ham/hamview"
|
|
"git.maze.io/ham/hamview/cmd"
|
|
)
|
|
|
|
var logger *logrus.Logger
|
|
|
|
func init() {
|
|
logger = cmd.NewLogger(nil)
|
|
}
|
|
|
|
func main() {
|
|
cmd := &cli.Command{
|
|
Name: "hamview-collector",
|
|
Usage: "Collector for HAM radio protocols",
|
|
Action: run,
|
|
Before: cmd.ConfigureLogging(&logger),
|
|
Flags: cmd.AllFlags("hamview-collector.yaml"),
|
|
}
|
|
|
|
if err := cmd.Run(context.Background(), os.Args); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
type collectorConfig struct {
|
|
hamview.CollectorConfig `yaml:",inline"`
|
|
Broker hamview.BrokerConfig `yaml:"broker"`
|
|
Include []string `yaml:"include"`
|
|
MeshCore struct {
|
|
Group meshCoreGroupConfig `yaml:"group"`
|
|
} `yaml:"meshcore"`
|
|
}
|
|
|
|
func (config *collectorConfig) Includes() []string {
|
|
includes := config.Include
|
|
config.Include = nil
|
|
return includes
|
|
}
|
|
|
|
type meshCoreGroupConfig struct {
|
|
Secret map[string]string `yaml:"secret"`
|
|
Public []string `yaml:"public"`
|
|
}
|
|
|
|
func run(ctx context.Context, command *cli.Command) error {
|
|
var config collectorConfig
|
|
if err := cmd.Load(logger, command.String(cmd.FlagConfig), &config); err != nil {
|
|
return err
|
|
}
|
|
|
|
collector, err := hamview.NewCollector(&config.CollectorConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
broker, err := hamview.NewBroker(&config.Broker)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err = broker.Start(); err != nil {
|
|
return err
|
|
}
|
|
defer broker.Close()
|
|
|
|
for _, proto := range []string{
|
|
protocol.APRS,
|
|
protocol.MeshCore,
|
|
} {
|
|
go func() {
|
|
if err := collector.Collect(broker, proto+"/packet/+"); err != nil {
|
|
logger.Fatalf("Error collecting %s packets: %v", proto, err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
return cmd.WaitForInterrupt(logger, "collector")
|
|
}
|