From b84c16945a9a18f7b30caffcaa4012102083836a Mon Sep 17 00:00:00 2001
From: i-moonlight <infinite.moonlight0506@gmail.com>
Date: Tue, 19 Dec 2023 18:32:33 +0100
Subject: [PATCH] Add SchemaGeneratorFactory property to
 OpenApiDocumentGeneratorSettings,  #4524

---
 src/NSwag.Generation/OpenApiDocumentGenerator.cs         | 3 ++-
 src/NSwag.Generation/OpenApiDocumentGeneratorSettings.cs | 4 ++++
 src/NSwag.Generation/OpenApiSchemaGenerator.cs           | 7 ++++---
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/NSwag.Generation/OpenApiDocumentGenerator.cs b/src/NSwag.Generation/OpenApiDocumentGenerator.cs
index 47ea6bf17..424a5c501 100644
--- a/src/NSwag.Generation/OpenApiDocumentGenerator.cs
+++ b/src/NSwag.Generation/OpenApiDocumentGenerator.cs
@@ -25,7 +25,8 @@ public class OpenApiDocumentGenerator
         /// <param name="schemaResolver">The schema resolver.</param>
         public OpenApiDocumentGenerator(OpenApiDocumentGeneratorSettings settings, JsonSchemaResolver schemaResolver)
         {
-            SchemaGenerator = new OpenApiSchemaGenerator(settings);
+            SchemaGenerator = settings.SchemaGeneratorFactory();
+
             _schemaResolver = schemaResolver;
             _settings = settings;
         }
diff --git a/src/NSwag.Generation/OpenApiDocumentGeneratorSettings.cs b/src/NSwag.Generation/OpenApiDocumentGeneratorSettings.cs
index 70c9b7079..94fabeb21 100644
--- a/src/NSwag.Generation/OpenApiDocumentGeneratorSettings.cs
+++ b/src/NSwag.Generation/OpenApiDocumentGeneratorSettings.cs
@@ -23,9 +23,13 @@ public class OpenApiDocumentGeneratorSettings
         /// <summary>Initializes a new instance of the <see cref="OpenApiDocumentGeneratorSettings"/> class.</summary>
         public OpenApiDocumentGeneratorSettings()
         {
+            SchemaGeneratorFactory = () => new OpenApiSchemaGenerator(this);
             DefaultResponseReferenceTypeNullHandling = ReferenceTypeNullHandling.NotNull;
         }
 
+        /// <summary>Gets or sets the JSON Schema generator factory (default: new instance of <see cref="OpenApiSchemaGenerator"/>.</summary>
+        public Func<OpenApiSchemaGenerator> SchemaGeneratorFactory { get; set; }
+
         /// <summary></summary>
         public JsonSchemaGeneratorSettings SchemaSettings { get; set; } = new SystemTextJsonSchemaGeneratorSettings
         {
diff --git a/src/NSwag.Generation/OpenApiSchemaGenerator.cs b/src/NSwag.Generation/OpenApiSchemaGenerator.cs
index 2131ac8cd..eb1ebc906 100644
--- a/src/NSwag.Generation/OpenApiSchemaGenerator.cs
+++ b/src/NSwag.Generation/OpenApiSchemaGenerator.cs
@@ -21,12 +21,13 @@ public class OpenApiSchemaGenerator : JsonSchemaGenerator
 
         /// <summary>Initializes a new instance of the <see cref="OpenApiSchemaGenerator" /> class.</summary>
         /// <param name="settings">The settings.</param>
-        public OpenApiSchemaGenerator(OpenApiDocumentGeneratorSettings settings) : base(settings.SchemaSettings)
+        public OpenApiSchemaGenerator(OpenApiDocumentGeneratorSettings settings) 
+            : base(settings.SchemaSettings)
         {
         }
 
         /// <summary>Generates the properties for the given type and schema.</summary>
-        /// <param name="typeDescription">The type desription.</param>
+        /// <param name="typeDescription">The type description.</param>
         /// <param name="schema">The properties</param>
         /// <param name="schemaResolver">The schema resolver.</param>
         /// <returns></returns>
@@ -51,7 +52,7 @@ protected override void GenerateObject(JsonSchema schema, JsonTypeDescription ty
             }
         }
 
-        /// <summary>Generetes a schema directly or referenced for the requested schema type; also adds nullability if required.</summary>
+        /// <summary>Generates a schema directly or referenced for the requested schema type; also adds nullability if required.</summary>
         /// <typeparam name="TSchemaType">The resulted schema type which may reference the actual schema.</typeparam>
         /// <param name="contextualType">The type of the schema to generate.</param>
         /// <param name="isNullable">Specifies whether the property, parameter or requested schema type is nullable.</param>