From 0c23304d34eceba9bb571e43e009eb506f4edd40 Mon Sep 17 00:00:00 2001
From: mlamalle <mlamalle@cadoles.com>
Date: Thu, 13 Mar 2025 13:33:51 +0100
Subject: [PATCH] add email rotation regarding a limit

---
 README.md                        |  1 +
 cmd/fake-smtp/container.go       |  5 ++++
 cmd/fake-smtp/smtp.go            | 12 +++++++++
 internal/command/rotate_email.go | 44 ++++++++++++++++++++++++++++++++
 internal/config/config.go        |  6 +++--
 5 files changed, 66 insertions(+), 2 deletions(-)
 create mode 100644 internal/command/rotate_email.go

diff --git a/README.md b/README.md
index 5512fcc..5855643 100644
--- a/README.md
+++ b/README.md
@@ -90,6 +90,7 @@ Les valeurs des variables d'environnement surchargent les valeurs présentes dan
 | `FAKESMTP_SMTP_MAXRECIPIENTS`         | `smtp.maxRecipients`                            |
 | `FAKESMTP_SMTP_ALLOWINSECUREAUTH`     | `smtp.allowInsecureAuth`                        |
 | `FAKESMTP_SMTP_DEBUG`                 | `smtp.debug`                                    |
+| `FAKESMTP_DATA_MAX_EMAIL`             | `data.maxEmail`                                 |
 | `FAKESMTP_DATA_PATH`                  | `data.path`                                     |
 | `FAKESMTP_RELAY_ENABLED`              | `relay.enabled`                                 |
 | `FAKESMTP_RELAY_ADDRESS`              | `relay.address`                                 |
diff --git a/cmd/fake-smtp/container.go b/cmd/fake-smtp/container.go
index 724a957..a31f8ce 100644
--- a/cmd/fake-smtp/container.go
+++ b/cmd/fake-smtp/container.go
@@ -43,6 +43,11 @@ func getServiceContainer(conf *config.Config) (*service.Container, error) {
 		cqrs.CommandHandlerFunc(command.HandleStoreEmail),
 	)
 
+	bus.RegisterCommand(
+		cqrs.MatchCommandRequest(&command.RotateEmailRequest{}),
+		cqrs.CommandHandlerFunc(command.HandleRotateEmail),
+	)
+
 	bus.RegisterCommand(
 		cqrs.MatchCommandRequest(&command.ClearInboxRequest{}),
 		cqrs.CommandHandlerFunc(command.HandleClearInbox),
diff --git a/cmd/fake-smtp/smtp.go b/cmd/fake-smtp/smtp.go
index a265ad8..4cc4756 100644
--- a/cmd/fake-smtp/smtp.go
+++ b/cmd/fake-smtp/smtp.go
@@ -92,6 +92,18 @@ func (s *Session) Data(r io.Reader) error {
 		}
 	}
 
+	if conf.Data.MaxEmail > 0 {
+		cmd := &command.RotateEmailRequest{
+			MaxEmail: conf.Data.MaxEmail,
+		}
+
+		if _, err := bus.Exec(s.ctx, cmd); err != nil {
+			logger.Error(s.ctx, "could not exec command", logger.E(err))
+
+			return errors.Wrapf(err, "could not exec '%T' command", cmd)
+		}
+	}
+
 	cmd := &command.StoreEmailRequest{
 		Envelope: env,
 	}
diff --git a/internal/command/rotate_email.go b/internal/command/rotate_email.go
new file mode 100644
index 0000000..ad49be8
--- /dev/null
+++ b/internal/command/rotate_email.go
@@ -0,0 +1,44 @@
+package command
+
+import (
+	"context"
+
+	"forge.cadoles.com/Cadoles/fake-smtp/internal/model"
+	"forge.cadoles.com/Cadoles/fake-smtp/internal/storm"
+	"github.com/pkg/errors"
+	"gitlab.com/wpetit/goweb/cqrs"
+	"gitlab.com/wpetit/goweb/middleware/container"
+)
+
+type RotateEmailRequest struct {
+	MaxEmail int
+}
+
+func HandleRotateEmail(ctx context.Context, cmd cqrs.Command) error {
+	req, ok := cmd.Request().(*RotateEmailRequest)
+	if !ok {
+		return cqrs.ErrUnexpectedRequest
+	}
+
+	ctn, err := container.From(ctx)
+	if err != nil {
+		return errors.Wrap(err, "could not retrieve service container")
+	}
+
+	db, err := storm.From(ctn)
+	if err != nil {
+		return errors.Wrap(err, "could not retrieve storm service")
+	}
+
+	emailCount, err := db.Select().Count(&model.Email{})
+	if err != nil {
+		return errors.Wrap(err, "could not count emails")
+	}
+
+	if emailCount >= req.MaxEmail {
+		var diff = emailCount - (req.MaxEmail - 1)
+		db.Select().OrderBy("ID").Limit(diff).Delete(&model.Email{})
+	}
+
+	return nil
+}
diff --git a/internal/config/config.go b/internal/config/config.go
index 49cf0ed..7725ad3 100644
--- a/internal/config/config.go
+++ b/internal/config/config.go
@@ -51,7 +51,8 @@ type RelayConfig struct {
 }
 
 type DataConfig struct {
-	Path string `yaml:"path" env:"FAKESMTP_DATA_PATH"`
+	MaxEmail int    `yaml:"maxEmail" env:"FAKESMTP_DATA_MAX_EMAIL"`
+	Path     string `yaml:"path" env:"FAKESMTP_DATA_PATH"`
 }
 
 // NewFromFile retrieves the configuration from the given file
@@ -103,7 +104,8 @@ func NewDefault() *Config {
 			Debug:             true,
 		},
 		Data: DataConfig{
-			Path: "fakesmtp.db",
+			MaxEmail: 0,
+			Path:     "fakesmtp.db",
 		},
 		Relay: RelayConfig{
 			Enabled: false,