use crate::*;
#[cfg(not(feature = "std"))]
use alloc::vec::Vec;
use core::convert::TryFrom;
use core::fmt::{Debug, Display};
use core::iter::FromIterator;
use core::ops::{Deref, DerefMut};
use self::debug::{trace, trace_generic};
#[derive(Debug, PartialEq)]
pub struct SequenceOf<T> {
pub(crate) items: Vec<T>,
}
impl<T> SequenceOf<T> {
#[inline]
pub const fn new(items: Vec<T>) -> Self {
SequenceOf { items }
}
#[inline]
pub fn into_vec(self) -> Vec<T> {
self.items
}
#[inline]
pub fn push(&mut self, item: T) {
self.items.push(item)
}
}
impl<T> AsRef<[T]> for SequenceOf<T> {
fn as_ref(&self) -> &[T] {
&self.items
}
}
impl<T> AsMut<[T]> for SequenceOf<T> {
fn as_mut(&mut self) -> &mut [T] {
&mut self.items
}
}
impl<T> Deref for SequenceOf<T> {
type Target = [T];
fn deref(&self) -> &Self::Target {
&self.items
}
}
impl<T> DerefMut for SequenceOf<T> {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.items
}
}
impl<T> From<SequenceOf<T>> for Vec<T> {
fn from(seq: SequenceOf<T>) -> Self {
seq.items
}
}
impl<T> FromIterator<T> for SequenceOf<T> {
fn from_iter<IT: IntoIterator<Item = T>>(iter: IT) -> Self {
let items = Vec::from_iter(iter);
SequenceOf::new(items)
}
}
impl<'a, T> TryFrom<Any<'a>> for SequenceOf<T>
where
T: FromBer<'a>,
{
type Error = Error;
fn try_from(any: Any<'a>) -> Result<Self> {
any.tag().assert_eq(Self::TAG)?;
if !any.header.is_constructed() {
return Err(Error::ConstructExpected);
}
let items = SequenceIterator::<T, BerParser>::new(any.data).collect::<Result<Vec<T>>>()?;
Ok(SequenceOf::new(items))
}
}
impl<T> CheckDerConstraints for SequenceOf<T>
where
T: CheckDerConstraints,
{
fn check_constraints(any: &Any) -> Result<()> {
any.tag().assert_eq(Self::TAG)?;
any.header.assert_constructed()?;
for item in SequenceIterator::<Any, DerParser>::new(any.data) {
let item = item?;
T::check_constraints(&item)?;
}
Ok(())
}
}
impl<'a, T, E> FromDer<'a, E> for SequenceOf<T>
where
T: FromDer<'a, E>,
E: From<Error> + Display + Debug,
{
fn from_der(bytes: &'a [u8]) -> ParseResult<Self, E> {
trace_generic(
core::any::type_name::<Self>(),
"SequenceOf::from_der",
|bytes| {
let (rem, any) = trace(core::any::type_name::<Self>(), parse_der_any, bytes)
.map_err(Err::convert)?;
any.header
.assert_tag(Self::TAG)
.map_err(|e| Err::Error(e.into()))?;
let items = SequenceIterator::<T, DerParser, E>::new(any.data)
.collect::<Result<Vec<T>, E>>()
.map_err(Err::Error)?;
Ok((rem, SequenceOf::new(items)))
},
bytes,
)
}
}
impl<T> Tagged for SequenceOf<T> {
const TAG: Tag = Tag::Sequence;
}
#[cfg(feature = "std")]
impl<T> ToDer for SequenceOf<T>
where
T: ToDer,
{
fn to_der_len(&self) -> Result<usize> {
self.items.to_der_len()
}
fn write_der_header(&self, writer: &mut dyn std::io::Write) -> SerializeResult<usize> {
self.items.write_der_header(writer)
}
fn write_der_content(&self, writer: &mut dyn std::io::Write) -> SerializeResult<usize> {
self.items.write_der_content(writer)
}
}
#[cfg(test)]
mod tests {
use crate::SequenceOf;
use core::iter::FromIterator;
#[test]
fn use_sequence_of() {
let mut set = SequenceOf::from_iter([1, 2, 3]);
set.push(4);
let sum: i32 = set.iter().sum();
assert_eq!(sum, 10);
assert_eq!(&set[1..3], &[2, 3]);
}
}