Add basic JSON API with filtering

internal/query/get_inbox.go

@@ -2,19 +2,33 @@ package query
 
 import (
 	"context"
+	"strings"
+	"time"
 
 	"forge.cadoles.com/wpetit/fake-smtp/internal/model"
 	"forge.cadoles.com/wpetit/fake-smtp/internal/storm"
+	stormdb "github.com/asdine/storm/v3"
+	"github.com/asdine/storm/v3/q"
 	"github.com/pkg/errors"
 	"gitlab.com/wpetit/goweb/cqrs"
 	"gitlab.com/wpetit/goweb/middleware/container"
 )
 
+type InboxSearch struct {
+	To      string
+	From    string
+	Body    string
+	Subject string
+	After   time.Time
+	Before  time.Time
+}
+
 type GetInboxRequest struct {
 	OrderBy string
 	Limit   int
 	Skip    int
 	Reverse bool
+	Search  *InboxSearch
 }
 
 type InboxData struct {
@@ -39,7 +53,26 @@ func HandleGetInbox(ctx context.Context, qry cqrs.Query) (interface{}, error) {
 
 	emails := make([]*model.Email, 0)
 
-	query := db.Select()
+	var query stormdb.Query
+
+	if req.Search != nil {
+		matchers := make([]q.Matcher, 0)
+
+		if req.Search.Body != "" {
+			matchers = append(matchers, q.Or(
+				q.Re("HTML", req.Search.Body),
+				q.Re("Text", req.Search.Body),
+			))
+		}
+
+		if req.Search.Subject != "" {
+			matchers = append(matchers, q.Re("Subject", req.Search.Subject))
+		}
+
+		query = db.Select(matchers...)
+	} else {
+		query = db.Select()
+	}
 
 	if req.OrderBy != "" {
 		query = query.OrderBy(req.OrderBy)
@@ -51,6 +84,14 @@ func HandleGetInbox(ctx context.Context, qry cqrs.Query) (interface{}, error) {
 		query = query.Reverse()
 	}
 
+	if req.Limit != 0 {
+		query = query.Limit(req.Limit)
+	}
+
+	if req.Skip != 0 {
+		query = query.Limit(req.Skip)
+	}
+
 	if err := query.Find(&emails); err != nil {
 		if err == storm.ErrNotFound {
 			return &InboxData{emails}, nil
@@ -59,5 +100,59 @@ func HandleGetInbox(ctx context.Context, qry cqrs.Query) (interface{}, error) {
 		return nil, errors.Wrap(err, "could not retrieve emails")
 	}
 
-	return &InboxData{emails}, nil
+	if req.Search == nil {
+		return &InboxData{emails}, nil
+	}
+
+	filtered := make([]*model.Email, 0, len(emails))
+
+	for _, eml := range emails {
+		match := true
+
+		if req.Search.To != "" {
+			found := false
+
+			for _, addr := range eml.To {
+				if strings.Contains(addr.Name, req.Search.To) || strings.Contains(addr.Address, req.Search.To) {
+					found = true
+
+					break
+				}
+			}
+
+			if !found {
+				match = false
+			}
+		}
+
+		if req.Search.From != "" {
+			found := false
+
+			for _, addr := range eml.From {
+				if strings.Contains(addr.Name, req.Search.From) || strings.Contains(addr.Address, req.Search.From) {
+					found = true
+
+					break
+				}
+			}
+
+			if !found {
+				match = false
+			}
+		}
+
+		if !req.Search.After.IsZero() && !eml.SentAt.After(req.Search.After) {
+			match = false
+		}
+
+		if !req.Search.Before.IsZero() && !eml.SentAt.Before(req.Search.Before) {
+			match = false
+		}
+
+		if match {
+			filtered = append(filtered, eml)
+		}
+	}
+
+	return &InboxData{filtered}, nil
 }