using System; using System.Collections.Generic; using System.IO; using NLog; using NzbDrone.Core.Providers.Core; using NzbDrone.Core.Repository; using PetaPoco; namespace NzbDrone.Core.Providers { public class SceneMappingProvider { private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); private readonly IDatabase _database; private readonly HttpProvider _httpProvider; public SceneMappingProvider(IDatabase database, HttpProvider httpProvider) { _database = database; _httpProvider = httpProvider; } public SceneMappingProvider() { } public virtual bool UpdateMappings() { try { var mapping = _httpProvider.DownloadString("http://vps.nzbdrone.com/SceneMappings.csv"); var newMaps = new List(); using (var reader = new StringReader(mapping)) { string line; while ((line = reader.ReadLine()) != null) { var split = line.Split(','); int seriesId; Int32.TryParse(split[1], out seriesId); var map = new SceneMapping(); map.CleanTitle = split[0]; map.SeriesId = seriesId; map.SceneName = split[2]; newMaps.Add(map); } } Logger.Debug("Deleting all existing Scene Mappings."); _database.Delete(String.Empty); Logger.Debug("Adding Scene Mappings"); _database.InsertMany(newMaps); } catch (Exception ex) { Logger.InfoException("Failed to Update Scene Mappings", ex); return false; } return true; } public virtual string GetSceneName(int seriesId) { var item = _database.FirstOrDefault("WHERE SeriesId = @0", seriesId); if (item == null) return null; return item.SceneName; } public virtual Nullable GetSeriesId(string cleanName) { var item = _database.SingleOrDefault("WHERE CleanTitle = @0", cleanName); if (item == null) return null; return item.SeriesId; } } }