context: add provide_if methods

crates/indicator/src/context/layer/mod.rs

@@ -206,6 +206,18 @@ where
  self.with(AddData::with_data(data))
  }
 
+ /// Optionally provide data to the context.
+ ///
+ /// If the data is `None`, the layer will be skipped,
+ /// so will not panic if the data is not in the context.
+ fn provide_if<D>(self, data: Option<D>) -> Stack<Self, OptionalLayer<AddData<D>>>
+ where
+ D: Clone + Send + Sync + 'static,
+ Self: Sized,
+ {
+ self.with_if(data.map(AddData::with_data))
+ }
+
  /// Provide data to the context with the given data provider.
  fn provide_with<D>(
  self,
@@ -218,6 +230,26 @@ where
  self.with(AddData::new(provider))
  }
 
+ /// Optionally provide data to the context with the given data provider.
+ ///
+ /// If not enabled, the layer will be skipped,
+ /// so will not panic if the data is not in the context.
+ fn provide_with_if<D>(
+ self,
+ enable: bool,
+ provider: impl Fn() -> Option<D> + Send + Sync + 'static,
+ ) -> Stack<Self, OptionalLayer<AddData<D>>>
+ where
+ D: Send + Sync + 'static,
+ Self: Sized,
+ {
+ self.with_if(if enable {
+ Some(AddData::new(provider))
+ } else {
+ None
+ })
+ }
+
  /// Declare that the data of type `D` is in the context.
  fn from_context<D>(self) -> Stack<Self, AddData<D>>
  where

crates/indicator/src/context/mod.rs

@@ -237,6 +237,18 @@ pub trait ContextOperatorExt<In>: ContextOperator<In> {
  self.with(AddData::with_data(data))
  }
 
+ /// Optionally provide data to the context.
+ ///
+ /// If data is `None`, the layer will be skipped,
+ /// so it won't panic if the data is not in the context.
+ fn provide_if<D>(self, data: Option<D>) -> Either<AddDataOperator<D, Self>, Self>
+ where
+ D: Clone + Send + Sync + 'static,
+ Self: Sized,
+ {
+ self.with_if(data.map(AddData::with_data))
+ }
+
  /// Provide data to the context with the given data provider.
  fn provide_with<D>(
  self,
@@ -249,6 +261,26 @@ pub trait ContextOperatorExt<In>: ContextOperator<In> {
  self.with(AddData::new(provider))
  }
 
+ /// Optionally provide data to the context with the given data provider.
+ ///
+ /// If not enabled, the layer will be skipped,
+ /// so it will not panic if the data is not in the context.
+ fn provide_with_if<D>(
+ self,
+ enable: bool,
+ provider: impl Fn() -> Option<D> + Send + Sync + 'static,
+ ) -> Either<AddDataOperator<D, Self>, Self>
+ where
+ D: Send + Sync + 'static,
+ Self: Sized,
+ {
+ self.with_if(if enable {
+ Some(AddData::new(provider))
+ } else {
+ None
+ })
+ }
+
  /// Declare that the data of type `D` is in the context.
  fn from_context<D>(self) -> AddDataOperator<D, Self>
  where