diff --git a/internal/query/get_inbox.go b/internal/query/get_inbox.go index cdd94ff..9bb508c 100644 --- a/internal/query/get_inbox.go +++ b/internal/query/get_inbox.go @@ -19,6 +19,7 @@ type InboxSearch struct { From string Body string Subject string + Headers map[string]string After time.Time Before time.Time } @@ -149,6 +150,38 @@ func HandleGetInbox(ctx context.Context, qry cqrs.Query) (interface{}, error) { match = false } + if req.Search.Headers != nil { + found := false + + for searchKey, searchValue := range req.Search.Headers { + for headerKey, headerValues := range eml.Headers { + if searchKey != headerKey { + continue + } + + for _, hv := range headerValues { + if strings.Contains(hv, searchValue) { + found = true + + break + } + } + + if found { + break + } + } + + if found { + break + } + } + + if !found { + match = false + } + } + if match { filtered = append(filtered, eml) } diff --git a/internal/route/helper.go b/internal/route/helper.go index 950a11b..0308995 100644 --- a/internal/route/helper.go +++ b/internal/route/helper.go @@ -1,6 +1,7 @@ package route import ( + "encoding/json" "net/http" "strconv" "time" @@ -75,6 +76,16 @@ func createInboxQueryFromRequest(r *http.Request) (*query.GetInboxRequest, error } } + var headers map[string]string + + rawHeaders := r.URL.Query().Get("headers") + if rawHeaders != "" { + headers = make(map[string]string) + if err := json.Unmarshal([]byte(rawHeaders), &headers); err != nil { + return nil, errors.WithStack(err) + } + } + search := &query.InboxSearch{} if to != "" { search.To = to @@ -96,6 +107,10 @@ func createInboxQueryFromRequest(r *http.Request) (*query.GetInboxRequest, error search.Before = before } + if rawHeaders != "" { + search.Headers = headers + } + inboxRequest := &query.GetInboxRequest{ OrderBy: orderBy, Reverse: reverse == "y", diff --git a/misc/api.http b/misc/api.http new file mode 100644 index 0000000..ca63531 --- /dev/null +++ b/misc/api.http @@ -0,0 +1,5 @@ +@baseURL = http://localhost:8080 + +### Filter emails via headers + +GET {{ baseURL }}/api/v1/emails?headers={"Mime-Version":"1.0"} \ No newline at end of file