From 991ac1c20f99187bc67058f2d427c25aa345479c Mon Sep 17 00:00:00 2001
From: Mikal Stordal <mikalstordal@gmail.com>
Date: Fri, 25 Oct 2024 10:37:38 +0200
Subject: [PATCH] fix: ensure only 1 xref per company exist

Closes #1191
---
 Shoko.Server/Providers/TMDB/TmdbMetadataService.cs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/Shoko.Server/Providers/TMDB/TmdbMetadataService.cs b/Shoko.Server/Providers/TMDB/TmdbMetadataService.cs
index 34d45d6d6..c1223e89f 100644
--- a/Shoko.Server/Providers/TMDB/TmdbMetadataService.cs
+++ b/Shoko.Server/Providers/TMDB/TmdbMetadataService.cs
@@ -1931,7 +1931,8 @@ private void PurgeTitlesAndOverviews(ForeignEntityType foreignType, int foreignI
     private async Task<bool> UpdateCompanies(IEntityMetadata tmdbEntity, List<ProductionCompany> companies)
     {
         var existingXrefs = _xrefTmdbCompanyEntity.GetByTmdbEntityTypeAndID(tmdbEntity.Type, tmdbEntity.Id)
-            .ToDictionary(xref => xref.TmdbCompanyID);
+            .GroupBy(xref => xref.TmdbCompanyID)
+            .ToDictionary(xref => xref.Key, groupBy => groupBy.ToList());
         var xrefsToAdd = 0;
         var xrefsToSkip = new HashSet<int>();
         var xrefsToSave = new List<TMDB_Company_Entity>();
@@ -1939,8 +1940,9 @@ private async Task<bool> UpdateCompanies(IEntityMetadata tmdbEntity, List<Produc
         foreach (var company in companies)
         {
             var currentIndex = indexCounter++;
-            if (existingXrefs.TryGetValue(company.Id, out var existingXref))
+            if (existingXrefs.TryGetValue(company.Id, out var existingXrefList))
             {
+                var existingXref = existingXrefList[0];
                 if (existingXref.Ordering != currentIndex || existingXref.ReleasedAt != tmdbEntity.ReleasedAt)
                 {
                     existingXref.Ordering = currentIndex;
@@ -1958,6 +1960,7 @@ private async Task<bool> UpdateCompanies(IEntityMetadata tmdbEntity, List<Produc
             await UpdateCompany(company);
         }
         var xrefsToRemove = existingXrefs.Values
+            .SelectMany(xrefs => xrefs)
             .ExceptBy(xrefsToSkip, o => o.TMDB_Company_EntityID)
             .ToList();