SqliteItemRepository.cs 64 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809
  1. using MediaBrowser.Common.Configuration;
  2. using MediaBrowser.Controller.Entities;
  3. using MediaBrowser.Controller.Entities.Audio;
  4. using MediaBrowser.Controller.Entities.Movies;
  5. using MediaBrowser.Controller.Entities.TV;
  6. using MediaBrowser.Controller.LiveTv;
  7. using MediaBrowser.Controller.Persistence;
  8. using MediaBrowser.Model.Entities;
  9. using MediaBrowser.Model.Logging;
  10. using MediaBrowser.Model.Querying;
  11. using MediaBrowser.Model.Serialization;
  12. using System;
  13. using System.Collections.Generic;
  14. using System.Data;
  15. using System.Globalization;
  16. using System.IO;
  17. using System.Linq;
  18. using System.Runtime.Serialization;
  19. using System.Threading;
  20. using System.Threading.Tasks;
  21. namespace MediaBrowser.Server.Implementations.Persistence
  22. {
  23. /// <summary>
  24. /// Class SQLiteItemRepository
  25. /// </summary>
  26. public class SqliteItemRepository : IItemRepository
  27. {
  28. private IDbConnection _connection;
  29. private readonly ILogger _logger;
  30. private readonly TypeMapper _typeMapper = new TypeMapper();
  31. /// <summary>
  32. /// Gets the name of the repository
  33. /// </summary>
  34. /// <value>The name.</value>
  35. public string Name
  36. {
  37. get
  38. {
  39. return "SQLite";
  40. }
  41. }
  42. /// <summary>
  43. /// Gets the json serializer.
  44. /// </summary>
  45. /// <value>The json serializer.</value>
  46. private readonly IJsonSerializer _jsonSerializer;
  47. /// <summary>
  48. /// The _app paths
  49. /// </summary>
  50. private readonly IApplicationPaths _appPaths;
  51. /// <summary>
  52. /// The _save item command
  53. /// </summary>
  54. private IDbCommand _saveItemCommand;
  55. private readonly string _criticReviewsPath;
  56. private SqliteChapterRepository _chapterRepository;
  57. private SqliteMediaStreamsRepository _mediaStreamsRepository;
  58. private IDbCommand _deleteChildrenCommand;
  59. private IDbCommand _saveChildrenCommand;
  60. private IDbCommand _deleteItemCommand;
  61. private IDbCommand _deletePeopleCommand;
  62. private IDbCommand _savePersonCommand;
  63. private const int LatestSchemaVersion = 11;
  64. /// <summary>
  65. /// Initializes a new instance of the <see cref="SqliteItemRepository"/> class.
  66. /// </summary>
  67. /// <param name="appPaths">The app paths.</param>
  68. /// <param name="jsonSerializer">The json serializer.</param>
  69. /// <param name="logManager">The log manager.</param>
  70. /// <exception cref="System.ArgumentNullException">
  71. /// appPaths
  72. /// or
  73. /// jsonSerializer
  74. /// </exception>
  75. public SqliteItemRepository(IApplicationPaths appPaths, IJsonSerializer jsonSerializer, ILogManager logManager)
  76. {
  77. if (appPaths == null)
  78. {
  79. throw new ArgumentNullException("appPaths");
  80. }
  81. if (jsonSerializer == null)
  82. {
  83. throw new ArgumentNullException("jsonSerializer");
  84. }
  85. _appPaths = appPaths;
  86. _jsonSerializer = jsonSerializer;
  87. _criticReviewsPath = Path.Combine(_appPaths.DataPath, "critic-reviews");
  88. _logger = logManager.GetLogger(GetType().Name);
  89. var chapterDbFile = Path.Combine(_appPaths.DataPath, "chapters.db");
  90. var chapterConnection = SqliteExtensions.ConnectToDb(chapterDbFile, _logger).Result;
  91. _chapterRepository = new SqliteChapterRepository(chapterConnection, logManager);
  92. var mediaStreamsDbFile = Path.Combine(_appPaths.DataPath, "mediainfo.db");
  93. var mediaStreamsConnection = SqliteExtensions.ConnectToDb(mediaStreamsDbFile, _logger).Result;
  94. _mediaStreamsRepository = new SqliteMediaStreamsRepository(mediaStreamsConnection, logManager);
  95. }
  96. /// <summary>
  97. /// Opens the connection to the database
  98. /// </summary>
  99. /// <returns>Task.</returns>
  100. public async Task Initialize()
  101. {
  102. var dbFile = Path.Combine(_appPaths.DataPath, "library.db");
  103. _connection = await SqliteExtensions.ConnectToDb(dbFile, _logger).ConfigureAwait(false);
  104. string[] queries = {
  105. "create table if not exists TypedBaseItems (guid GUID primary key, type TEXT, data BLOB)",
  106. "create index if not exists idx_TypedBaseItems on TypedBaseItems(guid)",
  107. "create table if not exists ChildrenIds (ParentId GUID, ItemId GUID, PRIMARY KEY (ParentId, ItemId))",
  108. "create index if not exists idx_ChildrenIds on ChildrenIds(ParentId,ItemId)",
  109. "create table if not exists People (ItemId GUID, Name TEXT NOT NULL, Role TEXT, PersonType TEXT, SortOrder int, ListOrder int)",
  110. //pragmas
  111. "pragma temp_store = memory",
  112. "pragma shrink_memory"
  113. };
  114. _connection.RunQueries(queries, _logger);
  115. _connection.AddColumn(_logger, "TypedBaseItems", "Path", "Text");
  116. _connection.AddColumn(_logger, "TypedBaseItems", "StartDate", "DATETIME");
  117. _connection.AddColumn(_logger, "TypedBaseItems", "EndDate", "DATETIME");
  118. _connection.AddColumn(_logger, "TypedBaseItems", "ChannelId", "Text");
  119. _connection.AddColumn(_logger, "TypedBaseItems", "IsMovie", "BIT");
  120. _connection.AddColumn(_logger, "TypedBaseItems", "IsSports", "BIT");
  121. _connection.AddColumn(_logger, "TypedBaseItems", "IsKids", "BIT");
  122. _connection.AddColumn(_logger, "TypedBaseItems", "CommunityRating", "Float");
  123. _connection.AddColumn(_logger, "TypedBaseItems", "CustomRating", "Text");
  124. _connection.AddColumn(_logger, "TypedBaseItems", "IndexNumber", "INT");
  125. _connection.AddColumn(_logger, "TypedBaseItems", "IsLocked", "BIT");
  126. _connection.AddColumn(_logger, "TypedBaseItems", "Name", "Text");
  127. _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRating", "Text");
  128. _connection.AddColumn(_logger, "TypedBaseItems", "MediaType", "Text");
  129. _connection.AddColumn(_logger, "TypedBaseItems", "Overview", "Text");
  130. _connection.AddColumn(_logger, "TypedBaseItems", "ParentIndexNumber", "INT");
  131. _connection.AddColumn(_logger, "TypedBaseItems", "PremiereDate", "DATETIME");
  132. _connection.AddColumn(_logger, "TypedBaseItems", "ProductionYear", "INT");
  133. _connection.AddColumn(_logger, "TypedBaseItems", "ParentId", "GUID");
  134. _connection.AddColumn(_logger, "TypedBaseItems", "Genres", "Text");
  135. _connection.AddColumn(_logger, "TypedBaseItems", "ParentalRatingValue", "INT");
  136. _connection.AddColumn(_logger, "TypedBaseItems", "SchemaVersion", "INT");
  137. _connection.AddColumn(_logger, "TypedBaseItems", "SortName", "Text");
  138. _connection.AddColumn(_logger, "TypedBaseItems", "RunTimeTicks", "BIGINT");
  139. _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRatingDescription", "Text");
  140. _connection.AddColumn(_logger, "TypedBaseItems", "HomePageUrl", "Text");
  141. _connection.AddColumn(_logger, "TypedBaseItems", "VoteCount", "INT");
  142. _connection.AddColumn(_logger, "TypedBaseItems", "DisplayMediaType", "Text");
  143. _connection.AddColumn(_logger, "TypedBaseItems", "DateCreated", "DATETIME");
  144. _connection.AddColumn(_logger, "TypedBaseItems", "DateModified", "DATETIME");
  145. _connection.AddColumn(_logger, "TypedBaseItems", "ForcedSortName", "Text");
  146. _connection.AddColumn(_logger, "TypedBaseItems", "IsOffline", "BIT");
  147. _connection.AddColumn(_logger, "TypedBaseItems", "LocationType", "Text");
  148. _connection.AddColumn(_logger, "TypedBaseItems", "IsSeries", "BIT");
  149. _connection.AddColumn(_logger, "TypedBaseItems", "IsLive", "BIT");
  150. _connection.AddColumn(_logger, "TypedBaseItems", "IsNews", "BIT");
  151. _connection.AddColumn(_logger, "TypedBaseItems", "IsPremiere", "BIT");
  152. _connection.AddColumn(_logger, "TypedBaseItems", "EpisodeTitle", "Text");
  153. _connection.AddColumn(_logger, "TypedBaseItems", "IsRepeat", "BIT");
  154. _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataLanguage", "Text");
  155. _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataCountryCode", "Text");
  156. PrepareStatements();
  157. _mediaStreamsRepository.Initialize();
  158. _chapterRepository.Initialize();
  159. }
  160. /// <summary>
  161. /// The _write lock
  162. /// </summary>
  163. private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
  164. private readonly string[] _retriveItemColumns =
  165. {
  166. "type",
  167. "data",
  168. "StartDate",
  169. "EndDate",
  170. "IsOffline",
  171. "ChannelId",
  172. "IsMovie",
  173. "IsSports",
  174. "IsKids",
  175. "IsSeries",
  176. "IsLive",
  177. "IsNews",
  178. "IsPremiere",
  179. "EpisodeTitle",
  180. "IsRepeat",
  181. "CommunityRating",
  182. "CustomRating",
  183. "IndexNumber",
  184. "IsLocked",
  185. "PreferredMetadataLanguage",
  186. "PreferredMetadataCountryCode"
  187. };
  188. /// <summary>
  189. /// Prepares the statements.
  190. /// </summary>
  191. private void PrepareStatements()
  192. {
  193. var saveColumns = new List<string>
  194. {
  195. "guid",
  196. "type",
  197. "data",
  198. "Path",
  199. "StartDate",
  200. "EndDate",
  201. "ChannelId",
  202. "IsKids",
  203. "IsMovie",
  204. "IsSports",
  205. "IsSeries",
  206. "IsLive",
  207. "IsNews",
  208. "IsPremiere",
  209. "EpisodeTitle",
  210. "IsRepeat",
  211. "CommunityRating",
  212. "CustomRating",
  213. "IndexNumber",
  214. "IsLocked",
  215. "Name",
  216. "OfficialRating",
  217. "MediaType",
  218. "Overview",
  219. "ParentIndexNumber",
  220. "PremiereDate",
  221. "ProductionYear",
  222. "ParentId",
  223. "Genres",
  224. "ParentalRatingValue",
  225. "SchemaVersion",
  226. "SortName",
  227. "RunTimeTicks",
  228. "OfficialRatingDescription",
  229. "HomePageUrl",
  230. "VoteCount",
  231. "DisplayMediaType",
  232. "DateCreated",
  233. "DateModified",
  234. "ForcedSortName",
  235. "IsOffline",
  236. "LocationType",
  237. "PreferredMetadataLanguage",
  238. "PreferredMetadataCountryCode"
  239. };
  240. _saveItemCommand = _connection.CreateCommand();
  241. _saveItemCommand.CommandText = "replace into TypedBaseItems (" + string.Join(",", saveColumns.ToArray()) + ") values (";
  242. for (var i = 1; i <= saveColumns.Count; i++)
  243. {
  244. if (i > 1)
  245. {
  246. _saveItemCommand.CommandText += ",";
  247. }
  248. _saveItemCommand.CommandText += "@" + i.ToString(CultureInfo.InvariantCulture);
  249. _saveItemCommand.Parameters.Add(_saveItemCommand, "@" + i.ToString(CultureInfo.InvariantCulture));
  250. }
  251. _saveItemCommand.CommandText += ")";
  252. _deleteChildrenCommand = _connection.CreateCommand();
  253. _deleteChildrenCommand.CommandText = "delete from ChildrenIds where ParentId=@ParentId";
  254. _deleteChildrenCommand.Parameters.Add(_deleteChildrenCommand, "@ParentId");
  255. _deleteItemCommand = _connection.CreateCommand();
  256. _deleteItemCommand.CommandText = "delete from TypedBaseItems where guid=@Id";
  257. _deleteItemCommand.Parameters.Add(_deleteItemCommand, "@Id");
  258. _saveChildrenCommand = _connection.CreateCommand();
  259. _saveChildrenCommand.CommandText = "replace into ChildrenIds (ParentId, ItemId) values (@ParentId, @ItemId)";
  260. _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ParentId");
  261. _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ItemId");
  262. _deletePeopleCommand = _connection.CreateCommand();
  263. _deletePeopleCommand.CommandText = "delete from People where ItemId=@Id";
  264. _deletePeopleCommand.Parameters.Add(_deletePeopleCommand, "@Id");
  265. _savePersonCommand = _connection.CreateCommand();
  266. _savePersonCommand.CommandText = "insert into People (ItemId, Name, Role, PersonType, SortOrder, ListOrder) values (@ItemId, @Name, @Role, @PersonType, @SortOrder, @ListOrder)";
  267. _savePersonCommand.Parameters.Add(_savePersonCommand, "@ItemId");
  268. _savePersonCommand.Parameters.Add(_savePersonCommand, "@Name");
  269. _savePersonCommand.Parameters.Add(_savePersonCommand, "@Role");
  270. _savePersonCommand.Parameters.Add(_savePersonCommand, "@PersonType");
  271. _savePersonCommand.Parameters.Add(_savePersonCommand, "@SortOrder");
  272. _savePersonCommand.Parameters.Add(_savePersonCommand, "@ListOrder");
  273. }
  274. /// <summary>
  275. /// Save a standard item in the repo
  276. /// </summary>
  277. /// <param name="item">The item.</param>
  278. /// <param name="cancellationToken">The cancellation token.</param>
  279. /// <returns>Task.</returns>
  280. /// <exception cref="System.ArgumentNullException">item</exception>
  281. public Task SaveItem(BaseItem item, CancellationToken cancellationToken)
  282. {
  283. if (item == null)
  284. {
  285. throw new ArgumentNullException("item");
  286. }
  287. return SaveItems(new[] { item }, cancellationToken);
  288. }
  289. /// <summary>
  290. /// Saves the items.
  291. /// </summary>
  292. /// <param name="items">The items.</param>
  293. /// <param name="cancellationToken">The cancellation token.</param>
  294. /// <returns>Task.</returns>
  295. /// <exception cref="System.ArgumentNullException">
  296. /// items
  297. /// or
  298. /// cancellationToken
  299. /// </exception>
  300. public async Task SaveItems(IEnumerable<BaseItem> items, CancellationToken cancellationToken)
  301. {
  302. if (items == null)
  303. {
  304. throw new ArgumentNullException("items");
  305. }
  306. cancellationToken.ThrowIfCancellationRequested();
  307. CheckDisposed();
  308. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  309. IDbTransaction transaction = null;
  310. try
  311. {
  312. transaction = _connection.BeginTransaction();
  313. foreach (var item in items)
  314. {
  315. cancellationToken.ThrowIfCancellationRequested();
  316. var index = 0;
  317. _saveItemCommand.GetParameter(index++).Value = item.Id;
  318. _saveItemCommand.GetParameter(index++).Value = item.GetType().FullName;
  319. _saveItemCommand.GetParameter(index++).Value = _jsonSerializer.SerializeToBytes(item);
  320. _saveItemCommand.GetParameter(index++).Value = item.Path;
  321. var hasStartDate = item as IHasStartDate;
  322. if (hasStartDate != null)
  323. {
  324. _saveItemCommand.GetParameter(index++).Value = hasStartDate.StartDate;
  325. }
  326. else
  327. {
  328. _saveItemCommand.GetParameter(index++).Value = null;
  329. }
  330. _saveItemCommand.GetParameter(index++).Value = item.EndDate;
  331. _saveItemCommand.GetParameter(index++).Value = item.ChannelId;
  332. var hasProgramAttributes = item as IHasProgramAttributes;
  333. if (hasProgramAttributes != null)
  334. {
  335. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsKids;
  336. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsMovie;
  337. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSports;
  338. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSeries;
  339. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsLive;
  340. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsNews;
  341. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsPremiere;
  342. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.EpisodeTitle;
  343. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsRepeat;
  344. }
  345. else
  346. {
  347. _saveItemCommand.GetParameter(index++).Value = null;
  348. _saveItemCommand.GetParameter(index++).Value = null;
  349. _saveItemCommand.GetParameter(index++).Value = null;
  350. _saveItemCommand.GetParameter(index++).Value = null;
  351. _saveItemCommand.GetParameter(index++).Value = null;
  352. _saveItemCommand.GetParameter(index++).Value = null;
  353. _saveItemCommand.GetParameter(index++).Value = null;
  354. _saveItemCommand.GetParameter(index++).Value = null;
  355. _saveItemCommand.GetParameter(index++).Value = null;
  356. }
  357. _saveItemCommand.GetParameter(index++).Value = item.CommunityRating;
  358. _saveItemCommand.GetParameter(index++).Value = item.CustomRating;
  359. _saveItemCommand.GetParameter(index++).Value = item.IndexNumber;
  360. _saveItemCommand.GetParameter(index++).Value = item.IsLocked;
  361. _saveItemCommand.GetParameter(index++).Value = item.Name;
  362. _saveItemCommand.GetParameter(index++).Value = item.OfficialRating;
  363. _saveItemCommand.GetParameter(index++).Value = item.MediaType;
  364. _saveItemCommand.GetParameter(index++).Value = item.Overview;
  365. _saveItemCommand.GetParameter(index++).Value = item.ParentIndexNumber;
  366. _saveItemCommand.GetParameter(index++).Value = item.PremiereDate;
  367. _saveItemCommand.GetParameter(index++).Value = item.ProductionYear;
  368. if (item.ParentId == Guid.Empty)
  369. {
  370. _saveItemCommand.GetParameter(index++).Value = null;
  371. }
  372. else
  373. {
  374. _saveItemCommand.GetParameter(index++).Value = item.ParentId;
  375. }
  376. _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.Genres.ToArray());
  377. _saveItemCommand.GetParameter(index++).Value = item.GetParentalRatingValue();
  378. _saveItemCommand.GetParameter(index++).Value = LatestSchemaVersion;
  379. _saveItemCommand.GetParameter(index++).Value = item.SortName;
  380. _saveItemCommand.GetParameter(index++).Value = item.RunTimeTicks;
  381. _saveItemCommand.GetParameter(index++).Value = item.OfficialRatingDescription;
  382. _saveItemCommand.GetParameter(index++).Value = item.HomePageUrl;
  383. _saveItemCommand.GetParameter(index++).Value = item.VoteCount;
  384. _saveItemCommand.GetParameter(index++).Value = item.DisplayMediaType;
  385. _saveItemCommand.GetParameter(index++).Value = item.DateCreated;
  386. _saveItemCommand.GetParameter(index++).Value = item.DateModified;
  387. _saveItemCommand.GetParameter(index++).Value = item.ForcedSortName;
  388. _saveItemCommand.GetParameter(index++).Value = item.IsOffline;
  389. _saveItemCommand.GetParameter(index++).Value = item.LocationType.ToString();
  390. _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataLanguage;
  391. _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataCountryCode;
  392. _saveItemCommand.Transaction = transaction;
  393. _saveItemCommand.ExecuteNonQuery();
  394. }
  395. transaction.Commit();
  396. }
  397. catch (OperationCanceledException)
  398. {
  399. if (transaction != null)
  400. {
  401. transaction.Rollback();
  402. }
  403. throw;
  404. }
  405. catch (Exception e)
  406. {
  407. _logger.ErrorException("Failed to save items:", e);
  408. if (transaction != null)
  409. {
  410. transaction.Rollback();
  411. }
  412. throw;
  413. }
  414. finally
  415. {
  416. if (transaction != null)
  417. {
  418. transaction.Dispose();
  419. }
  420. _writeLock.Release();
  421. }
  422. }
  423. /// <summary>
  424. /// Internal retrieve from items or users table
  425. /// </summary>
  426. /// <param name="id">The id.</param>
  427. /// <returns>BaseItem.</returns>
  428. /// <exception cref="System.ArgumentNullException">id</exception>
  429. /// <exception cref="System.ArgumentException"></exception>
  430. public BaseItem RetrieveItem(Guid id)
  431. {
  432. if (id == Guid.Empty)
  433. {
  434. throw new ArgumentNullException("id");
  435. }
  436. CheckDisposed();
  437. using (var cmd = _connection.CreateCommand())
  438. {
  439. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid = @guid";
  440. cmd.Parameters.Add(cmd, "@guid", DbType.Guid).Value = id;
  441. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
  442. {
  443. if (reader.Read())
  444. {
  445. return GetItem(reader);
  446. }
  447. }
  448. return null;
  449. }
  450. }
  451. private BaseItem GetItem(IDataReader reader)
  452. {
  453. var typeString = reader.GetString(0);
  454. var type = _typeMapper.GetType(typeString);
  455. if (type == null)
  456. {
  457. _logger.Debug("Unknown type {0}", typeString);
  458. return null;
  459. }
  460. BaseItem item;
  461. using (var stream = reader.GetMemoryStream(1))
  462. {
  463. try
  464. {
  465. item = _jsonSerializer.DeserializeFromStream(stream, type) as BaseItem;
  466. if (item == null)
  467. {
  468. return null;
  469. }
  470. }
  471. catch (SerializationException ex)
  472. {
  473. _logger.ErrorException("Error deserializing item", ex);
  474. return null;
  475. }
  476. }
  477. if (!reader.IsDBNull(2))
  478. {
  479. var hasStartDate = item as IHasStartDate;
  480. if (hasStartDate != null)
  481. {
  482. hasStartDate.StartDate = reader.GetDateTime(2).ToUniversalTime();
  483. }
  484. }
  485. if (!reader.IsDBNull(3))
  486. {
  487. item.EndDate = reader.GetDateTime(3).ToUniversalTime();
  488. }
  489. if (!reader.IsDBNull(4))
  490. {
  491. item.IsOffline = reader.GetBoolean(4);
  492. }
  493. if (!reader.IsDBNull(5))
  494. {
  495. item.ChannelId = reader.GetString(5);
  496. }
  497. var hasProgramAttributes = item as IHasProgramAttributes;
  498. if (hasProgramAttributes != null)
  499. {
  500. if (!reader.IsDBNull(6))
  501. {
  502. hasProgramAttributes.IsMovie = reader.GetBoolean(6);
  503. }
  504. if (!reader.IsDBNull(7))
  505. {
  506. hasProgramAttributes.IsSports = reader.GetBoolean(7);
  507. }
  508. if (!reader.IsDBNull(8))
  509. {
  510. hasProgramAttributes.IsKids = reader.GetBoolean(8);
  511. }
  512. if (!reader.IsDBNull(9))
  513. {
  514. hasProgramAttributes.IsSeries = reader.GetBoolean(9);
  515. }
  516. if (!reader.IsDBNull(10))
  517. {
  518. hasProgramAttributes.IsLive = reader.GetBoolean(10);
  519. }
  520. if (!reader.IsDBNull(11))
  521. {
  522. hasProgramAttributes.IsNews = reader.GetBoolean(11);
  523. }
  524. if (!reader.IsDBNull(12))
  525. {
  526. hasProgramAttributes.IsPremiere = reader.GetBoolean(12);
  527. }
  528. if (!reader.IsDBNull(13))
  529. {
  530. hasProgramAttributes.EpisodeTitle = reader.GetString(13);
  531. }
  532. if (!reader.IsDBNull(14))
  533. {
  534. hasProgramAttributes.IsRepeat = reader.GetBoolean(14);
  535. }
  536. }
  537. if (!reader.IsDBNull(15))
  538. {
  539. item.CommunityRating = reader.GetFloat(15);
  540. }
  541. if (!reader.IsDBNull(16))
  542. {
  543. item.CustomRating = reader.GetString(16);
  544. }
  545. if (!reader.IsDBNull(17))
  546. {
  547. item.IndexNumber = reader.GetInt32(17);
  548. }
  549. if (!reader.IsDBNull(18))
  550. {
  551. item.IsLocked = reader.GetBoolean(18);
  552. }
  553. if (!reader.IsDBNull(19))
  554. {
  555. item.PreferredMetadataLanguage = reader.GetString(19);
  556. }
  557. if (!reader.IsDBNull(20))
  558. {
  559. item.PreferredMetadataCountryCode = reader.GetString(20);
  560. }
  561. return item;
  562. }
  563. /// <summary>
  564. /// Gets the critic reviews.
  565. /// </summary>
  566. /// <param name="itemId">The item id.</param>
  567. /// <returns>Task{IEnumerable{ItemReview}}.</returns>
  568. public IEnumerable<ItemReview> GetCriticReviews(Guid itemId)
  569. {
  570. try
  571. {
  572. var path = Path.Combine(_criticReviewsPath, itemId + ".json");
  573. return _jsonSerializer.DeserializeFromFile<List<ItemReview>>(path);
  574. }
  575. catch (DirectoryNotFoundException)
  576. {
  577. return new List<ItemReview>();
  578. }
  579. catch (FileNotFoundException)
  580. {
  581. return new List<ItemReview>();
  582. }
  583. }
  584. private readonly Task _cachedTask = Task.FromResult(true);
  585. /// <summary>
  586. /// Saves the critic reviews.
  587. /// </summary>
  588. /// <param name="itemId">The item id.</param>
  589. /// <param name="criticReviews">The critic reviews.</param>
  590. /// <returns>Task.</returns>
  591. public Task SaveCriticReviews(Guid itemId, IEnumerable<ItemReview> criticReviews)
  592. {
  593. Directory.CreateDirectory(_criticReviewsPath);
  594. var path = Path.Combine(_criticReviewsPath, itemId + ".json");
  595. _jsonSerializer.SerializeToFile(criticReviews.ToList(), path);
  596. return _cachedTask;
  597. }
  598. /// <summary>
  599. /// Gets chapters for an item
  600. /// </summary>
  601. /// <param name="id">The id.</param>
  602. /// <returns>IEnumerable{ChapterInfo}.</returns>
  603. /// <exception cref="System.ArgumentNullException">id</exception>
  604. public IEnumerable<ChapterInfo> GetChapters(Guid id)
  605. {
  606. CheckDisposed();
  607. return _chapterRepository.GetChapters(id);
  608. }
  609. /// <summary>
  610. /// Gets a single chapter for an item
  611. /// </summary>
  612. /// <param name="id">The id.</param>
  613. /// <param name="index">The index.</param>
  614. /// <returns>ChapterInfo.</returns>
  615. /// <exception cref="System.ArgumentNullException">id</exception>
  616. public ChapterInfo GetChapter(Guid id, int index)
  617. {
  618. CheckDisposed();
  619. return _chapterRepository.GetChapter(id, index);
  620. }
  621. /// <summary>
  622. /// Saves the chapters.
  623. /// </summary>
  624. /// <param name="id">The id.</param>
  625. /// <param name="chapters">The chapters.</param>
  626. /// <param name="cancellationToken">The cancellation token.</param>
  627. /// <returns>Task.</returns>
  628. /// <exception cref="System.ArgumentNullException">
  629. /// id
  630. /// or
  631. /// chapters
  632. /// or
  633. /// cancellationToken
  634. /// </exception>
  635. public Task SaveChapters(Guid id, IEnumerable<ChapterInfo> chapters, CancellationToken cancellationToken)
  636. {
  637. CheckDisposed();
  638. return _chapterRepository.SaveChapters(id, chapters, cancellationToken);
  639. }
  640. /// <summary>
  641. /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
  642. /// </summary>
  643. public void Dispose()
  644. {
  645. Dispose(true);
  646. GC.SuppressFinalize(this);
  647. }
  648. private readonly object _disposeLock = new object();
  649. private bool _disposed;
  650. private void CheckDisposed()
  651. {
  652. if (_disposed)
  653. {
  654. throw new ObjectDisposedException(GetType().Name + " has been disposed and cannot be accessed.");
  655. }
  656. }
  657. /// <summary>
  658. /// Releases unmanaged and - optionally - managed resources.
  659. /// </summary>
  660. /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
  661. protected virtual void Dispose(bool dispose)
  662. {
  663. if (dispose)
  664. {
  665. _disposed = true;
  666. try
  667. {
  668. lock (_disposeLock)
  669. {
  670. _writeLock.Wait();
  671. if (_connection != null)
  672. {
  673. if (_connection.IsOpen())
  674. {
  675. _connection.Close();
  676. }
  677. _connection.Dispose();
  678. _connection = null;
  679. }
  680. if (_chapterRepository != null)
  681. {
  682. _chapterRepository.Dispose();
  683. _chapterRepository = null;
  684. }
  685. if (_mediaStreamsRepository != null)
  686. {
  687. _mediaStreamsRepository.Dispose();
  688. _mediaStreamsRepository = null;
  689. }
  690. }
  691. }
  692. catch (Exception ex)
  693. {
  694. _logger.ErrorException("Error disposing database", ex);
  695. }
  696. }
  697. }
  698. public IEnumerable<Guid> GetChildren(Guid parentId)
  699. {
  700. if (parentId == Guid.Empty)
  701. {
  702. throw new ArgumentNullException("parentId");
  703. }
  704. CheckDisposed();
  705. using (var cmd = _connection.CreateCommand())
  706. {
  707. cmd.CommandText = "select ItemId from ChildrenIds where ParentId = @ParentId";
  708. cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId;
  709. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  710. {
  711. while (reader.Read())
  712. {
  713. yield return reader.GetGuid(0);
  714. }
  715. }
  716. }
  717. }
  718. public IEnumerable<BaseItem> GetChildrenItems(Guid parentId)
  719. {
  720. if (parentId == Guid.Empty)
  721. {
  722. throw new ArgumentNullException("parentId");
  723. }
  724. CheckDisposed();
  725. using (var cmd = _connection.CreateCommand())
  726. {
  727. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid in (select ItemId from ChildrenIds where ParentId = @ParentId)";
  728. cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId;
  729. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  730. {
  731. while (reader.Read())
  732. {
  733. var item = GetItem(reader);
  734. if (item != null)
  735. {
  736. yield return item;
  737. }
  738. }
  739. }
  740. }
  741. }
  742. public IEnumerable<BaseItem> GetItemsOfType(Type type)
  743. {
  744. if (type == null)
  745. {
  746. throw new ArgumentNullException("type");
  747. }
  748. CheckDisposed();
  749. using (var cmd = _connection.CreateCommand())
  750. {
  751. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where type = @type";
  752. cmd.Parameters.Add(cmd, "@type", DbType.String).Value = type.FullName;
  753. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  754. {
  755. while (reader.Read())
  756. {
  757. var item = GetItem(reader);
  758. if (item != null)
  759. {
  760. yield return item;
  761. }
  762. }
  763. }
  764. }
  765. }
  766. public QueryResult<BaseItem> GetItems(InternalItemsQuery query)
  767. {
  768. if (query == null)
  769. {
  770. throw new ArgumentNullException("query");
  771. }
  772. CheckDisposed();
  773. using (var cmd = _connection.CreateCommand())
  774. {
  775. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems";
  776. var whereClauses = GetWhereClauses(query, cmd, false);
  777. var whereTextWithoutPaging = whereClauses.Count == 0 ?
  778. string.Empty :
  779. " where " + string.Join(" AND ", whereClauses.ToArray());
  780. whereClauses = GetWhereClauses(query, cmd, true);
  781. var whereText = whereClauses.Count == 0 ?
  782. string.Empty :
  783. " where " + string.Join(" AND ", whereClauses.ToArray());
  784. cmd.CommandText += whereText;
  785. cmd.CommandText += GetOrderByText(query);
  786. if (query.Limit.HasValue)
  787. {
  788. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  789. }
  790. cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
  791. _logger.Debug(cmd.CommandText);
  792. var list = new List<BaseItem>();
  793. var count = 0;
  794. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
  795. {
  796. while (reader.Read())
  797. {
  798. var item = GetItem(reader);
  799. if (item != null)
  800. {
  801. list.Add(item);
  802. }
  803. }
  804. if (reader.NextResult() && reader.Read())
  805. {
  806. count = reader.GetInt32(0);
  807. }
  808. }
  809. return new QueryResult<BaseItem>()
  810. {
  811. Items = list.ToArray(),
  812. TotalRecordCount = count
  813. };
  814. }
  815. }
  816. private string GetOrderByText(InternalItemsQuery query)
  817. {
  818. if (query.SortBy == null || query.SortBy.Length == 0)
  819. {
  820. return string.Empty;
  821. }
  822. var sortOrder = query.SortOrder == SortOrder.Descending ? "DESC" : "ASC";
  823. return " ORDER BY " + string.Join(",", query.SortBy.Select(i => MapOrderByField(i) + " " + sortOrder).ToArray());
  824. }
  825. private string MapOrderByField(string name)
  826. {
  827. return name;
  828. }
  829. public List<Guid> GetItemIdsList(InternalItemsQuery query)
  830. {
  831. if (query == null)
  832. {
  833. throw new ArgumentNullException("query");
  834. }
  835. CheckDisposed();
  836. using (var cmd = _connection.CreateCommand())
  837. {
  838. cmd.CommandText = "select guid from TypedBaseItems";
  839. var whereClauses = GetWhereClauses(query, cmd, true);
  840. var whereText = whereClauses.Count == 0 ?
  841. string.Empty :
  842. " where " + string.Join(" AND ", whereClauses.ToArray());
  843. cmd.CommandText += whereText;
  844. cmd.CommandText += GetOrderByText(query);
  845. if (query.Limit.HasValue)
  846. {
  847. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  848. }
  849. var list = new List<Guid>();
  850. _logger.Debug(cmd.CommandText);
  851. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
  852. {
  853. while (reader.Read())
  854. {
  855. list.Add(reader.GetGuid(0));
  856. }
  857. }
  858. return list;
  859. }
  860. }
  861. public QueryResult<Tuple<Guid, string>> GetItemIdsWithPath(InternalItemsQuery query)
  862. {
  863. if (query == null)
  864. {
  865. throw new ArgumentNullException("query");
  866. }
  867. CheckDisposed();
  868. using (var cmd = _connection.CreateCommand())
  869. {
  870. cmd.CommandText = "select guid,path from TypedBaseItems";
  871. var whereClauses = GetWhereClauses(query, cmd, false);
  872. var whereTextWithoutPaging = whereClauses.Count == 0 ?
  873. string.Empty :
  874. " where " + string.Join(" AND ", whereClauses.ToArray());
  875. whereClauses = GetWhereClauses(query, cmd, true);
  876. var whereText = whereClauses.Count == 0 ?
  877. string.Empty :
  878. " where " + string.Join(" AND ", whereClauses.ToArray());
  879. cmd.CommandText += whereText;
  880. cmd.CommandText += GetOrderByText(query);
  881. if (query.Limit.HasValue)
  882. {
  883. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  884. }
  885. cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
  886. var list = new List<Tuple<Guid, string>>();
  887. var count = 0;
  888. _logger.Debug(cmd.CommandText);
  889. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
  890. {
  891. while (reader.Read())
  892. {
  893. var id = reader.GetGuid(0);
  894. string path = null;
  895. if (!reader.IsDBNull(1))
  896. {
  897. path = reader.GetString(1);
  898. }
  899. list.Add(new Tuple<Guid, string>(id, path));
  900. }
  901. if (reader.NextResult() && reader.Read())
  902. {
  903. count = reader.GetInt32(0);
  904. }
  905. }
  906. return new QueryResult<Tuple<Guid, string>>()
  907. {
  908. Items = list.ToArray(),
  909. TotalRecordCount = count
  910. };
  911. }
  912. }
  913. public QueryResult<Guid> GetItemIds(InternalItemsQuery query)
  914. {
  915. if (query == null)
  916. {
  917. throw new ArgumentNullException("query");
  918. }
  919. CheckDisposed();
  920. using (var cmd = _connection.CreateCommand())
  921. {
  922. cmd.CommandText = "select guid from TypedBaseItems";
  923. var whereClauses = GetWhereClauses(query, cmd, false);
  924. var whereTextWithoutPaging = whereClauses.Count == 0 ?
  925. string.Empty :
  926. " where " + string.Join(" AND ", whereClauses.ToArray());
  927. whereClauses = GetWhereClauses(query, cmd, true);
  928. var whereText = whereClauses.Count == 0 ?
  929. string.Empty :
  930. " where " + string.Join(" AND ", whereClauses.ToArray());
  931. cmd.CommandText += whereText;
  932. cmd.CommandText += GetOrderByText(query);
  933. if (query.Limit.HasValue)
  934. {
  935. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  936. }
  937. cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
  938. var list = new List<Guid>();
  939. var count = 0;
  940. _logger.Debug(cmd.CommandText);
  941. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
  942. {
  943. while (reader.Read())
  944. {
  945. list.Add(reader.GetGuid(0));
  946. }
  947. if (reader.NextResult() && reader.Read())
  948. {
  949. count = reader.GetInt32(0);
  950. }
  951. }
  952. return new QueryResult<Guid>()
  953. {
  954. Items = list.ToArray(),
  955. TotalRecordCount = count
  956. };
  957. }
  958. }
  959. private List<string> GetWhereClauses(InternalItemsQuery query, IDbCommand cmd, bool addPaging)
  960. {
  961. var whereClauses = new List<string>();
  962. if (query.IsCurrentSchema.HasValue)
  963. {
  964. if (query.IsCurrentSchema.Value)
  965. {
  966. whereClauses.Add("(SchemaVersion not null AND SchemaVersion=@SchemaVersion)");
  967. }
  968. else
  969. {
  970. whereClauses.Add("(SchemaVersion is null or SchemaVersion<>@SchemaVersion)");
  971. }
  972. cmd.Parameters.Add(cmd, "@SchemaVersion", DbType.Int32).Value = LatestSchemaVersion;
  973. }
  974. if (query.IsOffline.HasValue)
  975. {
  976. whereClauses.Add("IsOffline=@IsOffline");
  977. cmd.Parameters.Add(cmd, "@IsOffline", DbType.Boolean).Value = query.IsOffline;
  978. }
  979. if (query.LocationType.HasValue)
  980. {
  981. whereClauses.Add("LocationType=@LocationType");
  982. cmd.Parameters.Add(cmd, "@LocationType", DbType.String).Value = query.LocationType.Value;
  983. }
  984. if (query.IsMovie.HasValue)
  985. {
  986. whereClauses.Add("IsMovie=@IsMovie");
  987. cmd.Parameters.Add(cmd, "@IsMovie", DbType.Boolean).Value = query.IsMovie;
  988. }
  989. if (query.IsKids.HasValue)
  990. {
  991. whereClauses.Add("IsKids=@IsKids");
  992. cmd.Parameters.Add(cmd, "@IsKids", DbType.Boolean).Value = query.IsKids;
  993. }
  994. if (query.IsSports.HasValue)
  995. {
  996. whereClauses.Add("IsSports=@IsSports");
  997. cmd.Parameters.Add(cmd, "@IsSports", DbType.Boolean).Value = query.IsSports;
  998. }
  999. var includeTypes = query.IncludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray();
  1000. if (includeTypes.Length == 1)
  1001. {
  1002. whereClauses.Add("type=@type");
  1003. cmd.Parameters.Add(cmd, "@type", DbType.String).Value = includeTypes[0];
  1004. }
  1005. else if (includeTypes.Length > 1)
  1006. {
  1007. var inClause = string.Join(",", includeTypes.Select(i => "'" + i + "'").ToArray());
  1008. whereClauses.Add(string.Format("type in ({0})", inClause));
  1009. }
  1010. var excludeTypes = query.ExcludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray();
  1011. if (excludeTypes.Length == 1)
  1012. {
  1013. whereClauses.Add("type<>@type");
  1014. cmd.Parameters.Add(cmd, "@type", DbType.String).Value = excludeTypes[0];
  1015. }
  1016. else if (excludeTypes.Length > 1)
  1017. {
  1018. var inClause = string.Join(",", excludeTypes.Select(i => "'" + i + "'").ToArray());
  1019. whereClauses.Add(string.Format("type not in ({0})", inClause));
  1020. }
  1021. if (query.ChannelIds.Length == 1)
  1022. {
  1023. whereClauses.Add("ChannelId=@ChannelId");
  1024. cmd.Parameters.Add(cmd, "@ChannelId", DbType.String).Value = query.ChannelIds[0];
  1025. }
  1026. if (query.ChannelIds.Length > 1)
  1027. {
  1028. var inClause = string.Join(",", query.ChannelIds.Select(i => "'" + i + "'").ToArray());
  1029. whereClauses.Add(string.Format("ChannelId in ({0})", inClause));
  1030. }
  1031. if (query.ParentId.HasValue)
  1032. {
  1033. whereClauses.Add("ParentId=@ParentId");
  1034. cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = query.ParentId.Value;
  1035. }
  1036. if (query.MinEndDate.HasValue)
  1037. {
  1038. whereClauses.Add("EndDate>=@MinEndDate");
  1039. cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = query.MinEndDate.Value;
  1040. }
  1041. if (query.MaxEndDate.HasValue)
  1042. {
  1043. whereClauses.Add("EndDate<=@MaxEndDate");
  1044. cmd.Parameters.Add(cmd, "@MaxEndDate", DbType.Date).Value = query.MaxEndDate.Value;
  1045. }
  1046. if (query.MinStartDate.HasValue)
  1047. {
  1048. whereClauses.Add("StartDate>=@MinStartDate");
  1049. cmd.Parameters.Add(cmd, "@MinStartDate", DbType.Date).Value = query.MinStartDate.Value;
  1050. }
  1051. if (query.MaxStartDate.HasValue)
  1052. {
  1053. whereClauses.Add("StartDate<=@MaxStartDate");
  1054. cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = query.MaxStartDate.Value;
  1055. }
  1056. if (query.IsAiring.HasValue)
  1057. {
  1058. if (query.IsAiring.Value)
  1059. {
  1060. whereClauses.Add("StartDate<=@MaxStartDate");
  1061. cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = DateTime.UtcNow;
  1062. whereClauses.Add("EndDate>=@MinEndDate");
  1063. cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = DateTime.UtcNow;
  1064. }
  1065. else
  1066. {
  1067. whereClauses.Add("(StartDate>@IsAiringDate OR EndDate < @IsAiringDate)");
  1068. cmd.Parameters.Add(cmd, "@IsAiringDate", DbType.Date).Value = DateTime.UtcNow;
  1069. }
  1070. }
  1071. if (!string.IsNullOrWhiteSpace(query.Person))
  1072. {
  1073. whereClauses.Add("Guid in (select ItemId from People where Name=@PersonName)");
  1074. cmd.Parameters.Add(cmd, "@PersonName", DbType.String).Value = query.Person;
  1075. }
  1076. if (!string.IsNullOrWhiteSpace(query.NameContains))
  1077. {
  1078. whereClauses.Add("Name like @NameContains");
  1079. cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%";
  1080. }
  1081. if (query.Genres.Length > 0)
  1082. {
  1083. var genres = new List<string>();
  1084. var index = 0;
  1085. foreach (var genre in query.Genres)
  1086. {
  1087. genres.Add("Genres like @Genres" + index);
  1088. cmd.Parameters.Add(cmd, "@Genres" + index, DbType.String).Value = "%" + genre + "%";
  1089. index++;
  1090. }
  1091. var genreCaluse = "(" + string.Join(" OR ", genres.ToArray()) + ")";
  1092. whereClauses.Add(genreCaluse);
  1093. }
  1094. if (query.MaxParentalRating.HasValue)
  1095. {
  1096. whereClauses.Add("(ParentalRatingValue is NULL OR ParentalRatingValue<=@MaxParentalRating)");
  1097. cmd.Parameters.Add(cmd, "@MaxParentalRating", DbType.Int32).Value = query.MaxParentalRating.Value;
  1098. }
  1099. if (query.HasParentalRating.HasValue)
  1100. {
  1101. if (query.HasParentalRating.Value)
  1102. {
  1103. whereClauses.Add("ParentalRatingValue NOT NULL");
  1104. }
  1105. else
  1106. {
  1107. whereClauses.Add("ParentalRatingValue IS NULL");
  1108. }
  1109. }
  1110. if (query.HasDeadParentId.HasValue)
  1111. {
  1112. if (query.HasDeadParentId.Value)
  1113. {
  1114. whereClauses.Add("ParentId NOT NULL AND ParentId NOT IN (select guid from TypedBaseItems)");
  1115. }
  1116. }
  1117. if (addPaging)
  1118. {
  1119. if (query.StartIndex.HasValue && query.StartIndex.Value > 0)
  1120. {
  1121. var pagingWhereText = whereClauses.Count == 0 ?
  1122. string.Empty :
  1123. " where " + string.Join(" AND ", whereClauses.ToArray());
  1124. var orderBy = GetOrderByText(query);
  1125. whereClauses.Add(string.Format("guid NOT IN (SELECT guid FROM TypedBaseItems {0}" + orderBy + " LIMIT {1})",
  1126. pagingWhereText,
  1127. query.StartIndex.Value.ToString(CultureInfo.InvariantCulture)));
  1128. }
  1129. }
  1130. return whereClauses;
  1131. }
  1132. private static readonly Type[] KnownTypes =
  1133. {
  1134. typeof(LiveTvProgram),
  1135. typeof(LiveTvChannel),
  1136. typeof(LiveTvVideoRecording),
  1137. typeof(LiveTvAudioRecording),
  1138. typeof(Series),
  1139. typeof(LiveTvAudioRecording),
  1140. typeof(LiveTvVideoRecording),
  1141. typeof(Audio),
  1142. typeof(MusicAlbum),
  1143. typeof(MusicArtist),
  1144. typeof(MusicGenre),
  1145. typeof(MusicVideo),
  1146. typeof(Movie),
  1147. typeof(BoxSet),
  1148. typeof(Episode),
  1149. typeof(Season),
  1150. typeof(Series),
  1151. typeof(Book),
  1152. typeof(CollectionFolder),
  1153. typeof(Folder),
  1154. typeof(Game),
  1155. typeof(GameGenre),
  1156. typeof(GameSystem),
  1157. typeof(Genre),
  1158. typeof(Person),
  1159. typeof(Photo),
  1160. typeof(PhotoAlbum),
  1161. typeof(Studio),
  1162. typeof(UserRootFolder),
  1163. typeof(UserView),
  1164. typeof(Video),
  1165. typeof(Year)
  1166. };
  1167. private static Dictionary<string, string[]> GetTypeMapDictionary()
  1168. {
  1169. var dict = new Dictionary<string, string[]>();
  1170. foreach (var t in KnownTypes)
  1171. {
  1172. dict[t.Name] = new[] { t.FullName };
  1173. }
  1174. dict["Recording"] = new[] { typeof(LiveTvAudioRecording).FullName, typeof(LiveTvVideoRecording).FullName };
  1175. dict["Program"] = new[] { typeof(LiveTvProgram).FullName };
  1176. dict["TvChannel"] = new[] { typeof(LiveTvChannel).FullName };
  1177. return dict;
  1178. }
  1179. // Not crazy about having this all the way down here, but at least it's in one place
  1180. readonly Dictionary<string, string[]> _types = GetTypeMapDictionary();
  1181. private IEnumerable<string> MapIncludeItemTypes(string value)
  1182. {
  1183. string[] result;
  1184. if (_types.TryGetValue(value, out result))
  1185. {
  1186. return result;
  1187. }
  1188. return new[] { value };
  1189. }
  1190. public async Task DeleteItem(Guid id, CancellationToken cancellationToken)
  1191. {
  1192. if (id == Guid.Empty)
  1193. {
  1194. throw new ArgumentNullException("id");
  1195. }
  1196. CheckDisposed();
  1197. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  1198. IDbTransaction transaction = null;
  1199. try
  1200. {
  1201. transaction = _connection.BeginTransaction();
  1202. // First delete children
  1203. _deleteChildrenCommand.GetParameter(0).Value = id;
  1204. _deleteChildrenCommand.Transaction = transaction;
  1205. _deleteChildrenCommand.ExecuteNonQuery();
  1206. // Delete people
  1207. _deletePeopleCommand.GetParameter(0).Value = id;
  1208. _deletePeopleCommand.Transaction = transaction;
  1209. _deletePeopleCommand.ExecuteNonQuery();
  1210. // Delete the item
  1211. _deleteItemCommand.GetParameter(0).Value = id;
  1212. _deleteItemCommand.Transaction = transaction;
  1213. _deleteItemCommand.ExecuteNonQuery();
  1214. transaction.Commit();
  1215. }
  1216. catch (OperationCanceledException)
  1217. {
  1218. if (transaction != null)
  1219. {
  1220. transaction.Rollback();
  1221. }
  1222. throw;
  1223. }
  1224. catch (Exception e)
  1225. {
  1226. _logger.ErrorException("Failed to save children:", e);
  1227. if (transaction != null)
  1228. {
  1229. transaction.Rollback();
  1230. }
  1231. throw;
  1232. }
  1233. finally
  1234. {
  1235. if (transaction != null)
  1236. {
  1237. transaction.Dispose();
  1238. }
  1239. _writeLock.Release();
  1240. }
  1241. }
  1242. public async Task SaveChildren(Guid parentId, IEnumerable<Guid> children, CancellationToken cancellationToken)
  1243. {
  1244. if (parentId == Guid.Empty)
  1245. {
  1246. throw new ArgumentNullException("parentId");
  1247. }
  1248. if (children == null)
  1249. {
  1250. throw new ArgumentNullException("children");
  1251. }
  1252. CheckDisposed();
  1253. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  1254. IDbTransaction transaction = null;
  1255. try
  1256. {
  1257. transaction = _connection.BeginTransaction();
  1258. // First delete
  1259. _deleteChildrenCommand.GetParameter(0).Value = parentId;
  1260. _deleteChildrenCommand.Transaction = transaction;
  1261. _deleteChildrenCommand.ExecuteNonQuery();
  1262. foreach (var id in children)
  1263. {
  1264. cancellationToken.ThrowIfCancellationRequested();
  1265. _saveChildrenCommand.GetParameter(0).Value = parentId;
  1266. _saveChildrenCommand.GetParameter(1).Value = id;
  1267. _saveChildrenCommand.Transaction = transaction;
  1268. _saveChildrenCommand.ExecuteNonQuery();
  1269. }
  1270. transaction.Commit();
  1271. }
  1272. catch (OperationCanceledException)
  1273. {
  1274. if (transaction != null)
  1275. {
  1276. transaction.Rollback();
  1277. }
  1278. throw;
  1279. }
  1280. catch (Exception e)
  1281. {
  1282. _logger.ErrorException("Failed to save children:", e);
  1283. if (transaction != null)
  1284. {
  1285. transaction.Rollback();
  1286. }
  1287. throw;
  1288. }
  1289. finally
  1290. {
  1291. if (transaction != null)
  1292. {
  1293. transaction.Dispose();
  1294. }
  1295. _writeLock.Release();
  1296. }
  1297. }
  1298. public IEnumerable<MediaStream> GetMediaStreams(MediaStreamQuery query)
  1299. {
  1300. CheckDisposed();
  1301. return _mediaStreamsRepository.GetMediaStreams(query);
  1302. }
  1303. public Task SaveMediaStreams(Guid id, IEnumerable<MediaStream> streams, CancellationToken cancellationToken)
  1304. {
  1305. CheckDisposed();
  1306. return _mediaStreamsRepository.SaveMediaStreams(id, streams, cancellationToken);
  1307. }
  1308. public List<string> GetPeopleNames(InternalPeopleQuery query)
  1309. {
  1310. if (query == null)
  1311. {
  1312. throw new ArgumentNullException("query");
  1313. }
  1314. CheckDisposed();
  1315. using (var cmd = _connection.CreateCommand())
  1316. {
  1317. cmd.CommandText = "select Distinct Name from People";
  1318. var whereClauses = GetPeopleWhereClauses(query, cmd);
  1319. if (whereClauses.Count > 0)
  1320. {
  1321. cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray());
  1322. }
  1323. cmd.CommandText += " order by ListOrder";
  1324. var list = new List<string>();
  1325. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1326. {
  1327. while (reader.Read())
  1328. {
  1329. list.Add(reader.GetString(0));
  1330. }
  1331. }
  1332. return list;
  1333. }
  1334. }
  1335. public List<PersonInfo> GetPeople(InternalPeopleQuery query)
  1336. {
  1337. if (query == null)
  1338. {
  1339. throw new ArgumentNullException("query");
  1340. }
  1341. CheckDisposed();
  1342. using (var cmd = _connection.CreateCommand())
  1343. {
  1344. cmd.CommandText = "select ItemId, Name, Role, PersonType, SortOrder from People";
  1345. var whereClauses = GetPeopleWhereClauses(query, cmd);
  1346. if (whereClauses.Count > 0)
  1347. {
  1348. cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray());
  1349. }
  1350. cmd.CommandText += " order by ListOrder";
  1351. var list = new List<PersonInfo>();
  1352. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1353. {
  1354. while (reader.Read())
  1355. {
  1356. list.Add(GetPerson(reader));
  1357. }
  1358. }
  1359. return list;
  1360. }
  1361. }
  1362. private List<string> GetPeopleWhereClauses(InternalPeopleQuery query, IDbCommand cmd)
  1363. {
  1364. var whereClauses = new List<string>();
  1365. if (query.ItemId != Guid.Empty)
  1366. {
  1367. whereClauses.Add("ItemId=@ItemId");
  1368. cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = query.ItemId;
  1369. }
  1370. if (query.AppearsInItemId != Guid.Empty)
  1371. {
  1372. whereClauses.Add("Name in (Select Name from People where ItemId=@AppearsInItemId)");
  1373. cmd.Parameters.Add(cmd, "@AppearsInItemId", DbType.Guid).Value = query.AppearsInItemId;
  1374. }
  1375. if (query.PersonTypes.Count == 1)
  1376. {
  1377. whereClauses.Add("PersonType=@PersonType");
  1378. cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.PersonTypes[0];
  1379. }
  1380. if (query.PersonTypes.Count > 1)
  1381. {
  1382. var val = string.Join(",", query.PersonTypes.Select(i => "'" + i + "'").ToArray());
  1383. whereClauses.Add("PersonType in (" + val + ")");
  1384. }
  1385. if (query.ExcludePersonTypes.Count == 1)
  1386. {
  1387. whereClauses.Add("PersonType<>@PersonType");
  1388. cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.ExcludePersonTypes[0];
  1389. }
  1390. if (query.ExcludePersonTypes.Count > 1)
  1391. {
  1392. var val = string.Join(",", query.ExcludePersonTypes.Select(i => "'" + i + "'").ToArray());
  1393. whereClauses.Add("PersonType not in (" + val + ")");
  1394. }
  1395. if (query.MaxListOrder.HasValue)
  1396. {
  1397. whereClauses.Add("ListOrder<=@MaxListOrder");
  1398. cmd.Parameters.Add(cmd, "@MaxListOrder", DbType.Int32).Value = query.MaxListOrder.Value;
  1399. }
  1400. if (!string.IsNullOrWhiteSpace(query.NameContains))
  1401. {
  1402. whereClauses.Add("Name like @NameContains");
  1403. cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%";
  1404. }
  1405. return whereClauses;
  1406. }
  1407. public async Task UpdatePeople(Guid itemId, List<PersonInfo> people)
  1408. {
  1409. if (itemId == Guid.Empty)
  1410. {
  1411. throw new ArgumentNullException("itemId");
  1412. }
  1413. if (people == null)
  1414. {
  1415. throw new ArgumentNullException("people");
  1416. }
  1417. CheckDisposed();
  1418. var cancellationToken = CancellationToken.None;
  1419. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  1420. IDbTransaction transaction = null;
  1421. try
  1422. {
  1423. transaction = _connection.BeginTransaction();
  1424. // First delete
  1425. _deletePeopleCommand.GetParameter(0).Value = itemId;
  1426. _deletePeopleCommand.Transaction = transaction;
  1427. _deletePeopleCommand.ExecuteNonQuery();
  1428. var listIndex = 0;
  1429. foreach (var person in people)
  1430. {
  1431. cancellationToken.ThrowIfCancellationRequested();
  1432. _savePersonCommand.GetParameter(0).Value = itemId;
  1433. _savePersonCommand.GetParameter(1).Value = person.Name;
  1434. _savePersonCommand.GetParameter(2).Value = person.Role;
  1435. _savePersonCommand.GetParameter(3).Value = person.Type;
  1436. _savePersonCommand.GetParameter(4).Value = person.SortOrder;
  1437. _savePersonCommand.GetParameter(5).Value = listIndex;
  1438. _savePersonCommand.Transaction = transaction;
  1439. _savePersonCommand.ExecuteNonQuery();
  1440. listIndex++;
  1441. }
  1442. transaction.Commit();
  1443. }
  1444. catch (OperationCanceledException)
  1445. {
  1446. if (transaction != null)
  1447. {
  1448. transaction.Rollback();
  1449. }
  1450. throw;
  1451. }
  1452. catch (Exception e)
  1453. {
  1454. _logger.ErrorException("Failed to save people:", e);
  1455. if (transaction != null)
  1456. {
  1457. transaction.Rollback();
  1458. }
  1459. throw;
  1460. }
  1461. finally
  1462. {
  1463. if (transaction != null)
  1464. {
  1465. transaction.Dispose();
  1466. }
  1467. _writeLock.Release();
  1468. }
  1469. }
  1470. private PersonInfo GetPerson(IDataReader reader)
  1471. {
  1472. var item = new PersonInfo();
  1473. item.ItemId = reader.GetGuid(0);
  1474. item.Name = reader.GetString(1);
  1475. if (!reader.IsDBNull(2))
  1476. {
  1477. item.Role = reader.GetString(2);
  1478. }
  1479. if (!reader.IsDBNull(3))
  1480. {
  1481. item.Type = reader.GetString(3);
  1482. }
  1483. if (!reader.IsDBNull(4))
  1484. {
  1485. item.SortOrder = reader.GetInt32(4);
  1486. }
  1487. return item;
  1488. }
  1489. }
  1490. }