Added further validation

BigBang1112 · Dec 12, 2022 · 5beb128 · 5beb128
1 parent 7ce8779
commit 5beb128
Showing 1 changed file with 44 additions and 9 deletions.
diff --git a/Src/RandomizerTMF.Logic/RandomizerEngine.cs b/Src/RandomizerTMF.Logic/RandomizerEngine.cs
@@ -784,22 +784,57 @@ public static void ValidateRules()
             throw new RuleValidationException("Time limit cannot be above 9:59:59");
         }
 
-        if (Config.Rules.RequestRules.PrimaryType is EPrimaryType.Platform
-        && (Config.Rules.RequestRules.Site.HasFlag(ESite.TMNF) || Config.Rules.RequestRules.Site.HasFlag(ESite.Nations)))
+        foreach (var primaryType in Enum.GetValues<EPrimaryType>())
         {
-            throw new RuleValidationException("Platform is not valid with TMNF or Nations Exchange");
+            if (primaryType is EPrimaryType.Race)
+            {
+                continue;
+            }
+
+            if (Config.Rules.RequestRules.PrimaryType == primaryType
+            && (Config.Rules.RequestRules.Site == ESite.Any
+             || Config.Rules.RequestRules.Site.HasFlag(ESite.TMNF) || Config.Rules.RequestRules.Site.HasFlag(ESite.Nations)))
+            {
+                throw new RuleValidationException($"{primaryType} is not valid with TMNF or Nations Exchange");
+            }
+        }
+
+        if (Config.Rules.RequestRules.Environment is not null || Config.Rules.RequestRules.Vehicle is not null)
+        {
+            foreach (var env in Enum.GetValues<EEnvironment>())
+            {
+                if (env is EEnvironment.Stadium)
+                {
+                    continue;
+                }
+
+                if (Config.Rules.RequestRules.Site != ESite.Any && !Config.Rules.RequestRules.Site.HasFlag(ESite.TMNF) && !Config.Rules.RequestRules.Site.HasFlag(ESite.Nations))
+                {
+                    continue;
+                }
+
+                if (Config.Rules.RequestRules.Environment?.Contains(env) == true)
+                {
+                    throw new RuleValidationException($"{env} is not valid with TMNF or Nations Exchange");
+                }
+
+                if (Config.Rules.RequestRules.Vehicle?.Contains(env) == true)
+                {
+                    throw new RuleValidationException($"{env}Car is not valid with TMNF or Nations Exchange");
+                }
+            }
         }
 
-        if (Config.Rules.RequestRules.PrimaryType is EPrimaryType.Stunts
-        && (Config.Rules.RequestRules.Site.HasFlag(ESite.TMNF) || Config.Rules.RequestRules.Site.HasFlag(ESite.Nations)))
+        if (Config.Rules.RequestRules.EqualEnvironmentDistribution
+         && Config.Rules.RequestRules.Site.HasFlag(ESite.TMNF) || Config.Rules.RequestRules.Site.HasFlag(ESite.Nations))
         {
-            throw new RuleValidationException("Stunts is not valid with TMNF or Nations Exchange");
+            throw new RuleValidationException($"Equal environment distribution is not valid with TMNF or Nations Exchange");
         }
 
-        if (Config.Rules.RequestRules.PrimaryType is EPrimaryType.Puzzle
-        && (Config.Rules.RequestRules.Site.HasFlag(ESite.TMNF) || Config.Rules.RequestRules.Site.HasFlag(ESite.Nations)))
+        if (Config.Rules.RequestRules.EqualVehicleDistribution
+         && Config.Rules.RequestRules.Site.HasFlag(ESite.TMNF) || Config.Rules.RequestRules.Site.HasFlag(ESite.Nations))
         {
-            throw new RuleValidationException("Puzzle is not valid with TMNF or Nations Exchange");
+            throw new RuleValidationException($"Equal vehicle distribution is not valid with TMNF or Nations Exchange");
         }
     }