Add Range structs to schema

jamesmunns · Dec 3, 2024 · 7c6fd97 · 7c6fd97
1 parent 109c223
commit 7c6fd97
Showing 1 changed file with 59 additions and 4 deletions.
diff --git a/source/postcard-schema/src/impls/builtins_nostd.rs b/source/postcard-schema/src/impls/builtins_nostd.rs
@@ -3,12 +3,15 @@
 //! These implementations are always available
 
 use crate::{
-    schema::{DataModelType, DataModelVariant, NamedType, NamedVariant},
+    schema::{DataModelType, DataModelVariant, NamedType, NamedValue, NamedVariant},
     Schema,
 };
-use core::num::{
-    NonZeroI128, NonZeroI16, NonZeroI32, NonZeroI64, NonZeroI8, NonZeroU128, NonZeroU16,
-    NonZeroU32, NonZeroU64, NonZeroU8,
+use core::{
+    num::{
+        NonZeroI128, NonZeroI16, NonZeroI32, NonZeroI64, NonZeroI8, NonZeroU128, NonZeroU16,
+        NonZeroU32, NonZeroU64, NonZeroU8,
+    },
+    ops::{Range, RangeFrom, RangeInclusive, RangeTo},
 };
 
 macro_rules! impl_schema {
@@ -100,3 +103,55 @@ impl<T: Schema, const N: usize> Schema for [T; N] {
         ty: &DataModelType::Tuple(&[T::SCHEMA; N]),
     };
 }
+
+impl<T: Schema> Schema for Range<T> {
+    const SCHEMA: &'static crate::schema::NamedType = &NamedType {
+        name: "Range<T>",
+        ty: &DataModelType::Struct(&[
+            &NamedValue {
+                name: "start",
+                ty: T::SCHEMA,
+            },
+            &NamedValue {
+                name: "end",
+                ty: T::SCHEMA,
+            },
+        ]),
+    };
+}
+
+impl<T: Schema> Schema for RangeInclusive<T> {
+    const SCHEMA: &'static crate::schema::NamedType = &NamedType {
+        name: "RangeInclusive<T>",
+        ty: &DataModelType::Struct(&[
+            &NamedValue {
+                name: "start",
+                ty: T::SCHEMA,
+            },
+            &NamedValue {
+                name: "end",
+                ty: T::SCHEMA,
+            },
+        ]),
+    };
+}
+
+impl<T: Schema> Schema for RangeFrom<T> {
+    const SCHEMA: &'static crate::schema::NamedType = &NamedType {
+        name: "RangeFrom<T>",
+        ty: &DataModelType::Struct(&[&NamedValue {
+            name: "start",
+            ty: T::SCHEMA,
+        }]),
+    };
+}
+
+impl<T: Schema> Schema for RangeTo<T> {
+    const SCHEMA: &'static crate::schema::NamedType = &NamedType {
+        name: "RangeTo<T>",
+        ty: &DataModelType::Struct(&[&NamedValue {
+            name: "end",
+            ty: T::SCHEMA,
+        }]),
+    };
+}