From 2b914f6f58d2aefae792ca67159d01c9a0a3b26c Mon Sep 17 00:00:00 2001
From: FlorianDahn <ich@florian-dahn.de>
Date: Wed, 5 Jan 2022 01:17:44 +0200
Subject: [PATCH] Changing CustomMessageLoop parameter

---
 TelegramBotBase/Builder/BotBaseBuilder.cs                  | 7 ++-----
 .../Builder/Interfaces/IMessageLoopSelectionStage.cs       | 4 ++--
 2 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/TelegramBotBase/Builder/BotBaseBuilder.cs b/TelegramBotBase/Builder/BotBaseBuilder.cs
index bf44005..9a021d7 100644
--- a/TelegramBotBase/Builder/BotBaseBuilder.cs
+++ b/TelegramBotBase/Builder/BotBaseBuilder.cs
@@ -54,12 +54,9 @@ namespace TelegramBotBase.Builder
             return this;
         }
 
-        public IStartFormSelectionStage CustomMessageLoop(Type messageLoopClass)
+        public IStartFormSelectionStage CustomMessageLoop(IMessageLoopFactory messageLoopClass)
         {
-            if (messageLoopClass.IsSubclassOf(typeof(IMessageLoopFactory)))
-                throw new ArgumentException($"Not a subclass of {nameof(IMessageLoopFactory)}");
-
-            _messageloopfactory = messageLoopClass.GetConstructor(new Type[] { })?.Invoke(new object[] { }) as IMessageLoopFactory;
+            _messageloopfactory = messageLoopClass;
 
             return this;
         }
diff --git a/TelegramBotBase/Builder/Interfaces/IMessageLoopSelectionStage.cs b/TelegramBotBase/Builder/Interfaces/IMessageLoopSelectionStage.cs
index 125945e..02cf87e 100644
--- a/TelegramBotBase/Builder/Interfaces/IMessageLoopSelectionStage.cs
+++ b/TelegramBotBase/Builder/Interfaces/IMessageLoopSelectionStage.cs
@@ -19,9 +19,9 @@ namespace TelegramBotBase.Builder.Interfaces
         /// <summary>
         /// Chooses a custom message loop.
         /// </summary>
-        /// <typeparam name="T"></typeparam>
+        /// <param name="startFormClass"></param>
         /// <returns></returns>
-        IStartFormSelectionStage CustomMessageLoop(Type startFormClass);
+        IStartFormSelectionStage CustomMessageLoop(IMessageLoopFactory startFormClass);
 
 
         /// <summary>