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
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102

use super::*;
use std::fmt;

use serde::de::{self, Deserializer, SeqAccess, Visitor};
use serde::ser::{SerializeSeq, Serializer};
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize)]
pub struct CheckLogRaw {
	pub address: ValueSubTree,
	pub identifier: ValueSubTree,

	#[serde(default)]
	#[serde(skip_serializing_if = "Vec::is_empty")]
	pub topics: Vec<ValueSubTree>,

	pub data: ValueSubTree,
}

pub enum CheckLogsRaw {
	Star,
	List(Vec<CheckLogRaw>),
	DefaultStar,
}

impl CheckLogsRaw {
	pub fn is_star(&self) -> bool {
		matches!(self, CheckLogsRaw::Star)
	}

	pub fn is_default(&self) -> bool {
		matches!(self, CheckLogsRaw::DefaultStar)
	}
}

impl Default for CheckLogsRaw {
	fn default() -> Self {
		CheckLogsRaw::DefaultStar
	}
}

impl Serialize for CheckLogsRaw {
	fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
	where
		S: Serializer,
	{
		match self {
			CheckLogsRaw::Star | CheckLogsRaw::DefaultStar => serializer.serialize_str("*"),
			CheckLogsRaw::List(l) => {
				let mut seq = serializer.serialize_seq(Some(l.len()))?;
				for item in l {
					seq.serialize_element(item)?;
				}
				seq.end()
			},
		}
	}
}

struct CheckLogsVisitor;

impl<'de> Visitor<'de> for CheckLogsVisitor {
	type Value = CheckLogsRaw;

	// Format a message stating what data this Visitor expects to receive.
	fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
		formatter.write_str("serialized object JSON representation of log check")
	}

	fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
	where
		E: de::Error,
	{
		if value == "*" {
			Ok(CheckLogsRaw::Star)
		} else {
			Err(de::Error::custom("only '*' allowed as logs string value"))
		}
	}

	fn visit_seq<A>(self, mut seq: A) -> Result<Self::Value, A::Error>
	where
		A: SeqAccess<'de>,
	{
		let mut list = Vec::<CheckLogRaw>::new();

		while let Some(item) = seq.next_element()? {
			list.push(item);
		}

		Ok(CheckLogsRaw::List(list))
	}
}

impl<'de> Deserialize<'de> for CheckLogsRaw {
	fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
	where
		D: Deserializer<'de>,
	{
		deserializer.deserialize_any(CheckLogsVisitor)
	}
}