Radarr/NzbDrone.Web/Controllers/AddSeriesController.cs

188 lines
6.5 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Web.Mvc;
using NLog;
2011-11-13 04:07:06 +00:00
using NzbDrone.Common;
2011-12-02 01:33:17 +00:00
using NzbDrone.Core.Jobs;
using NzbDrone.Core.Providers;
2011-04-04 03:50:12 +00:00
using NzbDrone.Core.Providers.Core;
using NzbDrone.Core.Repository;
using NzbDrone.Web.Filters;
using NzbDrone.Web.Models;
using TvdbLib.Exceptions;
namespace NzbDrone.Web.Controllers
{
public class AddSeriesController : Controller
{
private readonly ConfigProvider _configProvider;
private readonly QualityProvider _qualityProvider;
2011-04-10 02:44:01 +00:00
private readonly RootDirProvider _rootFolderProvider;
2011-04-08 23:55:23 +00:00
private readonly SeriesProvider _seriesProvider;
private readonly JobProvider _jobProvider;
2011-04-10 02:44:01 +00:00
private readonly TvDbProvider _tvDbProvider;
private readonly DiskProvider _diskProvider;
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
public AddSeriesController(RootDirProvider rootFolderProvider,
2011-04-10 02:44:01 +00:00
ConfigProvider configProvider,
QualityProvider qualityProvider, TvDbProvider tvDbProvider,
SeriesProvider seriesProvider, JobProvider jobProvider,
DiskProvider diskProvider)
{
_rootFolderProvider = rootFolderProvider;
_configProvider = configProvider;
_qualityProvider = qualityProvider;
_tvDbProvider = tvDbProvider;
_seriesProvider = seriesProvider;
_jobProvider = jobProvider;
_diskProvider = diskProvider;
}
public ActionResult Index()
{
return View();
}
public ActionResult AddNew()
{
2011-07-08 03:27:11 +00:00
ViewData["RootDirs"] = _rootFolderProvider.GetAll().Select(c => c.Path).OrderBy(e => e).ToList();
var defaultQuality = _configProvider.DefaultQualityProfile;
2011-07-08 05:41:08 +00:00
var qualityProfiles = _qualityProvider.All();
ViewData["qualityProfiles"] = new SelectList(
qualityProfiles,
"QualityProfileId",
"Name",
defaultQuality);
return View();
}
2011-07-27 22:59:48 +00:00
public ActionResult ExistingSeries()
{
2011-07-27 22:59:48 +00:00
var result = new ExistingSeriesModel();
var unmappedList = new List<String>();
2011-07-27 22:59:48 +00:00
foreach (var folder in _rootFolderProvider.GetAll())
{
unmappedList.AddRange(_rootFolderProvider.GetUnmappedFolders(folder.Path));
}
result.ExistingSeries = new List<Tuple<string, string, int>>();
2011-07-27 22:59:48 +00:00
foreach (var folder in unmappedList)
{
var foldername = new DirectoryInfo(folder).Name;
try
2011-07-27 22:59:48 +00:00
{
var tvdbResult = _tvDbProvider.SearchSeries(foldername).FirstOrDefault();
var title = String.Empty;
var seriesId = 0;
if (tvdbResult != null)
{
title = tvdbResult.SeriesName;
seriesId = tvdbResult.id;
}
result.ExistingSeries.Add(new Tuple<string, string, int>(folder, title, seriesId));
}
catch (Exception ex)
{
logger.WarnException("Failed to connect to TheTVDB to search for: " + foldername, ex);
return View();
}
2011-07-27 22:59:48 +00:00
}
2011-07-27 22:59:48 +00:00
var defaultQuality = Convert.ToInt32(_configProvider.DefaultQualityProfile);
result.Quality = new SelectList(_qualityProvider.All(), "QualityProfileId", "Name", defaultQuality);
2011-07-27 22:59:48 +00:00
return View(result);
}
[HttpPost]
2012-10-01 00:05:16 +00:00
[JsonErrorFilter]
2012-09-20 15:37:40 +00:00
public JsonResult AddNewSeries(string path, string seriesName, int seriesId, int qualityProfileId, string startDate)
2011-07-27 22:59:48 +00:00
{
if (string.IsNullOrWhiteSpace(path) || String.Equals(path, "null", StringComparison.InvariantCultureIgnoreCase))
return JsonNotificationResult.Error("Couldn't add " + seriesName, "You need a valid root folder");
2011-07-27 22:59:48 +00:00
path = Path.Combine(path, MediaFileProvider.CleanFilename(seriesName));
//Create the folder for the new series
//Use the created folder name when adding the series
path = _diskProvider.CreateDirectory(path);
2012-09-20 15:37:40 +00:00
return AddExistingSeries(path, seriesName, seriesId, qualityProfileId, startDate);
}
[HttpPost]
[JsonErrorFilter]
2012-09-20 15:37:40 +00:00
public JsonResult AddExistingSeries(string path, string seriesName, int seriesId, int qualityProfileId, string startDate)
{
if (seriesId == 0 || String.IsNullOrWhiteSpace(seriesName))
return JsonNotificationResult.Error("Add Existing series failed.", "Invalid Series information");
DateTime? date = null;
2012-09-20 15:37:40 +00:00
if (!String.IsNullOrWhiteSpace(startDate))
date = DateTime.Parse(startDate, null, DateTimeStyles.RoundtripKind);
_seriesProvider.AddSeries(seriesName, path, seriesId, qualityProfileId, date);
_jobProvider.QueueJob(typeof(ImportNewSeriesJob));
return JsonNotificationResult.Info(seriesName, "Was added successfully");
}
2011-07-27 22:59:48 +00:00
[HttpGet]
2012-10-01 00:05:16 +00:00
[JsonErrorFilter]
public JsonResult LookupSeries(string term)
{
return JsonNotificationResult.Info("Lookup Failed", "Unknown error while connecting to TheTVDB");
}
2011-07-01 07:23:07 +00:00
public ActionResult RootList()
2011-06-30 06:15:06 +00:00
{
2011-07-01 07:52:44 +00:00
IEnumerable<String> rootDir = _rootFolderProvider.GetAll().Select(c => c.Path).OrderBy(e => e);
2011-07-01 07:23:07 +00:00
return PartialView("RootList", rootDir);
}
public ActionResult RootDir()
{
return PartialView("RootDir");
2011-06-30 06:15:06 +00:00
}
[HttpPost]
[JsonErrorFilter]
public JsonResult SaveRootDir(string path)
{
if (String.IsNullOrWhiteSpace(path))
JsonNotificationResult.Error("Can't add root folder", "Path can not be empty");
_rootFolderProvider.Add(new RootDir { Path = path });
return JsonNotificationResult.Info("Root Folder saved", "Root folder saved successfully.");
}
[JsonErrorFilter]
2011-06-30 06:15:06 +00:00
public JsonResult DeleteRootDir(string path)
{
var id = _rootFolderProvider.GetAll().Where(c => c.Path == path).First().Id;
_rootFolderProvider.Remove(id);
return null;
}
}
2011-04-10 02:44:01 +00:00
}