Initial commit

internal/command/clear_inbox.go

@@ -0,0 +1,41 @@
+package command
+
+import (
+	"context"
+
+	"github.com/pkg/errors"
+
+	"forge.cadoles.com/wpetit/fake-smtp/internal/model"
+	"forge.cadoles.com/wpetit/fake-smtp/internal/storm"
+	"gitlab.com/wpetit/goweb/cqrs"
+	"gitlab.com/wpetit/goweb/middleware/container"
+)
+
+type ClearInboxRequest struct{}
+
+func HandleClearInbox(ctx context.Context, cmd cqrs.Command) error {
+	_, ok := cmd.Request().(*ClearInboxRequest)
+	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")
+	}
+
+	if err := db.Select().Delete(&model.Email{}); err != nil {
+		if err == storm.ErrNotFound {
+			return nil
+		}
+
+		return errors.Wrap(err, "could not delete emails")
+	}
+
+	return nil
+}

internal/command/delete_email.go

@@ -0,0 +1,43 @@
+package command
+
+import (
+	"context"
+
+	"github.com/pkg/errors"
+
+	"forge.cadoles.com/wpetit/fake-smtp/internal/model"
+	"forge.cadoles.com/wpetit/fake-smtp/internal/storm"
+	"gitlab.com/wpetit/goweb/cqrs"
+	"gitlab.com/wpetit/goweb/middleware/container"
+)
+
+type DeleteEmailRequest struct {
+	EmailID int
+}
+
+func HandleDeleteEmail(ctx context.Context, cmd cqrs.Command) error {
+	req, ok := cmd.Request().(*DeleteEmailRequest)
+	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")
+	}
+
+	if err := db.DeleteStruct(&model.Email{ID: req.EmailID}); err != nil {
+		if err == storm.ErrNotFound {
+			return nil
+		}
+
+		return errors.Wrap(err, "could not delete email")
+	}
+
+	return nil
+}

internal/command/store_email.go

@@ -0,0 +1,100 @@
+package command
+
+import (
+	"context"
+	"time"
+
+	"github.com/pkg/errors"
+
+	"forge.cadoles.com/wpetit/fake-smtp/internal/model"
+	"forge.cadoles.com/wpetit/fake-smtp/internal/storm"
+	"github.com/jhillyerd/enmime"
+	"gitlab.com/wpetit/goweb/cqrs"
+	"gitlab.com/wpetit/goweb/middleware/container"
+)
+
+type StoreEmailRequest struct {
+	Envelope *enmime.Envelope
+}
+
+func HandleStoreEmail(ctx context.Context, cmd cqrs.Command) error {
+	req, ok := cmd.Request().(*StoreEmailRequest)
+	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")
+	}
+
+	email := &model.Email{
+		Headers: make(map[string][]string),
+		SentAt:  time.Now(),
+	}
+
+	email.HTML = req.Envelope.HTML
+	email.Text = req.Envelope.Text
+
+	for _, k := range req.Envelope.GetHeaderKeys() {
+		switch k {
+		case "From":
+			from, err := req.Envelope.AddressList(k)
+			if err != nil {
+				return errors.Wrapf(err, "could not retrieve '%s' adresses", k)
+			}
+
+			email.From = from
+
+		case "To":
+			to, err := req.Envelope.AddressList(k)
+			if err != nil {
+				return errors.Wrapf(err, "could not retrieve '%s' adresses", k)
+			}
+
+			email.To = to
+
+		case "Cc":
+			cc, err := req.Envelope.AddressList(k)
+			if err != nil {
+				return errors.Wrapf(err, "could not retrieve '%s' adresses", k)
+			}
+
+			email.Cc = cc
+
+		case "Cci":
+			cci, err := req.Envelope.AddressList(k)
+			if err != nil {
+				return errors.Wrapf(err, "could not retrieve '%s' adresses", k)
+			}
+
+			email.Cci = cci
+
+		case "Subject":
+			email.Subject = req.Envelope.GetHeader(k)
+
+		default:
+			email.Headers[k] = req.Envelope.GetHeaderValues(k)
+		}
+	}
+
+	email.Attachments = make([]*model.Attachment, 0, len(req.Envelope.Attachments))
+	for _, a := range req.Envelope.Attachments {
+		email.Attachments = append(email.Attachments, &model.Attachment{
+			ContentType: a.ContentType,
+			Name:        a.FileName,
+			Data:        a.Content,
+		})
+	}
+
+	if err := db.Save(email); err != nil {
+		return errors.Wrap(err, "could not save email")
+	}
+
+	return nil
+}