Folder.cs 48 KB

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