sonarr-repo-only/NzbDrone.Web/Controllers/AddSeriesController.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Web;
using System.Web.Mvc;
using NzbDrone.Core.Providers;
using NzbDrone.Web.Models;

namespace NzbDrone.Web.Controllers
{
    public class AddSeriesController : Controller
    {
        public IConfigProvider ConfigProvider { get; set; }
        private readonly ISyncProvider _syncProvider;
        private readonly IRootDirProvider _rootFolderProvider;
        private readonly IConfigProvider _configProvider;
        private readonly IQualityProvider _qualityProvider;
        private readonly ITvDbProvider _tvDbProvider;
        private readonly ISeriesProvider _seriesProvider;

        public AddSeriesController(ISyncProvider syncProvider, IRootDirProvider rootFolderProvider, IConfigProvider configProvider,
            IQualityProvider qualityProvider, ITvDbProvider tvDbProvider, ISeriesProvider seriesProvider)
        {
            ConfigProvider = configProvider;
            _syncProvider = syncProvider;
            _rootFolderProvider = rootFolderProvider;
            _configProvider = configProvider;
            _qualityProvider = qualityProvider;
            _tvDbProvider = tvDbProvider;
            _seriesProvider = seriesProvider;
        }

        [HttpPost]
        public JsonResult ScanNewSeries()
        {
            _syncProvider.BeginUpdateNewSeries();
            return new JsonResult();
        }

        public ActionResult AddNew()
        {
            ViewData["RootDirs"] = _rootFolderProvider.GetAll();
            ViewData["DirSep"] = Path.DirectorySeparatorChar;

            var profiles = _qualityProvider.GetAllProfiles();
            var selectList = new SelectList(profiles, "QualityProfileId", "Name");
            var defaultQuality = Convert.ToInt32(_configProvider.DefaultQualityProfile);

            var model = new AddNewSeriesModel
            {
                DirectorySeparatorChar = Path.DirectorySeparatorChar.ToString(),
                RootDirectories = _rootFolderProvider.GetAll(),
                QualityProfileId = defaultQuality,
                QualitySelectList = selectList
            };

            return View(model);
        }

        public ActionResult AddExisting()
        {
            var defaultQuality = _configProvider.DefaultQualityProfile;
            var profiles = _qualityProvider.GetAllProfiles();
            var selectList = new SelectList(profiles, "QualityProfileId", "Name");

            ViewData["QualityProfileId"] = defaultQuality;
            ViewData["QualitySelectList"] = selectList;

            var unmappedList = new List<String>();

            foreach (var folder in _rootFolderProvider.GetAll())
            {
                unmappedList.AddRange(_syncProvider.GetUnmappedFolders(folder.Path));
            }

            return View(unmappedList);
        }

        public ActionResult AddExistingManual(string path)
        {
            var profiles = _qualityProvider.GetAllProfiles();
            var selectList = new SelectList(profiles, "QualityProfileId", "Name");
            var defaultQuality = _configProvider.DefaultQualityProfile;

            var model = new AddExistingManualModel();
            model.Path = path;
            model.FolderName = new DirectoryInfo(path).Name;
            model.QualityProfileId = defaultQuality;
            model.QualitySelectList = selectList;

            return View(model);
        }


        public ActionResult RenderPartial(string path)
        {
            var dataVal = _tvDbProvider.SearchSeries(new DirectoryInfo(path).Name);
            var names = dataVal.Select(tvdb => tvdb.SeriesName).ToList();

            if (dataVal.Count == 0) return null;
            var ids = dataVal.Select(tvdb => tvdb.Id).ToList();
            var list = new SelectList(dataVal, "Id", "SeriesName");

            ViewData["guid"] = Guid.NewGuid();
            ViewData["path"] = path;
            ViewData["javaPath"] = path.Replace(Path.DirectorySeparatorChar, '|').Replace(Path.VolumeSeparatorChar, '^');
            return PartialView("AddSeriesItem", list);

        }


        public JsonResult AddSeries(string path, int seriesId, int qualityProfileId)
        {
            //Get TVDB Series Name
            //Create new folder for series
            //Add the new series to the Database

            _seriesProvider.AddSeries(path.Replace('|', Path.DirectorySeparatorChar).Replace('^', Path.VolumeSeparatorChar), seriesId, qualityProfileId);
            ScanNewSeries();
            return new JsonResult() { Data = "ok" };
        }

    }
}