From f2beabcefb6e88186e10b8a117f8da020a63f670 Mon Sep 17 00:00:00 2001
From: JakkuSakura <qjk2001@gmail.com>
Date: Thu, 14 Dec 2023 12:54:28 +0900
Subject: [PATCH] fix(aide): fix inconsistency in extracting schema and cause
 redoc to crash

---
 crates/aide/src/gen.rs | 37 ++++++++++++++++++++-----------------
 1 file changed, 20 insertions(+), 17 deletions(-)

diff --git a/crates/aide/src/gen.rs b/crates/aide/src/gen.rs
index c4fe5c7..7b79914 100644
--- a/crates/aide/src/gen.rs
+++ b/crates/aide/src/gen.rs
@@ -50,18 +50,7 @@ pub fn on_error(handler: impl Fn(Error) + 'static) {
 /// [`OpenApi`]: crate::openapi::OpenApi
 pub fn extract_schemas(extract: bool) {
     in_context(|ctx| {
-        if extract {
-            ctx.schema = SchemaGenerator::new(SchemaSettings::draft07().with(|s| {
-                s.inline_subschemas = false;
-                s.definitions_path = "#/components/schemas/".into();
-            }));
-            ctx.extract_schemas = true;
-        } else {
-            ctx.schema = SchemaGenerator::new(SchemaSettings::draft07().with(|s| {
-                s.inline_subschemas = true;
-            }));
-            ctx.extract_schemas = false;
-        }
+        ctx.set_extract_schemas(extract)
     });
 }
 
@@ -143,22 +132,36 @@ impl GenContext {
             }
         }
 
-        Self {
-            schema: SchemaGenerator::new(
-                SchemaSettings::draft07().with(|s| s.inline_subschemas = false),
-            ),
+        let mut this = Self {
+            schema: SchemaGenerator::new(SchemaSettings::draft07()),
             infer_responses: true,
             all_error_responses: false,
             extract_schemas: true,
             show_error: default_error_filter,
             error_handler: None,
             no_content_status,
-        }
+        };
+        this.set_extract_schemas(true);
+        this
     }
 
     pub(crate) fn reset_error_filter(&mut self) {
         self.show_error = default_error_filter;
     }
+    fn set_extract_schemas(&mut self, extract: bool) {
+        if extract {
+            self.schema = SchemaGenerator::new(SchemaSettings::draft07().with(|s| {
+                s.inline_subschemas = false;
+                s.definitions_path = "#/components/schemas/".into();
+            }));
+            self.extract_schemas = true;
+        } else {
+            self.schema = SchemaGenerator::new(SchemaSettings::draft07().with(|s| {
+                s.inline_subschemas = true;
+            }));
+            self.extract_schemas = false;
+        }
+    }
 
     /// Add an error in the current context.
     #[tracing::instrument(skip_all)]