Add one method

skbkontur · May 16, 2024 · a3d9429 · a3d9429
1 parent eaa21eb
commit a3d9429
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/TypeScript.ContractGenerator/CustomTypeGenerator.cs b/TypeScript.ContractGenerator/CustomTypeGenerator.cs
@@ -107,6 +107,12 @@ public CustomTypeGenerator WithTypeBuildingContext(Func<ITypeInfo, bool> accept,
             typeBuildingContextRules.Add(new TypeBuildingContextRule(accept, createContext));
             return this;
         }
+
+        public CustomTypeGenerator WithTypeBuildingContext(Func<ITypeInfo, bool> accept, Func<ITypeInfo, ITypeBuildingContext> createContext)
+        {
+            typeBuildingContextRules.Add(new TypeBuildingContextRule(accept, createContext));
+            return this;
+        }
 
         public CustomTypeGenerator WithPropertyResolver(IPropertyResolver propertyResolver)
         {