From 9f57eca933ebb844b85fad9dd06e911222bd1424 Mon Sep 17 00:00:00 2001
From: "Scott R. Parish" <sparish@peak6.com>
Date: Mon, 18 Jan 2010 12:43:05 -0600
Subject: [PATCH] make field selector a list rather then a BsonDoc

Requiring callers to construct {foo: 1, bar: 1} style field selectors
is just dumb. Instead we'll let them pass in the field selector as a list
of strings and then construct the above form automatically for them.
---
 Database/MongoDB.hs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/Database/MongoDB.hs b/Database/MongoDB.hs
index 3ae22b5..93a0974 100644
--- a/Database/MongoDB.hs
+++ b/Database/MongoDB.hs
@@ -43,6 +43,7 @@ import Data.Binary.Put
 import Data.Bits
 import Data.ByteString.Char8 hiding (find)
 import qualified Data.ByteString.Lazy as L
+import qualified Data.ByteString.Lazy.UTF8 as L8
 import Data.Int
 import Data.IORef
 import qualified Data.List as List
@@ -131,7 +132,7 @@ toOpcode n = throw $ MongoDBInternalError $ "Got unexpected Opcode: " ++ show n
 
 type Collection = String
 type Selector = BsonDoc
-type FieldSelector = BsonDoc
+type FieldSelector = [L8.ByteString]
 type RequestID = Int32
 type NumToSkip = Int32
 type NumToReturn = Int32
@@ -194,7 +195,7 @@ insertMany c col docs = do
 {- | Open a cursor to find documents. If you need full functionality,
 see 'query' -}
 find :: Connection -> Collection -> Selector -> IO Cursor
-find c col sel = query c col [] 0 0 sel Nothing
+find c col sel = query c col [] 0 0 sel []
 
 {- | Perform a query and return the result as a lazy list. Be sure to
 understand the comments about using the lazy list given for 'allDocs'. -}
@@ -206,7 +207,7 @@ quickFind' :: Connection -> Collection -> Selector -> IO [BsonDoc]
 quickFind' c col sel = find c col sel >>= allDocs'
 
 query :: Connection -> Collection -> [QueryOpt] -> NumToSkip -> NumToReturn ->
-         Selector -> Maybe FieldSelector -> IO Cursor
+         Selector -> FieldSelector -> IO Cursor
 query c col opts nskip ret sel fsel = do
   let h = cHandle c
 
@@ -217,8 +218,8 @@ query c col opts nskip ret sel fsel = do
                putI32 ret
                put sel
                case fsel of
-                    Nothing -> putNothing
-                    Just _ -> put fsel
+                    [] -> putNothing
+                    _ -> put $ toBsonDoc $ List.zip fsel $ repeat $ BsonInt32 1
   (reqID, msg) <- packMsg c OP_QUERY body
   L.hPut h msg