Update RepeatLoop.cs

Author: Elektryk-Andrzej

Code/ContextSystem/Contexts/Control/Loops/RepeatLoop.cs

@@ -47,46 +47,27 @@ repeat 10 with $iter
 
     public override TryAddTokenRes TryAddToken(BaseToken token)
     {
-        switch (token)
+        if (token is not IValueToken valToken || !valToken.CapableOf<NumberValue>(out var getNumber))
         {
-            case NumberToken numberToken:
+            return TryAddTokenRes.Error($"Value '{token.RawRep}' cannot be interpreted as a number.");
+        }
+
+        _repeatCountExpression = () =>
+        {
+            if (getNumber().HasErrored(out var error, out var value))
             {
-                if (numberToken.Value < 0)
-                {
-                    return TryAddTokenRes.Error(
-                        $"Value '{numberToken.Value}' cannot be negative.");
-                }
-                
-                _repeatCount = (uint)numberToken.Value;
-                return TryAddTokenRes.End();
+                return error;
             }
-            
-            case IValueToken valToken:
-            {
-                if (!valToken.CapableOf<NumberValue>(out var getNumber))
-                {
-                    return TryAddTokenRes.Error($"Value '{token.RawRep}' returns a value, but.");   
-                }
-                
-                _repeatCountExpression = () =>
-                {
-                    if (getNumber().HasErrored(out var error, out var value))
-                    {
-                        return error;
-                    }
 
-                    if (value.Value < 0)
-                    {
-                        return $"Value '{value}' cannot be negative.";
-                    }
-
-                    return (uint)value.Value;
-                };
-                return TryAddTokenRes.End();
+            if (value.Value < 0)
+            {
+                return $"Value '{value}' cannot be negative.";
             }
-        }
 
-        return TryAddTokenRes.Error($"Value '{token.RawRep}' cannot be interpreted as a number.");
+            return (uint)value.Value;
+        };
+        
+        return TryAddTokenRes.End();
     }
 
     public override Result VerifyCurrentState()