From d3388ac3df96cc6f3f06a64711439a08625cc5ad Mon Sep 17 00:00:00 2001
From: "Benjamin S. Berman" <benjamin.s.berman@gmail.com>
Date: Thu, 11 Feb 2016 17:28:04 -0800
Subject: [PATCH] Add find syntax

---
 LiveData/Collection.cs | 30 +++++++++++++++++++++++++++++-
 LiveData/Cursor.cs     | 31 ++++++++++++++++++++++++++++---
 2 files changed, 57 insertions(+), 4 deletions(-)

diff --git a/LiveData/Collection.cs b/LiveData/Collection.cs
index c4e10e1..f2e5cc4 100644
--- a/LiveData/Collection.cs
+++ b/LiveData/Collection.cs
@@ -121,7 +121,6 @@ public Collection (string name) : base ()
 				throw new ArgumentException (string.Format ("A collection with name {0} already exists", name));
 			}
 
-
 			Collection<TRecordType>.Create (name, instance: this);
 		}
 
@@ -130,6 +129,35 @@ public Cursor<TRecordType> Find (Func<TRecordType, bool> selector = null)
 			return new Cursor<TRecordType> (collection: this, selector: selector);
 		}
 
+		public Cursor<TRecordType> Find (string id)
+		{
+			return new Cursor<TRecordType> (collection: this, id: id);
+		}
+
+		public Cursor<TRecordType> Find (IEnumerable<string> ids)
+		{
+			return new Cursor<TRecordType> (collection: this, ids: ids);
+		}
+
+		public TRecordType FindOne (string id)
+		{
+			return this [id];
+		}
+
+		public TRecordType FindOne (Func<TRecordType, bool> selector = null)
+		{
+			selector = selector ?? delegate(TRecordType arg) {
+				return true;
+			};
+
+			foreach (var record in this) {
+				if (selector (record)) {
+					return record;
+				}
+			}
+			return null;
+		}
+
 		public static Collection<TRecordType> Create (string name)
 		{
 			return Create (name, new Collection<TRecordType> ());
diff --git a/LiveData/Cursor.cs b/LiveData/Cursor.cs
index 4fa87cb..18a3355 100644
--- a/LiveData/Cursor.cs
+++ b/LiveData/Cursor.cs
@@ -11,6 +11,7 @@ namespace Meteor
 	public class Cursor<TRecordType>
 		where TRecordType : MongoDocument, new()
 	{
+
 		public Collection<TRecordType> collection {
 			get;
 			protected set;
@@ -21,12 +22,29 @@ public Func<TRecordType, bool> selector {
 			protected set;
 		}
 
+		public IEnumerable<string> ids {
+			get;
+			protected set;
+		}
+
 		public Cursor (Collection<TRecordType> collection, Func<TRecordType, bool> selector = null)
 		{
 			this.collection = collection;
 			this.selector = selector ?? SelectAll;
 		}
 
+		public Cursor (Collection<TRecordType> collection, string id)
+		{
+			this.collection = collection;
+			this.ids = new string[] { id };
+		}
+
+		public Cursor (Collection<TRecordType> collection, IEnumerable<string> ids)
+		{
+			this.collection = collection;
+			this.ids = ids;
+		}
+
 		public Observe<TRecordType> Observe (Action<string,TRecordType> added = null, Action<string,TRecordType,IDictionary,string[]> changed = null, Action<string> removed = null)
 		{
 			return new Observe<TRecordType> (collection: this.collection, added: added, changed: changed, removed: removed, selector: selector);
@@ -34,10 +52,17 @@ public Observe<TRecordType> Observe (Action<string,TRecordType> added = null, Ac
 
 		public IEnumerable<TRecordType> Fetch ()
 		{
-			foreach (var record in collection) {
-				if (selector (record)) {
-					yield return record;
+			if (ids == null) {
+				foreach (var record in collection) {
+					if (selector (record)) {
+						yield return record;
+					}
 				}
+				yield break;
+			}
+
+			foreach (var id in ids) {
+				yield return collection [id];
 			}
 
 			yield break;