2020-01-05 21:49:08 +00:00
|
|
|
using System.Collections.Generic;
|
2018-09-21 19:30:09 +00:00
|
|
|
using System.Linq;
|
2013-02-19 06:56:02 +00:00
|
|
|
using NzbDrone.Core.Datastore;
|
2013-09-14 06:36:07 +00:00
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2013-09-11 06:33:47 +00:00
|
|
|
|
2013-02-19 06:56:02 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Tv
|
|
|
|
{
|
2013-03-24 04:16:00 +00:00
|
|
|
public interface ISeriesRepository : IBasicRepository<Series>
|
2013-02-19 06:56:02 +00:00
|
|
|
{
|
|
|
|
bool SeriesPathExists(string path);
|
2013-04-15 01:41:39 +00:00
|
|
|
Series FindByTitle(string cleanTitle);
|
2013-10-31 23:50:39 +00:00
|
|
|
Series FindByTitle(string cleanTitle, int year);
|
2020-01-05 21:49:08 +00:00
|
|
|
List<Series> FindByTitleInexact(string cleanTitle);
|
2013-03-02 18:25:39 +00:00
|
|
|
Series FindByTvdbId(int tvdbId);
|
2013-08-19 05:53:26 +00:00
|
|
|
Series FindByTvRageId(int tvRageId);
|
2018-09-21 19:30:09 +00:00
|
|
|
Series FindByPath(string path);
|
2013-02-19 06:56:02 +00:00
|
|
|
}
|
|
|
|
|
2013-03-24 04:16:00 +00:00
|
|
|
public class SeriesRepository : BasicRepository<Series>, ISeriesRepository
|
2013-02-19 06:56:02 +00:00
|
|
|
{
|
2020-01-05 21:49:08 +00:00
|
|
|
protected IMainDatabase _database;
|
|
|
|
|
2015-05-03 19:46:21 +00:00
|
|
|
public SeriesRepository(IMainDatabase database, IEventAggregator eventAggregator)
|
2013-09-14 06:36:07 +00:00
|
|
|
: base(database, eventAggregator)
|
2013-02-19 06:56:02 +00:00
|
|
|
{
|
2020-01-05 21:49:08 +00:00
|
|
|
_database = database;
|
2013-02-19 06:56:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public bool SeriesPathExists(string path)
|
|
|
|
{
|
2014-02-22 08:53:29 +00:00
|
|
|
return Query.Where(c => c.Path == path).Any();
|
2013-02-19 06:56:02 +00:00
|
|
|
}
|
|
|
|
|
2013-04-15 01:41:39 +00:00
|
|
|
public Series FindByTitle(string cleanTitle)
|
2013-02-19 06:56:02 +00:00
|
|
|
{
|
2014-02-22 08:53:29 +00:00
|
|
|
cleanTitle = cleanTitle.ToLowerInvariant();
|
|
|
|
|
|
|
|
return Query.Where(s => s.CleanTitle == cleanTitle)
|
|
|
|
.SingleOrDefault();
|
2013-02-19 06:56:02 +00:00
|
|
|
}
|
2013-03-02 18:25:39 +00:00
|
|
|
|
2013-10-31 23:50:39 +00:00
|
|
|
public Series FindByTitle(string cleanTitle, int year)
|
|
|
|
{
|
2014-02-22 08:53:29 +00:00
|
|
|
cleanTitle = cleanTitle.ToLowerInvariant();
|
|
|
|
|
|
|
|
return Query.Where(s => s.CleanTitle == cleanTitle)
|
|
|
|
.AndWhere(s => s.Year == year)
|
|
|
|
.SingleOrDefault();
|
2013-10-31 23:50:39 +00:00
|
|
|
}
|
|
|
|
|
2020-01-05 21:49:08 +00:00
|
|
|
public List<Series> FindByTitleInexact(string cleanTitle)
|
|
|
|
{
|
|
|
|
var mapper = _database.GetDataMapper();
|
|
|
|
mapper.AddParameter("CleanTitle", cleanTitle);
|
|
|
|
|
|
|
|
return mapper.Query<Series>().Where($"instr(@CleanTitle, [t0].[CleanTitle])");
|
|
|
|
}
|
|
|
|
|
2013-03-02 18:25:39 +00:00
|
|
|
public Series FindByTvdbId(int tvdbId)
|
|
|
|
{
|
2014-02-22 08:53:29 +00:00
|
|
|
return Query.Where(s => s.TvdbId == tvdbId).SingleOrDefault();
|
2013-03-02 18:25:39 +00:00
|
|
|
}
|
|
|
|
|
2013-08-19 05:53:26 +00:00
|
|
|
public Series FindByTvRageId(int tvRageId)
|
|
|
|
{
|
2014-02-22 08:53:29 +00:00
|
|
|
return Query.Where(s => s.TvRageId == tvRageId).SingleOrDefault();
|
2013-08-19 05:53:26 +00:00
|
|
|
}
|
2018-09-21 19:30:09 +00:00
|
|
|
|
|
|
|
public Series FindByPath(string path)
|
|
|
|
{
|
|
|
|
return Query.Where(s => s.Path == path)
|
|
|
|
.FirstOrDefault();
|
|
|
|
}
|
2013-02-19 06:56:02 +00:00
|
|
|
}
|
2018-09-21 19:30:09 +00:00
|
|
|
}
|