refactor: configuration into separate file
This commit is contained in:
parent
f1733d3670
commit
c829408095
2 changed files with 57 additions and 35 deletions
|
@ -3,7 +3,6 @@ package main
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
|
@ -23,6 +22,8 @@ import (
|
||||||
"github.com/containrrr/shoutrrr"
|
"github.com/containrrr/shoutrrr"
|
||||||
"github.com/emersion/go-sasl"
|
"github.com/emersion/go-sasl"
|
||||||
"github.com/emersion/go-smtp"
|
"github.com/emersion/go-smtp"
|
||||||
|
|
||||||
|
"git.nakama.town/fmartingr/smtp2shoutrrr"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ReceivedEmail struct {
|
type ReceivedEmail struct {
|
||||||
|
@ -82,39 +83,11 @@ func (re *ReceivedEmail) Body() (string, error) {
|
||||||
return re.body, nil
|
return re.body, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type Config struct {
|
|
||||||
Port int
|
|
||||||
Recipients []ConfigRecipient
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Config) SetDefaults() {
|
|
||||||
if c.Port == 0 {
|
|
||||||
c.Port = 11125
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type ConfigRecipient struct {
|
|
||||||
Addresses []string // email addresses
|
|
||||||
Target string // shoutrrr address
|
|
||||||
targetURL *url.URL
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cr ConfigRecipient) GetTargetURL() *url.URL {
|
|
||||||
if cr.targetURL == nil {
|
|
||||||
var err error
|
|
||||||
cr.targetURL, err = url.Parse(cr.Target)
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to parse shoutrrr target URL", slog.String("target", cr.Target), slog.String("err", err.Error()))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return cr.targetURL
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ model.Server = (*smtpServer)(nil)
|
var _ model.Server = (*smtpServer)(nil)
|
||||||
|
|
||||||
type smtpServer struct {
|
type smtpServer struct {
|
||||||
backend *smtp.Server
|
backend *smtp.Server
|
||||||
config Config
|
config smtp2shoutrrr.Config
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *smtpServer) IsEnabled() bool {
|
func (s *smtpServer) IsEnabled() bool {
|
||||||
|
@ -131,7 +104,7 @@ func (s *smtpServer) Stop(ctx context.Context) error {
|
||||||
return s.backend.Shutdown(ctx)
|
return s.backend.Shutdown(ctx)
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewSMTPServer(config *Config) model.Server {
|
func NewSMTPServer(config *smtp2shoutrrr.Config) model.Server {
|
||||||
be := &Backend{
|
be := &Backend{
|
||||||
config: config,
|
config: config,
|
||||||
}
|
}
|
||||||
|
@ -152,13 +125,14 @@ func NewSMTPServer(config *Config) model.Server {
|
||||||
|
|
||||||
// The Backend implements SMTP server methods.
|
// The Backend implements SMTP server methods.
|
||||||
type Backend struct {
|
type Backend struct {
|
||||||
config *Config
|
config *smtp2shoutrrr.Config
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewSession is called after client greeting (EHLO, HELO).
|
// NewSession is called after client greeting (EHLO, HELO).
|
||||||
func (bkd *Backend) NewSession(c *smtp.Conn) (smtp.Session, error) {
|
func (bkd *Backend) NewSession(c *smtp.Conn) (smtp.Session, error) {
|
||||||
return &Session{
|
return &Session{
|
||||||
forwarderFunc: bkd.forwardEmail,
|
forwarderFunc: bkd.forwardEmail,
|
||||||
|
config: bkd.config,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -200,6 +174,8 @@ type Session struct {
|
||||||
addresses []string
|
addresses []string
|
||||||
body string
|
body string
|
||||||
|
|
||||||
|
config *smtp2shoutrrr.Config
|
||||||
|
|
||||||
forwarderFunc func(ReceivedEmail) error
|
forwarderFunc func(ReceivedEmail) error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -212,8 +188,8 @@ func (s *Session) AuthMechanisms() []string {
|
||||||
// Auth is the handler for supported authenticators.
|
// Auth is the handler for supported authenticators.
|
||||||
func (s *Session) Auth(mech string) (sasl.Server, error) {
|
func (s *Session) Auth(mech string) (sasl.Server, error) {
|
||||||
return sasl.NewPlainServer(func(identity, username, password string) error {
|
return sasl.NewPlainServer(func(identity, username, password string) error {
|
||||||
if username != "username" || password != "password" {
|
if username != s.config.Username && password != s.config.Password {
|
||||||
return errors.New("Invalid username or password")
|
return fmt.Errorf("invalid credentials")
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}), nil
|
}), nil
|
||||||
|
@ -256,7 +232,7 @@ func (s *Session) Logout() error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
var config *Config
|
var config *smtp2shoutrrr.Config
|
||||||
configPath := "config.toml"
|
configPath := "config.toml"
|
||||||
|
|
||||||
f, err := os.Open(configPath)
|
f, err := os.Open(configPath)
|
||||||
|
|
46
config.go
Normal file
46
config.go
Normal file
|
@ -0,0 +1,46 @@
|
||||||
|
package smtp2shoutrrr
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log/slog"
|
||||||
|
"net/url"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
Port int
|
||||||
|
Username string
|
||||||
|
Password string
|
||||||
|
Recipients []ConfigRecipient
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Config) SetDefaults() {
|
||||||
|
if c.Port == 0 {
|
||||||
|
c.Port = 11125
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.Username == "" {
|
||||||
|
slog.Warn("no username provided, using default: username")
|
||||||
|
c.Username = "username"
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.Password == "" {
|
||||||
|
slog.Warn("no password provided, using default: password")
|
||||||
|
c.Password = "password"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type ConfigRecipient struct {
|
||||||
|
Addresses []string // email addresses
|
||||||
|
Target string // shoutrrr address
|
||||||
|
targetURL *url.URL
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cr ConfigRecipient) GetTargetURL() *url.URL {
|
||||||
|
if cr.targetURL == nil {
|
||||||
|
var err error
|
||||||
|
cr.targetURL, err = url.Parse(cr.Target)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("failed to parse shoutrrr target URL", slog.String("target", cr.Target), slog.String("err", err.Error()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return cr.targetURL
|
||||||
|
}
|
Loading…
Add table
Add a link
Reference in a new issue