From eb3bf311995d22bcf611138db38d552e29f23cc4 Mon Sep 17 00:00:00 2001
From: Charlie Marsh <charlie.r.marsh@gmail.com>
Date: Tue, 23 Jan 2024 22:29:58 -0500
Subject: [PATCH] Generate custom JSON schema for dynamic setting

---
 crates/ruff_python_formatter/src/options.rs | 31 +++++++++++++++++++--
 ruff.schema.json                            | 18 ++++++++++--
 2 files changed, 44 insertions(+), 5 deletions(-)

diff --git a/crates/ruff_python_formatter/src/options.rs b/crates/ruff_python_formatter/src/options.rs
index 122a9909e943a..e4022a58891d9 100644
--- a/crates/ruff_python_formatter/src/options.rs
+++ b/crates/ruff_python_formatter/src/options.rs
@@ -1,10 +1,11 @@
+use std::fmt;
+use std::path::Path;
+use std::str::FromStr;
+
 use ruff_formatter::printer::{LineEnding, PrinterOptions, SourceMapGeneration};
 use ruff_formatter::{FormatOptions, IndentStyle, IndentWidth, LineWidth};
 use ruff_macros::CacheKey;
 use ruff_python_ast::PySourceType;
-use std::fmt;
-use std::path::Path;
-use std::str::FromStr;
 
 /// Resolved options for formatting one individual file. The difference to `FormatterSettings`
 /// is that `FormatterSettings` stores the settings for multiple files (the entire project, a subdirectory, ..)
@@ -367,15 +368,39 @@ impl fmt::Display for DocstringCode {
 #[cfg_attr(feature = "serde", serde(untagged))]
 #[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
 pub enum DocstringCodeLineWidth {
+    /// Wrap docstring code examples at a fixed line width.
     Fixed(LineWidth),
+
+    /// Respect the line length limit setting for the surrounding Python code.
     #[default]
     #[cfg_attr(
         feature = "serde",
         serde(deserialize_with = "deserialize_docstring_code_line_width_dynamic")
     )]
+    #[cfg_attr(feature = "schemars", schemars(with = "DynamicSchema"))]
     Dynamic,
 }
 
+/// A dummy type that is used to generate a schema for `DocstringCodeLineWidth::Dynamic`.
+#[cfg(feature = "schemars")]
+struct DynamicSchema;
+
+#[cfg(feature = "schemars")]
+impl schemars::JsonSchema for DynamicSchema {
+    fn schema_name() -> String {
+        "Dynamic".to_string()
+    }
+
+    fn json_schema(_: &mut schemars::gen::SchemaGenerator) -> schemars::schema::Schema {
+        schemars::schema::SchemaObject {
+            instance_type: Some(schemars::schema::InstanceType::String.into()),
+            const_value: Some("dynamic".to_string().into()),
+            ..Default::default()
+        }
+        .into()
+    }
+}
+
 impl fmt::Debug for DocstringCodeLineWidth {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
         match *self {
diff --git a/ruff.schema.json b/ruff.schema.json
index 65a83ce31c241..54e7656c6d7c5 100644
--- a/ruff.schema.json
+++ b/ruff.schema.json
@@ -750,13 +750,27 @@
     "DocstringCodeLineWidth": {
       "anyOf": [
         {
-          "$ref": "#/definitions/LineWidth"
+          "description": "Wrap docstring code examples at a fixed line width.",
+          "allOf": [
+            {
+              "$ref": "#/definitions/LineWidth"
+            }
+          ]
         },
         {
-          "type": "null"
+          "description": "Respect the line length limit setting for the surrounding Python code.",
+          "allOf": [
+            {
+              "$ref": "#/definitions/Dynamic"
+            }
+          ]
         }
       ]
     },
+    "Dynamic": {
+      "type": "string",
+      "const": "dynamic"
+    },
     "Flake8AnnotationsOptions": {
       "type": "object",
       "properties": {