From 86ab26e2d9bb0bbfae0971176dd34d507634080f Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Sun, 19 Jun 2011 20:08:09 -0700
Subject: [PATCH 1/6] UpcomingProvider now gets Series in call for episodes.

---
 .../Providers/UpcomingEpisodesProvider.cs     |  22 +++-
 NzbDrone.Core/Repository/Episode.cs           |   2 +-
 .../Controllers/UpcomingController.cs         | 112 +++++++-----------
 3 files changed, 57 insertions(+), 79 deletions(-)

diff --git a/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs b/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs
index 2b0d9c9f1..423f33947 100644
--- a/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs
+++ b/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs
@@ -20,8 +20,10 @@ namespace NzbDrone.Core.Providers
 
         public virtual UpcomingEpisodesModel Upcoming()
         {
-            var allEps = _database.Fetch<Episode>("WHERE AirDate BETWEEN @0 AND @1", DateTime.Today.AddDays(-1),
-                                              DateTime.Today.AddDays(8));
+            var allEps = _database.Fetch<Episode, Series>(@"SELECT * FROM Episodes 
+                                                        INNER JOIN Series ON Episodes.SeriesId = Series.SeriesId
+                                                        WHERE AirDate BETWEEN @0 AND @1",
+                                                        DateTime.Today.AddDays(-1), DateTime.Today.AddDays(8));
 
             var yesterday = allEps.Where(e => e.AirDate == DateTime.Today.AddDays(-1)).ToList();
             var today = allEps.Where(e => e.AirDate == DateTime.Today).ToList();
@@ -32,22 +34,30 @@ namespace NzbDrone.Core.Providers
 
         public virtual List<Episode> Yesterday()
         {
-            return _database.Fetch<Episode>("WHERE AirDate = @0", DateTime.Today.AddDays(-1));
+            return _database.Fetch<Episode, Series>(@"SELECT * FROM Episodes 
+                                                        INNER JOIN Series ON Episodes.SeriesId = Series.SeriesId
+                                                        WHERE AirDate = @0", DateTime.Today.AddDays(-1));
         }
 
         public virtual List<Episode> Today()
         {
-            return _database.Fetch<Episode>("WHERE AirDate = @0", DateTime.Today);
+            return _database.Fetch<Episode, Series>(@"SELECT * FROM Episodes 
+                                                        INNER JOIN Series ON Episodes.SeriesId = Series.SeriesId
+                                                        WHERE AirDate = @0", DateTime.Today);
         }
 
         public virtual List<Episode> Tomorrow()
         {
-            return _database.Fetch<Episode>("WHERE AirDate = @0", DateTime.Today.AddDays(1));
+            return _database.Fetch<Episode, Series>(@"SELECT * FROM Episodes 
+                                                        INNER JOIN Series ON Episodes.SeriesId = Series.SeriesId
+                                                        WHERE AirDate = @0", DateTime.Today.AddDays(1));
         }
 
         public virtual List<Episode> Week()
         {
-            return _database.Fetch<Episode>("WHERE AirDate BETWEEN @0 AND @1", DateTime.Today.AddDays(2), DateTime.Today.AddDays(8));
+            return _database.Fetch<Episode, Series>(@"SELECT * FROM Episodes 
+                                                        INNER JOIN Series ON Episodes.SeriesId = Series.SeriesId
+                                                        WHERE AirDate BETWEEN @0 AND @1", DateTime.Today.AddDays(2), DateTime.Today.AddDays(8));
         }
     }
 }
\ No newline at end of file
diff --git a/NzbDrone.Core/Repository/Episode.cs b/NzbDrone.Core/Repository/Episode.cs
index 6e482e2ec..f5ce16816 100644
--- a/NzbDrone.Core/Repository/Episode.cs
+++ b/NzbDrone.Core/Repository/Episode.cs
@@ -32,7 +32,7 @@ namespace NzbDrone.Core.Repository
         /// </remarks>
         public DateTime? GrabDate { get; set; }
 
-        [Ignore]
+        [ResultColumn]
         public EpisodeStatusType Status
         {
             get
diff --git a/NzbDrone.Web/Controllers/UpcomingController.cs b/NzbDrone.Web/Controllers/UpcomingController.cs
index d5c02f4a4..3ca9d69e6 100644
--- a/NzbDrone.Web/Controllers/UpcomingController.cs
+++ b/NzbDrone.Web/Controllers/UpcomingController.cs
@@ -30,25 +30,17 @@ namespace NzbDrone.Web.Controllers
         [GridAction]
         public ActionResult _AjaxBindingYesterday()
         {
-            var upcomingDb = _upcomingEpisodesProvider.Yesterday();
-            var upcoming = new List<UpcomingEpisodeModel>();
-                
-            foreach (var item in upcomingDb)
+            var upcoming = _upcomingEpisodesProvider.Yesterday().Select(u => new UpcomingEpisodeModel
             {
-                var series = _seriesProvider.GetSeries(item.SeriesId);
-
-                upcoming.Add(new UpcomingEpisodeModel
-                                                    {
-                                                        SeriesId = series.SeriesId,
-                                                        EpisodeId = item.EpisodeId,
-                                                        SeriesName = series.Title,
-                                                        SeasonNumber = item.SeasonNumber,
-                                                        EpisodeNumber = item.EpisodeNumber,
-                                                        Title = item.Title,
-                                                        Overview = item.Overview,
-                                                        AirDate = item.AirDate.Add(Convert.ToDateTime(series.AirTimes).TimeOfDay)
-                                                    });
-            }
+                SeriesId = u.Series.SeriesId,
+                EpisodeId = u.EpisodeId,
+                SeriesName = u.Series.Title,
+                SeasonNumber = u.SeasonNumber,
+                EpisodeNumber = u.EpisodeNumber,
+                Title = u.Title,
+                Overview = u.Overview,
+                AirDate = u.AirDate.Add(Convert.ToDateTime(u.Series.AirTimes).TimeOfDay)
+            });
 
             return View(new GridModel(upcoming));
         }
@@ -56,25 +48,17 @@ namespace NzbDrone.Web.Controllers
         [GridAction]
         public ActionResult _AjaxBindingToday()
         {
-            var upcomingDb = _upcomingEpisodesProvider.Today();
-            var upcoming = new List<UpcomingEpisodeModel>();
-
-            foreach (var item in upcomingDb)
+            var upcoming = _upcomingEpisodesProvider.Today().Select(u => new UpcomingEpisodeModel
             {
-                var series = _seriesProvider.GetSeries(item.SeriesId);
-
-                upcoming.Add(new UpcomingEpisodeModel
-                {
-                    SeriesId = series.SeriesId,
-                    EpisodeId = item.EpisodeId,
-                    SeriesName = series.Title,
-                    SeasonNumber = item.SeasonNumber,
-                    EpisodeNumber = item.EpisodeNumber,
-                    Title = item.Title,
-                    Overview = item.Overview,
-                    AirDate = item.AirDate.Add(Convert.ToDateTime(series.AirTimes).TimeOfDay)
-                });
-            }
+                SeriesId = u.Series.SeriesId,
+                EpisodeId = u.EpisodeId,
+                SeriesName = u.Series.Title,
+                SeasonNumber = u.SeasonNumber,
+                EpisodeNumber = u.EpisodeNumber,
+                Title = u.Title,
+                Overview = u.Overview,
+                AirDate = u.AirDate.Add(Convert.ToDateTime(u.Series.AirTimes).TimeOfDay)
+            });
 
             return View(new GridModel(upcoming));
         }
@@ -82,25 +66,17 @@ namespace NzbDrone.Web.Controllers
         [GridAction]
         public ActionResult _AjaxBindingTomorrow()
         {
-            var upcomingDb = _upcomingEpisodesProvider.Tomorrow();
-            var upcoming = new List<UpcomingEpisodeModel>();
-
-            foreach (var item in upcomingDb)
+            var upcoming = _upcomingEpisodesProvider.Tomorrow().Select(u => new UpcomingEpisodeModel
             {
-                var series = _seriesProvider.GetSeries(item.SeriesId);
-
-                upcoming.Add(new UpcomingEpisodeModel
-                {
-                    SeriesId = series.SeriesId,
-                    EpisodeId = item.EpisodeId,
-                    SeriesName = series.Title,
-                    SeasonNumber = item.SeasonNumber,
-                    EpisodeNumber = item.EpisodeNumber,
-                    Title = item.Title,
-                    Overview = item.Overview,
-                    AirDate = item.AirDate.Add(Convert.ToDateTime(series.AirTimes).TimeOfDay)
-                });
-            }
+                SeriesId = u.Series.SeriesId,
+                EpisodeId = u.EpisodeId,
+                SeriesName = u.Series.Title,
+                SeasonNumber = u.SeasonNumber,
+                EpisodeNumber = u.EpisodeNumber,
+                Title = u.Title,
+                Overview = u.Overview,
+                AirDate = u.AirDate.Add(Convert.ToDateTime(u.Series.AirTimes).TimeOfDay)
+            });
 
             return View(new GridModel(upcoming));
         }
@@ -108,25 +84,17 @@ namespace NzbDrone.Web.Controllers
         [GridAction]
         public ActionResult _AjaxBindingWeek()
         {
-            var upcomingDb = _upcomingEpisodesProvider.Week();
-            var upcoming = new List<UpcomingEpisodeModel>();
-
-            foreach (var item in upcomingDb)
+            var upcoming = _upcomingEpisodesProvider.Week().Select(u => new UpcomingEpisodeModel
             {
-                var series = _seriesProvider.GetSeries(item.SeriesId);
-
-                upcoming.Add(new UpcomingEpisodeModel
-                {
-                    SeriesId = series.SeriesId,
-                    EpisodeId = item.EpisodeId,
-                    SeriesName = series.Title,
-                    SeasonNumber = item.SeasonNumber,
-                    EpisodeNumber = item.EpisodeNumber,
-                    Title = item.Title,
-                    Overview = item.Overview,
-                    AirDate = item.AirDate.Add(Convert.ToDateTime(series.AirTimes).TimeOfDay)
-                });
-            }
+                SeriesId = u.Series.SeriesId,
+                EpisodeId = u.EpisodeId,
+                SeriesName = u.Series.Title,
+                SeasonNumber = u.SeasonNumber,
+                EpisodeNumber = u.EpisodeNumber,
+                Title = u.Title,
+                Overview = u.Overview,
+                AirDate = u.AirDate.Add(Convert.ToDateTime(u.Series.AirTimes).TimeOfDay)
+            });
 
             return View(new GridModel(upcoming));
         }

From 5a99d374d99f58d2d449321c364788cdc5c1e4de Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Sun, 19 Jun 2011 20:21:54 -0700
Subject: [PATCH 2/6] Updated tests to for Upcoming Provider to make sure a
 series is returned.

---
 .../UpcomingEpisodesProviderTest.cs           | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/NzbDrone.Core.Test/UpcomingEpisodesProviderTest.cs b/NzbDrone.Core.Test/UpcomingEpisodesProviderTest.cs
index fdd982e83..3e862686b 100644
--- a/NzbDrone.Core.Test/UpcomingEpisodesProviderTest.cs
+++ b/NzbDrone.Core.Test/UpcomingEpisodesProviderTest.cs
@@ -1,7 +1,9 @@
 // ReSharper disable RedundantUsingDirective
 using System;
+using System.Collections.Generic;
 using AutoMoq;
 using FizzWare.NBuilder;
+using FluentAssertions;
 using NUnit.Framework;
 using NzbDrone.Core.Providers;
 using NzbDrone.Core.Repository;
@@ -18,6 +20,7 @@ namespace NzbDrone.Core.Test
         private Episode tomorrow;
         private Episode twoDays;
         private Episode sevenDays;
+        private Series series;
 
         [SetUp]
         public new void Setup()
@@ -25,33 +28,41 @@ namespace NzbDrone.Core.Test
             yesterday = Builder<Episode>.CreateNew()
                 .With(c => c.AirDate = DateTime.Today.AddDays(-1))
                 .With(c => c.Title = "Yesterday")
+                .With(c => c.SeriesId = 1)
                 .Build();
 
             today = Builder<Episode>.CreateNew()
                 .With(c => c.AirDate = DateTime.Today)
                 .With(c => c.Title = "Today")
+                .With(c => c.SeriesId = 1)
                 .Build();
 
             tomorrow = Builder<Episode>.CreateNew()
                 .With(c => c.AirDate = DateTime.Today.AddDays(1))
                 .With(c => c.Title = "Tomorrow")
+                .With(c => c.SeriesId = 1)
                 .Build();
 
             twoDays = Builder<Episode>.CreateNew()
                 .With(c => c.AirDate = DateTime.Today.AddDays(2))
                 .With(c => c.Title = "Two Days")
+                .With(c => c.SeriesId = 1)
                 .Build();
 
             sevenDays = Builder<Episode>.CreateNew()
                 .With(c => c.AirDate = DateTime.Today.AddDays(7))
                 .With(c => c.Title = "Seven Days")
+                .With(c => c.SeriesId = 1)
                 .Build();
 
             sevenDays = Builder<Episode>.CreateNew()
                 .With(c => c.AirDate = DateTime.Today.AddDays(8))
                 .With(c => c.Title = "Eight Days")
+                .With(c => c.SeriesId = 1)
                 .Build();
 
+            series = Builder<Series>.CreateNew().With(s => s.SeriesId = 1).Build();
+
             base.Setup();
         }
 
@@ -68,6 +79,7 @@ namespace NzbDrone.Core.Test
             database.Insert(tomorrow);
             database.Insert(twoDays);
             database.Insert(sevenDays);
+            database.Insert(series);
 
             //Act
             var result = mocker.Resolve<UpcomingEpisodesProvider>().Yesterday();
@@ -75,6 +87,8 @@ namespace NzbDrone.Core.Test
             //Assert
             Assert.AreEqual(1, result.Count);
             Assert.AreEqual(yesterday.Title, result[0].Title);
+            result[0].Series.Should().NotBeNull();
+            result[0].Series.SeriesId.Should().NotBe(0);
         }
 
         [Test]
@@ -90,6 +104,7 @@ namespace NzbDrone.Core.Test
             database.Insert(tomorrow);
             database.Insert(twoDays);
             database.Insert(sevenDays);
+            database.Insert(series);
 
             //Act
             var result = mocker.Resolve<UpcomingEpisodesProvider>().Today();
@@ -97,6 +112,8 @@ namespace NzbDrone.Core.Test
             //Assert
             Assert.AreEqual(1, result.Count);
             Assert.AreEqual(today.Title, result[0].Title);
+            result[0].Series.Should().NotBeNull();
+            result[0].Series.SeriesId.Should().NotBe(0);
         }
 
         [Test]
@@ -112,6 +129,7 @@ namespace NzbDrone.Core.Test
             database.Insert(tomorrow);
             database.Insert(twoDays);
             database.Insert(sevenDays);
+            database.Insert(series);
 
             //Act
             var result = mocker.Resolve<UpcomingEpisodesProvider>().Tomorrow();
@@ -119,6 +137,8 @@ namespace NzbDrone.Core.Test
             //Assert
             Assert.AreEqual(1, result.Count);
             Assert.AreEqual(tomorrow.Title, result[0].Title);
+            result[0].Series.Should().NotBeNull();
+            result[0].Series.SeriesId.Should().NotBe(0);
         }
 
         [Test]
@@ -134,12 +154,17 @@ namespace NzbDrone.Core.Test
             database.Insert(tomorrow);
             database.Insert(twoDays);
             database.Insert(sevenDays);
+            database.Insert(series);
 
             //Act
             var result = mocker.Resolve<UpcomingEpisodesProvider>().Week();
 
             //Assert
             Assert.AreEqual(2, result.Count);
+            result[0].Series.Should().NotBeNull();
+            result[0].Series.SeriesId.Should().NotBe(0);
+            result[1].Series.Should().NotBeNull();
+            result[1].Series.SeriesId.Should().NotBe(0);
         }
     }
 }

From 431d850d32a36e8bed47c1c0d393ae921c6fe3be Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Mon, 20 Jun 2011 00:13:17 -0700
Subject: [PATCH 3/6] SeriesProvider gets QualityProfile in single call to DB.

SeriesProvider.GetAllSeriesWithEpisodeCount gets seasonCount(with or without specials), total episode count & episodeWithFile count (excluding ignored episodes).
Added tests for SeriesWithEpisodeCount.
---
 NzbDrone.Core.Test/SeriesProviderTest.cs     | 141 ++++++++++++++++++-
 NzbDrone.Core/Providers/SeriesProvider.cs    |  46 ++++--
 NzbDrone.Core/Repository/Series.cs           |  16 ++-
 NzbDrone.Web/Controllers/SeriesController.cs |  49 +++----
 NzbDrone.Web/Models/SeriesModel.cs           |   2 +
 NzbDrone.Web/Views/Series/Index.cshtml       |  34 +----
 6 files changed, 197 insertions(+), 91 deletions(-)

diff --git a/NzbDrone.Core.Test/SeriesProviderTest.cs b/NzbDrone.Core.Test/SeriesProviderTest.cs
index 9064eb8d5..0ce55b3ab 100644
--- a/NzbDrone.Core.Test/SeriesProviderTest.cs
+++ b/NzbDrone.Core.Test/SeriesProviderTest.cs
@@ -65,11 +65,13 @@ namespace NzbDrone.Core.Test
             mocker.GetMock<ConfigProvider>()
                 .Setup(c => c.UseSeasonFolder).Returns(useSeasonFolder);
 
-            mocker.SetConstant(MockLib.GetEmptyDatabase());
-
-
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
 
+            var fakeProfiles = Builder<QualityProfile>.CreateListOfSize(2).Build();
 
+            db.InsertMany(fakeProfiles);
+            
             const string path = "C:\\Test\\";
             const int tvDbId = 1234;
             const int qualityProfileId = 2;
@@ -78,8 +80,6 @@ namespace NzbDrone.Core.Test
             var seriesProvider = mocker.Resolve<SeriesProvider>();
             seriesProvider.AddSeries(path, tvDbId, qualityProfileId);
 
-
-
             //Assert
             var series = seriesProvider.GetAllSeries();
             series.Should().HaveCount(1);
@@ -104,7 +104,6 @@ namespace NzbDrone.Core.Test
             Assert.IsNull(series);
         }
 
-
         [Test]
         [ExpectedException(typeof(InvalidOperationException), ExpectedMessage = "Sequence contains no elements")]
         public void Get_series_invalid_series_id_should_return_null()
@@ -233,6 +232,7 @@ namespace NzbDrone.Core.Test
                                                  .With(c => c.SeriesId = 11)
                                                  .Build());
 
+            db.InsertMany(Builder<QualityProfile>.CreateListOfSize(3).Build());
 
             //Act, Assert
             var provider = mocker.Resolve<SeriesProvider>();
@@ -240,5 +240,134 @@ namespace NzbDrone.Core.Test
             Assert.IsFalse(provider.IsMonitored(11));
             Assert.IsFalse(provider.IsMonitored(1));
         }
+
+        [Test]
+        public void Get_Series_With_Count()
+        {
+            var mocker = new AutoMoqer(MockBehavior.Strict);
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
+            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
+            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10).WhereAll().Have(e => e.SeriesId = fakeSeries.SeriesId).Have(e => e.Ignored = false).WhereRandom(5).Have(e => e.EpisodeFileId = 0).Build();
+
+            db.Insert(fakeSeries);
+            db.Insert(fakeQuality);
+            db.InsertMany(fakeEpisodes);
+
+            //Act
+            mocker.Resolve<QualityProvider>();
+            var series = mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount(true);
+
+            //Assert
+            series.Should().HaveCount(1);
+            Assert.AreEqual(10, series[0].EpisodeCount);
+            Assert.AreEqual(5, series[0].EpisodeFileCount);
+        }
+
+        [Test]
+        public void Get_Series_With_Count_AllIgnored()
+        {
+            var mocker = new AutoMoqer(MockBehavior.Strict);
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
+            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
+            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10).WhereAll().Have(e => e.SeriesId = fakeSeries.SeriesId).Have(e => e.Ignored = true).WhereRandom(5).Have(e => e.EpisodeFileId = 0).Build();
+
+            db.Insert(fakeSeries);
+            db.Insert(fakeQuality);
+            db.InsertMany(fakeEpisodes);
+
+            //Act
+            mocker.Resolve<QualityProvider>();
+            var series = mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount(true);
+
+            //Assert
+            series.Should().HaveCount(1);
+            Assert.AreEqual(0, series[0].EpisodeCount);
+            Assert.AreEqual(0, series[0].EpisodeFileCount);
+        }
+
+        [Test]
+        public void Get_Series_With_Count_AllDownloaded()
+        {
+            var mocker = new AutoMoqer(MockBehavior.Strict);
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
+            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
+            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10).WhereAll().Have(e => e.SeriesId = fakeSeries.SeriesId).Have(e => e.Ignored = false).Build();
+
+            db.Insert(fakeSeries);
+            db.Insert(fakeQuality);
+            db.InsertMany(fakeEpisodes);
+
+            //Act
+            mocker.Resolve<QualityProvider>();
+            var series = mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount(true);
+
+            //Assert
+            series.Should().HaveCount(1);
+            Assert.AreEqual(10, series[0].EpisodeCount);
+            Assert.AreEqual(10, series[0].EpisodeFileCount);
+        }
+
+        [Test]
+        public void Get_Series_With_Count_Half_Ignored()
+        {
+            var mocker = new AutoMoqer(MockBehavior.Strict);
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
+            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
+            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
+                .WhereAll().Have(e => e.SeriesId = fakeSeries.SeriesId)
+                .WhereTheFirst(5).Have(e => e.Ignored = false)
+                .WhereTheLast(5).Have(e => e.Ignored = true)
+                .Build();
+
+            db.Insert(fakeSeries);
+            db.Insert(fakeQuality);
+            db.InsertMany(fakeEpisodes);
+
+            //Act
+            mocker.Resolve<QualityProvider>();
+            var series = mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount(true);
+
+            //Assert
+            series.Should().HaveCount(1);
+            Assert.AreEqual(5, series[0].EpisodeCount);
+            Assert.AreEqual(5, series[0].EpisodeFileCount);
+        }
+
+        [Test]
+        public void Get_Single_Series()
+        {
+            var mocker = new AutoMoqer(MockBehavior.Strict);
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
+            var fakeSeries = Builder<Series>.CreateNew()
+                .With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
+                .With(e => e.SeriesId = 1)
+                .Build();
+
+            db.Insert(fakeSeries);
+            db.Insert(fakeQuality);
+
+            //Act
+            mocker.Resolve<QualityProvider>();
+            var series = mocker.Resolve<SeriesProvider>().GetSeries(1);
+
+            //Assert
+            series.QualityProfile.Should().NotBeNull();
+            series.QualityProfileId.Should().Be(fakeQuality.QualityProfileId);
+        }
     }
 }
\ No newline at end of file
diff --git a/NzbDrone.Core/Providers/SeriesProvider.cs b/NzbDrone.Core/Providers/SeriesProvider.cs
index 127e9f08a..611e1e6c4 100644
--- a/NzbDrone.Core/Providers/SeriesProvider.cs
+++ b/NzbDrone.Core/Providers/SeriesProvider.cs
@@ -6,6 +6,7 @@ using System.Text.RegularExpressions;
 using NLog;
 using NzbDrone.Core.Providers.Core;
 using NzbDrone.Core.Repository;
+using NzbDrone.Core.Repository.Quality;
 using PetaPoco;
 using TvdbLib.Data;
 
@@ -17,17 +18,15 @@ namespace NzbDrone.Core.Providers
         private readonly ConfigProvider _configProvider;
         private readonly TvDbProvider _tvDbProvider;
         private readonly IDatabase _database;
-        private readonly QualityProvider _qualityProvider;
         private readonly SceneMappingProvider _sceneNameMappingProvider;
         private static readonly Regex TimeRegex = new Regex(@"^(?<time>\d+:?\d*)\W*(?<meridiem>am|pm)?", RegexOptions.IgnoreCase | RegexOptions.Compiled);
 
-        public SeriesProvider(IDatabase database, ConfigProvider configProviderProvider, QualityProvider qualityProvider,
+        public SeriesProvider(IDatabase database, ConfigProvider configProviderProvider,
                                 TvDbProvider tvDbProviderProvider, SceneMappingProvider sceneNameMappingProvider)
         {
             _database = database;
             _configProvider = configProviderProvider;
             _tvDbProvider = tvDbProviderProvider;
-            _qualityProvider = qualityProvider;
             _sceneNameMappingProvider = sceneNameMappingProvider;
         }
 
@@ -37,15 +36,36 @@ namespace NzbDrone.Core.Providers
 
         public virtual IList<Series> GetAllSeries()
         {
-            var series = _database.Fetch<Series>();
-            series.ForEach(c => c.QualityProfile = _qualityProvider.Get(c.QualityProfileId));
+            var series = _database.Fetch<Series, QualityProfile>(@"SELECT * FROM Series 
+                            INNER JOIN QualityProfiles ON Series.QualityProfileId = QualityProfiles.QualityProfileId");
+
+            return series;
+        }
+
+        public virtual IList<Series> GetAllSeriesWithEpisodeCount(bool ignoreSpecials)
+        {
+            var seasonNumber = 0;
+
+            if (!ignoreSpecials)
+                seasonNumber = -1;
+
+            var series = _database.Fetch<Series, QualityProfile>(@"SELECT Series.*, COUNT (NULLIF(Ignored, 1)) AS EpisodeCount,
+                                                    SUM(CASE WHEN Ignored = 0 AND EpisodeFileId > 0 THEN 1 ELSE 0 END) as EpisodeFileCount,
+                                                    COUNT (DISTINCT(NULLIF(SeasonNumber, @0))) as SeasonCount,
+                                                    QualityProfiles.*
+                                                    FROM Series
+                                                    INNER JOIN QualityProfiles ON Series.QualityProfileId = QualityProfiles.QualityProfileId
+                                                    JOIN Episodes ON Series.SeriesId = Episodes.SeriesId
+                                                    GROUP BY seriesId", seasonNumber);
+
             return series;
         }
 
         public virtual Series GetSeries(int seriesId)
         {
-            var series = _database.Single<Series>("WHERE seriesId= @0", seriesId);
-            series.QualityProfile = _qualityProvider.Get(series.QualityProfileId);
+            var series = _database.Fetch<Series, QualityProfile>(@"SELECT * FROM Series
+                            INNER JOIN QualityProfiles ON Series.QualityProfileId = QualityProfiles.QualityProfileId
+                            WHERE seriesId= @0", seriesId).Single();
 
             return series;
         }
@@ -117,15 +137,11 @@ namespace NzbDrone.Core.Providers
                 return GetSeries(seriesId.Value);
             }
 
-            var series = _database.FirstOrDefault<Series>("WHERE CleanTitle = @0", normalizeTitle);
-
-            if (series != null)
-            {
-                series.QualityProfile = _qualityProvider.Get(series.QualityProfileId);
-                return series;
-            }
+            var series = _database.Fetch<Series, QualityProfile>(@"SELECT * FROM Series
+                            INNER JOIN QualityProfiles ON Series.QualityProfileId = QualityProfiles.QualityProfileId
+                            WHERE CleanTitle = @0", normalizeTitle).FirstOrDefault();
 
-            return null;
+            return series;
         }
 
         public virtual void UpdateSeries(Series series)
diff --git a/NzbDrone.Core/Repository/Series.cs b/NzbDrone.Core/Repository/Series.cs
index 9206026ea..a8b189fe1 100644
--- a/NzbDrone.Core/Repository/Series.cs
+++ b/NzbDrone.Core/Repository/Series.cs
@@ -10,32 +10,25 @@ namespace NzbDrone.Core.Repository
     {
         public virtual int SeriesId { get; set; }
 
-
         public string Title { get; set; }
 
-
         public string CleanTitle { get; set; }
 
-
         public string Status { get; set; }
 
-
         public string Overview { get; set; }
 
         [DisplayName("Air on")]
         public DayOfWeek? AirsDayOfWeek { get; set; }
 
-
         public String AirTimes { get; set; }
 
-
         public string Language { get; set; }
 
         public string Path { get; set; }
 
         public bool Monitored { get; set; }
 
-
         public virtual int QualityProfileId { get; set; }
 
         public bool SeasonFolder { get; set; }
@@ -55,5 +48,14 @@ namespace NzbDrone.Core.Repository
 
         [Ignore]
         public QualityProfile QualityProfile { get; set; }
+
+        [ResultColumn]
+        public int EpisodeCount { get; set; }
+
+        [ResultColumn]
+        public int EpisodeFileCount { get; set; }
+
+        [ResultColumn]
+        public int SeasonCount { get; set; }
     }
 }
\ No newline at end of file
diff --git a/NzbDrone.Web/Controllers/SeriesController.cs b/NzbDrone.Web/Controllers/SeriesController.cs
index 20b8c4e4f..38e1eaa64 100644
--- a/NzbDrone.Web/Controllers/SeriesController.cs
+++ b/NzbDrone.Web/Controllers/SeriesController.cs
@@ -80,8 +80,7 @@ namespace NzbDrone.Web.Controllers
         [GridAction]
         public ActionResult _AjaxSeriesGrid()
         {
-            var series = GetSeriesModels(_seriesProvider.GetAllSeries().ToList());
-
+            var series = GetSeriesModels(_seriesProvider.GetAllSeriesWithEpisodeCount(true).ToList());
             return View(new GridModel(series));
         }
 
@@ -124,17 +123,6 @@ namespace NzbDrone.Web.Controllers
             return View(new GridModel(episodes));
         }
 
-        public JsonResult GetEpisodeCount(int seriesId)
-        {
-            var count = _mediaFileProvider.GetEpisodeFilesCount(seriesId);
-
-            return Json(new
-            {
-                Episodes = count.Item1,
-                EpisodeTotal = count.Item2
-            }, JsonRequestBehavior.AllowGet);
-        }
-
         public ActionResult SearchForSeries(string seriesName)
         {
             var model = new List<SeriesSearchResultModel>();
@@ -206,25 +194,22 @@ namespace NzbDrone.Web.Controllers
 
         private List<SeriesModel> GetSeriesModels(List<Series> seriesInDb)
         {
-            var series = new List<SeriesModel>();
-
-            foreach (var s in seriesInDb)
-            {
-                series.Add(new SeriesModel
-                               {
-                                   SeriesId = s.SeriesId,
-                                   Title = s.Title,
-                                   AirsDayOfWeek = s.AirsDayOfWeek.ToString(),
-                                   Monitored = s.Monitored,
-                                   Overview = s.Overview,
-                                   Path = s.Path,
-                                   QualityProfileId = s.QualityProfileId,
-                                   QualityProfileName = s.QualityProfile.Name,
-                                   SeasonFolder = s.SeasonFolder,
-                                   Status = s.Status,
-                                   SeasonsCount = _episodeProvider.GetSeasons(s.SeriesId).Where(n => n != 0).Count()
-                               });
-            }
+            var series = seriesInDb.Select(s => new SeriesModel
+                                                    {
+                                                        SeriesId = s.SeriesId,
+                                                        Title = s.Title,
+                                                        AirsDayOfWeek = s.AirsDayOfWeek.ToString(),
+                                                        Monitored = s.Monitored,
+                                                        Overview = s.Overview,
+                                                        Path = s.Path,
+                                                        QualityProfileId = s.QualityProfileId,
+                                                        QualityProfileName = s.QualityProfile.Name,
+                                                        SeasonFolder = s.SeasonFolder,
+                                                        Status = s.Status,
+                                                        SeasonsCount = s.SeasonCount,
+                                                        EpisodeCount = s.EpisodeCount,
+                                                        EpisodeFileCount = s.EpisodeFileCount
+                                                    }).ToList();
 
             return series;
         }
diff --git a/NzbDrone.Web/Models/SeriesModel.cs b/NzbDrone.Web/Models/SeriesModel.cs
index 785ba0691..7a2429acb 100644
--- a/NzbDrone.Web/Models/SeriesModel.cs
+++ b/NzbDrone.Web/Models/SeriesModel.cs
@@ -14,6 +14,8 @@ namespace NzbDrone.Web.Models
         //View Only
         public string Title { get; set; }
         public int SeasonsCount { get; set; }
+        public int EpisodeCount { get; set; }
+        public int EpisodeFileCount { get; set; }
         public string Status { get; set; }
         public string AirsDayOfWeek { get; set; }
         public string QualityProfileName { get; set; }
diff --git a/NzbDrone.Web/Views/Series/Index.cshtml b/NzbDrone.Web/Views/Series/Index.cshtml
index 4bb581d94..3b7e380b1 100644
--- a/NzbDrone.Web/Views/Series/Index.cshtml
+++ b/NzbDrone.Web/Views/Series/Index.cshtml
@@ -186,37 +186,9 @@
     function grid_rowBound(e) {
         var dataItem = e.dataItem;
         var seriesId = dataItem.SeriesId;
-        var getEpisodeCountUrl = '@Url.Action("GetEpisodeCount", "Series")';
+        var episodeCount = dataItem.EpisodeCount;
+        var episodeFileCount = dataItem.EpisodeFileCount;
 
-        $("#progressbar_" + seriesId).episodeProgress(0, 0);
-
-        $.ajax({
-            type: "GET",
-            url: getEpisodeCountUrl,
-            data: jQuery.param({ seriesId: seriesId }),
-            error: function (req, status, error) {
-                alert("Sorry! We could search for " + id + " at this time. " + error);
-            },
-            success: function (data, textStatus, jqXHR) {
-                var episodes = data.Episodes;
-                var episodeTotal = data.EpisodeTotal;
-
-                var counter = 0;
-                var max = episodes + 1;
-
-                $.doTimeout(10, function () {
-                    if (counter >= max) {
-                        $("#progressbar_" + seriesId).episodeProgress(episodes, episodeTotal);
-                        return false;
-                    }
-
-                    $("#progressbar_" + seriesId).episodeProgress(counter, episodeTotal);
-
-                    counter += 5;
-
-                    return true;
-                });
-            }
-        });
+        $("#progressbar_" + seriesId).episodeProgress(episodeFileCount, episodeCount);
     }
 </script>
\ No newline at end of file

From 244e0f04f4ef772bd2ec9537ea5f9f76bb740587 Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Mon, 20 Jun 2011 00:40:45 -0700
Subject: [PATCH 4/6] Fixed UnitTests (set EpisodeFileId to zero where no
 episo.de file is expected back)

---
 NzbDrone.Core.Test/EpisodeProviderTest.cs  |  6 ++--
 NzbDrone.Core.Test/SeriesProviderTest.cs   | 40 ++--------------------
 NzbDrone.Core/Providers/EpisodeProvider.cs |  1 +
 3 files changed, 7 insertions(+), 40 deletions(-)

diff --git a/NzbDrone.Core.Test/EpisodeProviderTest.cs b/NzbDrone.Core.Test/EpisodeProviderTest.cs
index 47b9a38a9..16e1c9658 100644
--- a/NzbDrone.Core.Test/EpisodeProviderTest.cs
+++ b/NzbDrone.Core.Test/EpisodeProviderTest.cs
@@ -28,7 +28,7 @@ namespace NzbDrone.Core.Test
 
             var fakeSeries = Builder<Series>.CreateNew().Build();
             var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
-                .WhereAll().Have(e => e.SeriesId = 1).Build();
+                .WhereAll().Have(e => e.SeriesId = 1).Have(e => e.EpisodeFileId = 0).Build();
 
 
             db.InsertMany(fakeEpisodes);
@@ -57,8 +57,8 @@ namespace NzbDrone.Core.Test
             var fakeEpisodes = Builder<Episode>.CreateNew()
                 .With(e => e.SeriesId = fakeSeries.SeriesId)
                 .With(e => e.EpisodeNumber = 1)
-                .And(e => e.SeasonNumber = 2).Build();
-
+                .And(e => e.SeasonNumber = 2)
+                .With(e => e.EpisodeFileId = 0).Build();
 
             db.Insert(fakeEpisodes);
 
diff --git a/NzbDrone.Core.Test/SeriesProviderTest.cs b/NzbDrone.Core.Test/SeriesProviderTest.cs
index 326b70acb..8e10dfeb6 100644
--- a/NzbDrone.Core.Test/SeriesProviderTest.cs
+++ b/NzbDrone.Core.Test/SeriesProviderTest.cs
@@ -19,43 +19,6 @@ namespace NzbDrone.Core.Test
     [TestFixture]
     public class SeriesProviderTest : TestBase
     {
-        [Test]
-        public void Map_path_to_series()
-        {
-            //Arrange
-            var fakeSeries = Builder<TvdbSeries>.CreateNew()
-                .With(f => f.SeriesName = "The Simpsons")
-                .Build();
-
-            var fakeSearch = Builder<TvdbSearchResult>.CreateNew()
-                .With(s => s.Id = fakeSeries.Id)
-                .With(s => s.SeriesName = fakeSeries.SeriesName)
-                .Build();
-
-
-            var mocker = new AutoMoqer();
-
-            mocker.GetMock<IDatabase>()
-                .Setup(f => f.Exists<Series>(It.IsAny<String>()))
-                .Returns(false);
-
-            mocker.GetMock<TvDbProvider>()
-                .Setup(f => f.GetSeries(It.IsAny<String>()))
-                .Returns(fakeSearch);
-            mocker.GetMock<TvDbProvider>()
-                .Setup(f => f.GetSeries(fakeSeries.Id, false))
-                .Returns(fakeSeries)
-                .Verifiable();
-
-            //Act
-
-            var mappedSeries = mocker.Resolve<SeriesProvider>().MapPathToSeries(@"D:\TV Shows\The Simpsons");
-
-            //Assert
-            mocker.GetMock<TvDbProvider>().VerifyAll();
-            Assert.AreEqual(fakeSeries, mappedSeries);
-        }
-
         [TestCase(true)]
         [TestCase(false)]
         public void Add_new_series(bool useSeasonFolder)
@@ -129,6 +92,9 @@ namespace NzbDrone.Core.Test
 
             var fakeSeries = Builder<Series>.CreateNew()
                 .With(c => c.QualityProfileId = 1)
+                .With(c => c.EpisodeCount = 0)
+                .With(c => c.EpisodeFileCount = 0)
+                .With(c => c.SeasonCount = 0)
                 .Build();
             var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
 
diff --git a/NzbDrone.Core/Providers/EpisodeProvider.cs b/NzbDrone.Core/Providers/EpisodeProvider.cs
index 8bdca33be..edc435c2b 100644
--- a/NzbDrone.Core/Providers/EpisodeProvider.cs
+++ b/NzbDrone.Core/Providers/EpisodeProvider.cs
@@ -38,6 +38,7 @@ namespace NzbDrone.Core.Providers
             var episode =  AttachSeries(_database.Fetch<Episode, EpisodeFile>(@"SELECT * FROM Episodes 
                                                             LEFT JOIN EpisodeFiles ON Episodes.EpisodeFileId = EpisodeFiles.EpisodeFileId
                                                             WHERE EpisodeId = @0", id).Single());
+
             if (episode.EpisodeFileId == 0)
                 episode.EpisodeFile = null;
 

From 208feb7d19fa32b703573b96f3d535484f73142f Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Mon, 20 Jun 2011 16:46:54 -0700
Subject: [PATCH 5/6] CleanFileName when getting new filename from
 MediaFileProvider.

---
 NzbDrone.Core.Test/MediaFileProviderTests.cs    | 10 ++++++++++
 NzbDrone.Core/Helpers/EpisodeRenameHelper.cs    | 12 ------------
 NzbDrone.Core/Providers/MediaFileProvider.cs    | 12 ++++++++++++
 NzbDrone.Web/Controllers/AddSeriesController.cs |  2 +-
 4 files changed, 23 insertions(+), 13 deletions(-)

diff --git a/NzbDrone.Core.Test/MediaFileProviderTests.cs b/NzbDrone.Core.Test/MediaFileProviderTests.cs
index 8f6ad61c1..cc0c63562 100644
--- a/NzbDrone.Core.Test/MediaFileProviderTests.cs
+++ b/NzbDrone.Core.Test/MediaFileProviderTests.cs
@@ -349,5 +349,15 @@ namespace NzbDrone.Core.Test
 
         }
 
+        [Test]
+        [TestCase("Law & Order: Criminal Intent - S10E07 - Icarus [HDTV]", "Law & Order- Criminal Intent - S10E07 - Icarus [HDTV]")]
+        public void CleanFileName(string name, string expectedName)
+        {
+            //Act
+            var result = MediaFileProvider.CleanFilename(name);
+
+            //Assert
+            Assert.AreEqual(expectedName, result);
+        }
     }
 }
\ No newline at end of file
diff --git a/NzbDrone.Core/Helpers/EpisodeRenameHelper.cs b/NzbDrone.Core/Helpers/EpisodeRenameHelper.cs
index d02781b55..f401c221a 100644
--- a/NzbDrone.Core/Helpers/EpisodeRenameHelper.cs
+++ b/NzbDrone.Core/Helpers/EpisodeRenameHelper.cs
@@ -55,17 +55,5 @@ namespace NzbDrone.Core.Helpers
             return String.Format("{0} - S{1:00}E{2} - {3}", erm.SeriesName, erm.EpisodeFile.Episodes[0].SeasonNumber,
                                  epNumberString, epNameString);
         }
-
-        public static string CleanFilename(string name)
-        {
-	        string result = name;
-	        string[] badCharacters = {"\\", "/", "<", ">", "?", "*", ":", "|", "\""};
-	        string[] goodCharacters = {"+", "+", "{", "}", "!", "@", "-", "#", "`"};
-
-	        for (int i = 0; i < badCharacters.Length; i++)
-		        result = result.Replace(badCharacters[i], goodCharacters[i]);
-
-	        return result.Trim();
-        }
     }
 }
\ No newline at end of file
diff --git a/NzbDrone.Core/Providers/MediaFileProvider.cs b/NzbDrone.Core/Providers/MediaFileProvider.cs
index 732d3a18d..d1dc1586a 100644
--- a/NzbDrone.Core/Providers/MediaFileProvider.cs
+++ b/NzbDrone.Core/Providers/MediaFileProvider.cs
@@ -129,6 +129,18 @@ namespace NzbDrone.Core.Providers
                 result = result.Replace(' ', '.');
 
             Logger.Debug("New File Name is: {0}", result.Trim());
+            return CleanFilename(result.Trim());
+        }
+
+        public static string CleanFilename(string name)
+        {
+            string result = name;
+            string[] badCharacters = { "\\", "/", "<", ">", "?", "*", ":", "|", "\"" };
+            string[] goodCharacters = { "+", "+", "{", "}", "!", "@", "-", "#", "`" };
+
+            for (int i = 0; i < badCharacters.Length; i++)
+                result = result.Replace(badCharacters[i], goodCharacters[i]);
+
             return result.Trim();
         }
     }
diff --git a/NzbDrone.Web/Controllers/AddSeriesController.cs b/NzbDrone.Web/Controllers/AddSeriesController.cs
index 6431d35da..8dd5eb4fe 100644
--- a/NzbDrone.Web/Controllers/AddSeriesController.cs
+++ b/NzbDrone.Web/Controllers/AddSeriesController.cs
@@ -129,7 +129,7 @@ namespace NzbDrone.Web.Controllers
                 var path =
                     rootPath.Replace('|', Path.DirectorySeparatorChar).Replace('^', Path.VolumeSeparatorChar).Replace(
                         '`', '\'') +
-                    Path.DirectorySeparatorChar + EpisodeRenameHelper.CleanFilename(seriesName);
+                    Path.DirectorySeparatorChar + MediaFileProvider.CleanFilename(seriesName);
 
                 //Create the folder for the new series and then Add it
                 _diskProvider.CreateDirectory(path);

From 68b7ba9209e3abbd9f60f841ca19b705d04f3747 Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Mon, 20 Jun 2011 18:49:16 -0700
Subject: [PATCH 6/6] Added method to delete orphaned episodeFiles and also
 cleanup episodes that have invalid episodeFiles.

Cleanup will run aforementioned methods before doing cleanup.
---
 NzbDrone.Core.Test/MediaFileProviderTests.cs | 42 ++++++++++++++++++++
 NzbDrone.Core/Providers/DiskScanProvider.cs  |  4 +-
 NzbDrone.Core/Providers/MediaFileProvider.cs | 20 ++++++++++
 3 files changed, 65 insertions(+), 1 deletion(-)

diff --git a/NzbDrone.Core.Test/MediaFileProviderTests.cs b/NzbDrone.Core.Test/MediaFileProviderTests.cs
index cc0c63562..95877762b 100644
--- a/NzbDrone.Core.Test/MediaFileProviderTests.cs
+++ b/NzbDrone.Core.Test/MediaFileProviderTests.cs
@@ -359,5 +359,47 @@ namespace NzbDrone.Core.Test
             //Assert
             Assert.AreEqual(expectedName, result);
         }
+
+        [Test]
+        public void CleanEpisodesWithNonExistantFiles()
+        {
+            //Setup
+            var episodes = Builder<Episode>.CreateListOfSize(10).Build();
+
+            var mocker = new AutoMoqer();
+            var database = MockLib.GetEmptyDatabase(true);
+            mocker.SetConstant(database);
+            database.InsertMany(episodes);
+
+            //Act
+            mocker.Resolve<MediaFileProvider>().CleanEpisodesWithNonExistantFiles();
+            var result = database.Fetch<Episode>();
+
+            //Assert
+            result.Should().HaveSameCount(episodes);
+            result.Should().OnlyContain(e => e.EpisodeFileId == 0);
+        }
+
+        [Test]
+        public void DeleteOrphanedEpisodeFiles()
+        {
+            //Setup
+            var episodeFiles = Builder<EpisodeFile>.CreateListOfSize(10).Build();
+            var episodes = Builder<Episode>.CreateListOfSize(5).Build();
+
+            var mocker = new AutoMoqer();
+            var database = MockLib.GetEmptyDatabase(true);
+            mocker.SetConstant(database);
+            database.InsertMany(episodes);
+            database.InsertMany(episodeFiles);
+
+            //Act
+            mocker.Resolve<MediaFileProvider>().DeleteOrphanedEpisodeFiles();
+            var result = database.Fetch<EpisodeFile>();
+
+            //Assert
+            result.Should().HaveCount(5);
+            result.Should().OnlyContain(e => e.EpisodeFileId > 0);
+        }
     }
 }
\ No newline at end of file
diff --git a/NzbDrone.Core/Providers/DiskScanProvider.cs b/NzbDrone.Core/Providers/DiskScanProvider.cs
index 7db6809b5..4db01ffd9 100644
--- a/NzbDrone.Core/Providers/DiskScanProvider.cs
+++ b/NzbDrone.Core/Providers/DiskScanProvider.cs
@@ -196,7 +196,9 @@ namespace NzbDrone.Core.Providers
         /// <param name = "files">list of files to verify</param>
         public virtual void CleanUp(List<EpisodeFile> files)
         {
-            //TODO: remove orphaned files. in files table but not linked to from episode table.
+            _mediaFileProvider.CleanEpisodesWithNonExistantFiles();
+            _mediaFileProvider.DeleteOrphanedEpisodeFiles();
+
             foreach (var episodeFile in files)
             {
                 if (!_diskProvider.FileExists(episodeFile.Path))
diff --git a/NzbDrone.Core/Providers/MediaFileProvider.cs b/NzbDrone.Core/Providers/MediaFileProvider.cs
index d1dc1586a..9d56da9fe 100644
--- a/NzbDrone.Core/Providers/MediaFileProvider.cs
+++ b/NzbDrone.Core/Providers/MediaFileProvider.cs
@@ -74,6 +74,26 @@ namespace NzbDrone.Core.Providers
             return new FileInfo(path);
         }
 
+        public virtual void CleanEpisodesWithNonExistantFiles()
+        {
+            _database.Execute(@"UPDATE Episodes SET EpisodeFileId = 0
+                                WHERE EpisodeFileId IN
+                                (SELECT Episodes.EpisodeFileId FROM Episodes
+                                LEFT OUTER JOIN EpisodeFiles
+                                ON Episodes.EpisodeFileId = EpisodeFiles.EpisodeFileId
+                                WHERE Episodes.EpisodeFileId > 0 AND EpisodeFiles.EpisodeFileId IS null)");
+        }
+
+        public virtual void DeleteOrphanedEpisodeFiles()
+        {
+            _database.Execute(@"DELETE FROM EpisodeFiles
+                                WHERE EpisodeFileId IN
+                                (SELECT EpisodeFiles.EpisodeFileId FROM EpisodeFiles
+                                LEFT OUTER JOIN Episodes
+                                ON EpisodeFiles.EpisodeFileId = Episodes.EpisodeFileId
+                                WHERE Episodes.EpisodeFileId IS null)");
+        }
+
         public virtual string GetNewFilename(IList<Episode> episodes, string seriesTitle, QualityTypes quality)
         {
             var separatorStyle = EpisodeSortingHelper.GetSeparatorStyle(_configProvider.SeparatorStyle);