Merge pull request #72 from skbkontur/d.kop/add-one-method

Add WithTypeBuildingContext method
skbkontur · May 16, 2024 · e9d6ace · e9d6ace
2 parents eaa21eb + aeecfa2
commit e9d6ace
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/TypeScript.ContractGenerator/CustomTypeGenerator.cs b/TypeScript.ContractGenerator/CustomTypeGenerator.cs
@@ -108,6 +108,12 @@ public CustomTypeGenerator WithTypeBuildingContext(Func<ITypeInfo, bool> accept,
             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)
         {
             propertyResolvers.Add(propertyResolver);