More config cleanups for #86

pkg/server/pop3/handler.go

@@ -496,7 +496,7 @@ func (ses *Session) sendMessageTop(msg storage.Message, lineCount int) {
 
 // Load the users mailbox
 func (ses *Session) loadMailbox() {
-	m, err := ses.server.dataStore.GetMessages(ses.user)
+	m, err := ses.server.store.GetMessages(ses.user)
 	if err != nil {
 		ses.logError("Failed to load messages for %v: %v", ses.user, err)
 	}
@@ -522,7 +522,7 @@ func (ses *Session) processDeletes() {
 	for i, msg := range ses.messages {
 		if !ses.retain[i] {
 			ses.logTrace("Deleting %v", msg)
-			if err := ses.server.dataStore.RemoveMessage(ses.user, msg.ID()); err != nil {
+			if err := ses.server.store.RemoveMessage(ses.user, msg.ID()); err != nil {
 				ses.logWarn("Error deleting %v: %v", msg, err)
 			}
 		}
@@ -536,7 +536,7 @@ func (ses *Session) enterState(state State) {
 
 // Calculate the next read or write deadline based on maxIdleSeconds
 func (ses *Session) nextDeadline() time.Time {
-	return time.Now().Add(ses.server.maxIdle)
+	return time.Now().Add(ses.server.timeout)
 }
 
 // Send requested message, store errors in Session.sendError

pkg/server/pop3/listener.go

@@ -15,20 +15,20 @@ import (
 type Server struct {
 	host           string
 	domain         string
-	maxIdle        time.Duration
-	dataStore      storage.Store
+	timeout        time.Duration
+	store          storage.Store
 	listener       net.Listener
 	globalShutdown chan bool
 	waitgroup      *sync.WaitGroup
 }
 
 // New creates a new Server struct
-func New(cfg config.POP3, shutdownChan chan bool, ds storage.Store) *Server {
+func New(cfg config.POP3, shutdownChan chan bool, store storage.Store) *Server {
 	return &Server{
 		host:           cfg.Addr,
 		domain:         cfg.Domain,
-		dataStore:      ds,
-		maxIdle:        cfg.MaxIdle,
+		store:          store,
+		timeout:        cfg.Timeout,
 		globalShutdown: shutdownChan,
 		waitgroup:      new(sync.WaitGroup),
 	}

pkg/server/smtp/handler.go

@@ -414,7 +414,7 @@ func (ss *Session) greet() {
 
 // Calculate the next read or write deadline based on maxIdle
 func (ss *Session) nextDeadline() time.Time {
-	return time.Now().Add(ss.server.maxIdle)
+	return time.Now().Add(ss.server.timeout)
 }
 
 // Send requested message, store errors in Session.sendError

pkg/server/smtp/handler_test.go

@@ -366,7 +366,7 @@ func setupSMTPServer(ds storage.Store) (s *Server, buf *bytes.Buffer, teardown f
 		Domain:          "inbucket.local",
 		DomainNoStore:   "bitbucket.local",
 		MaxRecipients:   5,
-		MaxIdle:         5,
+		Timeout:         5,
 		MaxMessageBytes: 5000,
 		StoreMessages:   true,
 	}

pkg/server/smtp/listener.go

@@ -42,9 +42,9 @@ type Server struct {
 	domain          string
 	domainNoStore   string
 	maxRecips       int
-	maxIdle         time.Duration
 	maxMessageBytes int
 	storeMessages   bool
+	timeout         time.Duration
 
 	// Dependencies
 	apolicy        *policy.Addressing // Address policy.
@@ -89,7 +89,7 @@ func NewServer(
 		domain:          cfg.Domain,
 		domainNoStore:   strings.ToLower(cfg.DomainNoStore),
 		maxRecips:       cfg.MaxRecipients,
-		maxIdle:         cfg.MaxIdle,
+		timeout:         cfg.Timeout,
 		maxMessageBytes: cfg.MaxMessageBytes,
 		storeMessages:   cfg.StoreMessages,
 		globalShutdown:  globalShutdown,