diff --git a/source/ChromeDevTools/MethodTypeMap.cs b/source/ChromeDevTools/MethodTypeMap.cs index 3f96d98c75d1fbd43b6379ccb81e303456908490..4f1a66849dfc0bb032c4f620e04f4d9466a92757 100644 --- a/source/ChromeDevTools/MethodTypeMap.cs +++ b/source/ChromeDevTools/MethodTypeMap.cs @@ -12,17 +12,25 @@ namespace MasterDevs.ChromeDevTools private readonly Dictionary<string, Type> _eventTypes = new Dictionary<string, Type>(); public MethodTypeMap() + : this("Chrome") { - LoadMethodTypeMap(); } - private void LoadMethodTypeMap() + public MethodTypeMap(string alias) + { + LoadMethodTypeMap(alias); + } + + private void LoadMethodTypeMap(string alias) { var assembly = Assembly.GetExecutingAssembly(); var assemblyTypes = assembly.GetTypes(); foreach (var type in assemblyTypes) { if (!type.IsClass) continue; + + if (!type.Namespace.StartsWith($"MasterDevs.ChromeDevTools.Protocol.{alias}")) continue; + if (type.Name.EndsWith("CommandResponse")) { var methodName = GetMethodName<CommandResponseAttribute>(type);