Add WriteError variant for custom errors

Signed-off-by: Facundo Tuesca <[email protected]>

Author: facutuesca

src/writer.rs

@@ -8,12 +8,14 @@ use alloc::{fmt, vec};
 #[derive(PartialEq, Eq, Debug)]
 pub enum WriteError {
     AllocationError,
+    CustomError(String),
 }
 
 impl fmt::Display for WriteError {
     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
         match self {
             WriteError::AllocationError => write!(f, "allocation error"),
+            WriteError::CustomError(s) => write!(f, "{s}"),
         }
     }
 }