read configuration file

This commit is contained in:
Felipe M 2024-11-23 19:53:14 +01:00
parent 92a994a073
commit fcf82bc359
Signed by: fmartingr
GPG key ID: CCFBC5637D4000A8
2 changed files with 46 additions and 27 deletions

View file

@ -1,3 +1,5 @@
Port = 11025
[[Recipients]] [[Recipients]]
Addresses = ["something@something.com"] Addresses = ["something@something.com"]
Target = "ntfy://ntfy.sh/fmartingr-dev" Target = "ntfy://ntfy.sh/fmartingr-dev"

71
main.go
View file

@ -8,10 +8,12 @@ import (
"log/slog" "log/slog"
"net/mail" "net/mail"
"net/url" "net/url"
"os"
"slices" "slices"
"strings" "strings"
"time" "time"
"git.nakama.town/fmartingr/gotoolkit/encoding"
"git.nakama.town/fmartingr/gotoolkit/model" "git.nakama.town/fmartingr/gotoolkit/model"
"git.nakama.town/fmartingr/gotoolkit/service" "git.nakama.town/fmartingr/gotoolkit/service"
"github.com/containrrr/shoutrrr" "github.com/containrrr/shoutrrr"
@ -44,21 +46,28 @@ func (re *ReceivedEmail) Body() string {
} }
type Config struct { type Config struct {
Port int
Recipients []ConfigRecipient Recipients []ConfigRecipient
} }
func (c *Config) SetDefaults() {
if c.Port == 0 {
c.Port = 11125
}
}
type ConfigRecipient struct { type ConfigRecipient struct {
Addresses []string // email addresses Addresses []string // email addresses
target string // shoutrrr address Target string // shoutrrr address
targetURL *url.URL targetURL *url.URL
} }
func (cr ConfigRecipient) Target() *url.URL { func (cr ConfigRecipient) GetTargetURL() *url.URL {
if cr.targetURL == nil { if cr.targetURL == nil {
var err error var err error
cr.targetURL, err = url.Parse(cr.target) cr.targetURL, err = url.Parse(cr.Target)
if err != nil { if err != nil {
slog.Error("failed to parse shoutrrr target URL", slog.String("target", cr.target), slog.String("err", err.Error())) slog.Error("failed to parse shoutrrr target URL", slog.String("target", cr.Target), slog.String("err", err.Error()))
} }
} }
return cr.targetURL return cr.targetURL
@ -85,9 +94,22 @@ func (s *smtpServer) Stop(ctx context.Context) error {
return s.backend.Shutdown(ctx) return s.backend.Shutdown(ctx)
} }
func NewSMTPServer(backend *smtp.Server) model.Server { func NewSMTPServer(config *Config) model.Server {
be := &Backend{
config: config,
}
smtpBackend := smtp.NewServer(be)
smtpBackend.Addr = fmt.Sprintf(":%d", config.Port)
// smtpBackend.Domain = "localhost"
smtpBackend.WriteTimeout = 10 * time.Second
smtpBackend.ReadTimeout = 10 * time.Second
smtpBackend.MaxMessageBytes = 1024 * 1024
smtpBackend.MaxRecipients = 50
smtpBackend.AllowInsecureAuth = true
return &smtpServer{ return &smtpServer{
backend: backend, backend: smtpBackend,
} }
} }
@ -104,7 +126,7 @@ func (bkd *Backend) NewSession(c *smtp.Conn) (smtp.Session, error) {
} }
func (bkd *Backend) forwardEmail(email ReceivedEmail) error { func (bkd *Backend) forwardEmail(email ReceivedEmail) error {
slog.Info("forwading message", slog.String("email", email.String())) slog.Info("forwading message", slog.String("to", strings.Join(email.Recipients, ",")))
for _, r := range bkd.config.Recipients { for _, r := range bkd.config.Recipients {
for _, a := range email.Recipients { for _, a := range email.Recipients {
@ -113,7 +135,7 @@ func (bkd *Backend) forwardEmail(email ReceivedEmail) error {
"title": {email.Msg.Header.Get("Subject")}, "title": {email.Msg.Header.Get("Subject")},
} }
destinationURL := r.Target() destinationURL := r.GetTargetURL()
destinationURL.RawQuery = urlParams.Encode() destinationURL.RawQuery = urlParams.Encode()
if err := shoutrrr.Send(destinationURL.String(), email.Body()); err != nil { if err := shoutrrr.Send(destinationURL.String(), email.Body()); err != nil {
@ -191,32 +213,27 @@ func (s *Session) Logout() error {
} }
func main() { func main() {
config := &Config{ var config *Config
Recipients: []ConfigRecipient{{ configPath := "config.toml"
Addresses: []string{"test@fmartingr.com", "caca@caca.com"},
target: "ntfy://ntfy.sh/fmartingr-dev", f, err := os.Open(configPath)
}, { if err != nil {
Addresses: []string{"something@something.com"}, slog.Error("Error opening config file", slog.String("err", err.Error()), slog.String("path", configPath))
target: "ntfy://ntfy.sh/fmartingr-dev", return
}},
} }
be := &Backend{ enc := encoding.NewTOMLEncoding()
config: config, if err := enc.DecodeReader(f, &config); err != nil {
slog.Error("Error decoding config file", slog.String("err", err.Error()), slog.String("path", configPath))
return
} }
smtpBackend := smtp.NewServer(be) config.SetDefaults()
smtpBackend.Addr = "localhost:11025" slog.Info("config loaded", slog.Int("recipients", len(config.Recipients)))
smtpBackend.Domain = "localhost"
smtpBackend.WriteTimeout = 10 * time.Second
smtpBackend.ReadTimeout = 10 * time.Second
smtpBackend.MaxMessageBytes = 1024 * 1024
smtpBackend.MaxRecipients = 50
smtpBackend.AllowInsecureAuth = true
ctx := context.Background() ctx := context.Background()
smtpServer := NewSMTPServer(smtpBackend) smtpServer := NewSMTPServer(config)
svc, err := service.NewService([]model.Server{ svc, err := service.NewService([]model.Server{
smtpServer, smtpServer,