package smtpd

import (
	"bytes"
	"net"
	"net/mail"

	"github.com/axllent/mailpit/config"
	"github.com/axllent/mailpit/logger"
	"github.com/axllent/mailpit/storage"
	s "github.com/mhale/smtpd"
)

func mailHandler(origin net.Addr, from string, to []string, data []byte) error {
	msg, err := mail.ReadMessage(bytes.NewReader(data))
	if err != nil {
		logger.Log().Errorf("error parsing message: %s", err.Error())
		return err
	}

	if _, err := storage.Store(storage.DefaultMailbox, data); err != nil {
		logger.Log().Errorf("error storing message: %s", err.Error())
		return err
	}

	subject := msg.Header.Get("Subject")
	logger.Log().Debugf("[smtp] received mail from %s for %s with subject %s", from, to[0], subject)
	return nil
}

// Listen starts the SMTPD server
func Listen() error {
	logger.Log().Infof("[smtp] starting on %s", config.SMTPListen)
	if err := s.ListenAndServe(config.SMTPListen, mailHandler, "Mailpit", ""); err != nil {
		return err
	}

	return nil
}