Support using overridden customFields in templates

icpc · Nov 4, 2023 · 650afac · 650afac
1 parent 057a850
commit 650afac
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/cds/src/main/kotlin/org/icpclive/cds/adapters/AdvancedPropertiesAdapter.kt b/src/cds/src/main/kotlin/org/icpclive/cds/adapters/AdvancedPropertiesAdapter.kt
@@ -251,10 +251,12 @@ internal fun applyAdvancedProperties(
         }
     }
 
+    val teamInfoWithCustomFields = teamInfosPrelim
+        .mergeTeams(overrides.teamOverrides?.filterValues { it.customFields != null }?.mapValues { TeamInfoOverride(customFields = it.value.customFields) })
 
-    @Suppress("DEPRECATION") val teamInfos = teamInfosPrelim
-        .mergeTeams(overrides.teamMediaTemplate?.instantiateTemplate(teamInfosPrelim, TeamInfo::templateValueGetter))
-        .mergeTeams(overrides.teamOverrideTemplate?.instantiateTemplate(teamInfosPrelim, TeamInfo::templateValueGetter))
+    @Suppress("DEPRECATION") val teamInfos = teamInfoWithCustomFields
+        .mergeTeams(overrides.teamMediaTemplate?.instantiateTemplate(teamInfoWithCustomFields, TeamInfo::templateValueGetter))
+        .mergeTeams(overrides.teamOverrideTemplate?.instantiateTemplate(teamInfoWithCustomFields, TeamInfo::templateValueGetter))
         .mergeTeams(overrides.teamOverrides)
     val problemInfos = mergeProblems(info.problemList, overrides.problemOverrides)