Merge pull request #2350 from MediaBrowser/beta

Beta

1 files changed, 28 insertions, 0 deletions

diff --git a/RSSDP/IEnumerableExtensions.cs b/RSSDP/IEnumerableExtensions.cs
new file mode 100644
index 000000000..f72073949
--- /dev/null
+++ b/RSSDP/IEnumerableExtensions.cs
@@ -0,0 +1,28 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+
+namespace Rssdp.Infrastructure
+{
+	internal static class IEnumerableExtensions
+	{
+		public static IEnumerable<T> SelectManyRecursive<T>(this IEnumerable<T> source, Func<T, IEnumerable<T>> selector)
+		{
+			if (source == null) throw new ArgumentNullException("source");
+			if (selector == null) throw new ArgumentNullException("selector");
+
+			return !source.Any() ? source :
+					source.Concat(
+							source
+							.SelectMany(i => selector(i).EmptyIfNull())
+							.SelectManyRecursive(selector)
+					);
+		}
+
+		public static IEnumerable<T> EmptyIfNull<T>(this IEnumerable<T> source)
+		{
+			return source ?? Enumerable.Empty<T>();
+		}
+	}
+}