From 8f85178a7ae0dbb4b9b73b2c13c7e71986ca019b Mon Sep 17 00:00:00 2001
From: Graham Esau <gesau@hotmail.co.uk>
Date: Fri, 12 Jun 2020 21:38:13 +0100
Subject: [PATCH] Add SetSingleExample visitor for openapi 3.0

---
 schemars/src/gen.rs   |  3 +++
 schemars/src/visit.rs | 26 ++++++++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/schemars/src/gen.rs b/schemars/src/gen.rs
index 27616de..04992ab 100644
--- a/schemars/src/gen.rs
+++ b/schemars/src/gen.rs
@@ -79,6 +79,9 @@ impl SchemaSettings {
                 Box::new(ReplaceBoolSchemas {
                     skip_additional_properties: true,
                 }),
+                Box::new(SetSingleExample {
+                    retain_examples: false,
+                }),
             ],
             _hidden: (),
         }
diff --git a/schemars/src/visit.rs b/schemars/src/visit.rs
index b9e3d50..a041e16 100644
--- a/schemars/src/visit.rs
+++ b/schemars/src/visit.rs
@@ -93,6 +93,7 @@ fn visit_single_or_vec<V: Visitor + ?Sized>(v: &mut V, target: &mut Option<Singl
 /// TODO document
 #[derive(Debug, Clone)]
 pub struct ReplaceBoolSchemas {
+    /// TODO document
     pub skip_additional_properties: bool,
 }
 
@@ -146,3 +147,28 @@ impl Visitor for RemoveRefSiblings {
         }
     }
 }
+
+/// TODO document
+#[derive(Debug, Clone)]
+pub struct SetSingleExample {
+    /// TODO document
+    pub retain_examples: bool,
+}
+
+impl Visitor for SetSingleExample {
+    fn visit_schema_object(&mut self, schema: &mut SchemaObject) {
+        visit_schema_object(self, schema);
+
+        let first_example = schema.metadata.as_mut().and_then(|m| {
+            if self.retain_examples {
+                m.examples.first().cloned()
+            } else {
+                m.examples.drain(..).next()
+            }
+        });
+
+        if let Some(example) = first_example {
+            schema.extensions.insert("example".to_owned(), example);
+        }
+    }
+}