PeopleHelper.cs 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using MediaBrowser.Model.Entities;
  5. namespace MediaBrowser.Controller.Entities
  6. {
  7. public static class PeopleHelper
  8. {
  9. public static void AddPerson(List<PersonInfo> people, PersonInfo person)
  10. {
  11. if (person == null)
  12. {
  13. throw new ArgumentNullException(nameof(person));
  14. }
  15. if (string.IsNullOrEmpty(person.Name))
  16. {
  17. throw new ArgumentException("The person's name was empty or null.", nameof(person));
  18. }
  19. // Normalize
  20. if (string.Equals(person.Role, PersonType.GuestStar, StringComparison.OrdinalIgnoreCase))
  21. {
  22. person.Type = PersonType.GuestStar;
  23. }
  24. else if (string.Equals(person.Role, PersonType.Director, StringComparison.OrdinalIgnoreCase))
  25. {
  26. person.Type = PersonType.Director;
  27. }
  28. else if (string.Equals(person.Role, PersonType.Producer, StringComparison.OrdinalIgnoreCase))
  29. {
  30. person.Type = PersonType.Producer;
  31. }
  32. else if (string.Equals(person.Role, PersonType.Writer, StringComparison.OrdinalIgnoreCase))
  33. {
  34. person.Type = PersonType.Writer;
  35. }
  36. // If the type is GuestStar and there's already an Actor entry, then update it to avoid dupes
  37. if (string.Equals(person.Type, PersonType.GuestStar, StringComparison.OrdinalIgnoreCase))
  38. {
  39. var existing = people.FirstOrDefault(p => p.Name.Equals(person.Name, StringComparison.OrdinalIgnoreCase) && p.Type.Equals(PersonType.Actor, StringComparison.OrdinalIgnoreCase));
  40. if (existing != null)
  41. {
  42. existing.Type = PersonType.GuestStar;
  43. MergeExisting(existing, person);
  44. return;
  45. }
  46. }
  47. if (string.Equals(person.Type, PersonType.Actor, StringComparison.OrdinalIgnoreCase))
  48. {
  49. // If the actor already exists without a role and we have one, fill it in
  50. var existing = people.FirstOrDefault(p => p.Name.Equals(person.Name, StringComparison.OrdinalIgnoreCase) && (p.Type.Equals(PersonType.Actor, StringComparison.OrdinalIgnoreCase) || p.Type.Equals(PersonType.GuestStar, StringComparison.OrdinalIgnoreCase)));
  51. if (existing == null)
  52. {
  53. // Wasn't there - add it
  54. people.Add(person);
  55. }
  56. else
  57. {
  58. // Was there, if no role and we have one - fill it in
  59. if (string.IsNullOrEmpty(existing.Role) && !string.IsNullOrEmpty(person.Role))
  60. {
  61. existing.Role = person.Role;
  62. }
  63. MergeExisting(existing, person);
  64. }
  65. }
  66. else
  67. {
  68. var existing = people.FirstOrDefault(p =>
  69. string.Equals(p.Name, person.Name, StringComparison.OrdinalIgnoreCase) &&
  70. string.Equals(p.Type, person.Type, StringComparison.OrdinalIgnoreCase));
  71. // Check for dupes based on the combination of Name and Type
  72. if (existing == null)
  73. {
  74. people.Add(person);
  75. }
  76. else
  77. {
  78. MergeExisting(existing, person);
  79. }
  80. }
  81. }
  82. private static void MergeExisting(PersonInfo existing, PersonInfo person)
  83. {
  84. existing.SortOrder = person.SortOrder ?? existing.SortOrder;
  85. existing.ImageUrl = person.ImageUrl ?? existing.ImageUrl;
  86. foreach (var id in person.ProviderIds)
  87. {
  88. existing.SetProviderId(id.Key, id.Value);
  89. }
  90. }
  91. public static bool ContainsPerson(List<PersonInfo> people, string name)
  92. {
  93. if (string.IsNullOrEmpty(name))
  94. {
  95. throw new ArgumentNullException(nameof(name));
  96. }
  97. foreach (var i in people)
  98. {
  99. if (string.Equals(i.Name, name, StringComparison.OrdinalIgnoreCase))
  100. {
  101. return true;
  102. }
  103. }
  104. return false;
  105. }
  106. }
  107. }