server.go

package smtpd

import (
	"io"
	"log"
	"os"
)

type Server struct {
	Hostname string
	Mailer   Mailer
	IOLoop   IOLoop

	commands map[string]commandFactory
}

func NewServer() *Server {
	host, err := os.Hostname()
	if err != nil {
		log.Println("No hostname available. Using 'local'")
		host = "local"
	}

	return &Server{
		Hostname: host,
		commands: defaultCommands(),
	}
}

func (s *Server) Run() {
	err := s.IOLoop.Run(s.accept)
	if err != nil {
		panic(err)
	}
}

func (s *Server) accept(r io.ReadCloser, w io.Writer) {
	channel := &WriterChannel{w: w, host: s.Hostname}
	exchange := NewExchange(s.Mailer, r, channel)
	loop := &protocolLoop{
		ex:       exchange,
		commands: s.commands,
		next:     []string{CommandHelo},
		waiting:  true,
	}

	loop.Run()
}