1
0
mirror of https://github.com/Radarr/Radarr.git synced 2024-11-05 18:42:42 +01:00
Radarr/NzbDrone.Core/Providers/SeriesProvider.cs

75 lines
2.6 KiB
C#
Raw Normal View History

using System.IO;
2010-09-23 05:19:47 +02:00
using System.Linq;
using log4net;
using NzbDrone.Core.Repository;
using SubSonic.Repository;
using TvdbLib.Data;
namespace NzbDrone.Core.Providers
2010-09-23 05:19:47 +02:00
{
public class SeriesProvider : ISeriesProvider
2010-09-23 05:19:47 +02:00
{
private readonly IConfigProvider _config;
private readonly IDiskProvider _diskProvider;
2010-09-23 05:19:47 +02:00
private readonly ILog _logger;
private readonly IRepository _sonioRepo;
private readonly ITvDbProvider _tvDb;
2010-09-23 05:19:47 +02:00
public SeriesProvider(ILog logger, IDiskProvider diskProvider, IConfigProvider configProvider, IRepository dataRepository, ITvDbProvider tvDbProvider)
2010-09-23 05:19:47 +02:00
{
_logger = logger;
_diskProvider = diskProvider;
_config = configProvider;
2010-09-23 05:19:47 +02:00
_sonioRepo = dataRepository;
_tvDb = tvDbProvider;
2010-09-23 05:19:47 +02:00
}
#region ISeriesProvider Members
2010-09-23 05:19:47 +02:00
public IQueryable<Series> GetSeries()
{
return _sonioRepo.All<Series>();
}
2010-09-24 09:14:42 +02:00
public Series GetSeries(int tvdbId)
{
return _sonioRepo.Single<Series>(s => s.TvdbId == tvdbId.ToString());
2010-09-24 09:14:42 +02:00
}
2010-09-23 05:19:47 +02:00
public void SyncSeriesWithDisk()
{
foreach (string seriesFolder in _diskProvider.GetDirectories(_config.SeriesRoot))
2010-09-23 05:19:47 +02:00
{
2010-09-28 07:01:54 +02:00
var cleanPath = DiskProvider.CleanPath(new DirectoryInfo(seriesFolder).FullName);
2010-09-24 09:14:42 +02:00
if (!_sonioRepo.Exists<Series>(s => s.Path == cleanPath))
2010-09-23 05:19:47 +02:00
{
2010-09-24 09:14:42 +02:00
_logger.InfoFormat("Folder '{0} isn't mapped to a series in the database. Trying to map it.'", cleanPath);
AddShow(cleanPath);
2010-09-23 05:19:47 +02:00
}
}
}
#endregion
private void AddShow(string path)
{
2010-09-28 02:27:02 +02:00
var searchResults = _tvDb.SearchSeries(new DirectoryInfo(path).Name);
2010-09-28 08:09:24 +02:00
if (searchResults.Count != 0 && !_sonioRepo.Exists<Series>(s => s.TvdbId == searchResults[0].Id.ToString()))
AddShow(path, _tvDb.GetSeries(searchResults[0].Id, searchResults[0].Language));
2010-09-23 05:19:47 +02:00
}
private void AddShow(string path, TvdbSeries series)
{
var repoSeries = new Series();
repoSeries.TvdbId = series.Id.ToString();
repoSeries.SeriesName = series.SeriesName;
repoSeries.AirTimes = series.AirsTime;
repoSeries.AirsDayOfWeek = series.AirsDayOfWeek;
repoSeries.Overview = series.Overview;
repoSeries.Status = series.Status;
repoSeries.Language = series.Language != null ? series.Language.Abbriviation : string.Empty;
repoSeries.Path = path;
_sonioRepo.Add(repoSeries);
2010-09-23 05:19:47 +02:00
}
}
}