From 3157ee4abad6aa5f211ea811c00493a789776b7e Mon Sep 17 00:00:00 2001
From: markus101 <markus.mcd5@gmail.com>
Date: Sun, 10 Oct 2010 14:20:43 -0700
Subject: [PATCH] Updated EpisodeProvider methods

---
 NzbDrone.Core/Providers/EpisodeProvider.cs  | 29 ++++++++++++---------
 NzbDrone.Core/Providers/IEpisodeProvider.cs | 10 +++----
 NzbDrone.Core/Providers/ISeasonProvider.cs  |  2 +-
 NzbDrone.Core/Providers/SeasonProvider.cs   |  2 +-
 4 files changed, 24 insertions(+), 19 deletions(-)

diff --git a/NzbDrone.Core/Providers/EpisodeProvider.cs b/NzbDrone.Core/Providers/EpisodeProvider.cs
index a7ef2f8a4..3ce2c81e0 100644
--- a/NzbDrone.Core/Providers/EpisodeProvider.cs
+++ b/NzbDrone.Core/Providers/EpisodeProvider.cs
@@ -40,27 +40,34 @@ namespace NzbDrone.Core.Providers
             _series = seriesProvider;
             _tvDb = tvDbProvider;
             _seasons = seasonProvider;
-
         }
 
-        public BasicEpisode GetEpisode(long id)
+        public EpisodeInfo GetEpisode(long id)
         {
-            throw new NotImplementedException();
+            return _sonicRepo.Single<EpisodeInfo>(e => e.EpisodeId == id);
         }
 
-        public BasicEpisode UpdateEpisode(BasicEpisode episode)
+        public void UpdateEpisode(EpisodeInfo episode)
         {
-            throw new NotImplementedException();
+            var episodeToUpdate = _sonicRepo.Single<EpisodeInfo>(e => e.EpisodeId == episode.EpisodeId);
+
+            episodeToUpdate.AirDate = episode.AirDate;
+            episodeToUpdate.Overview = episode.Overview;
+            episodeToUpdate.Title = episode.Title;
+            episodeToUpdate.EpisodeNumber = episode.EpisodeNumber;
+            episodeToUpdate.SeasonNumber = episode.SeasonNumber;
+
+            _sonicRepo.Update<EpisodeInfo>(episodeToUpdate);
         }
 
-        public IList<BasicEpisode> GetEpisodesBySeason(long seasonId)
+        public IList<EpisodeInfo> GetEpisodesBySeason(long seasonId)
         {
-            throw new NotImplementedException();
+            return _sonicRepo.Find<EpisodeInfo>(e => e.SeasonId == seasonId);
         }
 
-        public IList<BasicEpisode> GetEpisodeBySeries(long seriesId)
+        public IList<EpisodeInfo> GetEpisodeBySeries(long seriesId)
         {
-            throw new NotImplementedException();
+            return _sonicRepo.Find<EpisodeInfo>(e => e.SeriesId == seriesId);
         }
 
         public String GetSabTitle(BasicEpisode episode)
@@ -70,7 +77,6 @@ namespace NzbDrone.Core.Providers
 
             //TODO: This method should return a standard title for the sab episode.
             throw new NotImplementedException();
-
         }
 
         /// <summary>
@@ -78,7 +84,7 @@ namespace NzbDrone.Core.Providers
         /// </summary>
         /// <param name="episode">Episode that needs to be checked</param>
         /// <returns></returns>
-        public bool IsNeeded(BasicEpisode episode)
+        public bool IsNeeded(RemoteEpisode episode)
         {
             throw new NotImplementedException();
         }
@@ -153,6 +159,5 @@ namespace NzbDrone.Core.Providers
 
             return result;
         }
-
     }
 }
\ No newline at end of file
diff --git a/NzbDrone.Core/Providers/IEpisodeProvider.cs b/NzbDrone.Core/Providers/IEpisodeProvider.cs
index c658195bc..a332e206c 100644
--- a/NzbDrone.Core/Providers/IEpisodeProvider.cs
+++ b/NzbDrone.Core/Providers/IEpisodeProvider.cs
@@ -6,10 +6,10 @@ namespace NzbDrone.Core.Providers
 {
     public interface IEpisodeProvider
     {
-        BasicEpisode GetEpisode(long id);
-        BasicEpisode UpdateEpisode(BasicEpisode episode);
-        IList<BasicEpisode> GetEpisodesBySeason(long seasonId);
-        IList<BasicEpisode> GetEpisodeBySeries(long seriesId);
+        EpisodeInfo GetEpisode(long id);
+        void UpdateEpisode(EpisodeInfo episode);
+        IList<EpisodeInfo> GetEpisodesBySeason(long seasonId);
+        IList<EpisodeInfo> GetEpisodeBySeries(long seriesId);
         String GetSabTitle(BasicEpisode episode);
 
         /// <summary>
@@ -17,7 +17,7 @@ namespace NzbDrone.Core.Providers
         /// </summary>
         /// <param name="episode">Episode that needs to be checked</param>
         /// <returns></returns>
-        bool IsNeeded(BasicEpisode episode);
+        bool IsNeeded(RemoteEpisode episode);
 
         void RefreshSeries(int seriesId);
     }
diff --git a/NzbDrone.Core/Providers/ISeasonProvider.cs b/NzbDrone.Core/Providers/ISeasonProvider.cs
index 811147863..354de4426 100644
--- a/NzbDrone.Core/Providers/ISeasonProvider.cs
+++ b/NzbDrone.Core/Providers/ISeasonProvider.cs
@@ -6,7 +6,7 @@ namespace NzbDrone.Core.Providers
     public interface ISeasonProvider
     {
         Season GetSeason(int seasonId);
-        List<Season> GetSeasongs(int seriesId);
+        List<Season> GetSeasons(int seriesId);
         void EnsureSeason(int seriesId, int seasonId, int seasonNumber);
         int SaveSeason(Season season);
     }
diff --git a/NzbDrone.Core/Providers/SeasonProvider.cs b/NzbDrone.Core/Providers/SeasonProvider.cs
index 989502fd9..fc04cf436 100644
--- a/NzbDrone.Core/Providers/SeasonProvider.cs
+++ b/NzbDrone.Core/Providers/SeasonProvider.cs
@@ -21,7 +21,7 @@ namespace NzbDrone.Core.Providers
             throw new NotImplementedException();
         }
 
-        public List<Season> GetSeasongs(int seriesId)
+        public List<Season> GetSeasons(int seriesId)
         {
             throw new NotImplementedException();
         }