Initial commit

internal/query/get_outbox.go

@@ -0,0 +1,133 @@
+package query
+
+import (
+	"context"
+	"strings"
+	"time"
+
+	"forge.cadoles.com/Cadoles/fake-sms/internal/model"
+	"forge.cadoles.com/Cadoles/fake-sms/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 OutboxSearch struct {
+	Recipient string
+	Body      string
+	After     time.Time
+	Before    time.Time
+}
+
+type GetOutboxRequest struct {
+	OrderBy string
+	Limit   int
+	Skip    int
+	Reverse bool
+	Search  *OutboxSearch
+}
+
+type OutboxData struct {
+	Messages []*model.SMS
+}
+
+func HandleGetOutbox(ctx context.Context, qry cqrs.Query) (interface{}, error) {
+	req, ok := qry.Request().(*GetOutboxRequest)
+	if !ok {
+		return nil, cqrs.ErrUnexpectedRequest
+	}
+
+	ctn, err := container.From(ctx)
+	if err != nil {
+		return nil, errors.Wrap(err, "could not retrieve service container")
+	}
+
+	db, err := storm.From(ctn)
+	if err != nil {
+		return nil, errors.Wrap(err, "could not retrieve storm service")
+	}
+
+	messages := make([]*model.SMS, 0)
+
+	var query stormdb.Query
+
+	if req.Search != nil {
+		matchers := make([]q.Matcher, 0)
+
+		if req.Search.Body != "" {
+			matchers = append(matchers, q.Or(
+				q.Re("Body", req.Search.Body),
+			))
+		}
+
+		query = db.Select(matchers...)
+	} else {
+		query = db.Select()
+	}
+
+	if req.OrderBy != "" {
+		query = query.OrderBy(req.OrderBy)
+	} else {
+		query = query.OrderBy("SentAt").Reverse()
+	}
+
+	if req.Reverse {
+		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(&messages); err != nil {
+		if err == storm.ErrNotFound {
+			return &OutboxData{messages}, nil
+		}
+
+		return nil, errors.Wrap(err, "could not retrieve emails")
+	}
+
+	if req.Search == nil {
+		return &OutboxData{messages}, nil
+	}
+
+	filtered := make([]*model.SMS, 0, len(messages))
+
+	for _, sms := range messages {
+		match := true
+
+		if req.Search.Recipient != "" {
+			found := false
+
+			if strings.Contains(sms.Recipient, req.Search.Recipient) {
+				found = true
+
+				break
+			}
+
+			if !found {
+				match = false
+			}
+		}
+
+		if !req.Search.After.IsZero() && !sms.SentAt.After(req.Search.After) {
+			match = false
+		}
+
+		if !req.Search.Before.IsZero() && !sms.SentAt.Before(req.Search.Before) {
+			match = false
+		}
+
+		if match {
+			filtered = append(filtered, sms)
+		}
+	}
+
+	return &OutboxData{filtered}, nil
+}

internal/query/open_sms.go

@@ -0,0 +1,44 @@
+package query
+
+import (
+	"context"
+
+	"forge.cadoles.com/Cadoles/fake-sms/internal/model"
+	"forge.cadoles.com/Cadoles/fake-sms/internal/storm"
+	"github.com/pkg/errors"
+	"gitlab.com/wpetit/goweb/cqrs"
+	"gitlab.com/wpetit/goweb/middleware/container"
+)
+
+type OpenSMSRequest struct {
+	SMSID int
+}
+
+type OpenSMSData struct {
+	SMS *model.SMS
+}
+
+func HandleOpenSMS(ctx context.Context, qry cqrs.Query) (interface{}, error) {
+	req, ok := qry.Request().(*OpenSMSRequest)
+	if !ok {
+		return nil, cqrs.ErrUnexpectedRequest
+	}
+
+	ctn, err := container.From(ctx)
+	if err != nil {
+		return nil, errors.Wrap(err, "could not retrieve service container")
+	}
+
+	db, err := storm.From(ctn)
+	if err != nil {
+		return nil, errors.Wrap(err, "could not retrieve storm service")
+	}
+
+	sms := &model.SMS{}
+
+	if err := db.One("ID", req.SMSID, sms); err != nil {
+		return nil, errors.Wrap(err, "could not find email")
+	}
+
+	return &OpenSMSData{sms}, nil
+}