1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use super::{reader::PemReader, PemLabel};
use crate::{Decode, Reader};

/// Decoding trait for PEM documents.
///
/// This is an extension trait which is auto-impl'd for types which impl the
/// [`Decode`], [`PemLabel`], and [`Sized`] traits.
pub trait DecodePem: Decode + PemLabel + Sized {
    /// Decode the provided PEM-encoded string, interpreting the Base64-encoded
    /// body of the document using the [`Decode`] trait.
    fn decode_pem(pem: impl AsRef<[u8]>) -> Result<Self, Self::Error>;
}

impl<T: Decode + PemLabel + Sized> DecodePem for T {
    fn decode_pem(pem: impl AsRef<[u8]>) -> Result<Self, Self::Error> {
        let mut reader = PemReader::new(pem.as_ref()).map_err(crate::Error::from)?;
        Self::validate_pem_label(reader.type_label()).map_err(crate::Error::from)?;

        let ret = Self::decode(&mut reader)?;
        Ok(reader.finish(ret)?)
    }
}