SqliteItemRepository.cs 62 KB

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