Folder.cs 49 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442
  1. using MediaBrowser.Common.Progress;
  2. using MediaBrowser.Controller.Library;
  3. using MediaBrowser.Controller.Providers;
  4. using MediaBrowser.Model.Dto;
  5. using MediaBrowser.Model.Entities;
  6. using MediaBrowser.Model.Querying;
  7. using System;
  8. using System.Collections.Generic;
  9. using System.IO;
  10. using System.Linq;
  11. using System.Runtime.Serialization;
  12. using System.Threading;
  13. using System.Threading.Tasks;
  14. using CommonIO;
  15. using MediaBrowser.Controller.Channels;
  16. using MediaBrowser.Controller.Entities.TV;
  17. using MediaBrowser.Model.Channels;
  18. namespace MediaBrowser.Controller.Entities
  19. {
  20. /// <summary>
  21. /// Class Folder
  22. /// </summary>
  23. public class Folder : BaseItem, IHasThemeMedia
  24. {
  25. public static IUserManager UserManager { get; set; }
  26. public static IUserViewManager UserViewManager { get; set; }
  27. public List<Guid> ThemeSongIds { get; set; }
  28. public List<Guid> ThemeVideoIds { get; set; }
  29. [IgnoreDataMember]
  30. public DateTime? DateLastMediaAdded { get; set; }
  31. public Folder()
  32. {
  33. LinkedChildren = new List<LinkedChild>();
  34. ThemeSongIds = new List<Guid>();
  35. ThemeVideoIds = new List<Guid>();
  36. }
  37. [IgnoreDataMember]
  38. public virtual bool IsPreSorted
  39. {
  40. get { return false; }
  41. }
  42. /// <summary>
  43. /// Gets a value indicating whether this instance is folder.
  44. /// </summary>
  45. /// <value><c>true</c> if this instance is folder; otherwise, <c>false</c>.</value>
  46. [IgnoreDataMember]
  47. public override bool IsFolder
  48. {
  49. get
  50. {
  51. return true;
  52. }
  53. }
  54. [IgnoreDataMember]
  55. public virtual bool SupportsCumulativeRunTimeTicks
  56. {
  57. get
  58. {
  59. return false;
  60. }
  61. }
  62. [IgnoreDataMember]
  63. public virtual bool SupportsDateLastMediaAdded
  64. {
  65. get
  66. {
  67. return false;
  68. }
  69. }
  70. public override bool RequiresRefresh()
  71. {
  72. var baseResult = base.RequiresRefresh();
  73. if (SupportsCumulativeRunTimeTicks && !RunTimeTicks.HasValue)
  74. {
  75. baseResult = true;
  76. }
  77. return baseResult;
  78. }
  79. [IgnoreDataMember]
  80. public override string FileNameWithoutExtension
  81. {
  82. get
  83. {
  84. if (LocationType == LocationType.FileSystem)
  85. {
  86. return System.IO.Path.GetFileName(Path);
  87. }
  88. return null;
  89. }
  90. }
  91. protected override bool IsAllowTagFilterEnforced()
  92. {
  93. if (this is ICollectionFolder)
  94. {
  95. return false;
  96. }
  97. if (this is UserView)
  98. {
  99. return false;
  100. }
  101. return true;
  102. }
  103. /// <summary>
  104. /// Gets or sets a value indicating whether this instance is physical root.
  105. /// </summary>
  106. /// <value><c>true</c> if this instance is physical root; otherwise, <c>false</c>.</value>
  107. public bool IsPhysicalRoot { get; set; }
  108. /// <summary>
  109. /// Gets or sets a value indicating whether this instance is root.
  110. /// </summary>
  111. /// <value><c>true</c> if this instance is root; otherwise, <c>false</c>.</value>
  112. public bool IsRoot { get; set; }
  113. public virtual List<LinkedChild> LinkedChildren { get; set; }
  114. [IgnoreDataMember]
  115. protected virtual bool SupportsShortcutChildren
  116. {
  117. get { return false; }
  118. }
  119. /// <summary>
  120. /// Adds the child.
  121. /// </summary>
  122. /// <param name="item">The item.</param>
  123. /// <param name="cancellationToken">The cancellation token.</param>
  124. /// <returns>Task.</returns>
  125. /// <exception cref="System.InvalidOperationException">Unable to add + item.Name</exception>
  126. public async Task AddChild(BaseItem item, CancellationToken cancellationToken)
  127. {
  128. item.SetParent(this);
  129. if (item.Id == Guid.Empty)
  130. {
  131. item.Id = LibraryManager.GetNewItemId(item.Path, item.GetType());
  132. }
  133. if (ActualChildren.Any(i => i.Id == item.Id))
  134. {
  135. throw new ArgumentException(string.Format("A child with the Id {0} already exists.", item.Id));
  136. }
  137. if (item.DateCreated == DateTime.MinValue)
  138. {
  139. item.DateCreated = DateTime.UtcNow;
  140. }
  141. if (item.DateModified == DateTime.MinValue)
  142. {
  143. item.DateModified = DateTime.UtcNow;
  144. }
  145. await LibraryManager.CreateItem(item, cancellationToken).ConfigureAwait(false);
  146. }
  147. /// <summary>
  148. /// Removes the child.
  149. /// </summary>
  150. /// <param name="item">The item.</param>
  151. public void RemoveChild(BaseItem item)
  152. {
  153. item.SetParent(null);
  154. }
  155. #region Indexing
  156. /// <summary>
  157. /// Returns the valid set of index by options for this folder type.
  158. /// Override or extend to modify.
  159. /// </summary>
  160. /// <returns>Dictionary{System.StringFunc{UserIEnumerable{BaseItem}}}.</returns>
  161. protected virtual IEnumerable<string> GetIndexByOptions()
  162. {
  163. return new List<string> {
  164. {"None"},
  165. {"Performer"},
  166. {"Genre"},
  167. {"Director"},
  168. {"Year"},
  169. {"Studio"}
  170. };
  171. }
  172. /// <summary>
  173. /// Get the list of indexy by choices for this folder (localized).
  174. /// </summary>
  175. /// <value>The index by option strings.</value>
  176. [IgnoreDataMember]
  177. public IEnumerable<string> IndexByOptionStrings
  178. {
  179. get { return GetIndexByOptions(); }
  180. }
  181. #endregion
  182. /// <summary>
  183. /// Gets the actual children.
  184. /// </summary>
  185. /// <value>The actual children.</value>
  186. [IgnoreDataMember]
  187. protected virtual IEnumerable<BaseItem> ActualChildren
  188. {
  189. get
  190. {
  191. return LoadChildren();
  192. }
  193. }
  194. /// <summary>
  195. /// thread-safe access to the actual children of this folder - without regard to user
  196. /// </summary>
  197. /// <value>The children.</value>
  198. [IgnoreDataMember]
  199. public IEnumerable<BaseItem> Children
  200. {
  201. get { return ActualChildren.ToList(); }
  202. }
  203. /// <summary>
  204. /// thread-safe access to all recursive children of this folder - without regard to user
  205. /// </summary>
  206. /// <value>The recursive children.</value>
  207. [IgnoreDataMember]
  208. public IEnumerable<BaseItem> RecursiveChildren
  209. {
  210. get { return GetRecursiveChildren(); }
  211. }
  212. public override bool IsVisible(User user)
  213. {
  214. if (this is ICollectionFolder && !(this is BasePluginFolder))
  215. {
  216. if (user.Policy.BlockedMediaFolders != null)
  217. {
  218. if (user.Policy.BlockedMediaFolders.Contains(Id.ToString("N"), StringComparer.OrdinalIgnoreCase) ||
  219. // Backwards compatibility
  220. user.Policy.BlockedMediaFolders.Contains(Name, StringComparer.OrdinalIgnoreCase))
  221. {
  222. return false;
  223. }
  224. }
  225. else
  226. {
  227. if (!user.Policy.EnableAllFolders && !user.Policy.EnabledFolders.Contains(Id.ToString("N"), StringComparer.OrdinalIgnoreCase))
  228. {
  229. return false;
  230. }
  231. }
  232. }
  233. return base.IsVisible(user);
  234. }
  235. /// <summary>
  236. /// Loads our children. Validation will occur externally.
  237. /// We want this sychronous.
  238. /// </summary>
  239. protected virtual IEnumerable<BaseItem> LoadChildren()
  240. {
  241. //Logger.Debug("Loading children from {0} {1} {2}", GetType().Name, Id, Path);
  242. //just load our children from the repo - the library will be validated and maintained in other processes
  243. return GetCachedChildren();
  244. }
  245. public Task ValidateChildren(IProgress<double> progress, CancellationToken cancellationToken)
  246. {
  247. return ValidateChildren(progress, cancellationToken, new MetadataRefreshOptions(new DirectoryService(Logger, FileSystem)));
  248. }
  249. /// <summary>
  250. /// Validates that the children of the folder still exist
  251. /// </summary>
  252. /// <param name="progress">The progress.</param>
  253. /// <param name="cancellationToken">The cancellation token.</param>
  254. /// <param name="metadataRefreshOptions">The metadata refresh options.</param>
  255. /// <param name="recursive">if set to <c>true</c> [recursive].</param>
  256. /// <returns>Task.</returns>
  257. public Task ValidateChildren(IProgress<double> progress, CancellationToken cancellationToken, MetadataRefreshOptions metadataRefreshOptions, bool recursive = true)
  258. {
  259. return ValidateChildrenInternal(progress, cancellationToken, recursive, true, metadataRefreshOptions, metadataRefreshOptions.DirectoryService);
  260. }
  261. private Dictionary<Guid, BaseItem> GetActualChildrenDictionary()
  262. {
  263. var dictionary = new Dictionary<Guid, BaseItem>();
  264. foreach (var child in ActualChildren)
  265. {
  266. var id = child.Id;
  267. if (dictionary.ContainsKey(id))
  268. {
  269. Logger.Error("Found folder containing items with duplicate id. Path: {0}, Child Name: {1}",
  270. Path ?? Name,
  271. child.Path ?? child.Name);
  272. }
  273. else
  274. {
  275. dictionary[id] = child;
  276. }
  277. }
  278. return dictionary;
  279. }
  280. private bool IsValidFromResolver(BaseItem current, BaseItem newItem)
  281. {
  282. return current.IsValidFromResolver(newItem);
  283. }
  284. /// <summary>
  285. /// Validates the children internal.
  286. /// </summary>
  287. /// <param name="progress">The progress.</param>
  288. /// <param name="cancellationToken">The cancellation token.</param>
  289. /// <param name="recursive">if set to <c>true</c> [recursive].</param>
  290. /// <param name="refreshChildMetadata">if set to <c>true</c> [refresh child metadata].</param>
  291. /// <param name="refreshOptions">The refresh options.</param>
  292. /// <param name="directoryService">The directory service.</param>
  293. /// <returns>Task.</returns>
  294. protected async virtual Task ValidateChildrenInternal(IProgress<double> progress, CancellationToken cancellationToken, bool recursive, bool refreshChildMetadata, MetadataRefreshOptions refreshOptions, IDirectoryService directoryService)
  295. {
  296. var locationType = LocationType;
  297. cancellationToken.ThrowIfCancellationRequested();
  298. var validChildren = new List<BaseItem>();
  299. if (locationType != LocationType.Remote && locationType != LocationType.Virtual)
  300. {
  301. IEnumerable<BaseItem> nonCachedChildren;
  302. try
  303. {
  304. nonCachedChildren = GetNonCachedChildren(directoryService);
  305. }
  306. catch (IOException ex)
  307. {
  308. nonCachedChildren = new BaseItem[] { };
  309. Logger.ErrorException("Error getting file system entries for {0}", ex, Path);
  310. }
  311. if (nonCachedChildren == null) return; //nothing to validate
  312. progress.Report(5);
  313. //build a dictionary of the current children we have now by Id so we can compare quickly and easily
  314. var currentChildren = GetActualChildrenDictionary();
  315. //create a list for our validated children
  316. var newItems = new List<BaseItem>();
  317. cancellationToken.ThrowIfCancellationRequested();
  318. foreach (var child in nonCachedChildren)
  319. {
  320. BaseItem currentChild;
  321. if (currentChildren.TryGetValue(child.Id, out currentChild) && IsValidFromResolver(currentChild, child))
  322. {
  323. await UpdateIsOffline(currentChild, false).ConfigureAwait(false);
  324. validChildren.Add(currentChild);
  325. continue;
  326. }
  327. // Brand new item - needs to be added
  328. child.SetParent(this);
  329. newItems.Add(child);
  330. validChildren.Add(child);
  331. }
  332. // If any items were added or removed....
  333. if (newItems.Count > 0 || currentChildren.Count != validChildren.Count)
  334. {
  335. // That's all the new and changed ones - now see if there are any that are missing
  336. var itemsRemoved = currentChildren.Values.Except(validChildren).ToList();
  337. var actualRemovals = new List<BaseItem>();
  338. foreach (var item in itemsRemoved)
  339. {
  340. var itemLocationType = item.LocationType;
  341. if (itemLocationType == LocationType.Virtual ||
  342. itemLocationType == LocationType.Remote)
  343. {
  344. }
  345. else if (!string.IsNullOrEmpty(item.Path) && IsPathOffline(item.Path))
  346. {
  347. await UpdateIsOffline(item, true).ConfigureAwait(false);
  348. }
  349. else
  350. {
  351. actualRemovals.Add(item);
  352. }
  353. }
  354. if (actualRemovals.Count > 0)
  355. {
  356. foreach (var item in actualRemovals)
  357. {
  358. Logger.Debug("Removed item: " + item.Path);
  359. item.SetParent(null);
  360. item.IsOffline = false;
  361. await LibraryManager.DeleteItem(item, new DeleteOptions { DeleteFileLocation = false }).ConfigureAwait(false);
  362. LibraryManager.ReportItemRemoved(item);
  363. }
  364. }
  365. await LibraryManager.CreateItems(newItems, cancellationToken).ConfigureAwait(false);
  366. }
  367. }
  368. progress.Report(10);
  369. cancellationToken.ThrowIfCancellationRequested();
  370. if (recursive)
  371. {
  372. await ValidateSubFolders(ActualChildren.OfType<Folder>().ToList(), directoryService, progress, cancellationToken).ConfigureAwait(false);
  373. }
  374. progress.Report(20);
  375. if (refreshChildMetadata)
  376. {
  377. var container = this as IMetadataContainer;
  378. var innerProgress = new ActionableProgress<double>();
  379. innerProgress.RegisterAction(p => progress.Report(.80 * p + 20));
  380. if (container != null)
  381. {
  382. await container.RefreshAllMetadata(refreshOptions, innerProgress, cancellationToken).ConfigureAwait(false);
  383. }
  384. else
  385. {
  386. await RefreshMetadataRecursive(refreshOptions, recursive, innerProgress, cancellationToken);
  387. }
  388. }
  389. progress.Report(100);
  390. }
  391. private Task UpdateIsOffline(BaseItem item, bool newValue)
  392. {
  393. if (item.IsOffline != newValue)
  394. {
  395. item.IsOffline = newValue;
  396. return item.UpdateToRepository(ItemUpdateType.None, CancellationToken.None);
  397. }
  398. return Task.FromResult(true);
  399. }
  400. private async Task RefreshMetadataRecursive(MetadataRefreshOptions refreshOptions, bool recursive, IProgress<double> progress, CancellationToken cancellationToken)
  401. {
  402. var children = ActualChildren.ToList();
  403. var percentages = new Dictionary<Guid, double>(children.Count);
  404. var numComplete = 0;
  405. var count = children.Count;
  406. foreach (var child in children)
  407. {
  408. cancellationToken.ThrowIfCancellationRequested();
  409. if (child.IsFolder)
  410. {
  411. var innerProgress = new ActionableProgress<double>();
  412. // Avoid implicitly captured closure
  413. var currentChild = child;
  414. innerProgress.RegisterAction(p =>
  415. {
  416. lock (percentages)
  417. {
  418. percentages[currentChild.Id] = p / 100;
  419. var innerPercent = percentages.Values.Sum();
  420. innerPercent /= count;
  421. innerPercent *= 100;
  422. progress.Report(innerPercent);
  423. }
  424. });
  425. await RefreshChildMetadata(child, refreshOptions, recursive, innerProgress, cancellationToken)
  426. .ConfigureAwait(false);
  427. }
  428. else
  429. {
  430. await RefreshChildMetadata(child, refreshOptions, false, new Progress<double>(), cancellationToken)
  431. .ConfigureAwait(false);
  432. }
  433. numComplete++;
  434. double percent = numComplete;
  435. percent /= count;
  436. percent *= 100;
  437. progress.Report(percent);
  438. }
  439. progress.Report(100);
  440. }
  441. private async Task RefreshChildMetadata(BaseItem child, MetadataRefreshOptions refreshOptions, bool recursive, IProgress<double> progress, CancellationToken cancellationToken)
  442. {
  443. var container = child as IMetadataContainer;
  444. if (container != null)
  445. {
  446. await container.RefreshAllMetadata(refreshOptions, progress, cancellationToken).ConfigureAwait(false);
  447. }
  448. else
  449. {
  450. await child.RefreshMetadata(refreshOptions, cancellationToken).ConfigureAwait(false);
  451. if (recursive)
  452. {
  453. var folder = child as Folder;
  454. if (folder != null)
  455. {
  456. await folder.RefreshMetadataRecursive(refreshOptions, true, progress, cancellationToken);
  457. }
  458. }
  459. }
  460. progress.Report(100);
  461. }
  462. /// <summary>
  463. /// Refreshes the children.
  464. /// </summary>
  465. /// <param name="children">The children.</param>
  466. /// <param name="directoryService">The directory service.</param>
  467. /// <param name="progress">The progress.</param>
  468. /// <param name="cancellationToken">The cancellation token.</param>
  469. /// <returns>Task.</returns>
  470. private async Task ValidateSubFolders(IList<Folder> children, IDirectoryService directoryService, IProgress<double> progress, CancellationToken cancellationToken)
  471. {
  472. var list = children;
  473. var childCount = list.Count;
  474. var percentages = new Dictionary<Guid, double>(list.Count);
  475. foreach (var item in list)
  476. {
  477. cancellationToken.ThrowIfCancellationRequested();
  478. var child = item;
  479. var innerProgress = new ActionableProgress<double>();
  480. innerProgress.RegisterAction(p =>
  481. {
  482. lock (percentages)
  483. {
  484. percentages[child.Id] = p / 100;
  485. var percent = percentages.Values.Sum();
  486. percent /= childCount;
  487. progress.Report(10 * percent + 10);
  488. }
  489. });
  490. await child.ValidateChildrenInternal(innerProgress, cancellationToken, true, false, null, directoryService)
  491. .ConfigureAwait(false);
  492. }
  493. }
  494. /// <summary>
  495. /// Determines whether the specified path is offline.
  496. /// </summary>
  497. /// <param name="path">The path.</param>
  498. /// <returns><c>true</c> if the specified path is offline; otherwise, <c>false</c>.</returns>
  499. public static bool IsPathOffline(string path)
  500. {
  501. if (FileSystem.FileExists(path))
  502. {
  503. return false;
  504. }
  505. var originalPath = path;
  506. // Depending on whether the path is local or unc, it may return either null or '\' at the top
  507. while (!string.IsNullOrEmpty(path) && path.Length > 1)
  508. {
  509. if (FileSystem.DirectoryExists(path))
  510. {
  511. return false;
  512. }
  513. path = System.IO.Path.GetDirectoryName(path);
  514. }
  515. if (ContainsPath(LibraryManager.GetVirtualFolders(), originalPath))
  516. {
  517. return true;
  518. }
  519. return false;
  520. }
  521. /// <summary>
  522. /// Determines whether the specified folders contains path.
  523. /// </summary>
  524. /// <param name="folders">The folders.</param>
  525. /// <param name="path">The path.</param>
  526. /// <returns><c>true</c> if the specified folders contains path; otherwise, <c>false</c>.</returns>
  527. private static bool ContainsPath(IEnumerable<VirtualFolderInfo> folders, string path)
  528. {
  529. return folders.SelectMany(i => i.Locations).Any(i => ContainsPath(i, path));
  530. }
  531. private static bool ContainsPath(string parent, string path)
  532. {
  533. return string.Equals(parent, path, StringComparison.OrdinalIgnoreCase) || FileSystem.ContainsSubPath(parent, path);
  534. }
  535. /// <summary>
  536. /// Get the children of this folder from the actual file system
  537. /// </summary>
  538. /// <returns>IEnumerable{BaseItem}.</returns>
  539. protected virtual IEnumerable<BaseItem> GetNonCachedChildren(IDirectoryService directoryService)
  540. {
  541. var collectionType = LibraryManager.GetContentType(this);
  542. var libraryOptions = LibraryManager.GetLibraryOptions(this);
  543. return LibraryManager.ResolvePaths(GetFileSystemChildren(directoryService), directoryService, this, libraryOptions, collectionType);
  544. }
  545. /// <summary>
  546. /// Get our children from the repo - stubbed for now
  547. /// </summary>
  548. /// <returns>IEnumerable{BaseItem}.</returns>
  549. protected IEnumerable<BaseItem> GetCachedChildren()
  550. {
  551. return ItemRepository.GetItemList(new InternalItemsQuery
  552. {
  553. ParentId = Id,
  554. GroupByPresentationUniqueKey = false
  555. });
  556. }
  557. public virtual int GetChildCount(User user)
  558. {
  559. if (LinkedChildren.Count > 0)
  560. {
  561. if (!(this is ICollectionFolder))
  562. {
  563. return GetChildren(user, true).Count();
  564. }
  565. }
  566. var result = GetItems(new InternalItemsQuery(user)
  567. {
  568. Recursive = false,
  569. Limit = 0,
  570. ParentId = Id
  571. }).Result;
  572. return result.TotalRecordCount;
  573. }
  574. public QueryResult<BaseItem> QueryRecursive(InternalItemsQuery query)
  575. {
  576. var user = query.User;
  577. if (!query.ForceDirect && RequiresPostFiltering(query))
  578. {
  579. IEnumerable<BaseItem> items;
  580. Func<BaseItem, bool> filter = i => UserViewBuilder.Filter(i, user, query, UserDataManager, LibraryManager);
  581. if (query.User == null)
  582. {
  583. items = GetRecursiveChildren(filter);
  584. }
  585. else
  586. {
  587. items = GetRecursiveChildren(user, query);
  588. }
  589. return PostFilterAndSort(items, query);
  590. }
  591. if (!(this is UserRootFolder) && !(this is AggregateFolder))
  592. {
  593. query.ParentId = query.ParentId ?? Id;
  594. }
  595. return LibraryManager.GetItemsResult(query);
  596. }
  597. private bool RequiresPostFiltering(InternalItemsQuery query)
  598. {
  599. if (LinkedChildren.Count > 0)
  600. {
  601. if (!(this is ICollectionFolder))
  602. {
  603. Logger.Debug("Query requires post-filtering due to LinkedChildren. Type: " + GetType().Name);
  604. return true;
  605. }
  606. }
  607. if (query.SortBy != null && query.SortBy.Length > 0)
  608. {
  609. if (query.SortBy.Contains(ItemSortBy.AiredEpisodeOrder, StringComparer.OrdinalIgnoreCase))
  610. {
  611. Logger.Debug("Query requires post-filtering due to ItemSortBy.AiredEpisodeOrder");
  612. return true;
  613. }
  614. if (query.SortBy.Contains(ItemSortBy.Budget, StringComparer.OrdinalIgnoreCase))
  615. {
  616. Logger.Debug("Query requires post-filtering due to ItemSortBy.Budget");
  617. return true;
  618. }
  619. if (query.SortBy.Contains(ItemSortBy.GameSystem, StringComparer.OrdinalIgnoreCase))
  620. {
  621. Logger.Debug("Query requires post-filtering due to ItemSortBy.GameSystem");
  622. return true;
  623. }
  624. if (query.SortBy.Contains(ItemSortBy.Metascore, StringComparer.OrdinalIgnoreCase))
  625. {
  626. Logger.Debug("Query requires post-filtering due to ItemSortBy.Metascore");
  627. return true;
  628. }
  629. if (query.SortBy.Contains(ItemSortBy.Players, StringComparer.OrdinalIgnoreCase))
  630. {
  631. Logger.Debug("Query requires post-filtering due to ItemSortBy.Players");
  632. return true;
  633. }
  634. if (query.SortBy.Contains(ItemSortBy.Revenue, StringComparer.OrdinalIgnoreCase))
  635. {
  636. Logger.Debug("Query requires post-filtering due to ItemSortBy.Revenue");
  637. return true;
  638. }
  639. if (query.SortBy.Contains(ItemSortBy.VideoBitRate, StringComparer.OrdinalIgnoreCase))
  640. {
  641. Logger.Debug("Query requires post-filtering due to ItemSortBy.VideoBitRate");
  642. return true;
  643. }
  644. }
  645. if (query.ItemIds.Length > 0)
  646. {
  647. Logger.Debug("Query requires post-filtering due to ItemIds");
  648. return true;
  649. }
  650. if (query.IsInBoxSet.HasValue)
  651. {
  652. Logger.Debug("Query requires post-filtering due to IsInBoxSet");
  653. return true;
  654. }
  655. // Filter by Video3DFormat
  656. if (query.Is3D.HasValue)
  657. {
  658. Logger.Debug("Query requires post-filtering due to Is3D");
  659. return true;
  660. }
  661. if (query.HasOfficialRating.HasValue)
  662. {
  663. Logger.Debug("Query requires post-filtering due to HasOfficialRating");
  664. return true;
  665. }
  666. if (query.IsPlaceHolder.HasValue)
  667. {
  668. Logger.Debug("Query requires post-filtering due to IsPlaceHolder");
  669. return true;
  670. }
  671. if (query.HasSpecialFeature.HasValue)
  672. {
  673. Logger.Debug("Query requires post-filtering due to HasSpecialFeature");
  674. return true;
  675. }
  676. if (query.HasSubtitles.HasValue)
  677. {
  678. Logger.Debug("Query requires post-filtering due to HasSubtitles");
  679. return true;
  680. }
  681. if (query.HasTrailer.HasValue)
  682. {
  683. Logger.Debug("Query requires post-filtering due to HasTrailer");
  684. return true;
  685. }
  686. if (query.HasThemeSong.HasValue)
  687. {
  688. Logger.Debug("Query requires post-filtering due to HasThemeSong");
  689. return true;
  690. }
  691. if (query.HasThemeVideo.HasValue)
  692. {
  693. Logger.Debug("Query requires post-filtering due to HasThemeVideo");
  694. return true;
  695. }
  696. // Filter by VideoType
  697. if (query.VideoTypes.Length > 0)
  698. {
  699. Logger.Debug("Query requires post-filtering due to VideoTypes");
  700. return true;
  701. }
  702. // Apply person filter
  703. if (query.ItemIdsFromPersonFilters != null)
  704. {
  705. Logger.Debug("Query requires post-filtering due to ItemIdsFromPersonFilters");
  706. return true;
  707. }
  708. if (query.MinPlayers.HasValue)
  709. {
  710. Logger.Debug("Query requires post-filtering due to MinPlayers");
  711. return true;
  712. }
  713. if (query.MaxPlayers.HasValue)
  714. {
  715. Logger.Debug("Query requires post-filtering due to MaxPlayers");
  716. return true;
  717. }
  718. if (UserViewBuilder.CollapseBoxSetItems(query, this, query.User, ConfigurationManager))
  719. {
  720. Logger.Debug("Query requires post-filtering due to CollapseBoxSetItems");
  721. return true;
  722. }
  723. if (!string.IsNullOrWhiteSpace(query.AdjacentTo))
  724. {
  725. Logger.Debug("Query requires post-filtering due to AdjacentTo");
  726. return true;
  727. }
  728. if (query.AirDays.Length > 0)
  729. {
  730. Logger.Debug("Query requires post-filtering due to AirDays");
  731. return true;
  732. }
  733. if (query.SeriesStatuses.Length > 0)
  734. {
  735. Logger.Debug("Query requires post-filtering due to SeriesStatuses");
  736. return true;
  737. }
  738. if (query.AiredDuringSeason.HasValue)
  739. {
  740. Logger.Debug("Query requires post-filtering due to AiredDuringSeason");
  741. return true;
  742. }
  743. if (!string.IsNullOrWhiteSpace(query.AlbumArtistStartsWithOrGreater))
  744. {
  745. Logger.Debug("Query requires post-filtering due to AlbumArtistStartsWithOrGreater");
  746. return true;
  747. }
  748. if (query.IsPlayed.HasValue)
  749. {
  750. if (query.IncludeItemTypes.Length == 1 && query.IncludeItemTypes.Contains(typeof(Series).Name))
  751. {
  752. Logger.Debug("Query requires post-filtering due to IsPlayed");
  753. return true;
  754. }
  755. }
  756. return false;
  757. }
  758. public Task<QueryResult<BaseItem>> GetItems(InternalItemsQuery query)
  759. {
  760. if (query.ItemIds.Length > 0)
  761. {
  762. var specificItems = query.ItemIds.Select(LibraryManager.GetItemById).Where(i => i != null).ToList();
  763. return Task.FromResult(PostFilterAndSort(specificItems, query));
  764. }
  765. return GetItemsInternal(query);
  766. }
  767. protected virtual async Task<QueryResult<BaseItem>> GetItemsInternal(InternalItemsQuery query)
  768. {
  769. if (SourceType == SourceType.Channel)
  770. {
  771. try
  772. {
  773. // Don't blow up here because it could cause parent screens with other content to fail
  774. return await ChannelManager.GetChannelItemsInternal(new ChannelItemQuery
  775. {
  776. ChannelId = ChannelId,
  777. FolderId = Id.ToString("N"),
  778. Limit = query.Limit,
  779. StartIndex = query.StartIndex,
  780. UserId = query.User.Id.ToString("N"),
  781. SortBy = query.SortBy,
  782. SortOrder = query.SortOrder
  783. }, new Progress<double>(), CancellationToken.None);
  784. }
  785. catch
  786. {
  787. // Already logged at lower levels
  788. return new QueryResult<BaseItem>();
  789. }
  790. }
  791. if (query.Recursive)
  792. {
  793. return QueryRecursive(query);
  794. }
  795. var user = query.User;
  796. Func<BaseItem, bool> filter = i => UserViewBuilder.Filter(i, user, query, UserDataManager, LibraryManager);
  797. IEnumerable<BaseItem> items;
  798. if (query.User == null)
  799. {
  800. items = query.Recursive
  801. ? GetRecursiveChildren(filter)
  802. : Children.Where(filter);
  803. }
  804. else
  805. {
  806. items = query.Recursive
  807. ? GetRecursiveChildren(user, query)
  808. : GetChildren(user, true).Where(filter);
  809. }
  810. return PostFilterAndSort(items, query);
  811. }
  812. protected QueryResult<BaseItem> PostFilterAndSort(IEnumerable<BaseItem> items, InternalItemsQuery query)
  813. {
  814. return UserViewBuilder.PostFilterAndSort(items, this, null, query, LibraryManager, ConfigurationManager);
  815. }
  816. public virtual IEnumerable<BaseItem> GetChildren(User user, bool includeLinkedChildren)
  817. {
  818. if (user == null)
  819. {
  820. throw new ArgumentNullException();
  821. }
  822. //the true root should return our users root folder children
  823. if (IsPhysicalRoot) return user.RootFolder.GetChildren(user, includeLinkedChildren);
  824. var result = new Dictionary<Guid, BaseItem>();
  825. AddChildren(user, includeLinkedChildren, result, false, null);
  826. return result.Values;
  827. }
  828. protected virtual IEnumerable<BaseItem> GetEligibleChildrenForRecursiveChildren(User user)
  829. {
  830. return Children;
  831. }
  832. /// <summary>
  833. /// Adds the children to list.
  834. /// </summary>
  835. /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
  836. private void AddChildren(User user, bool includeLinkedChildren, Dictionary<Guid, BaseItem> result, bool recursive, InternalItemsQuery query)
  837. {
  838. foreach (var child in GetEligibleChildrenForRecursiveChildren(user))
  839. {
  840. if (child.IsVisible(user))
  841. {
  842. if (query == null || UserViewBuilder.FilterItem(child, query))
  843. {
  844. result[child.Id] = child;
  845. }
  846. if (recursive && child.IsFolder)
  847. {
  848. var folder = (Folder)child;
  849. folder.AddChildren(user, includeLinkedChildren, result, true, query);
  850. }
  851. }
  852. }
  853. if (includeLinkedChildren)
  854. {
  855. foreach (var child in GetLinkedChildren(user))
  856. {
  857. if (child.IsVisible(user))
  858. {
  859. if (query == null || UserViewBuilder.FilterItem(child, query))
  860. {
  861. result[child.Id] = child;
  862. }
  863. }
  864. }
  865. }
  866. }
  867. /// <summary>
  868. /// Gets allowed recursive children of an item
  869. /// </summary>
  870. /// <param name="user">The user.</param>
  871. /// <param name="includeLinkedChildren">if set to <c>true</c> [include linked children].</param>
  872. /// <returns>IEnumerable{BaseItem}.</returns>
  873. /// <exception cref="System.ArgumentNullException"></exception>
  874. public IEnumerable<BaseItem> GetRecursiveChildren(User user, bool includeLinkedChildren = true)
  875. {
  876. return GetRecursiveChildren(user, null);
  877. }
  878. public virtual IEnumerable<BaseItem> GetRecursiveChildren(User user, InternalItemsQuery query)
  879. {
  880. if (user == null)
  881. {
  882. throw new ArgumentNullException("user");
  883. }
  884. var result = new Dictionary<Guid, BaseItem>();
  885. AddChildren(user, true, result, true, query);
  886. return result.Values;
  887. }
  888. /// <summary>
  889. /// Gets the recursive children.
  890. /// </summary>
  891. /// <returns>IList{BaseItem}.</returns>
  892. public IList<BaseItem> GetRecursiveChildren()
  893. {
  894. return GetRecursiveChildren(i => true);
  895. }
  896. public IList<BaseItem> GetRecursiveChildren(Func<BaseItem, bool> filter)
  897. {
  898. var result = new Dictionary<Guid, BaseItem>();
  899. AddChildrenToList(result, true, true, filter);
  900. return result.Values.ToList();
  901. }
  902. /// <summary>
  903. /// Adds the children to list.
  904. /// </summary>
  905. private void AddChildrenToList(Dictionary<Guid, BaseItem> result, bool includeLinkedChildren, bool recursive, Func<BaseItem, bool> filter)
  906. {
  907. foreach (var child in Children)
  908. {
  909. if (filter == null || filter(child))
  910. {
  911. result[child.Id] = child;
  912. }
  913. if (recursive && child.IsFolder)
  914. {
  915. var folder = (Folder)child;
  916. // We can only support includeLinkedChildren for the first folder, or we might end up stuck in a loop of linked items
  917. folder.AddChildrenToList(result, false, true, filter);
  918. }
  919. }
  920. if (includeLinkedChildren)
  921. {
  922. foreach (var child in GetLinkedChildren())
  923. {
  924. if (filter == null || filter(child))
  925. {
  926. result[child.Id] = child;
  927. }
  928. }
  929. }
  930. }
  931. /// <summary>
  932. /// Gets the linked children.
  933. /// </summary>
  934. /// <returns>IEnumerable{BaseItem}.</returns>
  935. public IEnumerable<BaseItem> GetLinkedChildren()
  936. {
  937. return LinkedChildren
  938. .Select(GetLinkedChild)
  939. .Where(i => i != null);
  940. }
  941. protected virtual bool FilterLinkedChildrenPerUser
  942. {
  943. get
  944. {
  945. return false;
  946. }
  947. }
  948. public IEnumerable<BaseItem> GetLinkedChildren(User user)
  949. {
  950. if (!FilterLinkedChildrenPerUser || user == null)
  951. {
  952. return GetLinkedChildren();
  953. }
  954. var locations = user.RootFolder
  955. .Children
  956. .OfType<CollectionFolder>()
  957. .Where(i => i.IsVisible(user))
  958. .SelectMany(i => i.PhysicalLocations)
  959. .ToList();
  960. return LinkedChildren
  961. .Select(i =>
  962. {
  963. var requiresPostFilter = true;
  964. if (!string.IsNullOrWhiteSpace(i.Path))
  965. {
  966. requiresPostFilter = false;
  967. if (!locations.Any(l => FileSystem.ContainsSubPath(l, i.Path)))
  968. {
  969. return null;
  970. }
  971. }
  972. var child = GetLinkedChild(i);
  973. if (requiresPostFilter && child != null)
  974. {
  975. if (string.IsNullOrWhiteSpace(child.Path))
  976. {
  977. Logger.Debug("Found LinkedChild with null path: {0}", child.Name);
  978. return child;
  979. }
  980. if (!locations.Any(l => FileSystem.ContainsSubPath(l, child.Path)))
  981. {
  982. return null;
  983. }
  984. }
  985. return child;
  986. })
  987. .Where(i => i != null);
  988. }
  989. /// <summary>
  990. /// Gets the linked children.
  991. /// </summary>
  992. /// <returns>IEnumerable{BaseItem}.</returns>
  993. public IEnumerable<Tuple<LinkedChild, BaseItem>> GetLinkedChildrenInfos()
  994. {
  995. return LinkedChildren
  996. .Select(i => new Tuple<LinkedChild, BaseItem>(i, GetLinkedChild(i)))
  997. .Where(i => i.Item2 != null);
  998. }
  999. [IgnoreDataMember]
  1000. protected override bool SupportsOwnedItems
  1001. {
  1002. get
  1003. {
  1004. return base.SupportsOwnedItems || SupportsShortcutChildren;
  1005. }
  1006. }
  1007. protected override async Task<bool> RefreshedOwnedItems(MetadataRefreshOptions options, List<FileSystemMetadata> fileSystemChildren, CancellationToken cancellationToken)
  1008. {
  1009. var changesFound = false;
  1010. if (LocationType == LocationType.FileSystem)
  1011. {
  1012. if (RefreshLinkedChildren(fileSystemChildren))
  1013. {
  1014. changesFound = true;
  1015. }
  1016. }
  1017. var baseHasChanges = await base.RefreshedOwnedItems(options, fileSystemChildren, cancellationToken).ConfigureAwait(false);
  1018. return baseHasChanges || changesFound;
  1019. }
  1020. /// <summary>
  1021. /// Refreshes the linked children.
  1022. /// </summary>
  1023. /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
  1024. private bool RefreshLinkedChildren(IEnumerable<FileSystemMetadata> fileSystemChildren)
  1025. {
  1026. var currentManualLinks = LinkedChildren.Where(i => i.Type == LinkedChildType.Manual).ToList();
  1027. var currentShortcutLinks = LinkedChildren.Where(i => i.Type == LinkedChildType.Shortcut).ToList();
  1028. List<LinkedChild> newShortcutLinks;
  1029. if (SupportsShortcutChildren)
  1030. {
  1031. newShortcutLinks = fileSystemChildren
  1032. .Where(i => (i.Attributes & FileAttributes.Directory) != FileAttributes.Directory && FileSystem.IsShortcut(i.FullName))
  1033. .Select(i =>
  1034. {
  1035. try
  1036. {
  1037. Logger.Debug("Found shortcut at {0}", i.FullName);
  1038. var resolvedPath = FileSystem.ResolveShortcut(i.FullName);
  1039. if (!string.IsNullOrEmpty(resolvedPath))
  1040. {
  1041. return new LinkedChild
  1042. {
  1043. Path = resolvedPath,
  1044. Type = LinkedChildType.Shortcut
  1045. };
  1046. }
  1047. Logger.Error("Error resolving shortcut {0}", i.FullName);
  1048. return null;
  1049. }
  1050. catch (IOException ex)
  1051. {
  1052. Logger.ErrorException("Error resolving shortcut {0}", ex, i.FullName);
  1053. return null;
  1054. }
  1055. })
  1056. .Where(i => i != null)
  1057. .ToList();
  1058. }
  1059. else { newShortcutLinks = new List<LinkedChild>(); }
  1060. if (!newShortcutLinks.SequenceEqual(currentShortcutLinks, new LinkedChildComparer()))
  1061. {
  1062. Logger.Info("Shortcut links have changed for {0}", Path);
  1063. newShortcutLinks.AddRange(currentManualLinks);
  1064. LinkedChildren = newShortcutLinks;
  1065. return true;
  1066. }
  1067. foreach (var child in LinkedChildren)
  1068. {
  1069. // Reset the cached value
  1070. child.ItemId = null;
  1071. }
  1072. return false;
  1073. }
  1074. /// <summary>
  1075. /// Folders need to validate and refresh
  1076. /// </summary>
  1077. /// <returns>Task.</returns>
  1078. public override async Task ChangedExternally()
  1079. {
  1080. var progress = new Progress<double>();
  1081. await ValidateChildren(progress, CancellationToken.None).ConfigureAwait(false);
  1082. await base.ChangedExternally().ConfigureAwait(false);
  1083. }
  1084. /// <summary>
  1085. /// Marks the played.
  1086. /// </summary>
  1087. /// <param name="user">The user.</param>
  1088. /// <param name="datePlayed">The date played.</param>
  1089. /// <param name="resetPosition">if set to <c>true</c> [reset position].</param>
  1090. /// <returns>Task.</returns>
  1091. public override async Task MarkPlayed(User user,
  1092. DateTime? datePlayed,
  1093. bool resetPosition)
  1094. {
  1095. var query = new InternalItemsQuery
  1096. {
  1097. User = user,
  1098. Recursive = true,
  1099. IsFolder = false,
  1100. EnableTotalRecordCount = false
  1101. };
  1102. if (!user.Configuration.DisplayMissingEpisodes || !user.Configuration.DisplayUnairedEpisodes)
  1103. {
  1104. query.ExcludeLocationTypes = new[] { LocationType.Virtual };
  1105. }
  1106. var itemsResult = await GetItems(query).ConfigureAwait(false);
  1107. // Sweep through recursively and update status
  1108. var tasks = itemsResult.Items.Select(c => c.MarkPlayed(user, datePlayed, resetPosition));
  1109. await Task.WhenAll(tasks).ConfigureAwait(false);
  1110. }
  1111. /// <summary>
  1112. /// Marks the unplayed.
  1113. /// </summary>
  1114. /// <param name="user">The user.</param>
  1115. /// <returns>Task.</returns>
  1116. public override async Task MarkUnplayed(User user)
  1117. {
  1118. var itemsResult = await GetItems(new InternalItemsQuery
  1119. {
  1120. User = user,
  1121. Recursive = true,
  1122. IsFolder = false,
  1123. EnableTotalRecordCount = false
  1124. }).ConfigureAwait(false);
  1125. // Sweep through recursively and update status
  1126. var tasks = itemsResult.Items.Select(c => c.MarkUnplayed(user));
  1127. await Task.WhenAll(tasks).ConfigureAwait(false);
  1128. }
  1129. public override bool IsPlayed(User user)
  1130. {
  1131. var itemsResult = GetItems(new InternalItemsQuery(user)
  1132. {
  1133. Recursive = true,
  1134. IsFolder = false,
  1135. ExcludeLocationTypes = new[] { LocationType.Virtual },
  1136. EnableTotalRecordCount = false
  1137. }).Result;
  1138. return itemsResult.Items
  1139. .All(i => i.IsPlayed(user));
  1140. }
  1141. public override bool IsUnplayed(User user)
  1142. {
  1143. return !IsPlayed(user);
  1144. }
  1145. [IgnoreDataMember]
  1146. public virtual bool SupportsUserDataFromChildren
  1147. {
  1148. get
  1149. {
  1150. // These are just far too slow.
  1151. if (this is ICollectionFolder)
  1152. {
  1153. return false;
  1154. }
  1155. if (this is UserView)
  1156. {
  1157. return false;
  1158. }
  1159. if (this is UserRootFolder)
  1160. {
  1161. return false;
  1162. }
  1163. if (this is Channel)
  1164. {
  1165. return false;
  1166. }
  1167. if (SourceType != SourceType.Library)
  1168. {
  1169. return false;
  1170. }
  1171. return true;
  1172. }
  1173. }
  1174. public override async Task FillUserDataDtoValues(UserItemDataDto dto, UserItemData userData, BaseItemDto itemDto, User user)
  1175. {
  1176. if (!SupportsUserDataFromChildren)
  1177. {
  1178. return;
  1179. }
  1180. var unplayedQueryResult = await GetItems(new InternalItemsQuery(user)
  1181. {
  1182. Recursive = true,
  1183. IsFolder = false,
  1184. IsVirtualItem = false,
  1185. EnableTotalRecordCount = true,
  1186. Limit = 0,
  1187. IsPlayed = false
  1188. }).ConfigureAwait(false);
  1189. var allItemsQueryResult = await GetItems(new InternalItemsQuery(user)
  1190. {
  1191. Recursive = true,
  1192. IsFolder = false,
  1193. IsVirtualItem = false,
  1194. EnableTotalRecordCount = true,
  1195. Limit = 0
  1196. }).ConfigureAwait(false);
  1197. if (itemDto != null)
  1198. {
  1199. itemDto.RecursiveItemCount = allItemsQueryResult.TotalRecordCount;
  1200. }
  1201. double recursiveItemCount = allItemsQueryResult.TotalRecordCount;
  1202. double unplayedCount = unplayedQueryResult.TotalRecordCount;
  1203. if (recursiveItemCount > 0)
  1204. {
  1205. var unplayedPercentage = (unplayedCount / recursiveItemCount) * 100;
  1206. dto.PlayedPercentage = 100 - unplayedPercentage;
  1207. dto.Played = dto.PlayedPercentage.Value >= 100;
  1208. dto.UnplayedItemCount = unplayedQueryResult.TotalRecordCount;
  1209. }
  1210. }
  1211. }
  1212. }