Signalr errors will not be shown in the UI.

New: Search History, review recent searches and force specific reports to download (Under history)
2025-08-14 10:47:08 -07:00 · 2012-04-23 00:38:42 -07:00 · 2012-04-23 00:38:42 -07:00 · 38927e3ca1
commit 38927e3ca1
parent 06df8a86b6
16 changed files with 100 additions and 99 deletions
--- a/NzbDrone.Core/Providers/SearchHistoryProvider.cs
+++ b/NzbDrone.Core/Providers/SearchHistoryProvider.cs
@ -0,0 +1,117 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using NLog;
+using Ninject;
+using NzbDrone.Core.Repository;
+using NzbDrone.Core.Repository.Search;
+using PetaPoco;
+
+namespace NzbDrone.Core.Providers
+{
+    public class SearchHistoryProvider
+    {
+        private readonly IDatabase _database;
+        private readonly SeriesProvider _seriesProvider;
+        private readonly DownloadProvider _downloadProvider;
+        private readonly EpisodeProvider _episodeProvider;
+
+        private static readonly Logger logger = LogManager.GetCurrentClassLogger();
+
+        [Inject]
+        public SearchHistoryProvider(IDatabase database, SeriesProvider seriesProvider,
+                                        DownloadProvider downloadProvider, EpisodeProvider episodeProvider)
+        {
+            _database = database;
+            _seriesProvider = seriesProvider;
+            _downloadProvider = downloadProvider;
+            _episodeProvider = episodeProvider;
+        }
+
+        public SearchHistoryProvider()
+        {
+            
+        }
+
+        public virtual void Add(SearchHistory searchResult)
+        {
+            logger.Trace("Adding new search result");
+            searchResult.SuccessfulDownload = searchResult.SearchHistoryItems.Any(s => s.Success);
+            var id = Convert.ToInt32(_database.Insert(searchResult));
+
+            searchResult.SearchHistoryItems.ForEach(s => s.SearchHistoryId = id);
+            logger.Trace("Adding search result items");
+            _database.InsertMany(searchResult.SearchHistoryItems);
+        }
+
+        public virtual void Delete(int id)
+        {
+            logger.Trace("Deleting search result items attached to: {0}", id);
+            _database.Execute("DELETE FROM SearchHistoryItems WHERE SearchHistoryId = @0", id);
+
+            logger.Trace("Deleting search result: {0}", id);
+            _database.Delete<SearchHistory>(id);
+        }
+
+        public virtual List<SearchHistory> AllSearchHistory()
+        {
+            var sql = @"SELECT SearchHistory.Id, SearchHistory.SeriesId, SearchHistory.SeasonNumber,
+                        SearchHistory.EpisodeId, SearchHistory.SearchTime,
+                        Series.Title as SeriesTitle, Series.IsDaily,
+                        Episodes.EpisodeNumber, Episodes.SeasonNumber, Episodes.Title as EpisodeTitle,
+                        Episodes.AirDate,
+                        Count(SearchHistoryItems.Id) as TotalItems,
+                        SUM(CASE WHEN SearchHistoryItems.Success = 1 THEN 1 ELSE 0 END) as SuccessfulCount
+                        FROM SearchHistory
+                        INNER JOIN Series
+                        ON Series.SeriesId = SearchHistory.SeriesId
+                        LEFT JOIN Episodes
+                        ON Episodes.EpisodeId = SearchHistory.EpisodeId
+                        INNER JOIN SearchHistoryItems
+                        ON SearchHistoryItems.SearchHistoryId = SearchHistory.Id
+                        GROUP BY SearchHistory.Id, SearchHistory.SeriesId, SearchHistory.SeasonNumber,
+                        SearchHistory.EpisodeId, SearchHistory.SearchTime,
+                        Series.Title, Series.IsDaily,
+                        Episodes.EpisodeNumber, Episodes.SeasonNumber, Episodes.Title,
+                        Episodes.AirDate";
+
+            return _database.Fetch<SearchHistory>(sql);
+        }
+
+        public virtual SearchHistory GetSearchHistory(int id)
+        {
+            var sql = @"SELECT SearchHistory.Id, SearchHistory.SeriesId, SearchHistory.SeasonNumber,
+                        SearchHistory.EpisodeId, SearchHistory.SearchTime,
+                        Series.Title as SeriesTitle, Series.IsDaily,
+                        Episodes.EpisodeNumber, Episodes.SeasonNumber, Episodes.Title as EpisodeTitle,
+                        Episodes.AirDate
+                        FROM SearchHistory
+                        INNER JOIN Series
+                        ON Series.SeriesId = SearchHistory.SeriesId
+                        LEFT JOIN Episodes
+                        ON Episodes.EpisodeId = SearchHistory.EpisodeId
+                        WHERE SearchHistory.Id = @0";
+
+            var result = _database.Single<SearchHistory>(sql, id);
+            result.SearchHistoryItems = _database.Fetch<SearchHistoryItem>("WHERE SearchHistoryId = @0", id);
+
+            return result;
+        }
+
+        public virtual void ForceDownload(int itemId)
+        {
+            var item = _database.Single<SearchHistoryItem>(itemId);
+            var searchResult = _database.Single<SearchHistory>(item.SearchHistoryId);
+            var series = _seriesProvider.GetSeries(searchResult.SeriesId);
+            
+            var parseResult = Parser.ParseTitle(item.ReportTitle);
+            parseResult.NzbUrl = item.NzbUrl;
+            parseResult.Series = series;
+            parseResult.Indexer = item.Indexer;
+            var episodes = _episodeProvider.GetEpisodesByParseResult(parseResult);
+
+            _downloadProvider.DownloadReport(parseResult);
+        }
+    }
+}