flashmob.go-guerrilla/backends/p_debugger.go

package backends

import (
	"github.com/flashmob/go-guerrilla/mail"
	"strings"
	"time"
)

// ----------------------------------------------------------------------------------
// Processor Name: debugger
// ----------------------------------------------------------------------------------
// Description   : Log received emails
// ----------------------------------------------------------------------------------
// Config Options: log_received_mails bool - log if true
// --------------:-------------------------------------------------------------------
// Input         : e.MailFrom, e.RcptTo, e.Header
// ----------------------------------------------------------------------------------
// Output        : none (only output to the log if enabled)
// ----------------------------------------------------------------------------------
func init() {
	processors[strings.ToLower(defaultProcessor)] = func() Decorator {
		return Debugger()
	}
}

type debuggerConfig struct {
	LogReceivedMails bool `json:"log_received_mails"`
	SleepSec         int  `json:"sleep_seconds,omitempty"`
}

func Debugger() Decorator {
	var config *debuggerConfig
	initFunc := InitializeWith(func(backendConfig BackendConfig) error {
		configType := BaseConfig(&debuggerConfig{})
		bcfg, err := Svc.ExtractConfig(backendConfig, configType)
		if err != nil {
			return err
		}
		config = bcfg.(*debuggerConfig)
		return nil
	})
	Svc.AddInitializer(initFunc)
	return func(p Processor) Processor {
		return ProcessWith(func(e *mail.Envelope, task SelectTask) (Result, error) {
			if task == TaskSaveMail {
				if config.LogReceivedMails {
					Log().Infof("Mail from: %s / to: %v", e.MailFrom.String(), e.RcptTo)
					Log().Info("Headers are:", e.Header)
				}

				if config.SleepSec > 0 {
					Log().Infof("sleeping for %d", config.SleepSec)
					time.Sleep(time.Second * time.Duration(config.SleepSec))
					Log().Infof("woke up")

					if config.SleepSec == 1 {
						panic("panic on purpose")
					}

				}

				// continue to the next Processor in the decorator stack
				return p.Process(e, task)
			} else {
				return p.Process(e, task)
			}
		})
	}
}