SqliteItemRepository.cs 57 KB

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