From 2436627fdd21648a71e62cf3128e154c199c5cc6 Mon Sep 17 00:00:00 2001
From: samuraiGH <anonimajzer3@mail.ru>
Date: Fri, 28 Jun 2024 19:18:32 +0300
Subject: [PATCH] fixing method filtering

---
 .../TreeConversion/syntax_tree_visitor.cs     | 24 +++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/TreeConverter/TreeConversion/syntax_tree_visitor.cs b/TreeConverter/TreeConversion/syntax_tree_visitor.cs
index 2744eb567..6cbac5a69 100644
--- a/TreeConverter/TreeConversion/syntax_tree_visitor.cs
+++ b/TreeConverter/TreeConversion/syntax_tree_visitor.cs
@@ -18483,7 +18483,7 @@ namespace PascalABCCompiler.TreeConverter
             return false;
         }
 
-        private bool IsIEnumInterface(Type t)
+        private bool IsIEnumerableInterface(Type t)
 		{
             var IEnumType = typeof(IEnumerable);
 			var IEnumTypedType = typeof(IEnumerable<>);
@@ -18496,18 +18496,31 @@ namespace PascalABCCompiler.TreeConverter
             return isIEnumType || isIEnumTypedType;
 		}
 
+        public bool IsIEnumeratorInterface(Type t)
+        {
+			var IEnumType = typeof(IEnumerator);
+			var IEnumTypedType = typeof(IEnumerator<>);
+
+			var isIEnumType = (t == IEnumType);
+			var isIEnumTypedType = t.IsGenericType
+				&& (t.GetGenericTypeDefinition() == IEnumTypedType
+			);
+
+			return isIEnumType || isIEnumTypedType;
+		}
+
         public Type FindIEnumerableInterfaceInCompiledType(Type compiledType)
         {
 			var IEnumType = typeof(IEnumerable);
 			var IEnumTypedType = typeof(IEnumerable<>);
 
 			// если тип является интерфейсом IEnumerable или IEnumerable<T> то берём его
-			if (IsIEnumInterface(compiledType))
+			if (IsIEnumerableInterface(compiledType))
                 return compiledType;
 
             // иначе ищем подходящие интерфейсы в иерархии
             var filterdInterfaces = compiledType.FindInterfaces(
-                (item, _) => IsIEnumInterface(item),
+                (item, _) => IsIEnumerableInterface(item),
                 null
             );
 
@@ -18527,7 +18540,10 @@ namespace PascalABCCompiler.TreeConverter
 			}
 
             // для класса/записи необходимо учитывать явные реализации GetEnumerator
-            var methods = compiledType.GetMethods().Where(item => item.Name == "GetEnumerator").ToArray();
+            var methods = compiledType.GetMethods().Where(item => 
+                item.Name == "GetEnumerator"
+                && IsIEnumeratorInterface(item.ReturnType)
+            ).ToArray();
 
             switch (methods.Length)
             {