SqliteItemRepository.cs 100 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835
  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. using MediaBrowser.Model.LiveTv;
  23. namespace MediaBrowser.Server.Implementations.Persistence
  24. {
  25. /// <summary>
  26. /// Class SQLiteItemRepository
  27. /// </summary>
  28. public class SqliteItemRepository : IItemRepository
  29. {
  30. private IDbConnection _connection;
  31. private readonly ILogger _logger;
  32. private readonly TypeMapper _typeMapper = new TypeMapper();
  33. /// <summary>
  34. /// Gets the name of the repository
  35. /// </summary>
  36. /// <value>The name.</value>
  37. public string Name
  38. {
  39. get
  40. {
  41. return "SQLite";
  42. }
  43. }
  44. /// <summary>
  45. /// Gets the json serializer.
  46. /// </summary>
  47. /// <value>The json serializer.</value>
  48. private readonly IJsonSerializer _jsonSerializer;
  49. /// <summary>
  50. /// The _app paths
  51. /// </summary>
  52. private readonly IApplicationPaths _appPaths;
  53. /// <summary>
  54. /// The _save item command
  55. /// </summary>
  56. private IDbCommand _saveItemCommand;
  57. private readonly string _criticReviewsPath;
  58. private IDbCommand _deleteChildrenCommand;
  59. private IDbCommand _saveChildrenCommand;
  60. private IDbCommand _deleteItemCommand;
  61. private IDbCommand _deletePeopleCommand;
  62. private IDbCommand _savePersonCommand;
  63. private IDbCommand _deleteChaptersCommand;
  64. private IDbCommand _saveChapterCommand;
  65. private IDbCommand _deleteStreamsCommand;
  66. private IDbCommand _saveStreamCommand;
  67. private IDbCommand _deleteAncestorsCommand;
  68. private IDbCommand _saveAncestorCommand;
  69. private IDbCommand _updateInheritedRatingCommand;
  70. private const int LatestSchemaVersion = 32;
  71. /// <summary>
  72. /// Initializes a new instance of the <see cref="SqliteItemRepository"/> class.
  73. /// </summary>
  74. /// <param name="appPaths">The app paths.</param>
  75. /// <param name="jsonSerializer">The json serializer.</param>
  76. /// <param name="logManager">The log manager.</param>
  77. /// <exception cref="System.ArgumentNullException">
  78. /// appPaths
  79. /// or
  80. /// jsonSerializer
  81. /// </exception>
  82. public SqliteItemRepository(IApplicationPaths appPaths, IJsonSerializer jsonSerializer, ILogManager logManager)
  83. {
  84. if (appPaths == null)
  85. {
  86. throw new ArgumentNullException("appPaths");
  87. }
  88. if (jsonSerializer == null)
  89. {
  90. throw new ArgumentNullException("jsonSerializer");
  91. }
  92. _appPaths = appPaths;
  93. _jsonSerializer = jsonSerializer;
  94. _criticReviewsPath = Path.Combine(_appPaths.DataPath, "critic-reviews");
  95. _logger = logManager.GetLogger(GetType().Name);
  96. }
  97. private const string ChaptersTableName = "Chapters2";
  98. /// <summary>
  99. /// Opens the connection to the database
  100. /// </summary>
  101. /// <returns>Task.</returns>
  102. public async Task Initialize()
  103. {
  104. var dbFile = Path.Combine(_appPaths.DataPath, "library.db");
  105. _connection = await SqliteExtensions.ConnectToDb(dbFile, _logger).ConfigureAwait(false);
  106. var createMediaStreamsTableCommand
  107. = "create table if not exists mediastreams (ItemId GUID, StreamIndex INT, StreamType TEXT, Codec TEXT, Language TEXT, ChannelLayout TEXT, Profile TEXT, AspectRatio TEXT, Path TEXT, IsInterlaced BIT, BitRate INT NULL, Channels INT NULL, SampleRate INT NULL, IsDefault BIT, IsForced BIT, IsExternal BIT, Height INT NULL, Width INT NULL, AverageFrameRate FLOAT NULL, RealFrameRate FLOAT NULL, Level FLOAT NULL, PixelFormat TEXT, BitDepth INT NULL, IsAnamorphic BIT NULL, RefFrames INT NULL, IsCabac BIT NULL, KeyFrames TEXT NULL, CodecTag TEXT NULL, PRIMARY KEY (ItemId, StreamIndex))";
  108. string[] queries = {
  109. "create table if not exists TypedBaseItems (guid GUID primary key, type TEXT, data BLOB, ParentId GUID)",
  110. "create index if not exists idx_TypedBaseItems on TypedBaseItems(guid)",
  111. "create index if not exists idx_ParentIdTypedBaseItems on TypedBaseItems(ParentId)",
  112. "create table if not exists AncestorIds (ItemId GUID, AncestorId GUID, AncestorIdText TEXT, PRIMARY KEY (ItemId, AncestorId))",
  113. "create index if not exists idx_AncestorIds1 on AncestorIds(AncestorId)",
  114. "create index if not exists idx_AncestorIds2 on AncestorIds(AncestorIdText)",
  115. "create table if not exists ChildrenIds (ParentId GUID, ItemId GUID, PRIMARY KEY (ParentId, ItemId))",
  116. "create index if not exists idx_ChildrenIds on ChildrenIds(ParentId,ItemId)",
  117. "create table if not exists People (ItemId GUID, Name TEXT NOT NULL, Role TEXT, PersonType TEXT, SortOrder int, ListOrder int)",
  118. "create index if not exists idxPeopleItemId on People(ItemId)",
  119. "create index if not exists idxPeopleName on People(Name)",
  120. "create table if not exists "+ChaptersTableName+" (ItemId GUID, ChapterIndex INT, StartPositionTicks BIGINT, Name TEXT, ImagePath TEXT, PRIMARY KEY (ItemId, ChapterIndex))",
  121. "create index if not exists idx_"+ChaptersTableName+" on "+ChaptersTableName+"(ItemId, ChapterIndex)",
  122. createMediaStreamsTableCommand,
  123. "create index if not exists idx_mediastreams on mediastreams(ItemId, StreamIndex)",
  124. //pragmas
  125. "pragma temp_store = memory",
  126. "pragma shrink_memory"
  127. };
  128. _connection.RunQueries(queries, _logger);
  129. _connection.AddColumn(_logger, "AncestorIds", "AncestorIdText", "Text");
  130. _connection.AddColumn(_logger, "TypedBaseItems", "Path", "Text");
  131. _connection.AddColumn(_logger, "TypedBaseItems", "StartDate", "DATETIME");
  132. _connection.AddColumn(_logger, "TypedBaseItems", "EndDate", "DATETIME");
  133. _connection.AddColumn(_logger, "TypedBaseItems", "ChannelId", "Text");
  134. _connection.AddColumn(_logger, "TypedBaseItems", "IsMovie", "BIT");
  135. _connection.AddColumn(_logger, "TypedBaseItems", "IsSports", "BIT");
  136. _connection.AddColumn(_logger, "TypedBaseItems", "IsKids", "BIT");
  137. _connection.AddColumn(_logger, "TypedBaseItems", "CommunityRating", "Float");
  138. _connection.AddColumn(_logger, "TypedBaseItems", "CustomRating", "Text");
  139. _connection.AddColumn(_logger, "TypedBaseItems", "IndexNumber", "INT");
  140. _connection.AddColumn(_logger, "TypedBaseItems", "IsLocked", "BIT");
  141. _connection.AddColumn(_logger, "TypedBaseItems", "Name", "Text");
  142. _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRating", "Text");
  143. _connection.AddColumn(_logger, "TypedBaseItems", "MediaType", "Text");
  144. _connection.AddColumn(_logger, "TypedBaseItems", "Overview", "Text");
  145. _connection.AddColumn(_logger, "TypedBaseItems", "ParentIndexNumber", "INT");
  146. _connection.AddColumn(_logger, "TypedBaseItems", "PremiereDate", "DATETIME");
  147. _connection.AddColumn(_logger, "TypedBaseItems", "ProductionYear", "INT");
  148. _connection.AddColumn(_logger, "TypedBaseItems", "ParentId", "GUID");
  149. _connection.AddColumn(_logger, "TypedBaseItems", "Genres", "Text");
  150. _connection.AddColumn(_logger, "TypedBaseItems", "ParentalRatingValue", "INT");
  151. _connection.AddColumn(_logger, "TypedBaseItems", "SchemaVersion", "INT");
  152. _connection.AddColumn(_logger, "TypedBaseItems", "SortName", "Text");
  153. _connection.AddColumn(_logger, "TypedBaseItems", "RunTimeTicks", "BIGINT");
  154. _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRatingDescription", "Text");
  155. _connection.AddColumn(_logger, "TypedBaseItems", "HomePageUrl", "Text");
  156. _connection.AddColumn(_logger, "TypedBaseItems", "VoteCount", "INT");
  157. _connection.AddColumn(_logger, "TypedBaseItems", "DisplayMediaType", "Text");
  158. _connection.AddColumn(_logger, "TypedBaseItems", "DateCreated", "DATETIME");
  159. _connection.AddColumn(_logger, "TypedBaseItems", "DateModified", "DATETIME");
  160. _connection.AddColumn(_logger, "TypedBaseItems", "ForcedSortName", "Text");
  161. _connection.AddColumn(_logger, "TypedBaseItems", "IsOffline", "BIT");
  162. _connection.AddColumn(_logger, "TypedBaseItems", "LocationType", "Text");
  163. _connection.AddColumn(_logger, "TypedBaseItems", "IsSeries", "BIT");
  164. _connection.AddColumn(_logger, "TypedBaseItems", "IsLive", "BIT");
  165. _connection.AddColumn(_logger, "TypedBaseItems", "IsNews", "BIT");
  166. _connection.AddColumn(_logger, "TypedBaseItems", "IsPremiere", "BIT");
  167. _connection.AddColumn(_logger, "TypedBaseItems", "EpisodeTitle", "Text");
  168. _connection.AddColumn(_logger, "TypedBaseItems", "IsRepeat", "BIT");
  169. _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataLanguage", "Text");
  170. _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataCountryCode", "Text");
  171. _connection.AddColumn(_logger, "TypedBaseItems", "IsHD", "BIT");
  172. _connection.AddColumn(_logger, "TypedBaseItems", "ExternalEtag", "Text");
  173. _connection.AddColumn(_logger, "TypedBaseItems", "DateLastRefreshed", "DATETIME");
  174. _connection.AddColumn(_logger, "TypedBaseItems", "DateLastSaved", "DATETIME");
  175. _connection.AddColumn(_logger, "TypedBaseItems", "IsInMixedFolder", "BIT");
  176. _connection.AddColumn(_logger, "TypedBaseItems", "LockedFields", "Text");
  177. _connection.AddColumn(_logger, "TypedBaseItems", "Studios", "Text");
  178. _connection.AddColumn(_logger, "TypedBaseItems", "Audio", "Text");
  179. _connection.AddColumn(_logger, "TypedBaseItems", "ExternalServiceId", "Text");
  180. _connection.AddColumn(_logger, "TypedBaseItems", "Tags", "Text");
  181. _connection.AddColumn(_logger, "TypedBaseItems", "IsFolder", "BIT");
  182. _connection.AddColumn(_logger, "TypedBaseItems", "InheritedParentalRatingValue", "INT");
  183. _connection.AddColumn(_logger, "TypedBaseItems", "UnratedType", "Text");
  184. PrepareStatements();
  185. new MediaStreamColumns(_connection, _logger).AddColumns();
  186. var chapterDbFile = Path.Combine(_appPaths.DataPath, "chapters.db");
  187. if (File.Exists(chapterDbFile))
  188. {
  189. MigrateChapters(chapterDbFile);
  190. }
  191. var mediaStreamsDbFile = Path.Combine(_appPaths.DataPath, "mediainfo.db");
  192. if (File.Exists(mediaStreamsDbFile))
  193. {
  194. MigrateMediaStreams(mediaStreamsDbFile);
  195. }
  196. }
  197. private void MigrateMediaStreams(string file)
  198. {
  199. try
  200. {
  201. var backupFile = file + ".bak";
  202. File.Copy(file, backupFile, true);
  203. SqliteExtensions.Attach(_connection, backupFile, "MediaInfoOld");
  204. var columns = string.Join(",", _mediaStreamSaveColumns);
  205. string[] queries = {
  206. "REPLACE INTO mediastreams("+columns+") SELECT "+columns+" FROM MediaInfoOld.mediastreams;"
  207. };
  208. _connection.RunQueries(queries, _logger);
  209. }
  210. catch (Exception ex)
  211. {
  212. _logger.ErrorException("Error migrating media info database", ex);
  213. }
  214. finally
  215. {
  216. TryDeleteFile(file);
  217. }
  218. }
  219. private void MigrateChapters(string file)
  220. {
  221. try
  222. {
  223. var backupFile = file + ".bak";
  224. File.Copy(file, backupFile, true);
  225. SqliteExtensions.Attach(_connection, backupFile, "ChaptersOld");
  226. string[] queries = {
  227. "REPLACE INTO "+ChaptersTableName+"(ItemId, ChapterIndex, StartPositionTicks, Name, ImagePath) SELECT ItemId, ChapterIndex, StartPositionTicks, Name, ImagePath FROM ChaptersOld.Chapters;"
  228. };
  229. _connection.RunQueries(queries, _logger);
  230. }
  231. catch (Exception ex)
  232. {
  233. _logger.ErrorException("Error migrating chapter database", ex);
  234. }
  235. finally
  236. {
  237. TryDeleteFile(file);
  238. }
  239. }
  240. private void TryDeleteFile(string file)
  241. {
  242. try
  243. {
  244. File.Delete(file);
  245. }
  246. catch (Exception ex)
  247. {
  248. _logger.ErrorException("Error deleting file {0}", ex, file);
  249. }
  250. }
  251. /// <summary>
  252. /// The _write lock
  253. /// </summary>
  254. private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
  255. private readonly string[] _retriveItemColumns =
  256. {
  257. "type",
  258. "data",
  259. "StartDate",
  260. "EndDate",
  261. "IsOffline",
  262. "ChannelId",
  263. "IsMovie",
  264. "IsSports",
  265. "IsKids",
  266. "IsSeries",
  267. "IsLive",
  268. "IsNews",
  269. "IsPremiere",
  270. "EpisodeTitle",
  271. "IsRepeat",
  272. "CommunityRating",
  273. "CustomRating",
  274. "IndexNumber",
  275. "IsLocked",
  276. "PreferredMetadataLanguage",
  277. "PreferredMetadataCountryCode",
  278. "IsHD",
  279. "ExternalEtag",
  280. "DateLastRefreshed",
  281. "Name",
  282. "Path",
  283. "PremiereDate",
  284. "Overview",
  285. "ParentIndexNumber",
  286. "ProductionYear",
  287. "OfficialRating",
  288. "OfficialRatingDescription",
  289. "HomePageUrl",
  290. "DisplayMediaType",
  291. "ForcedSortName",
  292. "RunTimeTicks",
  293. "VoteCount",
  294. "DateCreated",
  295. "DateModified",
  296. "guid",
  297. "Genres",
  298. "ParentId",
  299. "Audio",
  300. "ExternalServiceId"
  301. };
  302. private readonly string[] _mediaStreamSaveColumns =
  303. {
  304. "ItemId",
  305. "StreamIndex",
  306. "StreamType",
  307. "Codec",
  308. "Language",
  309. "ChannelLayout",
  310. "Profile",
  311. "AspectRatio",
  312. "Path",
  313. "IsInterlaced",
  314. "BitRate",
  315. "Channels",
  316. "SampleRate",
  317. "IsDefault",
  318. "IsForced",
  319. "IsExternal",
  320. "Height",
  321. "Width",
  322. "AverageFrameRate",
  323. "RealFrameRate",
  324. "Level",
  325. "PixelFormat",
  326. "BitDepth",
  327. "IsAnamorphic",
  328. "RefFrames",
  329. "IsCabac",
  330. "KeyFrames",
  331. "CodecTag"
  332. };
  333. /// <summary>
  334. /// Prepares the statements.
  335. /// </summary>
  336. private void PrepareStatements()
  337. {
  338. var saveColumns = new List<string>
  339. {
  340. "guid",
  341. "type",
  342. "data",
  343. "Path",
  344. "StartDate",
  345. "EndDate",
  346. "ChannelId",
  347. "IsKids",
  348. "IsMovie",
  349. "IsSports",
  350. "IsSeries",
  351. "IsLive",
  352. "IsNews",
  353. "IsPremiere",
  354. "EpisodeTitle",
  355. "IsRepeat",
  356. "CommunityRating",
  357. "CustomRating",
  358. "IndexNumber",
  359. "IsLocked",
  360. "Name",
  361. "OfficialRating",
  362. "MediaType",
  363. "Overview",
  364. "ParentIndexNumber",
  365. "PremiereDate",
  366. "ProductionYear",
  367. "ParentId",
  368. "Genres",
  369. "ParentalRatingValue",
  370. "InheritedParentalRatingValue",
  371. "SchemaVersion",
  372. "SortName",
  373. "RunTimeTicks",
  374. "OfficialRatingDescription",
  375. "HomePageUrl",
  376. "VoteCount",
  377. "DisplayMediaType",
  378. "DateCreated",
  379. "DateModified",
  380. "ForcedSortName",
  381. "IsOffline",
  382. "LocationType",
  383. "PreferredMetadataLanguage",
  384. "PreferredMetadataCountryCode",
  385. "IsHD",
  386. "ExternalEtag",
  387. "DateLastRefreshed",
  388. "DateLastSaved",
  389. "IsInMixedFolder",
  390. "LockedFields",
  391. "Studios",
  392. "Audio",
  393. "ExternalServiceId",
  394. "Tags",
  395. "IsFolder",
  396. "UnratedType"
  397. };
  398. _saveItemCommand = _connection.CreateCommand();
  399. _saveItemCommand.CommandText = "replace into TypedBaseItems (" + string.Join(",", saveColumns.ToArray()) + ") values (";
  400. for (var i = 1; i <= saveColumns.Count; i++)
  401. {
  402. if (i > 1)
  403. {
  404. _saveItemCommand.CommandText += ",";
  405. }
  406. _saveItemCommand.CommandText += "@" + i.ToString(CultureInfo.InvariantCulture);
  407. _saveItemCommand.Parameters.Add(_saveItemCommand, "@" + i.ToString(CultureInfo.InvariantCulture));
  408. }
  409. _saveItemCommand.CommandText += ")";
  410. _deleteChildrenCommand = _connection.CreateCommand();
  411. _deleteChildrenCommand.CommandText = "delete from ChildrenIds where ParentId=@ParentId";
  412. _deleteChildrenCommand.Parameters.Add(_deleteChildrenCommand, "@ParentId");
  413. _deleteItemCommand = _connection.CreateCommand();
  414. _deleteItemCommand.CommandText = "delete from TypedBaseItems where guid=@Id";
  415. _deleteItemCommand.Parameters.Add(_deleteItemCommand, "@Id");
  416. _saveChildrenCommand = _connection.CreateCommand();
  417. _saveChildrenCommand.CommandText = "replace into ChildrenIds (ParentId, ItemId) values (@ParentId, @ItemId)";
  418. _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ParentId");
  419. _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ItemId");
  420. // People
  421. _deletePeopleCommand = _connection.CreateCommand();
  422. _deletePeopleCommand.CommandText = "delete from People where ItemId=@Id";
  423. _deletePeopleCommand.Parameters.Add(_deletePeopleCommand, "@Id");
  424. _savePersonCommand = _connection.CreateCommand();
  425. _savePersonCommand.CommandText = "insert into People (ItemId, Name, Role, PersonType, SortOrder, ListOrder) values (@ItemId, @Name, @Role, @PersonType, @SortOrder, @ListOrder)";
  426. _savePersonCommand.Parameters.Add(_savePersonCommand, "@ItemId");
  427. _savePersonCommand.Parameters.Add(_savePersonCommand, "@Name");
  428. _savePersonCommand.Parameters.Add(_savePersonCommand, "@Role");
  429. _savePersonCommand.Parameters.Add(_savePersonCommand, "@PersonType");
  430. _savePersonCommand.Parameters.Add(_savePersonCommand, "@SortOrder");
  431. _savePersonCommand.Parameters.Add(_savePersonCommand, "@ListOrder");
  432. // Ancestors
  433. _deleteAncestorsCommand = _connection.CreateCommand();
  434. _deleteAncestorsCommand.CommandText = "delete from AncestorIds where ItemId=@Id";
  435. _deleteAncestorsCommand.Parameters.Add(_deleteAncestorsCommand, "@Id");
  436. _saveAncestorCommand = _connection.CreateCommand();
  437. _saveAncestorCommand.CommandText = "insert into AncestorIds (ItemId, AncestorId, AncestorIdText) values (@ItemId, @AncestorId, @AncestorIdText)";
  438. _saveAncestorCommand.Parameters.Add(_saveAncestorCommand, "@ItemId");
  439. _saveAncestorCommand.Parameters.Add(_saveAncestorCommand, "@AncestorId");
  440. _saveAncestorCommand.Parameters.Add(_saveAncestorCommand, "@AncestorIdText");
  441. // Chapters
  442. _deleteChaptersCommand = _connection.CreateCommand();
  443. _deleteChaptersCommand.CommandText = "delete from " + ChaptersTableName + " where ItemId=@ItemId";
  444. _deleteChaptersCommand.Parameters.Add(_deleteChaptersCommand, "@ItemId");
  445. _saveChapterCommand = _connection.CreateCommand();
  446. _saveChapterCommand.CommandText = "replace into " + ChaptersTableName + " (ItemId, ChapterIndex, StartPositionTicks, Name, ImagePath) values (@ItemId, @ChapterIndex, @StartPositionTicks, @Name, @ImagePath)";
  447. _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@ItemId");
  448. _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@ChapterIndex");
  449. _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@StartPositionTicks");
  450. _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@Name");
  451. _saveChapterCommand.Parameters.Add(_saveChapterCommand, "@ImagePath");
  452. // MediaStreams
  453. _deleteStreamsCommand = _connection.CreateCommand();
  454. _deleteStreamsCommand.CommandText = "delete from mediastreams where ItemId=@ItemId";
  455. _deleteStreamsCommand.Parameters.Add(_deleteStreamsCommand, "@ItemId");
  456. _saveStreamCommand = _connection.CreateCommand();
  457. _saveStreamCommand.CommandText = string.Format("replace into mediastreams ({0}) values ({1})",
  458. string.Join(",", _mediaStreamSaveColumns),
  459. string.Join(",", _mediaStreamSaveColumns.Select(i => "@" + i).ToArray()));
  460. foreach (var col in _mediaStreamSaveColumns)
  461. {
  462. _saveStreamCommand.Parameters.Add(_saveStreamCommand, "@" + col);
  463. }
  464. _updateInheritedRatingCommand = _connection.CreateCommand();
  465. _updateInheritedRatingCommand.CommandText = "Update TypedBaseItems set InheritedParentalRatingValue=@InheritedParentalRatingValue where Guid=@Guid";
  466. _updateInheritedRatingCommand.Parameters.Add(_updateInheritedRatingCommand, "@InheritedParentalRatingValue");
  467. _updateInheritedRatingCommand.Parameters.Add(_updateInheritedRatingCommand, "@Guid");
  468. }
  469. /// <summary>
  470. /// Save a standard item in the repo
  471. /// </summary>
  472. /// <param name="item">The item.</param>
  473. /// <param name="cancellationToken">The cancellation token.</param>
  474. /// <returns>Task.</returns>
  475. /// <exception cref="System.ArgumentNullException">item</exception>
  476. public Task SaveItem(BaseItem item, CancellationToken cancellationToken)
  477. {
  478. if (item == null)
  479. {
  480. throw new ArgumentNullException("item");
  481. }
  482. return SaveItems(new[] { item }, cancellationToken);
  483. }
  484. /// <summary>
  485. /// Saves the items.
  486. /// </summary>
  487. /// <param name="items">The items.</param>
  488. /// <param name="cancellationToken">The cancellation token.</param>
  489. /// <returns>Task.</returns>
  490. /// <exception cref="System.ArgumentNullException">
  491. /// items
  492. /// or
  493. /// cancellationToken
  494. /// </exception>
  495. public async Task SaveItems(IEnumerable<BaseItem> items, CancellationToken cancellationToken)
  496. {
  497. if (items == null)
  498. {
  499. throw new ArgumentNullException("items");
  500. }
  501. cancellationToken.ThrowIfCancellationRequested();
  502. CheckDisposed();
  503. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  504. IDbTransaction transaction = null;
  505. try
  506. {
  507. transaction = _connection.BeginTransaction();
  508. foreach (var item in items)
  509. {
  510. cancellationToken.ThrowIfCancellationRequested();
  511. var index = 0;
  512. _saveItemCommand.GetParameter(index++).Value = item.Id;
  513. _saveItemCommand.GetParameter(index++).Value = item.GetType().FullName;
  514. _saveItemCommand.GetParameter(index++).Value = _jsonSerializer.SerializeToBytes(item);
  515. _saveItemCommand.GetParameter(index++).Value = item.Path;
  516. var hasStartDate = item as IHasStartDate;
  517. if (hasStartDate != null)
  518. {
  519. _saveItemCommand.GetParameter(index++).Value = hasStartDate.StartDate;
  520. }
  521. else
  522. {
  523. _saveItemCommand.GetParameter(index++).Value = null;
  524. }
  525. _saveItemCommand.GetParameter(index++).Value = item.EndDate;
  526. _saveItemCommand.GetParameter(index++).Value = item.ChannelId;
  527. var hasProgramAttributes = item as IHasProgramAttributes;
  528. if (hasProgramAttributes != null)
  529. {
  530. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsKids;
  531. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsMovie;
  532. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSports;
  533. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSeries;
  534. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsLive;
  535. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsNews;
  536. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsPremiere;
  537. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.EpisodeTitle;
  538. _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsRepeat;
  539. }
  540. else
  541. {
  542. _saveItemCommand.GetParameter(index++).Value = null;
  543. _saveItemCommand.GetParameter(index++).Value = null;
  544. _saveItemCommand.GetParameter(index++).Value = null;
  545. _saveItemCommand.GetParameter(index++).Value = null;
  546. _saveItemCommand.GetParameter(index++).Value = null;
  547. _saveItemCommand.GetParameter(index++).Value = null;
  548. _saveItemCommand.GetParameter(index++).Value = null;
  549. _saveItemCommand.GetParameter(index++).Value = null;
  550. _saveItemCommand.GetParameter(index++).Value = null;
  551. }
  552. _saveItemCommand.GetParameter(index++).Value = item.CommunityRating;
  553. _saveItemCommand.GetParameter(index++).Value = item.CustomRating;
  554. _saveItemCommand.GetParameter(index++).Value = item.IndexNumber;
  555. _saveItemCommand.GetParameter(index++).Value = item.IsLocked;
  556. _saveItemCommand.GetParameter(index++).Value = item.Name;
  557. _saveItemCommand.GetParameter(index++).Value = item.OfficialRating;
  558. _saveItemCommand.GetParameter(index++).Value = item.MediaType;
  559. _saveItemCommand.GetParameter(index++).Value = item.Overview;
  560. _saveItemCommand.GetParameter(index++).Value = item.ParentIndexNumber;
  561. _saveItemCommand.GetParameter(index++).Value = item.PremiereDate;
  562. _saveItemCommand.GetParameter(index++).Value = item.ProductionYear;
  563. if (item.ParentId == Guid.Empty)
  564. {
  565. _saveItemCommand.GetParameter(index++).Value = null;
  566. }
  567. else
  568. {
  569. _saveItemCommand.GetParameter(index++).Value = item.ParentId;
  570. }
  571. _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.Genres.ToArray());
  572. _saveItemCommand.GetParameter(index++).Value = item.GetParentalRatingValue() ?? 0;
  573. _saveItemCommand.GetParameter(index++).Value = item.GetInheritedParentalRatingValue() ?? 0;
  574. _saveItemCommand.GetParameter(index++).Value = LatestSchemaVersion;
  575. _saveItemCommand.GetParameter(index++).Value = item.SortName;
  576. _saveItemCommand.GetParameter(index++).Value = item.RunTimeTicks;
  577. _saveItemCommand.GetParameter(index++).Value = item.OfficialRatingDescription;
  578. _saveItemCommand.GetParameter(index++).Value = item.HomePageUrl;
  579. _saveItemCommand.GetParameter(index++).Value = item.VoteCount;
  580. _saveItemCommand.GetParameter(index++).Value = item.DisplayMediaType;
  581. _saveItemCommand.GetParameter(index++).Value = item.DateCreated;
  582. _saveItemCommand.GetParameter(index++).Value = item.DateModified;
  583. _saveItemCommand.GetParameter(index++).Value = item.ForcedSortName;
  584. _saveItemCommand.GetParameter(index++).Value = item.IsOffline;
  585. _saveItemCommand.GetParameter(index++).Value = item.LocationType.ToString();
  586. _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataLanguage;
  587. _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataCountryCode;
  588. _saveItemCommand.GetParameter(index++).Value = item.IsHD;
  589. _saveItemCommand.GetParameter(index++).Value = item.ExternalEtag;
  590. if (item.DateLastRefreshed == default(DateTime))
  591. {
  592. _saveItemCommand.GetParameter(index++).Value = null;
  593. }
  594. else
  595. {
  596. _saveItemCommand.GetParameter(index++).Value = item.DateLastRefreshed;
  597. }
  598. _saveItemCommand.GetParameter(index++).Value = item.DateLastSaved;
  599. _saveItemCommand.GetParameter(index++).Value = item.IsInMixedFolder;
  600. _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.LockedFields.Select(i => i.ToString()).ToArray());
  601. _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.Studios.ToArray());
  602. if (item.Audio.HasValue)
  603. {
  604. _saveItemCommand.GetParameter(index++).Value = item.Audio.Value.ToString();
  605. }
  606. else
  607. {
  608. _saveItemCommand.GetParameter(index++).Value = null;
  609. }
  610. var tvItem = item as ILiveTvItem;
  611. if (tvItem != null)
  612. {
  613. _saveItemCommand.GetParameter(index++).Value = tvItem.ServiceName;
  614. }
  615. else
  616. {
  617. _saveItemCommand.GetParameter(index++).Value = null;
  618. }
  619. _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.Tags.ToArray());
  620. _saveItemCommand.GetParameter(index++).Value = item.IsFolder;
  621. _saveItemCommand.GetParameter(index++).Value = item.GetBlockUnratedType().ToString();
  622. _saveItemCommand.Transaction = transaction;
  623. _saveItemCommand.ExecuteNonQuery();
  624. if (item.SupportsAncestors)
  625. {
  626. UpdateAncestors(item.Id, item.GetAncestorIds().Distinct().ToList(), transaction);
  627. }
  628. }
  629. transaction.Commit();
  630. }
  631. catch (OperationCanceledException)
  632. {
  633. if (transaction != null)
  634. {
  635. transaction.Rollback();
  636. }
  637. throw;
  638. }
  639. catch (Exception e)
  640. {
  641. _logger.ErrorException("Failed to save items:", e);
  642. if (transaction != null)
  643. {
  644. transaction.Rollback();
  645. }
  646. throw;
  647. }
  648. finally
  649. {
  650. if (transaction != null)
  651. {
  652. transaction.Dispose();
  653. }
  654. _writeLock.Release();
  655. }
  656. }
  657. /// <summary>
  658. /// Internal retrieve from items or users table
  659. /// </summary>
  660. /// <param name="id">The id.</param>
  661. /// <returns>BaseItem.</returns>
  662. /// <exception cref="System.ArgumentNullException">id</exception>
  663. /// <exception cref="System.ArgumentException"></exception>
  664. public BaseItem RetrieveItem(Guid id)
  665. {
  666. if (id == Guid.Empty)
  667. {
  668. throw new ArgumentNullException("id");
  669. }
  670. CheckDisposed();
  671. using (var cmd = _connection.CreateCommand())
  672. {
  673. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid = @guid";
  674. cmd.Parameters.Add(cmd, "@guid", DbType.Guid).Value = id;
  675. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
  676. {
  677. if (reader.Read())
  678. {
  679. return GetItem(reader);
  680. }
  681. }
  682. return null;
  683. }
  684. }
  685. private BaseItem GetItem(IDataReader reader)
  686. {
  687. var typeString = reader.GetString(0);
  688. var type = _typeMapper.GetType(typeString);
  689. if (type == null)
  690. {
  691. _logger.Debug("Unknown type {0}", typeString);
  692. return null;
  693. }
  694. BaseItem item = null;
  695. using (var stream = reader.GetMemoryStream(1))
  696. {
  697. try
  698. {
  699. item = _jsonSerializer.DeserializeFromStream(stream, type) as BaseItem;
  700. }
  701. catch (SerializationException ex)
  702. {
  703. _logger.ErrorException("Error deserializing item", ex);
  704. }
  705. if (item == null)
  706. {
  707. try
  708. {
  709. item = Activator.CreateInstance(type) as BaseItem;
  710. }
  711. catch
  712. {
  713. }
  714. }
  715. if (item == null)
  716. {
  717. return null;
  718. }
  719. }
  720. if (!reader.IsDBNull(2))
  721. {
  722. var hasStartDate = item as IHasStartDate;
  723. if (hasStartDate != null)
  724. {
  725. hasStartDate.StartDate = reader.GetDateTime(2).ToUniversalTime();
  726. }
  727. }
  728. if (!reader.IsDBNull(3))
  729. {
  730. item.EndDate = reader.GetDateTime(3).ToUniversalTime();
  731. }
  732. if (!reader.IsDBNull(4))
  733. {
  734. item.IsOffline = reader.GetBoolean(4);
  735. }
  736. if (!reader.IsDBNull(5))
  737. {
  738. item.ChannelId = reader.GetString(5);
  739. }
  740. var hasProgramAttributes = item as IHasProgramAttributes;
  741. if (hasProgramAttributes != null)
  742. {
  743. if (!reader.IsDBNull(6))
  744. {
  745. hasProgramAttributes.IsMovie = reader.GetBoolean(6);
  746. }
  747. if (!reader.IsDBNull(7))
  748. {
  749. hasProgramAttributes.IsSports = reader.GetBoolean(7);
  750. }
  751. if (!reader.IsDBNull(8))
  752. {
  753. hasProgramAttributes.IsKids = reader.GetBoolean(8);
  754. }
  755. if (!reader.IsDBNull(9))
  756. {
  757. hasProgramAttributes.IsSeries = reader.GetBoolean(9);
  758. }
  759. if (!reader.IsDBNull(10))
  760. {
  761. hasProgramAttributes.IsLive = reader.GetBoolean(10);
  762. }
  763. if (!reader.IsDBNull(11))
  764. {
  765. hasProgramAttributes.IsNews = reader.GetBoolean(11);
  766. }
  767. if (!reader.IsDBNull(12))
  768. {
  769. hasProgramAttributes.IsPremiere = reader.GetBoolean(12);
  770. }
  771. if (!reader.IsDBNull(13))
  772. {
  773. hasProgramAttributes.EpisodeTitle = reader.GetString(13);
  774. }
  775. if (!reader.IsDBNull(14))
  776. {
  777. hasProgramAttributes.IsRepeat = reader.GetBoolean(14);
  778. }
  779. }
  780. if (!reader.IsDBNull(15))
  781. {
  782. item.CommunityRating = reader.GetFloat(15);
  783. }
  784. if (!reader.IsDBNull(16))
  785. {
  786. item.CustomRating = reader.GetString(16);
  787. }
  788. if (!reader.IsDBNull(17))
  789. {
  790. item.IndexNumber = reader.GetInt32(17);
  791. }
  792. if (!reader.IsDBNull(18))
  793. {
  794. item.IsLocked = reader.GetBoolean(18);
  795. }
  796. if (!reader.IsDBNull(19))
  797. {
  798. item.PreferredMetadataLanguage = reader.GetString(19);
  799. }
  800. if (!reader.IsDBNull(20))
  801. {
  802. item.PreferredMetadataCountryCode = reader.GetString(20);
  803. }
  804. if (!reader.IsDBNull(21))
  805. {
  806. item.IsHD = reader.GetBoolean(21);
  807. }
  808. if (!reader.IsDBNull(22))
  809. {
  810. item.ExternalEtag = reader.GetString(22);
  811. }
  812. if (!reader.IsDBNull(23))
  813. {
  814. item.DateLastRefreshed = reader.GetDateTime(23).ToUniversalTime();
  815. }
  816. if (!reader.IsDBNull(24))
  817. {
  818. item.Name = reader.GetString(24);
  819. }
  820. if (!reader.IsDBNull(25))
  821. {
  822. item.Path = reader.GetString(25);
  823. }
  824. if (!reader.IsDBNull(26))
  825. {
  826. item.PremiereDate = reader.GetDateTime(26).ToUniversalTime();
  827. }
  828. if (!reader.IsDBNull(27))
  829. {
  830. item.Overview = reader.GetString(27);
  831. }
  832. if (!reader.IsDBNull(28))
  833. {
  834. item.ParentIndexNumber = reader.GetInt32(28);
  835. }
  836. if (!reader.IsDBNull(29))
  837. {
  838. item.ProductionYear = reader.GetInt32(29);
  839. }
  840. if (!reader.IsDBNull(30))
  841. {
  842. item.OfficialRating = reader.GetString(30);
  843. }
  844. if (!reader.IsDBNull(31))
  845. {
  846. item.OfficialRating = reader.GetString(31);
  847. }
  848. if (!reader.IsDBNull(32))
  849. {
  850. item.HomePageUrl = reader.GetString(32);
  851. }
  852. if (!reader.IsDBNull(33))
  853. {
  854. item.DisplayMediaType = reader.GetString(33);
  855. }
  856. if (!reader.IsDBNull(34))
  857. {
  858. item.ForcedSortName = reader.GetString(34);
  859. }
  860. if (!reader.IsDBNull(35))
  861. {
  862. item.RunTimeTicks = reader.GetInt64(35);
  863. }
  864. if (!reader.IsDBNull(36))
  865. {
  866. item.VoteCount = reader.GetInt32(36);
  867. }
  868. if (!reader.IsDBNull(37))
  869. {
  870. item.DateCreated = reader.GetDateTime(37).ToUniversalTime();
  871. }
  872. if (!reader.IsDBNull(38))
  873. {
  874. item.DateModified = reader.GetDateTime(38).ToUniversalTime();
  875. }
  876. item.Id = reader.GetGuid(39);
  877. if (!reader.IsDBNull(40))
  878. {
  879. item.Genres = reader.GetString(40).Split('|').Where(i => !string.IsNullOrWhiteSpace(i)).ToList();
  880. }
  881. if (!reader.IsDBNull(41))
  882. {
  883. item.ParentId = reader.GetGuid(41);
  884. }
  885. if (!reader.IsDBNull(42))
  886. {
  887. item.Audio = (ProgramAudio)Enum.Parse(typeof(ProgramAudio), reader.GetString(42), true);
  888. }
  889. if (!reader.IsDBNull(43))
  890. {
  891. var tvItem = item as ILiveTvItem;
  892. if (tvItem != null)
  893. {
  894. tvItem.ServiceName = reader.GetString(43);
  895. }
  896. }
  897. return item;
  898. }
  899. /// <summary>
  900. /// Gets the critic reviews.
  901. /// </summary>
  902. /// <param name="itemId">The item id.</param>
  903. /// <returns>Task{IEnumerable{ItemReview}}.</returns>
  904. public IEnumerable<ItemReview> GetCriticReviews(Guid itemId)
  905. {
  906. try
  907. {
  908. var path = Path.Combine(_criticReviewsPath, itemId + ".json");
  909. return _jsonSerializer.DeserializeFromFile<List<ItemReview>>(path);
  910. }
  911. catch (DirectoryNotFoundException)
  912. {
  913. return new List<ItemReview>();
  914. }
  915. catch (FileNotFoundException)
  916. {
  917. return new List<ItemReview>();
  918. }
  919. }
  920. private readonly Task _cachedTask = Task.FromResult(true);
  921. /// <summary>
  922. /// Saves the critic reviews.
  923. /// </summary>
  924. /// <param name="itemId">The item id.</param>
  925. /// <param name="criticReviews">The critic reviews.</param>
  926. /// <returns>Task.</returns>
  927. public Task SaveCriticReviews(Guid itemId, IEnumerable<ItemReview> criticReviews)
  928. {
  929. Directory.CreateDirectory(_criticReviewsPath);
  930. var path = Path.Combine(_criticReviewsPath, itemId + ".json");
  931. _jsonSerializer.SerializeToFile(criticReviews.ToList(), path);
  932. return _cachedTask;
  933. }
  934. /// <summary>
  935. /// Gets chapters for an item
  936. /// </summary>
  937. /// <param name="id">The id.</param>
  938. /// <returns>IEnumerable{ChapterInfo}.</returns>
  939. /// <exception cref="System.ArgumentNullException">id</exception>
  940. public IEnumerable<ChapterInfo> GetChapters(Guid id)
  941. {
  942. CheckDisposed();
  943. if (id == Guid.Empty)
  944. {
  945. throw new ArgumentNullException("id");
  946. }
  947. using (var cmd = _connection.CreateCommand())
  948. {
  949. cmd.CommandText = "select StartPositionTicks,Name,ImagePath from " + ChaptersTableName + " where ItemId = @ItemId order by ChapterIndex asc";
  950. cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = id;
  951. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  952. {
  953. while (reader.Read())
  954. {
  955. yield return GetChapter(reader);
  956. }
  957. }
  958. }
  959. }
  960. /// <summary>
  961. /// Gets a single chapter for an item
  962. /// </summary>
  963. /// <param name="id">The id.</param>
  964. /// <param name="index">The index.</param>
  965. /// <returns>ChapterInfo.</returns>
  966. /// <exception cref="System.ArgumentNullException">id</exception>
  967. public ChapterInfo GetChapter(Guid id, int index)
  968. {
  969. CheckDisposed();
  970. if (id == Guid.Empty)
  971. {
  972. throw new ArgumentNullException("id");
  973. }
  974. using (var cmd = _connection.CreateCommand())
  975. {
  976. cmd.CommandText = "select StartPositionTicks,Name,ImagePath from " + ChaptersTableName + " where ItemId = @ItemId and ChapterIndex=@ChapterIndex";
  977. cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = id;
  978. cmd.Parameters.Add(cmd, "@ChapterIndex", DbType.Int32).Value = index;
  979. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
  980. {
  981. if (reader.Read())
  982. {
  983. return GetChapter(reader);
  984. }
  985. }
  986. return null;
  987. }
  988. }
  989. /// <summary>
  990. /// Gets the chapter.
  991. /// </summary>
  992. /// <param name="reader">The reader.</param>
  993. /// <returns>ChapterInfo.</returns>
  994. private ChapterInfo GetChapter(IDataReader reader)
  995. {
  996. var chapter = new ChapterInfo
  997. {
  998. StartPositionTicks = reader.GetInt64(0)
  999. };
  1000. if (!reader.IsDBNull(1))
  1001. {
  1002. chapter.Name = reader.GetString(1);
  1003. }
  1004. if (!reader.IsDBNull(2))
  1005. {
  1006. chapter.ImagePath = reader.GetString(2);
  1007. }
  1008. return chapter;
  1009. }
  1010. /// <summary>
  1011. /// Saves the chapters.
  1012. /// </summary>
  1013. /// <param name="id">The id.</param>
  1014. /// <param name="chapters">The chapters.</param>
  1015. /// <param name="cancellationToken">The cancellation token.</param>
  1016. /// <returns>Task.</returns>
  1017. /// <exception cref="System.ArgumentNullException">
  1018. /// id
  1019. /// or
  1020. /// chapters
  1021. /// or
  1022. /// cancellationToken
  1023. /// </exception>
  1024. public async Task SaveChapters(Guid id, IEnumerable<ChapterInfo> chapters, CancellationToken cancellationToken)
  1025. {
  1026. CheckDisposed();
  1027. if (id == Guid.Empty)
  1028. {
  1029. throw new ArgumentNullException("id");
  1030. }
  1031. if (chapters == null)
  1032. {
  1033. throw new ArgumentNullException("chapters");
  1034. }
  1035. cancellationToken.ThrowIfCancellationRequested();
  1036. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  1037. IDbTransaction transaction = null;
  1038. try
  1039. {
  1040. transaction = _connection.BeginTransaction();
  1041. // First delete chapters
  1042. _deleteChaptersCommand.GetParameter(0).Value = id;
  1043. _deleteChaptersCommand.Transaction = transaction;
  1044. _deleteChaptersCommand.ExecuteNonQuery();
  1045. var index = 0;
  1046. foreach (var chapter in chapters)
  1047. {
  1048. cancellationToken.ThrowIfCancellationRequested();
  1049. _saveChapterCommand.GetParameter(0).Value = id;
  1050. _saveChapterCommand.GetParameter(1).Value = index;
  1051. _saveChapterCommand.GetParameter(2).Value = chapter.StartPositionTicks;
  1052. _saveChapterCommand.GetParameter(3).Value = chapter.Name;
  1053. _saveChapterCommand.GetParameter(4).Value = chapter.ImagePath;
  1054. _saveChapterCommand.Transaction = transaction;
  1055. _saveChapterCommand.ExecuteNonQuery();
  1056. index++;
  1057. }
  1058. transaction.Commit();
  1059. }
  1060. catch (OperationCanceledException)
  1061. {
  1062. if (transaction != null)
  1063. {
  1064. transaction.Rollback();
  1065. }
  1066. throw;
  1067. }
  1068. catch (Exception e)
  1069. {
  1070. _logger.ErrorException("Failed to save chapters:", e);
  1071. if (transaction != null)
  1072. {
  1073. transaction.Rollback();
  1074. }
  1075. throw;
  1076. }
  1077. finally
  1078. {
  1079. if (transaction != null)
  1080. {
  1081. transaction.Dispose();
  1082. }
  1083. _writeLock.Release();
  1084. }
  1085. }
  1086. /// <summary>
  1087. /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
  1088. /// </summary>
  1089. public void Dispose()
  1090. {
  1091. Dispose(true);
  1092. GC.SuppressFinalize(this);
  1093. }
  1094. private readonly object _disposeLock = new object();
  1095. private bool _disposed;
  1096. private void CheckDisposed()
  1097. {
  1098. if (_disposed)
  1099. {
  1100. throw new ObjectDisposedException(GetType().Name + " has been disposed and cannot be accessed.");
  1101. }
  1102. }
  1103. /// <summary>
  1104. /// Releases unmanaged and - optionally - managed resources.
  1105. /// </summary>
  1106. /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
  1107. protected virtual void Dispose(bool dispose)
  1108. {
  1109. if (dispose)
  1110. {
  1111. _disposed = true;
  1112. try
  1113. {
  1114. lock (_disposeLock)
  1115. {
  1116. _writeLock.Wait();
  1117. if (_connection != null)
  1118. {
  1119. if (_connection.IsOpen())
  1120. {
  1121. _connection.Close();
  1122. }
  1123. _connection.Dispose();
  1124. _connection = null;
  1125. }
  1126. }
  1127. }
  1128. catch (Exception ex)
  1129. {
  1130. _logger.ErrorException("Error disposing database", ex);
  1131. }
  1132. }
  1133. }
  1134. public IEnumerable<Guid> GetChildren(Guid parentId)
  1135. {
  1136. if (parentId == Guid.Empty)
  1137. {
  1138. throw new ArgumentNullException("parentId");
  1139. }
  1140. CheckDisposed();
  1141. using (var cmd = _connection.CreateCommand())
  1142. {
  1143. cmd.CommandText = "select ItemId from ChildrenIds where ParentId = @ParentId";
  1144. cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId;
  1145. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1146. {
  1147. while (reader.Read())
  1148. {
  1149. yield return reader.GetGuid(0);
  1150. }
  1151. }
  1152. }
  1153. }
  1154. public IEnumerable<BaseItem> GetChildrenItems(Guid parentId)
  1155. {
  1156. if (parentId == Guid.Empty)
  1157. {
  1158. throw new ArgumentNullException("parentId");
  1159. }
  1160. CheckDisposed();
  1161. using (var cmd = _connection.CreateCommand())
  1162. {
  1163. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid in (select ItemId from ChildrenIds where ParentId = @ParentId)";
  1164. cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId;
  1165. //_logger.Debug(cmd.CommandText);
  1166. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1167. {
  1168. while (reader.Read())
  1169. {
  1170. var item = GetItem(reader);
  1171. if (item != null)
  1172. {
  1173. yield return item;
  1174. }
  1175. }
  1176. }
  1177. }
  1178. }
  1179. public IEnumerable<BaseItem> GetItemsOfType(Type type)
  1180. {
  1181. if (type == null)
  1182. {
  1183. throw new ArgumentNullException("type");
  1184. }
  1185. CheckDisposed();
  1186. using (var cmd = _connection.CreateCommand())
  1187. {
  1188. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where type = @type";
  1189. cmd.Parameters.Add(cmd, "@type", DbType.String).Value = type.FullName;
  1190. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1191. {
  1192. while (reader.Read())
  1193. {
  1194. var item = GetItem(reader);
  1195. if (item != null)
  1196. {
  1197. yield return item;
  1198. }
  1199. }
  1200. }
  1201. }
  1202. }
  1203. public IEnumerable<BaseItem> GetItemList(InternalItemsQuery query)
  1204. {
  1205. if (query == null)
  1206. {
  1207. throw new ArgumentNullException("query");
  1208. }
  1209. CheckDisposed();
  1210. using (var cmd = _connection.CreateCommand())
  1211. {
  1212. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems";
  1213. var whereClauses = GetWhereClauses(query, cmd, true);
  1214. var whereText = whereClauses.Count == 0 ?
  1215. string.Empty :
  1216. " where " + string.Join(" AND ", whereClauses.ToArray());
  1217. cmd.CommandText += whereText;
  1218. cmd.CommandText += GetOrderByText(query);
  1219. if (query.Limit.HasValue)
  1220. {
  1221. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  1222. }
  1223. //_logger.Debug(cmd.CommandText);
  1224. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1225. {
  1226. while (reader.Read())
  1227. {
  1228. var item = GetItem(reader);
  1229. if (item != null)
  1230. {
  1231. yield return item;
  1232. }
  1233. }
  1234. }
  1235. }
  1236. }
  1237. public QueryResult<BaseItem> GetItems(InternalItemsQuery query)
  1238. {
  1239. if (query == null)
  1240. {
  1241. throw new ArgumentNullException("query");
  1242. }
  1243. CheckDisposed();
  1244. using (var cmd = _connection.CreateCommand())
  1245. {
  1246. cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems";
  1247. var whereClauses = GetWhereClauses(query, cmd, false);
  1248. var whereTextWithoutPaging = whereClauses.Count == 0 ?
  1249. string.Empty :
  1250. " where " + string.Join(" AND ", whereClauses.ToArray());
  1251. whereClauses = GetWhereClauses(query, cmd, true);
  1252. var whereText = whereClauses.Count == 0 ?
  1253. string.Empty :
  1254. " where " + string.Join(" AND ", whereClauses.ToArray());
  1255. cmd.CommandText += whereText;
  1256. cmd.CommandText += GetOrderByText(query);
  1257. if (query.Limit.HasValue)
  1258. {
  1259. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  1260. }
  1261. cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
  1262. _logger.Debug(cmd.CommandText);
  1263. var list = new List<BaseItem>();
  1264. var count = 0;
  1265. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
  1266. {
  1267. while (reader.Read())
  1268. {
  1269. var item = GetItem(reader);
  1270. if (item != null)
  1271. {
  1272. list.Add(item);
  1273. }
  1274. }
  1275. if (reader.NextResult() && reader.Read())
  1276. {
  1277. count = reader.GetInt32(0);
  1278. }
  1279. }
  1280. return new QueryResult<BaseItem>()
  1281. {
  1282. Items = list.ToArray(),
  1283. TotalRecordCount = count
  1284. };
  1285. }
  1286. }
  1287. private string GetOrderByText(InternalItemsQuery query)
  1288. {
  1289. if (query.SortBy == null || query.SortBy.Length == 0)
  1290. {
  1291. return string.Empty;
  1292. }
  1293. var sortOrder = query.SortOrder == SortOrder.Descending ? "DESC" : "ASC";
  1294. return " ORDER BY " + string.Join(",", query.SortBy.Select(i => MapOrderByField(i) + " " + sortOrder).ToArray());
  1295. }
  1296. private string MapOrderByField(string name)
  1297. {
  1298. if (string.Equals(name, "airtime", StringComparison.OrdinalIgnoreCase))
  1299. {
  1300. // TODO
  1301. return "SortName";
  1302. }
  1303. return name;
  1304. }
  1305. public List<Guid> GetItemIdsList(InternalItemsQuery query)
  1306. {
  1307. if (query == null)
  1308. {
  1309. throw new ArgumentNullException("query");
  1310. }
  1311. CheckDisposed();
  1312. using (var cmd = _connection.CreateCommand())
  1313. {
  1314. cmd.CommandText = "select guid from TypedBaseItems";
  1315. var whereClauses = GetWhereClauses(query, cmd, true);
  1316. var whereText = whereClauses.Count == 0 ?
  1317. string.Empty :
  1318. " where " + string.Join(" AND ", whereClauses.ToArray());
  1319. cmd.CommandText += whereText;
  1320. cmd.CommandText += GetOrderByText(query);
  1321. if (query.Limit.HasValue)
  1322. {
  1323. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  1324. }
  1325. var list = new List<Guid>();
  1326. _logger.Debug(cmd.CommandText);
  1327. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1328. {
  1329. while (reader.Read())
  1330. {
  1331. list.Add(reader.GetGuid(0));
  1332. }
  1333. }
  1334. return list;
  1335. }
  1336. }
  1337. public QueryResult<Tuple<Guid, string>> GetItemIdsWithPath(InternalItemsQuery query)
  1338. {
  1339. if (query == null)
  1340. {
  1341. throw new ArgumentNullException("query");
  1342. }
  1343. CheckDisposed();
  1344. using (var cmd = _connection.CreateCommand())
  1345. {
  1346. cmd.CommandText = "select guid,path from TypedBaseItems";
  1347. var whereClauses = GetWhereClauses(query, cmd, false);
  1348. var whereTextWithoutPaging = whereClauses.Count == 0 ?
  1349. string.Empty :
  1350. " where " + string.Join(" AND ", whereClauses.ToArray());
  1351. whereClauses = GetWhereClauses(query, cmd, true);
  1352. var whereText = whereClauses.Count == 0 ?
  1353. string.Empty :
  1354. " where " + string.Join(" AND ", whereClauses.ToArray());
  1355. cmd.CommandText += whereText;
  1356. cmd.CommandText += GetOrderByText(query);
  1357. if (query.Limit.HasValue)
  1358. {
  1359. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  1360. }
  1361. cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
  1362. var list = new List<Tuple<Guid, string>>();
  1363. var count = 0;
  1364. _logger.Debug(cmd.CommandText);
  1365. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
  1366. {
  1367. while (reader.Read())
  1368. {
  1369. var id = reader.GetGuid(0);
  1370. string path = null;
  1371. if (!reader.IsDBNull(1))
  1372. {
  1373. path = reader.GetString(1);
  1374. }
  1375. list.Add(new Tuple<Guid, string>(id, path));
  1376. }
  1377. if (reader.NextResult() && reader.Read())
  1378. {
  1379. count = reader.GetInt32(0);
  1380. }
  1381. }
  1382. return new QueryResult<Tuple<Guid, string>>()
  1383. {
  1384. Items = list.ToArray(),
  1385. TotalRecordCount = count
  1386. };
  1387. }
  1388. }
  1389. public QueryResult<Guid> GetItemIds(InternalItemsQuery query)
  1390. {
  1391. if (query == null)
  1392. {
  1393. throw new ArgumentNullException("query");
  1394. }
  1395. CheckDisposed();
  1396. using (var cmd = _connection.CreateCommand())
  1397. {
  1398. cmd.CommandText = "select guid from TypedBaseItems";
  1399. var whereClauses = GetWhereClauses(query, cmd, false);
  1400. var whereTextWithoutPaging = whereClauses.Count == 0 ?
  1401. string.Empty :
  1402. " where " + string.Join(" AND ", whereClauses.ToArray());
  1403. whereClauses = GetWhereClauses(query, cmd, true);
  1404. var whereText = whereClauses.Count == 0 ?
  1405. string.Empty :
  1406. " where " + string.Join(" AND ", whereClauses.ToArray());
  1407. cmd.CommandText += whereText;
  1408. cmd.CommandText += GetOrderByText(query);
  1409. if (query.Limit.HasValue)
  1410. {
  1411. cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
  1412. }
  1413. cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
  1414. var list = new List<Guid>();
  1415. var count = 0;
  1416. _logger.Debug(cmd.CommandText);
  1417. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
  1418. {
  1419. while (reader.Read())
  1420. {
  1421. list.Add(reader.GetGuid(0));
  1422. }
  1423. if (reader.NextResult() && reader.Read())
  1424. {
  1425. count = reader.GetInt32(0);
  1426. }
  1427. }
  1428. return new QueryResult<Guid>()
  1429. {
  1430. Items = list.ToArray(),
  1431. TotalRecordCount = count
  1432. };
  1433. }
  1434. }
  1435. private List<string> GetWhereClauses(InternalItemsQuery query, IDbCommand cmd, bool addPaging)
  1436. {
  1437. var whereClauses = new List<string>();
  1438. if (query.IsCurrentSchema.HasValue)
  1439. {
  1440. if (query.IsCurrentSchema.Value)
  1441. {
  1442. whereClauses.Add("(SchemaVersion not null AND SchemaVersion=@SchemaVersion)");
  1443. }
  1444. else
  1445. {
  1446. whereClauses.Add("(SchemaVersion is null or SchemaVersion<>@SchemaVersion)");
  1447. }
  1448. cmd.Parameters.Add(cmd, "@SchemaVersion", DbType.Int32).Value = LatestSchemaVersion;
  1449. }
  1450. if (query.IsOffline.HasValue)
  1451. {
  1452. whereClauses.Add("IsOffline=@IsOffline");
  1453. cmd.Parameters.Add(cmd, "@IsOffline", DbType.Boolean).Value = query.IsOffline;
  1454. }
  1455. if (query.LocationType.HasValue)
  1456. {
  1457. whereClauses.Add("LocationType=@LocationType");
  1458. cmd.Parameters.Add(cmd, "@LocationType", DbType.String).Value = query.LocationType.Value;
  1459. }
  1460. if (query.IsMovie.HasValue)
  1461. {
  1462. whereClauses.Add("IsMovie=@IsMovie");
  1463. cmd.Parameters.Add(cmd, "@IsMovie", DbType.Boolean).Value = query.IsMovie;
  1464. }
  1465. if (query.IsKids.HasValue)
  1466. {
  1467. whereClauses.Add("IsKids=@IsKids");
  1468. cmd.Parameters.Add(cmd, "@IsKids", DbType.Boolean).Value = query.IsKids;
  1469. }
  1470. if (query.IsSports.HasValue)
  1471. {
  1472. whereClauses.Add("IsSports=@IsSports");
  1473. cmd.Parameters.Add(cmd, "@IsSports", DbType.Boolean).Value = query.IsSports;
  1474. }
  1475. if (query.IsFolder.HasValue)
  1476. {
  1477. whereClauses.Add("IsFolder=@IsFolder");
  1478. cmd.Parameters.Add(cmd, "@IsFolder", DbType.Boolean).Value = query.IsFolder;
  1479. }
  1480. var includeTypes = query.IncludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray();
  1481. if (includeTypes.Length == 1)
  1482. {
  1483. whereClauses.Add("type=@type");
  1484. cmd.Parameters.Add(cmd, "@type", DbType.String).Value = includeTypes[0];
  1485. }
  1486. else if (includeTypes.Length > 1)
  1487. {
  1488. var inClause = string.Join(",", includeTypes.Select(i => "'" + i + "'").ToArray());
  1489. whereClauses.Add(string.Format("type in ({0})", inClause));
  1490. }
  1491. var excludeTypes = query.ExcludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray();
  1492. if (excludeTypes.Length == 1)
  1493. {
  1494. whereClauses.Add("type<>@type");
  1495. cmd.Parameters.Add(cmd, "@type", DbType.String).Value = excludeTypes[0];
  1496. }
  1497. else if (excludeTypes.Length > 1)
  1498. {
  1499. var inClause = string.Join(",", excludeTypes.Select(i => "'" + i + "'").ToArray());
  1500. whereClauses.Add(string.Format("type not in ({0})", inClause));
  1501. }
  1502. if (query.ChannelIds.Length == 1)
  1503. {
  1504. whereClauses.Add("ChannelId=@ChannelId");
  1505. cmd.Parameters.Add(cmd, "@ChannelId", DbType.String).Value = query.ChannelIds[0];
  1506. }
  1507. if (query.ChannelIds.Length > 1)
  1508. {
  1509. var inClause = string.Join(",", query.ChannelIds.Select(i => "'" + i + "'").ToArray());
  1510. whereClauses.Add(string.Format("ChannelId in ({0})", inClause));
  1511. }
  1512. if (query.ParentId.HasValue)
  1513. {
  1514. whereClauses.Add("ParentId=@ParentId");
  1515. cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = query.ParentId.Value;
  1516. }
  1517. if (query.MinEndDate.HasValue)
  1518. {
  1519. whereClauses.Add("EndDate>=@MinEndDate");
  1520. cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = query.MinEndDate.Value;
  1521. }
  1522. if (query.MaxEndDate.HasValue)
  1523. {
  1524. whereClauses.Add("EndDate<=@MaxEndDate");
  1525. cmd.Parameters.Add(cmd, "@MaxEndDate", DbType.Date).Value = query.MaxEndDate.Value;
  1526. }
  1527. if (query.MinStartDate.HasValue)
  1528. {
  1529. whereClauses.Add("StartDate>=@MinStartDate");
  1530. cmd.Parameters.Add(cmd, "@MinStartDate", DbType.Date).Value = query.MinStartDate.Value;
  1531. }
  1532. if (query.MinPremiereDate.HasValue)
  1533. {
  1534. whereClauses.Add("PremiereDate>=@MinPremiereDate");
  1535. cmd.Parameters.Add(cmd, "@MinPremiereDate", DbType.Date).Value = query.MinPremiereDate.Value;
  1536. }
  1537. if (query.MaxStartDate.HasValue)
  1538. {
  1539. whereClauses.Add("StartDate<=@MaxStartDate");
  1540. cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = query.MaxStartDate.Value;
  1541. }
  1542. if (query.IsAiring.HasValue)
  1543. {
  1544. if (query.IsAiring.Value)
  1545. {
  1546. whereClauses.Add("StartDate<=@MaxStartDate");
  1547. cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = DateTime.UtcNow;
  1548. whereClauses.Add("EndDate>=@MinEndDate");
  1549. cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = DateTime.UtcNow;
  1550. }
  1551. else
  1552. {
  1553. whereClauses.Add("(StartDate>@IsAiringDate OR EndDate < @IsAiringDate)");
  1554. cmd.Parameters.Add(cmd, "@IsAiringDate", DbType.Date).Value = DateTime.UtcNow;
  1555. }
  1556. }
  1557. if (!string.IsNullOrWhiteSpace(query.Person))
  1558. {
  1559. whereClauses.Add("Guid in (select ItemId from People where Name=@PersonName)");
  1560. cmd.Parameters.Add(cmd, "@PersonName", DbType.String).Value = query.Person;
  1561. }
  1562. if (!string.IsNullOrWhiteSpace(query.NameContains))
  1563. {
  1564. whereClauses.Add("Name like @NameContains");
  1565. cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%";
  1566. }
  1567. if (query.Genres.Length > 0)
  1568. {
  1569. var genres = new List<string>();
  1570. var index = 0;
  1571. foreach (var genre in query.Genres)
  1572. {
  1573. genres.Add("Genres like @Genres" + index);
  1574. cmd.Parameters.Add(cmd, "@Genres" + index, DbType.String).Value = "%" + genre + "%";
  1575. index++;
  1576. }
  1577. var genreCaluse = "(" + string.Join(" OR ", genres.ToArray()) + ")";
  1578. whereClauses.Add(genreCaluse);
  1579. }
  1580. if (query.MaxParentalRating.HasValue)
  1581. {
  1582. whereClauses.Add("InheritedParentalRatingValue<=@MaxParentalRating");
  1583. cmd.Parameters.Add(cmd, "@MaxParentalRating", DbType.Int32).Value = query.MaxParentalRating.Value;
  1584. }
  1585. if (query.HasParentalRating.HasValue)
  1586. {
  1587. if (query.HasParentalRating.Value)
  1588. {
  1589. whereClauses.Add("InheritedParentalRatingValue > 0");
  1590. }
  1591. else
  1592. {
  1593. whereClauses.Add("InheritedParentalRatingValue = 0");
  1594. }
  1595. }
  1596. if (query.HasDeadParentId.HasValue)
  1597. {
  1598. if (query.HasDeadParentId.Value)
  1599. {
  1600. whereClauses.Add("ParentId NOT NULL AND ParentId NOT IN (select guid from TypedBaseItems)");
  1601. }
  1602. }
  1603. if (query.ExcludeLocationTypes.Length == 1)
  1604. {
  1605. whereClauses.Add("LocationType<>@LocationType");
  1606. cmd.Parameters.Add(cmd, "@LocationType", DbType.String).Value = query.ExcludeLocationTypes[0].ToString();
  1607. }
  1608. if (query.ExcludeLocationTypes.Length > 1)
  1609. {
  1610. var val = string.Join(",", query.ExcludeLocationTypes.Select(i => "'" + i + "'").ToArray());
  1611. whereClauses.Add("LocationType not in (" + val + ")");
  1612. }
  1613. if (query.AncestorIds.Length == 1)
  1614. {
  1615. whereClauses.Add("Guid in (select itemId from AncestorIds where AncestorId=@AncestorId)");
  1616. cmd.Parameters.Add(cmd, "@AncestorId", DbType.Guid).Value = new Guid(query.AncestorIds[0]);
  1617. }
  1618. if (query.AncestorIds.Length > 1)
  1619. {
  1620. var inClause = string.Join(",", query.AncestorIds.Select(i => "'" + new Guid(i).ToString("N") + "'").ToArray());
  1621. whereClauses.Add(string.Format("Guid in (select itemId from AncestorIds where AncestorIdText in ({0}))", inClause));
  1622. }
  1623. if (query.BlockUnratedItems.Length == 1)
  1624. {
  1625. whereClauses.Add("(InheritedParentalRatingValue > 0 or UnratedType <> @UnratedType)");
  1626. cmd.Parameters.Add(cmd, "@UnratedType", DbType.String).Value = query.BlockUnratedItems[0].ToString();
  1627. }
  1628. if (query.BlockUnratedItems.Length > 1)
  1629. {
  1630. var inClause = string.Join(",", query.BlockUnratedItems.Select(i => "'" + i.ToString() + "'").ToArray());
  1631. whereClauses.Add(string.Format("(InheritedParentalRatingValue > 0 or UnratedType not in ({0}))", inClause));
  1632. }
  1633. if (addPaging)
  1634. {
  1635. if (query.StartIndex.HasValue && query.StartIndex.Value > 0)
  1636. {
  1637. var pagingWhereText = whereClauses.Count == 0 ?
  1638. string.Empty :
  1639. " where " + string.Join(" AND ", whereClauses.ToArray());
  1640. var orderBy = GetOrderByText(query);
  1641. whereClauses.Add(string.Format("guid NOT IN (SELECT guid FROM TypedBaseItems {0}" + orderBy + " LIMIT {1})",
  1642. pagingWhereText,
  1643. query.StartIndex.Value.ToString(CultureInfo.InvariantCulture)));
  1644. }
  1645. }
  1646. return whereClauses;
  1647. }
  1648. private static readonly Type[] KnownTypes =
  1649. {
  1650. typeof(LiveTvProgram),
  1651. typeof(LiveTvChannel),
  1652. typeof(LiveTvVideoRecording),
  1653. typeof(LiveTvAudioRecording),
  1654. typeof(Series),
  1655. typeof(Audio),
  1656. typeof(MusicAlbum),
  1657. typeof(MusicArtist),
  1658. typeof(MusicGenre),
  1659. typeof(MusicVideo),
  1660. typeof(Movie),
  1661. typeof(BoxSet),
  1662. typeof(Episode),
  1663. typeof(ChannelVideoItem),
  1664. typeof(Season),
  1665. typeof(Series),
  1666. typeof(Book),
  1667. typeof(CollectionFolder),
  1668. typeof(Folder),
  1669. typeof(Game),
  1670. typeof(GameGenre),
  1671. typeof(GameSystem),
  1672. typeof(Genre),
  1673. typeof(Person),
  1674. typeof(Photo),
  1675. typeof(PhotoAlbum),
  1676. typeof(Studio),
  1677. typeof(UserRootFolder),
  1678. typeof(UserView),
  1679. typeof(Video),
  1680. typeof(Year),
  1681. typeof(Channel),
  1682. typeof(AggregateFolder)
  1683. };
  1684. public async Task UpdateInheritedValues(CancellationToken cancellationToken)
  1685. {
  1686. var newValues = new List<Tuple<Guid, int>>();
  1687. using (var cmd = _connection.CreateCommand())
  1688. {
  1689. cmd.CommandText = "select Guid,InheritedParentalRatingValue,(select Max(ParentalRatingValue, (select COALESCE(MAX(ParentalRatingValue),0) from TypedBaseItems where guid in (Select AncestorId from AncestorIds where ItemId=Outer.guid)))) as NewInheritedParentalRatingValue from typedbaseitems as Outer where InheritedParentalRatingValue <> NewInheritedParentalRatingValue";
  1690. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1691. {
  1692. while (reader.Read())
  1693. {
  1694. var id = reader.GetGuid(0);
  1695. var newValue = reader.GetInt32(2);
  1696. newValues.Add(new Tuple<Guid, int>(id, newValue));
  1697. }
  1698. }
  1699. }
  1700. if (newValues.Count == 0)
  1701. {
  1702. return;
  1703. }
  1704. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  1705. IDbTransaction transaction = null;
  1706. try
  1707. {
  1708. transaction = _connection.BeginTransaction();
  1709. foreach (var item in newValues)
  1710. {
  1711. _updateInheritedRatingCommand.GetParameter(0).Value = item.Item1;
  1712. _updateInheritedRatingCommand.GetParameter(1).Value = item.Item2;
  1713. _updateInheritedRatingCommand.Transaction = transaction;
  1714. _updateInheritedRatingCommand.ExecuteNonQuery();
  1715. _updateInheritedRatingCommand.ExecuteNonQuery();
  1716. }
  1717. transaction.Commit();
  1718. }
  1719. catch (OperationCanceledException)
  1720. {
  1721. if (transaction != null)
  1722. {
  1723. transaction.Rollback();
  1724. }
  1725. throw;
  1726. }
  1727. catch (Exception e)
  1728. {
  1729. _logger.ErrorException("Error running query:", e);
  1730. if (transaction != null)
  1731. {
  1732. transaction.Rollback();
  1733. }
  1734. throw;
  1735. }
  1736. finally
  1737. {
  1738. if (transaction != null)
  1739. {
  1740. transaction.Dispose();
  1741. }
  1742. _writeLock.Release();
  1743. }
  1744. }
  1745. private static Dictionary<string, string[]> GetTypeMapDictionary()
  1746. {
  1747. var dict = new Dictionary<string, string[]>();
  1748. foreach (var t in KnownTypes)
  1749. {
  1750. dict[t.Name] = new[] { t.FullName };
  1751. }
  1752. dict["ChannelItem"] = new[] { typeof(ChannelVideoItem).FullName, typeof(ChannelAudioItem).FullName, typeof(ChannelFolderItem).FullName };
  1753. dict["LiveTvItem"] = new[] { typeof(LiveTvAudioRecording).FullName, typeof(LiveTvVideoRecording).FullName, typeof(LiveTvChannel).FullName, typeof(LiveTvProgram).FullName };
  1754. dict["Recording"] = new[] { typeof(LiveTvAudioRecording).FullName, typeof(LiveTvVideoRecording).FullName };
  1755. dict["Program"] = new[] { typeof(LiveTvProgram).FullName };
  1756. dict["TvChannel"] = new[] { typeof(LiveTvChannel).FullName };
  1757. return dict;
  1758. }
  1759. // Not crazy about having this all the way down here, but at least it's in one place
  1760. readonly Dictionary<string, string[]> _types = GetTypeMapDictionary();
  1761. private IEnumerable<string> MapIncludeItemTypes(string value)
  1762. {
  1763. string[] result;
  1764. if (_types.TryGetValue(value, out result))
  1765. {
  1766. return result;
  1767. }
  1768. return new[] { value };
  1769. }
  1770. public async Task DeleteItem(Guid id, CancellationToken cancellationToken)
  1771. {
  1772. if (id == Guid.Empty)
  1773. {
  1774. throw new ArgumentNullException("id");
  1775. }
  1776. CheckDisposed();
  1777. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  1778. IDbTransaction transaction = null;
  1779. try
  1780. {
  1781. transaction = _connection.BeginTransaction();
  1782. // First delete children
  1783. _deleteChildrenCommand.GetParameter(0).Value = id;
  1784. _deleteChildrenCommand.Transaction = transaction;
  1785. _deleteChildrenCommand.ExecuteNonQuery();
  1786. // Delete people
  1787. _deletePeopleCommand.GetParameter(0).Value = id;
  1788. _deletePeopleCommand.Transaction = transaction;
  1789. _deletePeopleCommand.ExecuteNonQuery();
  1790. // Delete chapters
  1791. _deleteChaptersCommand.GetParameter(0).Value = id;
  1792. _deleteChaptersCommand.Transaction = transaction;
  1793. _deleteChaptersCommand.ExecuteNonQuery();
  1794. // Delete media streams
  1795. _deleteStreamsCommand.GetParameter(0).Value = id;
  1796. _deleteStreamsCommand.Transaction = transaction;
  1797. _deleteStreamsCommand.ExecuteNonQuery();
  1798. // Delete ancestors
  1799. _deleteAncestorsCommand.GetParameter(0).Value = id;
  1800. _deleteAncestorsCommand.Transaction = transaction;
  1801. _deleteAncestorsCommand.ExecuteNonQuery();
  1802. // Delete the item
  1803. _deleteItemCommand.GetParameter(0).Value = id;
  1804. _deleteItemCommand.Transaction = transaction;
  1805. _deleteItemCommand.ExecuteNonQuery();
  1806. transaction.Commit();
  1807. }
  1808. catch (OperationCanceledException)
  1809. {
  1810. if (transaction != null)
  1811. {
  1812. transaction.Rollback();
  1813. }
  1814. throw;
  1815. }
  1816. catch (Exception e)
  1817. {
  1818. _logger.ErrorException("Failed to save children:", e);
  1819. if (transaction != null)
  1820. {
  1821. transaction.Rollback();
  1822. }
  1823. throw;
  1824. }
  1825. finally
  1826. {
  1827. if (transaction != null)
  1828. {
  1829. transaction.Dispose();
  1830. }
  1831. _writeLock.Release();
  1832. }
  1833. }
  1834. public async Task SaveChildren(Guid parentId, IEnumerable<Guid> children, CancellationToken cancellationToken)
  1835. {
  1836. if (parentId == Guid.Empty)
  1837. {
  1838. throw new ArgumentNullException("parentId");
  1839. }
  1840. if (children == null)
  1841. {
  1842. throw new ArgumentNullException("children");
  1843. }
  1844. CheckDisposed();
  1845. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  1846. IDbTransaction transaction = null;
  1847. try
  1848. {
  1849. transaction = _connection.BeginTransaction();
  1850. // First delete
  1851. _deleteChildrenCommand.GetParameter(0).Value = parentId;
  1852. _deleteChildrenCommand.Transaction = transaction;
  1853. _deleteChildrenCommand.ExecuteNonQuery();
  1854. foreach (var id in children)
  1855. {
  1856. cancellationToken.ThrowIfCancellationRequested();
  1857. _saveChildrenCommand.GetParameter(0).Value = parentId;
  1858. _saveChildrenCommand.GetParameter(1).Value = id;
  1859. _saveChildrenCommand.Transaction = transaction;
  1860. _saveChildrenCommand.ExecuteNonQuery();
  1861. }
  1862. transaction.Commit();
  1863. }
  1864. catch (OperationCanceledException)
  1865. {
  1866. if (transaction != null)
  1867. {
  1868. transaction.Rollback();
  1869. }
  1870. throw;
  1871. }
  1872. catch (Exception e)
  1873. {
  1874. _logger.ErrorException("Failed to save children:", e);
  1875. if (transaction != null)
  1876. {
  1877. transaction.Rollback();
  1878. }
  1879. throw;
  1880. }
  1881. finally
  1882. {
  1883. if (transaction != null)
  1884. {
  1885. transaction.Dispose();
  1886. }
  1887. _writeLock.Release();
  1888. }
  1889. }
  1890. public List<string> GetPeopleNames(InternalPeopleQuery query)
  1891. {
  1892. if (query == null)
  1893. {
  1894. throw new ArgumentNullException("query");
  1895. }
  1896. CheckDisposed();
  1897. using (var cmd = _connection.CreateCommand())
  1898. {
  1899. cmd.CommandText = "select Distinct Name from People";
  1900. var whereClauses = GetPeopleWhereClauses(query, cmd);
  1901. if (whereClauses.Count > 0)
  1902. {
  1903. cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray());
  1904. }
  1905. cmd.CommandText += " order by ListOrder";
  1906. var list = new List<string>();
  1907. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1908. {
  1909. while (reader.Read())
  1910. {
  1911. list.Add(reader.GetString(0));
  1912. }
  1913. }
  1914. return list;
  1915. }
  1916. }
  1917. public List<PersonInfo> GetPeople(InternalPeopleQuery query)
  1918. {
  1919. if (query == null)
  1920. {
  1921. throw new ArgumentNullException("query");
  1922. }
  1923. CheckDisposed();
  1924. using (var cmd = _connection.CreateCommand())
  1925. {
  1926. cmd.CommandText = "select ItemId, Name, Role, PersonType, SortOrder from People";
  1927. var whereClauses = GetPeopleWhereClauses(query, cmd);
  1928. if (whereClauses.Count > 0)
  1929. {
  1930. cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray());
  1931. }
  1932. cmd.CommandText += " order by ListOrder";
  1933. var list = new List<PersonInfo>();
  1934. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  1935. {
  1936. while (reader.Read())
  1937. {
  1938. list.Add(GetPerson(reader));
  1939. }
  1940. }
  1941. return list;
  1942. }
  1943. }
  1944. private List<string> GetPeopleWhereClauses(InternalPeopleQuery query, IDbCommand cmd)
  1945. {
  1946. var whereClauses = new List<string>();
  1947. if (query.ItemId != Guid.Empty)
  1948. {
  1949. whereClauses.Add("ItemId=@ItemId");
  1950. cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = query.ItemId;
  1951. }
  1952. if (query.AppearsInItemId != Guid.Empty)
  1953. {
  1954. whereClauses.Add("Name in (Select Name from People where ItemId=@AppearsInItemId)");
  1955. cmd.Parameters.Add(cmd, "@AppearsInItemId", DbType.Guid).Value = query.AppearsInItemId;
  1956. }
  1957. if (query.PersonTypes.Count == 1)
  1958. {
  1959. whereClauses.Add("PersonType=@PersonType");
  1960. cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.PersonTypes[0];
  1961. }
  1962. if (query.PersonTypes.Count > 1)
  1963. {
  1964. var val = string.Join(",", query.PersonTypes.Select(i => "'" + i + "'").ToArray());
  1965. whereClauses.Add("PersonType in (" + val + ")");
  1966. }
  1967. if (query.ExcludePersonTypes.Count == 1)
  1968. {
  1969. whereClauses.Add("PersonType<>@PersonType");
  1970. cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.ExcludePersonTypes[0];
  1971. }
  1972. if (query.ExcludePersonTypes.Count > 1)
  1973. {
  1974. var val = string.Join(",", query.ExcludePersonTypes.Select(i => "'" + i + "'").ToArray());
  1975. whereClauses.Add("PersonType not in (" + val + ")");
  1976. }
  1977. if (query.MaxListOrder.HasValue)
  1978. {
  1979. whereClauses.Add("ListOrder<=@MaxListOrder");
  1980. cmd.Parameters.Add(cmd, "@MaxListOrder", DbType.Int32).Value = query.MaxListOrder.Value;
  1981. }
  1982. if (!string.IsNullOrWhiteSpace(query.NameContains))
  1983. {
  1984. whereClauses.Add("Name like @NameContains");
  1985. cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%";
  1986. }
  1987. return whereClauses;
  1988. }
  1989. private void UpdateAncestors(Guid itemId, List<Guid> ancestorIds, IDbTransaction transaction)
  1990. {
  1991. if (itemId == Guid.Empty)
  1992. {
  1993. throw new ArgumentNullException("itemId");
  1994. }
  1995. if (ancestorIds == null)
  1996. {
  1997. throw new ArgumentNullException("ancestorIds");
  1998. }
  1999. CheckDisposed();
  2000. // First delete
  2001. _deleteAncestorsCommand.GetParameter(0).Value = itemId;
  2002. _deleteAncestorsCommand.Transaction = transaction;
  2003. _deleteAncestorsCommand.ExecuteNonQuery();
  2004. foreach (var ancestorId in ancestorIds)
  2005. {
  2006. _saveAncestorCommand.GetParameter(0).Value = itemId;
  2007. _saveAncestorCommand.GetParameter(1).Value = ancestorId;
  2008. _saveAncestorCommand.GetParameter(2).Value = ancestorId.ToString("N");
  2009. _saveAncestorCommand.Transaction = transaction;
  2010. _saveAncestorCommand.ExecuteNonQuery();
  2011. }
  2012. }
  2013. public async Task UpdatePeople(Guid itemId, List<PersonInfo> people)
  2014. {
  2015. if (itemId == Guid.Empty)
  2016. {
  2017. throw new ArgumentNullException("itemId");
  2018. }
  2019. if (people == null)
  2020. {
  2021. throw new ArgumentNullException("people");
  2022. }
  2023. CheckDisposed();
  2024. var cancellationToken = CancellationToken.None;
  2025. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  2026. IDbTransaction transaction = null;
  2027. try
  2028. {
  2029. transaction = _connection.BeginTransaction();
  2030. // First delete
  2031. _deletePeopleCommand.GetParameter(0).Value = itemId;
  2032. _deletePeopleCommand.Transaction = transaction;
  2033. _deletePeopleCommand.ExecuteNonQuery();
  2034. var listIndex = 0;
  2035. foreach (var person in people)
  2036. {
  2037. cancellationToken.ThrowIfCancellationRequested();
  2038. _savePersonCommand.GetParameter(0).Value = itemId;
  2039. _savePersonCommand.GetParameter(1).Value = person.Name;
  2040. _savePersonCommand.GetParameter(2).Value = person.Role;
  2041. _savePersonCommand.GetParameter(3).Value = person.Type;
  2042. _savePersonCommand.GetParameter(4).Value = person.SortOrder;
  2043. _savePersonCommand.GetParameter(5).Value = listIndex;
  2044. _savePersonCommand.Transaction = transaction;
  2045. _savePersonCommand.ExecuteNonQuery();
  2046. listIndex++;
  2047. }
  2048. transaction.Commit();
  2049. }
  2050. catch (OperationCanceledException)
  2051. {
  2052. if (transaction != null)
  2053. {
  2054. transaction.Rollback();
  2055. }
  2056. throw;
  2057. }
  2058. catch (Exception e)
  2059. {
  2060. _logger.ErrorException("Failed to save people:", e);
  2061. if (transaction != null)
  2062. {
  2063. transaction.Rollback();
  2064. }
  2065. throw;
  2066. }
  2067. finally
  2068. {
  2069. if (transaction != null)
  2070. {
  2071. transaction.Dispose();
  2072. }
  2073. _writeLock.Release();
  2074. }
  2075. }
  2076. private PersonInfo GetPerson(IDataReader reader)
  2077. {
  2078. var item = new PersonInfo();
  2079. item.ItemId = reader.GetGuid(0);
  2080. item.Name = reader.GetString(1);
  2081. if (!reader.IsDBNull(2))
  2082. {
  2083. item.Role = reader.GetString(2);
  2084. }
  2085. if (!reader.IsDBNull(3))
  2086. {
  2087. item.Type = reader.GetString(3);
  2088. }
  2089. if (!reader.IsDBNull(4))
  2090. {
  2091. item.SortOrder = reader.GetInt32(4);
  2092. }
  2093. return item;
  2094. }
  2095. public IEnumerable<MediaStream> GetMediaStreams(MediaStreamQuery query)
  2096. {
  2097. CheckDisposed();
  2098. if (query == null)
  2099. {
  2100. throw new ArgumentNullException("query");
  2101. }
  2102. using (var cmd = _connection.CreateCommand())
  2103. {
  2104. var cmdText = "select " + string.Join(",", _mediaStreamSaveColumns) + " from mediastreams where";
  2105. cmdText += " ItemId=@ItemId";
  2106. cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = query.ItemId;
  2107. if (query.Type.HasValue)
  2108. {
  2109. cmdText += " AND StreamType=@StreamType";
  2110. cmd.Parameters.Add(cmd, "@StreamType", DbType.String).Value = query.Type.Value.ToString();
  2111. }
  2112. if (query.Index.HasValue)
  2113. {
  2114. cmdText += " AND StreamIndex=@StreamIndex";
  2115. cmd.Parameters.Add(cmd, "@StreamIndex", DbType.Int32).Value = query.Index.Value;
  2116. }
  2117. cmdText += " order by StreamIndex ASC";
  2118. cmd.CommandText = cmdText;
  2119. using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
  2120. {
  2121. while (reader.Read())
  2122. {
  2123. yield return GetMediaStream(reader);
  2124. }
  2125. }
  2126. }
  2127. }
  2128. public async Task SaveMediaStreams(Guid id, IEnumerable<MediaStream> streams, CancellationToken cancellationToken)
  2129. {
  2130. CheckDisposed();
  2131. if (id == Guid.Empty)
  2132. {
  2133. throw new ArgumentNullException("id");
  2134. }
  2135. if (streams == null)
  2136. {
  2137. throw new ArgumentNullException("streams");
  2138. }
  2139. cancellationToken.ThrowIfCancellationRequested();
  2140. await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
  2141. IDbTransaction transaction = null;
  2142. try
  2143. {
  2144. transaction = _connection.BeginTransaction();
  2145. // First delete chapters
  2146. _deleteStreamsCommand.GetParameter(0).Value = id;
  2147. _deleteStreamsCommand.Transaction = transaction;
  2148. _deleteStreamsCommand.ExecuteNonQuery();
  2149. foreach (var stream in streams)
  2150. {
  2151. cancellationToken.ThrowIfCancellationRequested();
  2152. var index = 0;
  2153. _saveStreamCommand.GetParameter(index++).Value = id;
  2154. _saveStreamCommand.GetParameter(index++).Value = stream.Index;
  2155. _saveStreamCommand.GetParameter(index++).Value = stream.Type.ToString();
  2156. _saveStreamCommand.GetParameter(index++).Value = stream.Codec;
  2157. _saveStreamCommand.GetParameter(index++).Value = stream.Language;
  2158. _saveStreamCommand.GetParameter(index++).Value = stream.ChannelLayout;
  2159. _saveStreamCommand.GetParameter(index++).Value = stream.Profile;
  2160. _saveStreamCommand.GetParameter(index++).Value = stream.AspectRatio;
  2161. _saveStreamCommand.GetParameter(index++).Value = stream.Path;
  2162. _saveStreamCommand.GetParameter(index++).Value = stream.IsInterlaced;
  2163. _saveStreamCommand.GetParameter(index++).Value = stream.BitRate;
  2164. _saveStreamCommand.GetParameter(index++).Value = stream.Channels;
  2165. _saveStreamCommand.GetParameter(index++).Value = stream.SampleRate;
  2166. _saveStreamCommand.GetParameter(index++).Value = stream.IsDefault;
  2167. _saveStreamCommand.GetParameter(index++).Value = stream.IsForced;
  2168. _saveStreamCommand.GetParameter(index++).Value = stream.IsExternal;
  2169. _saveStreamCommand.GetParameter(index++).Value = stream.Width;
  2170. _saveStreamCommand.GetParameter(index++).Value = stream.Height;
  2171. _saveStreamCommand.GetParameter(index++).Value = stream.AverageFrameRate;
  2172. _saveStreamCommand.GetParameter(index++).Value = stream.RealFrameRate;
  2173. _saveStreamCommand.GetParameter(index++).Value = stream.Level;
  2174. _saveStreamCommand.GetParameter(index++).Value = stream.PixelFormat;
  2175. _saveStreamCommand.GetParameter(index++).Value = stream.BitDepth;
  2176. _saveStreamCommand.GetParameter(index++).Value = stream.IsAnamorphic;
  2177. _saveStreamCommand.GetParameter(index++).Value = stream.RefFrames;
  2178. _saveStreamCommand.GetParameter(index++).Value = stream.IsCabac;
  2179. if (stream.KeyFrames == null || stream.KeyFrames.Count == 0)
  2180. {
  2181. _saveStreamCommand.GetParameter(index++).Value = null;
  2182. }
  2183. else
  2184. {
  2185. _saveStreamCommand.GetParameter(index++).Value = string.Join(",", stream.KeyFrames.Select(i => i.ToString(CultureInfo.InvariantCulture)).ToArray());
  2186. }
  2187. _saveStreamCommand.GetParameter(index++).Value = stream.CodecTag;
  2188. _saveStreamCommand.Transaction = transaction;
  2189. _saveStreamCommand.ExecuteNonQuery();
  2190. }
  2191. transaction.Commit();
  2192. }
  2193. catch (OperationCanceledException)
  2194. {
  2195. if (transaction != null)
  2196. {
  2197. transaction.Rollback();
  2198. }
  2199. throw;
  2200. }
  2201. catch (Exception e)
  2202. {
  2203. _logger.ErrorException("Failed to save media streams:", e);
  2204. if (transaction != null)
  2205. {
  2206. transaction.Rollback();
  2207. }
  2208. throw;
  2209. }
  2210. finally
  2211. {
  2212. if (transaction != null)
  2213. {
  2214. transaction.Dispose();
  2215. }
  2216. _writeLock.Release();
  2217. }
  2218. }
  2219. /// <summary>
  2220. /// Gets the chapter.
  2221. /// </summary>
  2222. /// <param name="reader">The reader.</param>
  2223. /// <returns>ChapterInfo.</returns>
  2224. private MediaStream GetMediaStream(IDataReader reader)
  2225. {
  2226. var item = new MediaStream
  2227. {
  2228. Index = reader.GetInt32(1)
  2229. };
  2230. item.Type = (MediaStreamType)Enum.Parse(typeof(MediaStreamType), reader.GetString(2), true);
  2231. if (!reader.IsDBNull(3))
  2232. {
  2233. item.Codec = reader.GetString(3);
  2234. }
  2235. if (!reader.IsDBNull(4))
  2236. {
  2237. item.Language = reader.GetString(4);
  2238. }
  2239. if (!reader.IsDBNull(5))
  2240. {
  2241. item.ChannelLayout = reader.GetString(5);
  2242. }
  2243. if (!reader.IsDBNull(6))
  2244. {
  2245. item.Profile = reader.GetString(6);
  2246. }
  2247. if (!reader.IsDBNull(7))
  2248. {
  2249. item.AspectRatio = reader.GetString(7);
  2250. }
  2251. if (!reader.IsDBNull(8))
  2252. {
  2253. item.Path = reader.GetString(8);
  2254. }
  2255. item.IsInterlaced = reader.GetBoolean(9);
  2256. if (!reader.IsDBNull(10))
  2257. {
  2258. item.BitRate = reader.GetInt32(10);
  2259. }
  2260. if (!reader.IsDBNull(11))
  2261. {
  2262. item.Channels = reader.GetInt32(11);
  2263. }
  2264. if (!reader.IsDBNull(12))
  2265. {
  2266. item.SampleRate = reader.GetInt32(12);
  2267. }
  2268. item.IsDefault = reader.GetBoolean(13);
  2269. item.IsForced = reader.GetBoolean(14);
  2270. item.IsExternal = reader.GetBoolean(15);
  2271. if (!reader.IsDBNull(16))
  2272. {
  2273. item.Width = reader.GetInt32(16);
  2274. }
  2275. if (!reader.IsDBNull(17))
  2276. {
  2277. item.Height = reader.GetInt32(17);
  2278. }
  2279. if (!reader.IsDBNull(18))
  2280. {
  2281. item.AverageFrameRate = reader.GetFloat(18);
  2282. }
  2283. if (!reader.IsDBNull(19))
  2284. {
  2285. item.RealFrameRate = reader.GetFloat(19);
  2286. }
  2287. if (!reader.IsDBNull(20))
  2288. {
  2289. item.Level = reader.GetFloat(20);
  2290. }
  2291. if (!reader.IsDBNull(21))
  2292. {
  2293. item.PixelFormat = reader.GetString(21);
  2294. }
  2295. if (!reader.IsDBNull(22))
  2296. {
  2297. item.BitDepth = reader.GetInt32(22);
  2298. }
  2299. if (!reader.IsDBNull(23))
  2300. {
  2301. item.IsAnamorphic = reader.GetBoolean(23);
  2302. }
  2303. if (!reader.IsDBNull(24))
  2304. {
  2305. item.RefFrames = reader.GetInt32(24);
  2306. }
  2307. if (!reader.IsDBNull(25))
  2308. {
  2309. item.IsCabac = reader.GetBoolean(25);
  2310. }
  2311. if (!reader.IsDBNull(26))
  2312. {
  2313. var frames = reader.GetString(26);
  2314. if (!string.IsNullOrWhiteSpace(frames))
  2315. {
  2316. item.KeyFrames = frames.Split(',').Select(i => int.Parse(i, CultureInfo.InvariantCulture)).ToList();
  2317. }
  2318. }
  2319. if (!reader.IsDBNull(27))
  2320. {
  2321. item.CodecTag = reader.GetString(27);
  2322. }
  2323. return item;
  2324. }
  2325. }
  2326. }