Read log level and bind address from config file

This commit is contained in:
Thorben Günther 2022-10-10 01:20:18 +02:00
parent 0bfa2cf7b9
commit 8347d6410f
No known key found for this signature in database
GPG key ID: 415CD778D8C5AFED
4 changed files with 65 additions and 4 deletions

36
config.go Normal file
View file

@ -0,0 +1,36 @@
package main
import "git.sr.ht/~emersion/go-scfg"
type config struct {
HTTPAddress string
LogLevel string
}
func readConfig(path string) (*config, error) {
cfg, err := scfg.Load(path)
if err != nil {
return nil, err
}
config := new(config)
// Set default values
config.HTTPAddress = "127.0.0.1:8080"
config.LogLevel = "info"
d := cfg.Get("log-level")
if d != nil {
if err := d.ParseParams(&config.LogLevel); err != nil {
return nil, err
}
}
d = cfg.Get("http-address")
if d != nil {
if err := d.ParseParams(&config.HTTPAddress); err != nil {
return nil, err
}
}
return config, nil
}

8
go.mod
View file

@ -2,6 +2,10 @@ module git.xenrox.net/~xenrox/ntfy-alertmanager
go 1.19
require git.xenrox.net/~xenrox/go-log v0.0.0-20221009121411-e93f4223b622
require (
git.sr.ht/~emersion/go-scfg v0.0.0-20211215104734-c2c7a15d6c99
git.xenrox.net/~xenrox/go-log v0.0.0-20221009121411-e93f4223b622
golang.org/x/text v0.3.7
)
require golang.org/x/text v0.3.7
require github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510 // indirect

6
go.sum
View file

@ -1,4 +1,10 @@
git.sr.ht/~emersion/go-scfg v0.0.0-20211215104734-c2c7a15d6c99 h1:1s8n5uisqkR+BzPgaum6xxIjKmzGrTykJdh+Y3f5Xao=
git.sr.ht/~emersion/go-scfg v0.0.0-20211215104734-c2c7a15d6c99/go.mod h1:t+Ww6SR24yYnXzEWiNlOY0AFo5E9B73X++10lrSpp4U=
git.xenrox.net/~xenrox/go-log v0.0.0-20221009121411-e93f4223b622 h1:XxHbaQOQllWEMoeSLocFWPyqMPWCDUDatDEtIdYIXhI=
git.xenrox.net/~xenrox/go-log v0.0.0-20221009121411-e93f4223b622/go.mod h1:d98WFDHGpxaEThKue5CfGtr9OrWgbaApprt3GH+OM4s=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510 h1:El6M4kTTCOh6aBiKaUGG7oYTSPP8MxqL4YI3kZKwcP4=
github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510/go.mod h1:pupxD2MaaD3pAXIBCelhxNneeOaAeabZDe5s4K6zSpQ=
golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk=
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=

19
main.go
View file

@ -2,6 +2,7 @@ package main
import (
"encoding/json"
"flag"
"fmt"
"net/http"
"strings"
@ -13,6 +14,7 @@ import (
)
type receiver struct {
cfg *config
logger *log.Logger
}
@ -100,10 +102,23 @@ func (rcv *receiver) handleWebhooks(w http.ResponseWriter, r *http.Request) {
}
func main() {
var configPath string
flag.StringVar(&configPath, "config", "/etc/ntfy-alertmanager/config", "config file path")
flag.Parse()
logger := log.NewDefaultLogger()
receiver := &receiver{logger: logger}
cfg, err := readConfig(configPath)
if err != nil {
logger.Fatalf("config: %v", err)
}
if err := logger.ParseLevel(cfg.LogLevel); err != nil {
logger.Errorf("config: %v", err)
}
receiver := &receiver{cfg: cfg, logger: logger}
http.HandleFunc("/", receiver.handleWebhooks)
logger.Fatal(http.ListenAndServe("127.0.0.1:8080", nil))
logger.Fatal(http.ListenAndServe(cfg.HTTPAddress, nil))
}