From 03cb13bf401df997422adb5978cb71dd572b2236 Mon Sep 17 00:00:00 2001
From: Sean Herrala <sherrala@topofmind.com>
Date: Thu, 17 Mar 2016 13:45:27 -0400
Subject: [PATCH] Add raw (UI needs) and implement searching for MongoDB
 storage

---
 mongodb.go | 29 ++++++++++++++++++++++++++---
 1 file changed, 26 insertions(+), 3 deletions(-)

diff --git a/mongodb.go b/mongodb.go
index cb131ad..015c2d5 100644
--- a/mongodb.go
+++ b/mongodb.go
@@ -2,7 +2,6 @@ package storage
 
 import (
 	"log"
-
 	"github.com/mailhog/data"
 	"labix.org/v2/mgo"
 	"labix.org/v2/mgo/bson"
@@ -52,8 +51,31 @@ func (mongo *MongoDB) Count() int {
 // Search finds messages matching the query
 func (mongo *MongoDB) Search(kind, query string, start, limit int) (*data.Messages, int, error) {
 	messages := &data.Messages{}
-	// FIXME
-	return messages, 0, nil
+	var count = 0
+	var field = "raw.data"
+	switch kind {
+		case "to":
+			field = "raw.to"
+		case "from":
+			field = "raw.from"
+	}
+	err := mongo.Collection.Find(bson.M{field: bson.RegEx{Pattern: query, Options: "i"}}).Skip(start).Limit(limit).Sort("-created").Select(bson.M{
+		"id":              1,
+		"_id":             1,
+		"from":            1,
+		"to":              1,
+		"content.headers": 1,
+		"content.size":    1,
+		"created":         1,
+		"raw":             1,
+	}).All(messages)
+	if err != nil {
+		log.Printf("Error loading messages: %s", err)
+		return nil, 0, err
+	}
+	count, _ = mongo.Collection.Find(bson.M{field: bson.RegEx{Pattern: query, Options: "i"}}).Count()
+
+	return messages, count, nil
 }
 
 // List returns a list of messages by index
@@ -67,6 +89,7 @@ func (mongo *MongoDB) List(start int, limit int) (*data.Messages, error) {
 		"content.headers": 1,
 		"content.size":    1,
 		"created":         1,
+		"raw":             1,
 	}).All(messages)
 	if err != nil {
 		log.Printf("Error loading messages: %s", err)