prost-protovalidate 0.3.0

Runtime validation for Protocol Buffer messages using buf.validate rules
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use prost_reflect::{DynamicMessage, EnumDescriptor};

use crate::config::ValidationConfig;
use crate::error::{Error, ValidationError};
use crate::violation::Violation;

use super::Evaluator;

/// Evaluator that checks if an enum value corresponds to a defined variant.
pub(crate) struct DefinedEnumEval {
    pub enum_descriptor: EnumDescriptor,
}

impl Evaluator for DefinedEnumEval {
    fn tautology(&self) -> bool {
        false
    }

    fn evaluate(
        &self,
        _msg: &DynamicMessage,
        val: &prost_reflect::Value,
        _cfg: &ValidationConfig,
    ) -> Result<(), Error> {
        let Some(enum_number) = val.as_enum_number() else {
            return Ok(());
        };

        if self.enum_descriptor.get_value(enum_number).is_none() {
            return Err(ValidationError::single(Violation::new(
                "",
                "enum.defined_only",
                "value must be one of the defined enum values",
            ))
            .into());
        }

        Ok(())
    }
}