Add stability extensions for openapi

compiler-rs/clients_schema_to_openapi/src/lib.rs

@@ -25,7 +25,7 @@ use std::io::{BufWriter, Write};
 use std::path::Path;
 use indexmap::IndexMap;
 
-use clients_schema::{Availabilities, Endpoint, IndexedModel};
+use clients_schema::{Availabilities, Endpoint, IndexedModel, Stability};
 use openapiv3::{Components, OpenAPI};
 use tracing::warn;
 
@@ -149,15 +149,25 @@ fn info(model: &IndexedModel) -> openapiv3::Info {
     }
 }
 
-pub fn availability_as_extension(availabilities: &Option<Availabilities>) -> IndexMap<String, serde_json::Value> {
+pub fn availability_as_extensions(availabilities: &Option<Availabilities>) -> IndexMap<String, serde_json::Value> {
     let mut result = IndexMap::new();
 
     if let Some(avails) = availabilities {
         // We may have several availabilities, but since generally exists only on stateful (stack)
         for (_, availability) in avails {
             if let Some(since) = &availability.since {
                 result.insert("x-available-since".to_string(), serde_json::Value::String(since.clone()));
-                break;
+            }
+            if let Some(stability) = &availability.stability {
+                match stability {
+                    Stability::Beta => {
+                        result.insert("x-beta".to_string(), serde_json::Value::Bool(true));
+                    }
+                    Stability::Experimental => {
+                        result.insert("x-technical-preview".to_string(), serde_json::Value::Bool(true));
+                    }
+                    _ => {}
+                }
             }
         }
     }

compiler-rs/clients_schema_to_openapi/src/paths.rs

@@ -211,7 +211,7 @@ pub fn add_endpoint(
             deprecated: endpoint.deprecation.is_some(),
             security: None,
             servers: vec![],
-            extensions: crate::availability_as_extension(&endpoint.availability),
+            extensions: crate::availability_as_extensions(&endpoint.availability),
         };
 
 

compiler-rs/clients_schema_to_openapi/src/schemas.rs

@@ -472,7 +472,7 @@ impl<'a> TypesAndComponents<'a> {
         data.external_docs = self.convert_external_docs(prop);
         data.deprecated = prop.deprecation.is_some();
         data.description = prop.description.clone();
-        data.extensions = crate::availability_as_extension(&prop.availability);
+        data.extensions = crate::availability_as_extensions(&prop.availability);
         // TODO: prop.aliases as extensions
         // TODO: prop.server_default as extension
         // TODO: prop.doc_id as extension (new representation of since and stability)