chore(purity): UniqueValidator 1

This commit is contained in:
yairm210 2025-08-03 23:11:42 +03:00
parent 5a583fd3e8
commit 3735364d81

View File

@ -15,6 +15,8 @@ import com.unciv.models.ruleset.unique.UniqueParameterType
import com.unciv.models.ruleset.unique.UniqueTarget import com.unciv.models.ruleset.unique.UniqueTarget
import com.unciv.models.ruleset.unique.UniqueType import com.unciv.models.ruleset.unique.UniqueType
import com.unciv.models.ruleset.unique.expressions.Expressions import com.unciv.models.ruleset.unique.expressions.Expressions
import yairm210.purity.annotations.Cache
import yairm210.purity.annotations.LocalState
import yairm210.purity.annotations.Readonly import yairm210.purity.annotations.Readonly
class UniqueValidator(val ruleset: Ruleset) { class UniqueValidator(val ruleset: Ruleset) {
@ -74,7 +76,7 @@ class UniqueValidator(val ruleset: Ruleset) {
val prefix by lazy { getUniqueContainerPrefix(uniqueContainer) + "\"${unique.text}\"" } val prefix by lazy { getUniqueContainerPrefix(uniqueContainer) + "\"${unique.text}\"" }
if (unique.type == null) return checkUntypedUnique(unique, tryFixUnknownUniques, uniqueContainer, prefix, reportRulesetSpecificErrors) if (unique.type == null) return checkUntypedUnique(unique, tryFixUnknownUniques, uniqueContainer, prefix, reportRulesetSpecificErrors)
val rulesetErrors = RulesetErrorList(ruleset) @LocalState val rulesetErrors = RulesetErrorList(ruleset)
if (uniqueContainer != null && if (uniqueContainer != null &&
!(unique.type.canAcceptUniqueTarget(uniqueContainer.getUniqueTarget()) || !(unique.type.canAcceptUniqueTarget(uniqueContainer.getUniqueTarget()) ||
@ -107,11 +109,11 @@ class UniqueValidator(val ruleset: Ruleset) {
complianceError.errorSeverity.getRulesetErrorSeverity(), uniqueContainer, unique complianceError.errorSeverity.getRulesetErrorSeverity(), uniqueContainer, unique
) )
addExpressionParseErrors(complianceError, rulesetErrors, uniqueContainer, unique) rulesetErrors += getExpressionParseErrors(complianceError, uniqueContainer, unique)
} }
for (conditional in unique.modifiers) { for (conditional in unique.modifiers) {
addConditionalErrors(conditional, rulesetErrors, prefix, unique, uniqueContainer, reportRulesetSpecificErrors) rulesetErrors += getConditionalErrors(conditional, prefix, unique, uniqueContainer, reportRulesetSpecificErrors)
} }
addUniqueTypeSpecificErrors(rulesetErrors, prefix, unique, uniqueContainer, reportRulesetSpecificErrors) addUniqueTypeSpecificErrors(rulesetErrors, prefix, unique, uniqueContainer, reportRulesetSpecificErrors)
@ -128,7 +130,6 @@ class UniqueValidator(val ruleset: Ruleset) {
} }
} }
if (unique.type in MapUnitCache.UnitMovementUniques if (unique.type in MapUnitCache.UnitMovementUniques
&& unique.modifiers.any { it.type != UniqueType.ConditionalOurUnit || it.params[0] !in Constants.all } && unique.modifiers.any { it.type != UniqueType.ConditionalOurUnit || it.params[0] !in Constants.all }
) )
@ -149,13 +150,14 @@ class UniqueValidator(val ruleset: Ruleset) {
return rulesetErrors return rulesetErrors
} }
private fun addExpressionParseErrors( @Readonly
private fun getExpressionParseErrors(
complianceError: UniqueComplianceError, complianceError: UniqueComplianceError,
rulesetErrors: RulesetErrorList,
uniqueContainer: IHasUniques?, uniqueContainer: IHasUniques?,
unique: Unique unique: Unique
) { ): RulesetErrorList {
if (!complianceError.acceptableParameterTypes.contains(UniqueParameterType.Countable)) return @LocalState val rulesetErrors = RulesetErrorList()
if (!complianceError.acceptableParameterTypes.contains(UniqueParameterType.Countable)) return rulesetErrors
val parseError = Expressions.getParsingError(complianceError.parameterName) val parseError = Expressions.getParsingError(complianceError.parameterName)
if (parseError != null) { if (parseError != null) {
@ -167,7 +169,7 @@ class UniqueValidator(val ruleset: Ruleset) {
val text = "\"${complianceError.parameterName}\" could not be parsed as an expression due to:" + val text = "\"${complianceError.parameterName}\" could not be parsed as an expression due to:" +
" ${parseError.message}. \n$parameterWithErrorLocationMarked" " ${parseError.message}. \n$parameterWithErrorLocationMarked"
rulesetErrors.add(text, RulesetErrorSeverity.WarningOptionsOnly, uniqueContainer, unique) rulesetErrors.add(text, RulesetErrorSeverity.WarningOptionsOnly, uniqueContainer, unique)
return return rulesetErrors
} }
val countableErrors = Expressions.getCountableErrors(complianceError.parameterName, ruleset) val countableErrors = Expressions.getCountableErrors(complianceError.parameterName, ruleset)
@ -176,6 +178,7 @@ class UniqueValidator(val ruleset: Ruleset) {
" ${countableErrors.joinToString(", ")}" " ${countableErrors.joinToString(", ")}"
rulesetErrors.add(text, RulesetErrorSeverity.WarningOptionsOnly, uniqueContainer, unique) rulesetErrors.add(text, RulesetErrorSeverity.WarningOptionsOnly, uniqueContainer, unique)
} }
return rulesetErrors
} }
private val resourceUniques = setOf(UniqueType.ProvidesResources, UniqueType.ConsumesResources, private val resourceUniques = setOf(UniqueType.ProvidesResources, UniqueType.ConsumesResources,
@ -188,21 +191,21 @@ class UniqueValidator(val ruleset: Ruleset) {
UniqueType.ConditionalWhenBelowAmountStatResource, UniqueType.ConditionalWhenBelowAmountStatResource,
) )
private fun addConditionalErrors( private fun getConditionalErrors(
conditional: Unique, conditional: Unique,
rulesetErrors: RulesetErrorList,
prefix: String, prefix: String,
unique: Unique, unique: Unique,
uniqueContainer: IHasUniques?, uniqueContainer: IHasUniques?,
reportRulesetSpecificErrors: Boolean reportRulesetSpecificErrors: Boolean
) { ): RulesetErrorList {
@LocalState val rulesetErrors = RulesetErrorList()
if (unique.hasFlag(UniqueFlag.NoConditionals)) { if (unique.hasFlag(UniqueFlag.NoConditionals)) {
rulesetErrors.add( rulesetErrors.add(
"$prefix contains the conditional \"${conditional.text}\"," + "$prefix contains the conditional \"${conditional.text}\"," +
" but the unique does not accept conditionals!", " but the unique does not accept conditionals!",
RulesetErrorSeverity.Error, uniqueContainer, unique RulesetErrorSeverity.Error, uniqueContainer, unique
) )
return return rulesetErrors
} }
if (conditional.type == null) { if (conditional.type == null) {
@ -221,7 +224,7 @@ class UniqueValidator(val ruleset: Ruleset) {
text, text,
RulesetErrorSeverity.Warning, uniqueContainer, unique RulesetErrorSeverity.Warning, uniqueContainer, unique
) )
return return rulesetErrors
} }
if (conditional.type.targetTypes.none { it.modifierType != UniqueTarget.ModifierType.None }) if (conditional.type.targetTypes.none { it.modifierType != UniqueTarget.ModifierType.None })
@ -276,10 +279,11 @@ class UniqueValidator(val ruleset: Ruleset) {
complianceError.errorSeverity.getRulesetErrorSeverity(), uniqueContainer, unique complianceError.errorSeverity.getRulesetErrorSeverity(), uniqueContainer, unique
) )
addExpressionParseErrors(complianceError, rulesetErrors, uniqueContainer, unique) getExpressionParseErrors(complianceError, uniqueContainer, unique)
} }
addDeprecationAnnotationErrors(conditional, "$prefix contains modifier \"${conditional.text}\" which", rulesetErrors, uniqueContainer) addDeprecationAnnotationErrors(conditional, "$prefix contains modifier \"${conditional.text}\" which", rulesetErrors, uniqueContainer)
return rulesetErrors
} }
private fun addUniqueTypeSpecificErrors( private fun addUniqueTypeSpecificErrors(
@ -334,11 +338,12 @@ class UniqueValidator(val ruleset: Ruleset) {
} }
/** Maps uncompliant parameters to their required types */ /** Maps uncompliant parameters to their required types */
@Readonly
private fun getComplianceErrors( private fun getComplianceErrors(
unique: Unique, unique: Unique,
): List<UniqueComplianceError> { ): List<UniqueComplianceError> {
if (unique.type == null) return emptyList() if (unique.type == null) return emptyList()
val errorList = ArrayList<UniqueComplianceError>() @LocalState val errorList = ArrayList<UniqueComplianceError>()
for ((index, param) in unique.params.withIndex()) { for ((index, param) in unique.params.withIndex()) {
// Trying to catch the error at #11404 // Trying to catch the error at #11404
if (unique.type.parameterTypeMap.size != unique.params.size) { if (unique.type.parameterTypeMap.size != unique.params.size) {
@ -363,11 +368,12 @@ class UniqueValidator(val ruleset: Ruleset) {
return errorList return errorList
} }
private val paramTypeErrorSeverityCache = HashMap<UniqueParameterType, HashMap<String, UniqueType.UniqueParameterErrorSeverity?>>() @Cache private val paramTypeErrorSeverityCache = HashMap<UniqueParameterType, HashMap<String, UniqueType.UniqueParameterErrorSeverity?>>()
@Readonly
private fun getParamTypeErrorSeverityCached(uniqueParameterType: UniqueParameterType, param: String): UniqueType.UniqueParameterErrorSeverity? { private fun getParamTypeErrorSeverityCached(uniqueParameterType: UniqueParameterType, param: String): UniqueType.UniqueParameterErrorSeverity? {
if (!paramTypeErrorSeverityCache.containsKey(uniqueParameterType)) if (!paramTypeErrorSeverityCache.containsKey(uniqueParameterType))
paramTypeErrorSeverityCache[uniqueParameterType] = hashMapOf() paramTypeErrorSeverityCache[uniqueParameterType] = hashMapOf()
val uniqueParamCache = paramTypeErrorSeverityCache[uniqueParameterType]!! @LocalState val uniqueParamCache = paramTypeErrorSeverityCache[uniqueParameterType]!!
if (uniqueParamCache.containsKey(param)) return uniqueParamCache[param] if (uniqueParamCache.containsKey(param)) return uniqueParamCache[param]
@ -376,6 +382,7 @@ class UniqueValidator(val ruleset: Ruleset) {
return severity return severity
} }
@Readonly
private fun checkUntypedUnique( private fun checkUntypedUnique(
unique: Unique, unique: Unique,
tryFixUnknownUniques: Boolean, tryFixUnknownUniques: Boolean,
@ -405,6 +412,7 @@ class UniqueValidator(val ruleset: Ruleset) {
) )
} }
@Readonly
private fun isFilteringUniqueAllowed(unique: Unique, reportRulesetSpecificErrors: Boolean): Boolean { private fun isFilteringUniqueAllowed(unique: Unique, reportRulesetSpecificErrors: Boolean): Boolean {
// Isolate this decision, to allow easy change of approach // Isolate this decision, to allow easy change of approach
// This says: Must have no conditionals or parameters, and is used in any "filtering" parameter of another Unique // This says: Must have no conditionals or parameters, and is used in any "filtering" parameter of another Unique
@ -413,6 +421,7 @@ class UniqueValidator(val ruleset: Ruleset) {
return unique.text in allUniqueParameters // referenced at least once from elsewhere return unique.text in allUniqueParameters // referenced at least once from elsewhere
} }
@Readonly
private fun tryFixUnknownUnique(unique: Unique, uniqueContainer: IHasUniques?, prefix: String): RulesetErrorList { private fun tryFixUnknownUnique(unique: Unique, uniqueContainer: IHasUniques?, prefix: String): RulesetErrorList {
val similarUniques = UniqueType.entries.filter { val similarUniques = UniqueType.entries.filter {
getRelativeTextDistance( getRelativeTextDistance(
@ -450,6 +459,7 @@ class UniqueValidator(val ruleset: Ruleset) {
companion object { companion object {
const val whichDoesNotFitParameterType = "which does not fit parameter type" const val whichDoesNotFitParameterType = "which does not fit parameter type"
@Readonly
internal fun getUniqueContainerPrefix(uniqueContainer: IHasUniques?) = internal fun getUniqueContainerPrefix(uniqueContainer: IHasUniques?) =
(if (uniqueContainer is IRulesetObject) "${uniqueContainer.originRuleset}: " else "") + (if (uniqueContainer is IRulesetObject) "${uniqueContainer.originRuleset}: " else "") +
(if (uniqueContainer == null) "The" else "(${uniqueContainer.getUniqueTarget().name}) ${uniqueContainer.name}'s") + (if (uniqueContainer == null) "The" else "(${uniqueContainer.getUniqueTarget().name}) ${uniqueContainer.name}'s") +