using MediaBrowser.Common.Configuration; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Entities.Audio; using MediaBrowser.Controller.Entities.Movies; using MediaBrowser.Controller.Entities.TV; using MediaBrowser.Controller.LiveTv; using MediaBrowser.Controller.Persistence; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using MediaBrowser.Model.Querying; using MediaBrowser.Model.Serialization; using System; using System.Collections.Generic; using System.Data; using System.Globalization; using System.IO; using System.Linq; using System.Runtime.Serialization; using System.Threading; using System.Threading.Tasks; using MediaBrowser.Controller.Channels; using MediaBrowser.Model.LiveTv; namespace MediaBrowser.Server.Implementations.Persistence { /// /// Class SQLiteItemRepository /// public class SqliteItemRepository : IItemRepository { private IDbConnection _connection; private readonly ILogger _logger; private readonly TypeMapper _typeMapper = new TypeMapper(); /// /// Gets the name of the repository /// /// The name. public string Name { get { return "SQLite"; } } /// /// Gets the json serializer. /// /// The json serializer. private readonly IJsonSerializer _jsonSerializer; /// /// The _app paths /// private readonly IApplicationPaths _appPaths; /// /// The _save item command /// private IDbCommand _saveItemCommand; private readonly string _criticReviewsPath; private IDbCommand _deleteChildrenCommand; private IDbCommand _saveChildrenCommand; private IDbCommand _deleteItemCommand; private IDbCommand _deletePeopleCommand; private IDbCommand _savePersonCommand; private IDbCommand _deleteChaptersCommand; private IDbCommand _saveChapterCommand; private IDbCommand _deleteStreamsCommand; private IDbCommand _saveStreamCommand; private const int LatestSchemaVersion = 17; /// /// Initializes a new instance of the class. /// /// The app paths. /// The json serializer. /// The log manager. /// /// appPaths /// or /// jsonSerializer /// public SqliteItemRepository(IApplicationPaths appPaths, IJsonSerializer jsonSerializer, ILogManager logManager) { if (appPaths == null) { throw new ArgumentNullException("appPaths"); } if (jsonSerializer == null) { throw new ArgumentNullException("jsonSerializer"); } _appPaths = appPaths; _jsonSerializer = jsonSerializer; _criticReviewsPath = Path.Combine(_appPaths.DataPath, "critic-reviews"); _logger = logManager.GetLogger(GetType().Name); } private const string ChaptersTableName = "Chapters2"; /// /// Opens the connection to the database /// /// Task. public async Task Initialize() { var dbFile = Path.Combine(_appPaths.DataPath, "library.db"); _connection = await SqliteExtensions.ConnectToDb(dbFile, _logger).ConfigureAwait(false); var createMediaStreamsTableCommand = "create table if not exists mediastreams (ItemId GUID, StreamIndex INT, StreamType TEXT, Codec TEXT, Language TEXT, ChannelLayout TEXT, Profile TEXT, AspectRatio TEXT, Path TEXT, IsInterlaced BIT, BitRate INT NULL, Channels INT NULL, SampleRate INT NULL, IsDefault BIT, IsForced BIT, IsExternal BIT, Height INT NULL, Width INT NULL, AverageFrameRate FLOAT NULL, RealFrameRate FLOAT NULL, Level FLOAT NULL, PixelFormat TEXT, BitDepth INT NULL, IsAnamorphic BIT NULL, RefFrames INT NULL, IsCabac BIT NULL, KeyFrames TEXT NULL, CodecTag TEXT NULL, PRIMARY KEY (ItemId, StreamIndex))"; string[] queries = { "create table if not exists TypedBaseItems (guid GUID primary key, type TEXT, data BLOB)", "create index if not exists idx_TypedBaseItems on TypedBaseItems(guid)", "create table if not exists ChildrenIds (ParentId GUID, ItemId GUID, PRIMARY KEY (ParentId, ItemId))", "create index if not exists idx_ChildrenIds on ChildrenIds(ParentId,ItemId)", "create table if not exists People (ItemId GUID, Name TEXT NOT NULL, Role TEXT, PersonType TEXT, SortOrder int, ListOrder int)", "create table if not exists "+ChaptersTableName+" (ItemId GUID, ChapterIndex INT, StartPositionTicks BIGINT, Name TEXT, ImagePath TEXT, PRIMARY KEY (ItemId, ChapterIndex))", "create index if not exists idx_"+ChaptersTableName+" on "+ChaptersTableName+"(ItemId, ChapterIndex)", createMediaStreamsTableCommand, "create index if not exists idx_mediastreams on mediastreams(ItemId, StreamIndex)", //pragmas "pragma temp_store = memory", "pragma shrink_memory" }; _connection.RunQueries(queries, _logger); _connection.AddColumn(_logger, "TypedBaseItems", "Path", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "StartDate", "DATETIME"); _connection.AddColumn(_logger, "TypedBaseItems", "EndDate", "DATETIME"); _connection.AddColumn(_logger, "TypedBaseItems", "ChannelId", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "IsMovie", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "IsSports", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "IsKids", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "CommunityRating", "Float"); _connection.AddColumn(_logger, "TypedBaseItems", "CustomRating", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "IndexNumber", "INT"); _connection.AddColumn(_logger, "TypedBaseItems", "IsLocked", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "Name", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRating", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "MediaType", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "Overview", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "ParentIndexNumber", "INT"); _connection.AddColumn(_logger, "TypedBaseItems", "PremiereDate", "DATETIME"); _connection.AddColumn(_logger, "TypedBaseItems", "ProductionYear", "INT"); _connection.AddColumn(_logger, "TypedBaseItems", "ParentId", "GUID"); _connection.AddColumn(_logger, "TypedBaseItems", "Genres", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "ParentalRatingValue", "INT"); _connection.AddColumn(_logger, "TypedBaseItems", "SchemaVersion", "INT"); _connection.AddColumn(_logger, "TypedBaseItems", "SortName", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "RunTimeTicks", "BIGINT"); _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRatingDescription", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "HomePageUrl", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "VoteCount", "INT"); _connection.AddColumn(_logger, "TypedBaseItems", "DisplayMediaType", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "DateCreated", "DATETIME"); _connection.AddColumn(_logger, "TypedBaseItems", "DateModified", "DATETIME"); _connection.AddColumn(_logger, "TypedBaseItems", "ForcedSortName", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "IsOffline", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "LocationType", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "IsSeries", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "IsLive", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "IsNews", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "IsPremiere", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "EpisodeTitle", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "IsRepeat", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataLanguage", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataCountryCode", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "IsHD", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "ExternalEtag", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "DateLastRefreshed", "DATETIME"); _connection.AddColumn(_logger, "TypedBaseItems", "DateLastSaved", "DATETIME"); _connection.AddColumn(_logger, "TypedBaseItems", "IsInMixedFolder", "BIT"); _connection.AddColumn(_logger, "TypedBaseItems", "LockedFields", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "Studios", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "Audio", "Text"); _connection.AddColumn(_logger, "TypedBaseItems", "ExternalServiceId", "Text"); PrepareStatements(); new MediaStreamColumns(_connection, _logger).AddColumns(); var chapterDbFile = Path.Combine(_appPaths.DataPath, "chapters.db"); if (File.Exists(chapterDbFile)) { MigrateChapters(chapterDbFile); } var mediaStreamsDbFile = Path.Combine(_appPaths.DataPath, "mediainfo.db"); if (File.Exists(mediaStreamsDbFile)) { MigrateMediaStreams(mediaStreamsDbFile); } } private void MigrateMediaStreams(string file) { try { var backupFile = file + ".bak"; File.Copy(file, backupFile, true); SqliteExtensions.Attach(_connection, backupFile, "MediaInfoOld"); var columns = string.Join(",", _mediaStreamSaveColumns); string[] queries = { "REPLACE INTO mediastreams("+columns+") SELECT "+columns+" FROM MediaInfoOld.mediastreams;" }; _connection.RunQueries(queries, _logger); } catch (Exception ex) { _logger.ErrorException("Error migrating media info database", ex); } finally { TryDeleteFile(file); } } private void MigrateChapters(string file) { try { var backupFile = file + ".bak"; File.Copy(file, backupFile, true); SqliteExtensions.Attach(_connection, backupFile, "ChaptersOld"); string[] queries = { "REPLACE INTO "+ChaptersTableName+"(ItemId, ChapterIndex, StartPositionTicks, Name, ImagePath) SELECT ItemId, ChapterIndex, StartPositionTicks, Name, ImagePath FROM ChaptersOld.Chapters;" }; _connection.RunQueries(queries, _logger); } catch (Exception ex) { _logger.ErrorException("Error migrating chapter database", ex); } finally { TryDeleteFile(file); } } private void TryDeleteFile(string file) { try { File.Delete(file); } catch (Exception ex) { _logger.ErrorException("Error deleting file {0}", ex, file); } } /// /// The _write lock /// private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1); private readonly string[] _retriveItemColumns = { "type", "data", "StartDate", "EndDate", "IsOffline", "ChannelId", "IsMovie", "IsSports", "IsKids", "IsSeries", "IsLive", "IsNews", "IsPremiere", "EpisodeTitle", "IsRepeat", "CommunityRating", "CustomRating", "IndexNumber", "IsLocked", "PreferredMetadataLanguage", "PreferredMetadataCountryCode", "IsHD", "ExternalEtag", "DateLastRefreshed", "Name", "Path", "PremiereDate", "Overview", "ParentIndexNumber", "ProductionYear", "OfficialRating", "OfficialRatingDescription", "HomePageUrl", "DisplayMediaType", "ForcedSortName", "RunTimeTicks", "VoteCount", "DateCreated", "DateModified", "guid", "Genres", "ParentId", "Audio", "ExternalServiceId" }; private readonly string[] _mediaStreamSaveColumns = { "ItemId", "StreamIndex", "StreamType", "Codec", "Language", "ChannelLayout", "Profile", "AspectRatio", "Path", "IsInterlaced", "BitRate", "Channels", "SampleRate", "IsDefault", "IsForced", "IsExternal", "Height", "Width", "AverageFrameRate", "RealFrameRate", "Level", "PixelFormat", "BitDepth", "IsAnamorphic", "RefFrames", "IsCabac", "KeyFrames", "CodecTag" }; /// /// Prepares the statements. /// private void PrepareStatements() { var saveColumns = new List { "guid", "type", "data", "Path", "StartDate", "EndDate", "ChannelId", "IsKids", "IsMovie", "IsSports", "IsSeries", "IsLive", "IsNews", "IsPremiere", "EpisodeTitle", "IsRepeat", "CommunityRating", "CustomRating", "IndexNumber", "IsLocked", "Name", "OfficialRating", "MediaType", "Overview", "ParentIndexNumber", "PremiereDate", "ProductionYear", "ParentId", "Genres", "ParentalRatingValue", "SchemaVersion", "SortName", "RunTimeTicks", "OfficialRatingDescription", "HomePageUrl", "VoteCount", "DisplayMediaType", "DateCreated", "DateModified", "ForcedSortName", "IsOffline", "LocationType", "PreferredMetadataLanguage", "PreferredMetadataCountryCode", "IsHD", "ExternalEtag", "DateLastRefreshed", "DateLastSaved", "IsInMixedFolder", "LockedFields", "Studios", "Audio", "ExternalServiceId" }; _saveItemCommand = _connection.CreateCommand(); _saveItemCommand.CommandText = "replace into TypedBaseItems (" + string.Join(",", saveColumns.ToArray()) + ") values ("; for (var i = 1; i <= saveColumns.Count; i++) { if (i > 1) { _saveItemCommand.CommandText += ","; } _saveItemCommand.CommandText += "@" + i.ToString(CultureInfo.InvariantCulture); _saveItemCommand.Parameters.Add(_saveItemCommand, "@" + i.ToString(CultureInfo.InvariantCulture)); } _saveItemCommand.CommandText += ")"; _deleteChildrenCommand = _connection.CreateCommand(); _deleteChildrenCommand.CommandText = "delete from ChildrenIds where ParentId=@ParentId"; _deleteChildrenCommand.Parameters.Add(_deleteChildrenCommand, "@ParentId"); _deleteItemCommand = _connection.CreateCommand(); _deleteItemCommand.CommandText = "delete from TypedBaseItems where guid=@Id"; _deleteItemCommand.Parameters.Add(_deleteItemCommand, "@Id"); _saveChildrenCommand = _connection.CreateCommand(); _saveChildrenCommand.CommandText = "replace into ChildrenIds (ParentId, ItemId) values (@ParentId, @ItemId)"; _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ParentId"); _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ItemId"); // People _deletePeopleCommand = _connection.CreateCommand(); _deletePeopleCommand.CommandText = "delete from People where ItemId=@Id"; _deletePeopleCommand.Parameters.Add(_deletePeopleCommand, "@Id"); _savePersonCommand = _connection.CreateCommand(); _savePersonCommand.CommandText = "insert into People (ItemId, Name, Role, PersonType, SortOrder, ListOrder) values (@ItemId, @Name, @Role, @PersonType, @SortOrder, @ListOrder)"; _savePersonCommand.Parameters.Add(_savePersonCommand, "@ItemId"); _savePersonCommand.Parameters.Add(_savePersonCommand, "@Name"); _savePersonCommand.Parameters.Add(_savePersonCommand, "@Role"); _savePersonCommand.Parameters.Add(_savePersonCommand, "@PersonType"); _savePersonCommand.Parameters.Add(_savePersonCommand, "@SortOrder"); _savePersonCommand.Parameters.Add(_savePersonCommand, "@ListOrder"); // Chapters _deleteChaptersCommand = _connection.CreateCommand(); _deleteChaptersCommand.CommandText = "delete from " + ChaptersTableName + " where ItemId=@ItemId"; _deleteChaptersCommand.Parameters.Add(_deleteChaptersCommand, "@ItemId"); _saveChapterCommand = _connection.CreateCommand(); _saveChapterCommand.CommandText = "replace into " + ChaptersTableName + " (ItemId, ChapterIndex, StartPositionTicks, Name, ImagePath) values (@ItemId, @ChapterIndex, @StartPositionTicks, @Name, @ImagePath)"; _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@ItemId"); _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@ChapterIndex"); _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@StartPositionTicks"); _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@Name"); _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@ImagePath"); // MediaStreams _deleteStreamsCommand = _connection.CreateCommand(); _deleteStreamsCommand.CommandText = "delete from mediastreams where ItemId=@ItemId"; _deleteStreamsCommand.Parameters.Add(_deleteStreamsCommand, "@ItemId"); _saveStreamCommand = _connection.CreateCommand(); _saveStreamCommand.CommandText = string.Format("replace into mediastreams ({0}) values ({1})", string.Join(",", _mediaStreamSaveColumns), string.Join(",", _mediaStreamSaveColumns.Select(i => "@" + i).ToArray())); foreach (var col in _mediaStreamSaveColumns) { _saveStreamCommand.Parameters.Add(_saveStreamCommand, "@" + col); } } /// /// Save a standard item in the repo /// /// The item. /// The cancellation token. /// Task. /// item public Task SaveItem(BaseItem item, CancellationToken cancellationToken) { if (item == null) { throw new ArgumentNullException("item"); } return SaveItems(new[] { item }, cancellationToken); } /// /// Saves the items. /// /// The items. /// The cancellation token. /// Task. /// /// items /// or /// cancellationToken /// public async Task SaveItems(IEnumerable items, CancellationToken cancellationToken) { if (items == null) { throw new ArgumentNullException("items"); } cancellationToken.ThrowIfCancellationRequested(); CheckDisposed(); await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false); IDbTransaction transaction = null; try { transaction = _connection.BeginTransaction(); foreach (var item in items) { cancellationToken.ThrowIfCancellationRequested(); var index = 0; _saveItemCommand.GetParameter(index++).Value = item.Id; _saveItemCommand.GetParameter(index++).Value = item.GetType().FullName; _saveItemCommand.GetParameter(index++).Value = _jsonSerializer.SerializeToBytes(item); _saveItemCommand.GetParameter(index++).Value = item.Path; var hasStartDate = item as IHasStartDate; if (hasStartDate != null) { _saveItemCommand.GetParameter(index++).Value = hasStartDate.StartDate; } else { _saveItemCommand.GetParameter(index++).Value = null; } _saveItemCommand.GetParameter(index++).Value = item.EndDate; _saveItemCommand.GetParameter(index++).Value = item.ChannelId; var hasProgramAttributes = item as IHasProgramAttributes; if (hasProgramAttributes != null) { _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsKids; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsMovie; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSports; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSeries; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsLive; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsNews; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsPremiere; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.EpisodeTitle; _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsRepeat; } else { _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; _saveItemCommand.GetParameter(index++).Value = null; } _saveItemCommand.GetParameter(index++).Value = item.CommunityRating; _saveItemCommand.GetParameter(index++).Value = item.CustomRating; _saveItemCommand.GetParameter(index++).Value = item.IndexNumber; _saveItemCommand.GetParameter(index++).Value = item.IsLocked; _saveItemCommand.GetParameter(index++).Value = item.Name; _saveItemCommand.GetParameter(index++).Value = item.OfficialRating; _saveItemCommand.GetParameter(index++).Value = item.MediaType; _saveItemCommand.GetParameter(index++).Value = item.Overview; _saveItemCommand.GetParameter(index++).Value = item.ParentIndexNumber; _saveItemCommand.GetParameter(index++).Value = item.PremiereDate; _saveItemCommand.GetParameter(index++).Value = item.ProductionYear; if (item.ParentId == Guid.Empty) { _saveItemCommand.GetParameter(index++).Value = null; } else { _saveItemCommand.GetParameter(index++).Value = item.ParentId; } _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.Genres.ToArray()); _saveItemCommand.GetParameter(index++).Value = item.GetParentalRatingValue(); _saveItemCommand.GetParameter(index++).Value = LatestSchemaVersion; _saveItemCommand.GetParameter(index++).Value = item.SortName; _saveItemCommand.GetParameter(index++).Value = item.RunTimeTicks; _saveItemCommand.GetParameter(index++).Value = item.OfficialRatingDescription; _saveItemCommand.GetParameter(index++).Value = item.HomePageUrl; _saveItemCommand.GetParameter(index++).Value = item.VoteCount; _saveItemCommand.GetParameter(index++).Value = item.DisplayMediaType; _saveItemCommand.GetParameter(index++).Value = item.DateCreated; _saveItemCommand.GetParameter(index++).Value = item.DateModified; _saveItemCommand.GetParameter(index++).Value = item.ForcedSortName; _saveItemCommand.GetParameter(index++).Value = item.IsOffline; _saveItemCommand.GetParameter(index++).Value = item.LocationType.ToString(); _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataLanguage; _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataCountryCode; _saveItemCommand.GetParameter(index++).Value = item.IsHD; _saveItemCommand.GetParameter(index++).Value = item.ExternalEtag; if (item.DateLastRefreshed == default(DateTime)) { _saveItemCommand.GetParameter(index++).Value = null; } else { _saveItemCommand.GetParameter(index++).Value = item.DateLastRefreshed; } _saveItemCommand.GetParameter(index++).Value = item.DateLastSaved; _saveItemCommand.GetParameter(index++).Value = item.IsInMixedFolder; _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.LockedFields.Select(i => i.ToString()).ToArray()); _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.Studios.ToArray()); if (item.Audio.HasValue) { _saveItemCommand.GetParameter(index++).Value = item.Audio.Value.ToString(); } else { _saveItemCommand.GetParameter(index++).Value = null; } var tvItem = item as ILiveTvItem; if (tvItem != null) { _saveItemCommand.GetParameter(index++).Value = tvItem.ServiceName; } else { _saveItemCommand.GetParameter(index++).Value = null; } _saveItemCommand.Transaction = transaction; _saveItemCommand.ExecuteNonQuery(); } transaction.Commit(); } catch (OperationCanceledException) { if (transaction != null) { transaction.Rollback(); } throw; } catch (Exception e) { _logger.ErrorException("Failed to save items:", e); if (transaction != null) { transaction.Rollback(); } throw; } finally { if (transaction != null) { transaction.Dispose(); } _writeLock.Release(); } } /// /// Internal retrieve from items or users table /// /// The id. /// BaseItem. /// id /// public BaseItem RetrieveItem(Guid id) { if (id == Guid.Empty) { throw new ArgumentNullException("id"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid = @guid"; cmd.Parameters.Add(cmd, "@guid", DbType.Guid).Value = id; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow)) { if (reader.Read()) { return GetItem(reader); } } return null; } } private BaseItem GetItem(IDataReader reader) { var typeString = reader.GetString(0); var type = _typeMapper.GetType(typeString); if (type == null) { _logger.Debug("Unknown type {0}", typeString); return null; } BaseItem item; using (var stream = reader.GetMemoryStream(1)) { try { item = _jsonSerializer.DeserializeFromStream(stream, type) as BaseItem; if (item == null) { return null; } } catch (SerializationException ex) { _logger.ErrorException("Error deserializing item", ex); return null; } } if (!reader.IsDBNull(2)) { var hasStartDate = item as IHasStartDate; if (hasStartDate != null) { hasStartDate.StartDate = reader.GetDateTime(2).ToUniversalTime(); } } if (!reader.IsDBNull(3)) { item.EndDate = reader.GetDateTime(3).ToUniversalTime(); } if (!reader.IsDBNull(4)) { item.IsOffline = reader.GetBoolean(4); } if (!reader.IsDBNull(5)) { item.ChannelId = reader.GetString(5); } var hasProgramAttributes = item as IHasProgramAttributes; if (hasProgramAttributes != null) { if (!reader.IsDBNull(6)) { hasProgramAttributes.IsMovie = reader.GetBoolean(6); } if (!reader.IsDBNull(7)) { hasProgramAttributes.IsSports = reader.GetBoolean(7); } if (!reader.IsDBNull(8)) { hasProgramAttributes.IsKids = reader.GetBoolean(8); } if (!reader.IsDBNull(9)) { hasProgramAttributes.IsSeries = reader.GetBoolean(9); } if (!reader.IsDBNull(10)) { hasProgramAttributes.IsLive = reader.GetBoolean(10); } if (!reader.IsDBNull(11)) { hasProgramAttributes.IsNews = reader.GetBoolean(11); } if (!reader.IsDBNull(12)) { hasProgramAttributes.IsPremiere = reader.GetBoolean(12); } if (!reader.IsDBNull(13)) { hasProgramAttributes.EpisodeTitle = reader.GetString(13); } if (!reader.IsDBNull(14)) { hasProgramAttributes.IsRepeat = reader.GetBoolean(14); } } if (!reader.IsDBNull(15)) { item.CommunityRating = reader.GetFloat(15); } if (!reader.IsDBNull(16)) { item.CustomRating = reader.GetString(16); } if (!reader.IsDBNull(17)) { item.IndexNumber = reader.GetInt32(17); } if (!reader.IsDBNull(18)) { item.IsLocked = reader.GetBoolean(18); } if (!reader.IsDBNull(19)) { item.PreferredMetadataLanguage = reader.GetString(19); } if (!reader.IsDBNull(20)) { item.PreferredMetadataCountryCode = reader.GetString(20); } if (!reader.IsDBNull(21)) { item.IsHD = reader.GetBoolean(21); } if (!reader.IsDBNull(22)) { item.ExternalEtag = reader.GetString(22); } if (!reader.IsDBNull(23)) { item.DateLastRefreshed = reader.GetDateTime(23).ToUniversalTime(); } if (!reader.IsDBNull(24)) { item.Name = reader.GetString(24); } if (!reader.IsDBNull(25)) { item.Path = reader.GetString(25); } if (!reader.IsDBNull(26)) { item.PremiereDate = reader.GetDateTime(26).ToUniversalTime(); } if (!reader.IsDBNull(27)) { item.Overview = reader.GetString(27); } if (!reader.IsDBNull(28)) { item.ParentIndexNumber = reader.GetInt32(28); } if (!reader.IsDBNull(29)) { item.ProductionYear = reader.GetInt32(29); } if (!reader.IsDBNull(30)) { item.OfficialRating = reader.GetString(30); } if (!reader.IsDBNull(31)) { item.OfficialRating = reader.GetString(31); } if (!reader.IsDBNull(32)) { item.HomePageUrl = reader.GetString(32); } if (!reader.IsDBNull(33)) { item.DisplayMediaType = reader.GetString(33); } if (!reader.IsDBNull(34)) { item.ForcedSortName = reader.GetString(34); } if (!reader.IsDBNull(35)) { item.RunTimeTicks = reader.GetInt64(35); } if (!reader.IsDBNull(36)) { item.VoteCount = reader.GetInt32(36); } if (!reader.IsDBNull(37)) { item.DateCreated = reader.GetDateTime(37).ToUniversalTime(); } if (!reader.IsDBNull(38)) { item.DateModified = reader.GetDateTime(38).ToUniversalTime(); } item.Id = reader.GetGuid(39); if (!reader.IsDBNull(40)) { item.Genres = reader.GetString(40).Split('|').Where(i => !string.IsNullOrWhiteSpace(i)).ToList(); } if (!reader.IsDBNull(41)) { item.ParentId = reader.GetGuid(41); } if (!reader.IsDBNull(42)) { item.Audio = (ProgramAudio)Enum.Parse(typeof(ProgramAudio), reader.GetString(42), true); } if (!reader.IsDBNull(43)) { var tvItem = item as ILiveTvItem; if (tvItem != null) { item.ForcedSortName = reader.GetString(43); } } return item; } /// /// Gets the critic reviews. /// /// The item id. /// Task{IEnumerable{ItemReview}}. public IEnumerable GetCriticReviews(Guid itemId) { try { var path = Path.Combine(_criticReviewsPath, itemId + ".json"); return _jsonSerializer.DeserializeFromFile>(path); } catch (DirectoryNotFoundException) { return new List(); } catch (FileNotFoundException) { return new List(); } } private readonly Task _cachedTask = Task.FromResult(true); /// /// Saves the critic reviews. /// /// The item id. /// The critic reviews. /// Task. public Task SaveCriticReviews(Guid itemId, IEnumerable criticReviews) { Directory.CreateDirectory(_criticReviewsPath); var path = Path.Combine(_criticReviewsPath, itemId + ".json"); _jsonSerializer.SerializeToFile(criticReviews.ToList(), path); return _cachedTask; } /// /// Gets chapters for an item /// /// The id. /// IEnumerable{ChapterInfo}. /// id public IEnumerable GetChapters(Guid id) { CheckDisposed(); if (id == Guid.Empty) { throw new ArgumentNullException("id"); } using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select StartPositionTicks,Name,ImagePath from " + ChaptersTableName + " where ItemId = @ItemId order by ChapterIndex asc"; cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = id; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult)) { while (reader.Read()) { yield return GetChapter(reader); } } } } /// /// Gets a single chapter for an item /// /// The id. /// The index. /// ChapterInfo. /// id public ChapterInfo GetChapter(Guid id, int index) { CheckDisposed(); if (id == Guid.Empty) { throw new ArgumentNullException("id"); } using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select StartPositionTicks,Name,ImagePath from " + ChaptersTableName + " where ItemId = @ItemId and ChapterIndex=@ChapterIndex"; cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = id; cmd.Parameters.Add(cmd, "@ChapterIndex", DbType.Int32).Value = index; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow)) { if (reader.Read()) { return GetChapter(reader); } } return null; } } /// /// Gets the chapter. /// /// The reader. /// ChapterInfo. private ChapterInfo GetChapter(IDataReader reader) { var chapter = new ChapterInfo { StartPositionTicks = reader.GetInt64(0) }; if (!reader.IsDBNull(1)) { chapter.Name = reader.GetString(1); } if (!reader.IsDBNull(2)) { chapter.ImagePath = reader.GetString(2); } return chapter; } /// /// Saves the chapters. /// /// The id. /// The chapters. /// The cancellation token. /// Task. /// /// id /// or /// chapters /// or /// cancellationToken /// public async Task SaveChapters(Guid id, IEnumerable chapters, CancellationToken cancellationToken) { CheckDisposed(); if (id == Guid.Empty) { throw new ArgumentNullException("id"); } if (chapters == null) { throw new ArgumentNullException("chapters"); } cancellationToken.ThrowIfCancellationRequested(); await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false); IDbTransaction transaction = null; try { transaction = _connection.BeginTransaction(); // First delete chapters _deleteChaptersCommand.GetParameter(0).Value = id; _deleteChaptersCommand.Transaction = transaction; _deleteChaptersCommand.ExecuteNonQuery(); var index = 0; foreach (var chapter in chapters) { cancellationToken.ThrowIfCancellationRequested(); _saveChapterCommand.GetParameter(0).Value = id; _saveChapterCommand.GetParameter(1).Value = index; _saveChapterCommand.GetParameter(2).Value = chapter.StartPositionTicks; _saveChapterCommand.GetParameter(3).Value = chapter.Name; _saveChapterCommand.GetParameter(4).Value = chapter.ImagePath; _saveChapterCommand.Transaction = transaction; _saveChapterCommand.ExecuteNonQuery(); index++; } transaction.Commit(); } catch (OperationCanceledException) { if (transaction != null) { transaction.Rollback(); } throw; } catch (Exception e) { _logger.ErrorException("Failed to save chapters:", e); if (transaction != null) { transaction.Rollback(); } throw; } finally { if (transaction != null) { transaction.Dispose(); } _writeLock.Release(); } } /// /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources. /// public void Dispose() { Dispose(true); GC.SuppressFinalize(this); } private readonly object _disposeLock = new object(); private bool _disposed; private void CheckDisposed() { if (_disposed) { throw new ObjectDisposedException(GetType().Name + " has been disposed and cannot be accessed."); } } /// /// Releases unmanaged and - optionally - managed resources. /// /// true to release both managed and unmanaged resources; false to release only unmanaged resources. protected virtual void Dispose(bool dispose) { if (dispose) { _disposed = true; try { lock (_disposeLock) { _writeLock.Wait(); if (_connection != null) { if (_connection.IsOpen()) { _connection.Close(); } _connection.Dispose(); _connection = null; } } } catch (Exception ex) { _logger.ErrorException("Error disposing database", ex); } } } public IEnumerable GetChildren(Guid parentId) { if (parentId == Guid.Empty) { throw new ArgumentNullException("parentId"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select ItemId from ChildrenIds where ParentId = @ParentId"; cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult)) { while (reader.Read()) { yield return reader.GetGuid(0); } } } } public IEnumerable GetChildrenItems(Guid parentId) { if (parentId == Guid.Empty) { throw new ArgumentNullException("parentId"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid in (select ItemId from ChildrenIds where ParentId = @ParentId)"; cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult)) { while (reader.Read()) { var item = GetItem(reader); if (item != null) { yield return item; } } } } } public IEnumerable GetItemsOfType(Type type) { if (type == null) { throw new ArgumentNullException("type"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where type = @type"; cmd.Parameters.Add(cmd, "@type", DbType.String).Value = type.FullName; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult)) { while (reader.Read()) { var item = GetItem(reader); if (item != null) { yield return item; } } } } } public QueryResult GetItems(InternalItemsQuery query) { if (query == null) { throw new ArgumentNullException("query"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems"; var whereClauses = GetWhereClauses(query, cmd, false); var whereTextWithoutPaging = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); whereClauses = GetWhereClauses(query, cmd, true); var whereText = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); cmd.CommandText += whereText; cmd.CommandText += GetOrderByText(query); if (query.Limit.HasValue) { cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture); } cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging; _logger.Debug(cmd.CommandText); var list = new List(); var count = 0; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess)) { while (reader.Read()) { var item = GetItem(reader); if (item != null) { list.Add(item); } } if (reader.NextResult() && reader.Read()) { count = reader.GetInt32(0); } } return new QueryResult() { Items = list.ToArray(), TotalRecordCount = count }; } } private string GetOrderByText(InternalItemsQuery query) { if (query.SortBy == null || query.SortBy.Length == 0) { return string.Empty; } var sortOrder = query.SortOrder == SortOrder.Descending ? "DESC" : "ASC"; return " ORDER BY " + string.Join(",", query.SortBy.Select(i => MapOrderByField(i) + " " + sortOrder).ToArray()); } private string MapOrderByField(string name) { return name; } public List GetItemIdsList(InternalItemsQuery query) { if (query == null) { throw new ArgumentNullException("query"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select guid from TypedBaseItems"; var whereClauses = GetWhereClauses(query, cmd, true); var whereText = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); cmd.CommandText += whereText; cmd.CommandText += GetOrderByText(query); if (query.Limit.HasValue) { cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture); } var list = new List(); _logger.Debug(cmd.CommandText); using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess)) { while (reader.Read()) { list.Add(reader.GetGuid(0)); } } return list; } } public QueryResult> GetItemIdsWithPath(InternalItemsQuery query) { if (query == null) { throw new ArgumentNullException("query"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select guid,path from TypedBaseItems"; var whereClauses = GetWhereClauses(query, cmd, false); var whereTextWithoutPaging = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); whereClauses = GetWhereClauses(query, cmd, true); var whereText = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); cmd.CommandText += whereText; cmd.CommandText += GetOrderByText(query); if (query.Limit.HasValue) { cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture); } cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging; var list = new List>(); var count = 0; _logger.Debug(cmd.CommandText); using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess)) { while (reader.Read()) { var id = reader.GetGuid(0); string path = null; if (!reader.IsDBNull(1)) { path = reader.GetString(1); } list.Add(new Tuple(id, path)); } if (reader.NextResult() && reader.Read()) { count = reader.GetInt32(0); } } return new QueryResult>() { Items = list.ToArray(), TotalRecordCount = count }; } } public QueryResult GetItemIds(InternalItemsQuery query) { if (query == null) { throw new ArgumentNullException("query"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select guid from TypedBaseItems"; var whereClauses = GetWhereClauses(query, cmd, false); var whereTextWithoutPaging = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); whereClauses = GetWhereClauses(query, cmd, true); var whereText = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); cmd.CommandText += whereText; cmd.CommandText += GetOrderByText(query); if (query.Limit.HasValue) { cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture); } cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging; var list = new List(); var count = 0; _logger.Debug(cmd.CommandText); using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess)) { while (reader.Read()) { list.Add(reader.GetGuid(0)); } if (reader.NextResult() && reader.Read()) { count = reader.GetInt32(0); } } return new QueryResult() { Items = list.ToArray(), TotalRecordCount = count }; } } private List GetWhereClauses(InternalItemsQuery query, IDbCommand cmd, bool addPaging) { var whereClauses = new List(); if (query.IsCurrentSchema.HasValue) { if (query.IsCurrentSchema.Value) { whereClauses.Add("(SchemaVersion not null AND SchemaVersion=@SchemaVersion)"); } else { whereClauses.Add("(SchemaVersion is null or SchemaVersion<>@SchemaVersion)"); } cmd.Parameters.Add(cmd, "@SchemaVersion", DbType.Int32).Value = LatestSchemaVersion; } if (query.IsOffline.HasValue) { whereClauses.Add("IsOffline=@IsOffline"); cmd.Parameters.Add(cmd, "@IsOffline", DbType.Boolean).Value = query.IsOffline; } if (query.LocationType.HasValue) { whereClauses.Add("LocationType=@LocationType"); cmd.Parameters.Add(cmd, "@LocationType", DbType.String).Value = query.LocationType.Value; } if (query.IsMovie.HasValue) { whereClauses.Add("IsMovie=@IsMovie"); cmd.Parameters.Add(cmd, "@IsMovie", DbType.Boolean).Value = query.IsMovie; } if (query.IsKids.HasValue) { whereClauses.Add("IsKids=@IsKids"); cmd.Parameters.Add(cmd, "@IsKids", DbType.Boolean).Value = query.IsKids; } if (query.IsSports.HasValue) { whereClauses.Add("IsSports=@IsSports"); cmd.Parameters.Add(cmd, "@IsSports", DbType.Boolean).Value = query.IsSports; } var includeTypes = query.IncludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray(); if (includeTypes.Length == 1) { whereClauses.Add("type=@type"); cmd.Parameters.Add(cmd, "@type", DbType.String).Value = includeTypes[0]; } else if (includeTypes.Length > 1) { var inClause = string.Join(",", includeTypes.Select(i => "'" + i + "'").ToArray()); whereClauses.Add(string.Format("type in ({0})", inClause)); } var excludeTypes = query.ExcludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray(); if (excludeTypes.Length == 1) { whereClauses.Add("type<>@type"); cmd.Parameters.Add(cmd, "@type", DbType.String).Value = excludeTypes[0]; } else if (excludeTypes.Length > 1) { var inClause = string.Join(",", excludeTypes.Select(i => "'" + i + "'").ToArray()); whereClauses.Add(string.Format("type not in ({0})", inClause)); } if (query.ChannelIds.Length == 1) { whereClauses.Add("ChannelId=@ChannelId"); cmd.Parameters.Add(cmd, "@ChannelId", DbType.String).Value = query.ChannelIds[0]; } if (query.ChannelIds.Length > 1) { var inClause = string.Join(",", query.ChannelIds.Select(i => "'" + i + "'").ToArray()); whereClauses.Add(string.Format("ChannelId in ({0})", inClause)); } if (query.ParentId.HasValue) { whereClauses.Add("ParentId=@ParentId"); cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = query.ParentId.Value; } if (query.MinEndDate.HasValue) { whereClauses.Add("EndDate>=@MinEndDate"); cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = query.MinEndDate.Value; } if (query.MaxEndDate.HasValue) { whereClauses.Add("EndDate<=@MaxEndDate"); cmd.Parameters.Add(cmd, "@MaxEndDate", DbType.Date).Value = query.MaxEndDate.Value; } if (query.MinStartDate.HasValue) { whereClauses.Add("StartDate>=@MinStartDate"); cmd.Parameters.Add(cmd, "@MinStartDate", DbType.Date).Value = query.MinStartDate.Value; } if (query.MinPremiereDate.HasValue) { whereClauses.Add("PremiereDate>=@MinPremiereDate"); cmd.Parameters.Add(cmd, "@MinPremiereDate", DbType.Date).Value = query.MinPremiereDate.Value; } if (query.MaxStartDate.HasValue) { whereClauses.Add("StartDate<=@MaxStartDate"); cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = query.MaxStartDate.Value; } if (query.IsAiring.HasValue) { if (query.IsAiring.Value) { whereClauses.Add("StartDate<=@MaxStartDate"); cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = DateTime.UtcNow; whereClauses.Add("EndDate>=@MinEndDate"); cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = DateTime.UtcNow; } else { whereClauses.Add("(StartDate>@IsAiringDate OR EndDate < @IsAiringDate)"); cmd.Parameters.Add(cmd, "@IsAiringDate", DbType.Date).Value = DateTime.UtcNow; } } if (!string.IsNullOrWhiteSpace(query.Person)) { whereClauses.Add("Guid in (select ItemId from People where Name=@PersonName)"); cmd.Parameters.Add(cmd, "@PersonName", DbType.String).Value = query.Person; } if (!string.IsNullOrWhiteSpace(query.NameContains)) { whereClauses.Add("Name like @NameContains"); cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%"; } if (query.Genres.Length > 0) { var genres = new List(); var index = 0; foreach (var genre in query.Genres) { genres.Add("Genres like @Genres" + index); cmd.Parameters.Add(cmd, "@Genres" + index, DbType.String).Value = "%" + genre + "%"; index++; } var genreCaluse = "(" + string.Join(" OR ", genres.ToArray()) + ")"; whereClauses.Add(genreCaluse); } if (query.MaxParentalRating.HasValue) { whereClauses.Add("(ParentalRatingValue is NULL OR ParentalRatingValue<=@MaxParentalRating)"); cmd.Parameters.Add(cmd, "@MaxParentalRating", DbType.Int32).Value = query.MaxParentalRating.Value; } if (query.HasParentalRating.HasValue) { if (query.HasParentalRating.Value) { whereClauses.Add("ParentalRatingValue NOT NULL"); } else { whereClauses.Add("ParentalRatingValue IS NULL"); } } if (query.HasDeadParentId.HasValue) { if (query.HasDeadParentId.Value) { whereClauses.Add("ParentId NOT NULL AND ParentId NOT IN (select guid from TypedBaseItems)"); } } if (addPaging) { if (query.StartIndex.HasValue && query.StartIndex.Value > 0) { var pagingWhereText = whereClauses.Count == 0 ? string.Empty : " where " + string.Join(" AND ", whereClauses.ToArray()); var orderBy = GetOrderByText(query); whereClauses.Add(string.Format("guid NOT IN (SELECT guid FROM TypedBaseItems {0}" + orderBy + " LIMIT {1})", pagingWhereText, query.StartIndex.Value.ToString(CultureInfo.InvariantCulture))); } } return whereClauses; } private static readonly Type[] KnownTypes = { typeof(LiveTvProgram), typeof(LiveTvChannel), typeof(LiveTvVideoRecording), typeof(LiveTvAudioRecording), typeof(Series), typeof(Audio), typeof(MusicAlbum), typeof(MusicArtist), typeof(MusicGenre), typeof(MusicVideo), typeof(Movie), typeof(BoxSet), typeof(Episode), typeof(Season), typeof(Series), typeof(Book), typeof(CollectionFolder), typeof(Folder), typeof(Game), typeof(GameGenre), typeof(GameSystem), typeof(Genre), typeof(Person), typeof(Photo), typeof(PhotoAlbum), typeof(Studio), typeof(UserRootFolder), typeof(UserView), typeof(Video), typeof(Year), typeof(Channel) }; private static Dictionary GetTypeMapDictionary() { var dict = new Dictionary(); foreach (var t in KnownTypes) { dict[t.Name] = new[] { t.FullName }; } dict["Recording"] = new[] { typeof(LiveTvAudioRecording).FullName, typeof(LiveTvVideoRecording).FullName }; dict["Program"] = new[] { typeof(LiveTvProgram).FullName }; dict["TvChannel"] = new[] { typeof(LiveTvChannel).FullName }; return dict; } // Not crazy about having this all the way down here, but at least it's in one place readonly Dictionary _types = GetTypeMapDictionary(); private IEnumerable MapIncludeItemTypes(string value) { string[] result; if (_types.TryGetValue(value, out result)) { return result; } return new[] { value }; } public async Task DeleteItem(Guid id, CancellationToken cancellationToken) { if (id == Guid.Empty) { throw new ArgumentNullException("id"); } CheckDisposed(); await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false); IDbTransaction transaction = null; try { transaction = _connection.BeginTransaction(); // First delete children _deleteChildrenCommand.GetParameter(0).Value = id; _deleteChildrenCommand.Transaction = transaction; _deleteChildrenCommand.ExecuteNonQuery(); // Delete people _deletePeopleCommand.GetParameter(0).Value = id; _deletePeopleCommand.Transaction = transaction; _deletePeopleCommand.ExecuteNonQuery(); // Delete chapters _deleteChaptersCommand.GetParameter(0).Value = id; _deleteChaptersCommand.Transaction = transaction; _deleteChaptersCommand.ExecuteNonQuery(); // Delete media streams _deleteStreamsCommand.GetParameter(0).Value = id; _deleteStreamsCommand.Transaction = transaction; _deleteStreamsCommand.ExecuteNonQuery(); // Delete the item _deleteItemCommand.GetParameter(0).Value = id; _deleteItemCommand.Transaction = transaction; _deleteItemCommand.ExecuteNonQuery(); transaction.Commit(); } catch (OperationCanceledException) { if (transaction != null) { transaction.Rollback(); } throw; } catch (Exception e) { _logger.ErrorException("Failed to save children:", e); if (transaction != null) { transaction.Rollback(); } throw; } finally { if (transaction != null) { transaction.Dispose(); } _writeLock.Release(); } } public async Task SaveChildren(Guid parentId, IEnumerable children, CancellationToken cancellationToken) { if (parentId == Guid.Empty) { throw new ArgumentNullException("parentId"); } if (children == null) { throw new ArgumentNullException("children"); } CheckDisposed(); await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false); IDbTransaction transaction = null; try { transaction = _connection.BeginTransaction(); // First delete _deleteChildrenCommand.GetParameter(0).Value = parentId; _deleteChildrenCommand.Transaction = transaction; _deleteChildrenCommand.ExecuteNonQuery(); foreach (var id in children) { cancellationToken.ThrowIfCancellationRequested(); _saveChildrenCommand.GetParameter(0).Value = parentId; _saveChildrenCommand.GetParameter(1).Value = id; _saveChildrenCommand.Transaction = transaction; _saveChildrenCommand.ExecuteNonQuery(); } transaction.Commit(); } catch (OperationCanceledException) { if (transaction != null) { transaction.Rollback(); } throw; } catch (Exception e) { _logger.ErrorException("Failed to save children:", e); if (transaction != null) { transaction.Rollback(); } throw; } finally { if (transaction != null) { transaction.Dispose(); } _writeLock.Release(); } } public List GetPeopleNames(InternalPeopleQuery query) { if (query == null) { throw new ArgumentNullException("query"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select Distinct Name from People"; var whereClauses = GetPeopleWhereClauses(query, cmd); if (whereClauses.Count > 0) { cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray()); } cmd.CommandText += " order by ListOrder"; var list = new List(); using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult)) { while (reader.Read()) { list.Add(reader.GetString(0)); } } return list; } } public List GetPeople(InternalPeopleQuery query) { if (query == null) { throw new ArgumentNullException("query"); } CheckDisposed(); using (var cmd = _connection.CreateCommand()) { cmd.CommandText = "select ItemId, Name, Role, PersonType, SortOrder from People"; var whereClauses = GetPeopleWhereClauses(query, cmd); if (whereClauses.Count > 0) { cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray()); } cmd.CommandText += " order by ListOrder"; var list = new List(); using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult)) { while (reader.Read()) { list.Add(GetPerson(reader)); } } return list; } } private List GetPeopleWhereClauses(InternalPeopleQuery query, IDbCommand cmd) { var whereClauses = new List(); if (query.ItemId != Guid.Empty) { whereClauses.Add("ItemId=@ItemId"); cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = query.ItemId; } if (query.AppearsInItemId != Guid.Empty) { whereClauses.Add("Name in (Select Name from People where ItemId=@AppearsInItemId)"); cmd.Parameters.Add(cmd, "@AppearsInItemId", DbType.Guid).Value = query.AppearsInItemId; } if (query.PersonTypes.Count == 1) { whereClauses.Add("PersonType=@PersonType"); cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.PersonTypes[0]; } if (query.PersonTypes.Count > 1) { var val = string.Join(",", query.PersonTypes.Select(i => "'" + i + "'").ToArray()); whereClauses.Add("PersonType in (" + val + ")"); } if (query.ExcludePersonTypes.Count == 1) { whereClauses.Add("PersonType<>@PersonType"); cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.ExcludePersonTypes[0]; } if (query.ExcludePersonTypes.Count > 1) { var val = string.Join(",", query.ExcludePersonTypes.Select(i => "'" + i + "'").ToArray()); whereClauses.Add("PersonType not in (" + val + ")"); } if (query.MaxListOrder.HasValue) { whereClauses.Add("ListOrder<=@MaxListOrder"); cmd.Parameters.Add(cmd, "@MaxListOrder", DbType.Int32).Value = query.MaxListOrder.Value; } if (!string.IsNullOrWhiteSpace(query.NameContains)) { whereClauses.Add("Name like @NameContains"); cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%"; } return whereClauses; } public async Task UpdatePeople(Guid itemId, List people) { if (itemId == Guid.Empty) { throw new ArgumentNullException("itemId"); } if (people == null) { throw new ArgumentNullException("people"); } CheckDisposed(); var cancellationToken = CancellationToken.None; await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false); IDbTransaction transaction = null; try { transaction = _connection.BeginTransaction(); // First delete _deletePeopleCommand.GetParameter(0).Value = itemId; _deletePeopleCommand.Transaction = transaction; _deletePeopleCommand.ExecuteNonQuery(); var listIndex = 0; foreach (var person in people) { cancellationToken.ThrowIfCancellationRequested(); _savePersonCommand.GetParameter(0).Value = itemId; _savePersonCommand.GetParameter(1).Value = person.Name; _savePersonCommand.GetParameter(2).Value = person.Role; _savePersonCommand.GetParameter(3).Value = person.Type; _savePersonCommand.GetParameter(4).Value = person.SortOrder; _savePersonCommand.GetParameter(5).Value = listIndex; _savePersonCommand.Transaction = transaction; _savePersonCommand.ExecuteNonQuery(); listIndex++; } transaction.Commit(); } catch (OperationCanceledException) { if (transaction != null) { transaction.Rollback(); } throw; } catch (Exception e) { _logger.ErrorException("Failed to save people:", e); if (transaction != null) { transaction.Rollback(); } throw; } finally { if (transaction != null) { transaction.Dispose(); } _writeLock.Release(); } } private PersonInfo GetPerson(IDataReader reader) { var item = new PersonInfo(); item.ItemId = reader.GetGuid(0); item.Name = reader.GetString(1); if (!reader.IsDBNull(2)) { item.Role = reader.GetString(2); } if (!reader.IsDBNull(3)) { item.Type = reader.GetString(3); } if (!reader.IsDBNull(4)) { item.SortOrder = reader.GetInt32(4); } return item; } public IEnumerable GetMediaStreams(MediaStreamQuery query) { CheckDisposed(); if (query == null) { throw new ArgumentNullException("query"); } using (var cmd = _connection.CreateCommand()) { var cmdText = "select " + string.Join(",", _mediaStreamSaveColumns) + " from mediastreams where"; cmdText += " ItemId=@ItemId"; cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = query.ItemId; if (query.Type.HasValue) { cmdText += " AND StreamType=@StreamType"; cmd.Parameters.Add(cmd, "@StreamType", DbType.String).Value = query.Type.Value.ToString(); } if (query.Index.HasValue) { cmdText += " AND StreamIndex=@StreamIndex"; cmd.Parameters.Add(cmd, "@StreamIndex", DbType.Int32).Value = query.Index.Value; } cmdText += " order by StreamIndex ASC"; cmd.CommandText = cmdText; using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult)) { while (reader.Read()) { yield return GetMediaStream(reader); } } } } public async Task SaveMediaStreams(Guid id, IEnumerable streams, CancellationToken cancellationToken) { CheckDisposed(); if (id == Guid.Empty) { throw new ArgumentNullException("id"); } if (streams == null) { throw new ArgumentNullException("streams"); } cancellationToken.ThrowIfCancellationRequested(); await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false); IDbTransaction transaction = null; try { transaction = _connection.BeginTransaction(); // First delete chapters _deleteStreamsCommand.GetParameter(0).Value = id; _deleteStreamsCommand.Transaction = transaction; _deleteStreamsCommand.ExecuteNonQuery(); foreach (var stream in streams) { cancellationToken.ThrowIfCancellationRequested(); var index = 0; _saveStreamCommand.GetParameter(index++).Value = id; _saveStreamCommand.GetParameter(index++).Value = stream.Index; _saveStreamCommand.GetParameter(index++).Value = stream.Type.ToString(); _saveStreamCommand.GetParameter(index++).Value = stream.Codec; _saveStreamCommand.GetParameter(index++).Value = stream.Language; _saveStreamCommand.GetParameter(index++).Value = stream.ChannelLayout; _saveStreamCommand.GetParameter(index++).Value = stream.Profile; _saveStreamCommand.GetParameter(index++).Value = stream.AspectRatio; _saveStreamCommand.GetParameter(index++).Value = stream.Path; _saveStreamCommand.GetParameter(index++).Value = stream.IsInterlaced; _saveStreamCommand.GetParameter(index++).Value = stream.BitRate; _saveStreamCommand.GetParameter(index++).Value = stream.Channels; _saveStreamCommand.GetParameter(index++).Value = stream.SampleRate; _saveStreamCommand.GetParameter(index++).Value = stream.IsDefault; _saveStreamCommand.GetParameter(index++).Value = stream.IsForced; _saveStreamCommand.GetParameter(index++).Value = stream.IsExternal; _saveStreamCommand.GetParameter(index++).Value = stream.Width; _saveStreamCommand.GetParameter(index++).Value = stream.Height; _saveStreamCommand.GetParameter(index++).Value = stream.AverageFrameRate; _saveStreamCommand.GetParameter(index++).Value = stream.RealFrameRate; _saveStreamCommand.GetParameter(index++).Value = stream.Level; _saveStreamCommand.GetParameter(index++).Value = stream.PixelFormat; _saveStreamCommand.GetParameter(index++).Value = stream.BitDepth; _saveStreamCommand.GetParameter(index++).Value = stream.IsAnamorphic; _saveStreamCommand.GetParameter(index++).Value = stream.RefFrames; _saveStreamCommand.GetParameter(index++).Value = stream.IsCabac; if (stream.KeyFrames == null || stream.KeyFrames.Count == 0) { _saveStreamCommand.GetParameter(index++).Value = null; } else { _saveStreamCommand.GetParameter(index++).Value = string.Join(",", stream.KeyFrames.Select(i => i.ToString(CultureInfo.InvariantCulture)).ToArray()); } _saveStreamCommand.GetParameter(index++).Value = stream.CodecTag; _saveStreamCommand.Transaction = transaction; _saveStreamCommand.ExecuteNonQuery(); } transaction.Commit(); } catch (OperationCanceledException) { if (transaction != null) { transaction.Rollback(); } throw; } catch (Exception e) { _logger.ErrorException("Failed to save media streams:", e); if (transaction != null) { transaction.Rollback(); } throw; } finally { if (transaction != null) { transaction.Dispose(); } _writeLock.Release(); } } /// /// Gets the chapter. /// /// The reader. /// ChapterInfo. private MediaStream GetMediaStream(IDataReader reader) { var item = new MediaStream { Index = reader.GetInt32(1) }; item.Type = (MediaStreamType)Enum.Parse(typeof(MediaStreamType), reader.GetString(2), true); if (!reader.IsDBNull(3)) { item.Codec = reader.GetString(3); } if (!reader.IsDBNull(4)) { item.Language = reader.GetString(4); } if (!reader.IsDBNull(5)) { item.ChannelLayout = reader.GetString(5); } if (!reader.IsDBNull(6)) { item.Profile = reader.GetString(6); } if (!reader.IsDBNull(7)) { item.AspectRatio = reader.GetString(7); } if (!reader.IsDBNull(8)) { item.Path = reader.GetString(8); } item.IsInterlaced = reader.GetBoolean(9); if (!reader.IsDBNull(10)) { item.BitRate = reader.GetInt32(10); } if (!reader.IsDBNull(11)) { item.Channels = reader.GetInt32(11); } if (!reader.IsDBNull(12)) { item.SampleRate = reader.GetInt32(12); } item.IsDefault = reader.GetBoolean(13); item.IsForced = reader.GetBoolean(14); item.IsExternal = reader.GetBoolean(15); if (!reader.IsDBNull(16)) { item.Width = reader.GetInt32(16); } if (!reader.IsDBNull(17)) { item.Height = reader.GetInt32(17); } if (!reader.IsDBNull(18)) { item.AverageFrameRate = reader.GetFloat(18); } if (!reader.IsDBNull(19)) { item.RealFrameRate = reader.GetFloat(19); } if (!reader.IsDBNull(20)) { item.Level = reader.GetFloat(20); } if (!reader.IsDBNull(21)) { item.PixelFormat = reader.GetString(21); } if (!reader.IsDBNull(22)) { item.BitDepth = reader.GetInt32(22); } if (!reader.IsDBNull(23)) { item.IsAnamorphic = reader.GetBoolean(23); } if (!reader.IsDBNull(24)) { item.RefFrames = reader.GetInt32(24); } if (!reader.IsDBNull(25)) { item.IsCabac = reader.GetBoolean(25); } if (!reader.IsDBNull(26)) { var frames = reader.GetString(26); if (!string.IsNullOrWhiteSpace(frames)) { item.KeyFrames = frames.Split(',').Select(i => int.Parse(i, CultureInfo.InvariantCulture)).ToList(); } } if (!reader.IsDBNull(27)) { item.CodecTag = reader.GetString(27); } return item; } } }