SqliteItemRepository.cs 63 KB

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