SqliteItemRepository.cs 65 KB

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