work
Avril 4 years ago
parent f1f3a30ac7
commit 9538f4bc46
Signed by: flanchan
GPG Key ID: 284488987C31F630

@ -4,257 +4,133 @@ use std::{
fmt, fmt,
}; };
//TODO: pub trait Context
// this was a complete failure, look at eyre instead or something
/// Context for an error
#[derive(Debug, Clone, PartialEq, Eq, Ord, PartialOrd, Hash)]
pub struct Context
{
line: Option<u32>,
column: Option<u32>,
file: &'static str,
}
impl Context
{
pub fn new(line: Option<u32>, column: Option<u32>, file: &'static str) -> Self
{
Self {
line,column,file
}
}
}
impl fmt::Display for Context
{ {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result fn line(&self) -> Option<u32>;
{ fn column(&self) -> Option<u32>;
write!(f, "{} ", self.file)?; fn file(&self) -> Option<&str>;
if let Some(line) = self.line {
write!(f, "{}:", line)?;
} else {
write!(f, "?:")?;
}
if let Some(column) = self.column {
write!(f, "{}", column)?;
} else {
write!(f, "?")?;
}
Ok(())
}
} }
/// A wrapper over an error with a nicer error message pub trait Reporter
///
/// Like doushio's `Muggle`.
pub struct Wrap<E,D>(E, D);
pub trait WrappedError
{ {
fn long_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result; type Error: error::Error + ?Sized;
fn short_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result;
}
impl<E,D> WrappedError for Wrap<E,D> #[inline] fn context(&self) -> Option<&dyn Context>
where D: fmt::Display,
E: error::Error + 'static
{
fn long_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{}", self.long_message()) None
} }
fn short_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result fn source(&self) -> &Self::Error;
fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result;
#[inline] fn long_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{}", self) self.short_message(f)
} }
} }
impl<E,D> Wrap<E,D> pub struct ShortMessage<R>(R)
where D: fmt::Display, where R: Reporter;
E: error::Error + 'static
{
/// Get all sources of the error
pub fn all_sources(&self) -> Vec<&(dyn error::Error + 'static)>
{
use std::error::Error;
std::iter::successors(self.source(), |x| x.source()).collect()
}
/// Into the inner error
pub fn into_inner(self) -> E
{
self.0
}
/// Get the full and long error message impl<R> fmt::Display for ShortMessage<R>
pub fn long_message(&self) -> impl fmt::Display + '_ where R: Reporter
{
#[inline] fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
struct LongOutput<'a, E,D>(&'a Wrap<E,D>); self.0.short_message(f)
impl<'a, E,D> fmt::Display for LongOutput<'a, E,D>
where Wrap<E,D>: error::Error
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{
writeln!(f, "{}", self.0)?;
use std::error::Error;
for (i, spec) in (1..).zip(std::iter::successors(self.0.source(), |x| x.source()))
{
writeln!(f, " [{}] -> {}", i, spec)?;
}
Ok(())
}
}
LongOutput(&self)
} }
} }
pub trait WrapExt: Sized pub struct LongMessage<R>(R)
{ where R: Reporter;
/// Wrap this error in another
fn wrap<T>(self, msg: T) -> Wrap<Self, T>
where T: fmt::Display;
}
impl<E> WrapExt for E where E: error::Error impl<R> fmt::Display for LongMessage<R>
where R: Reporter
{ {
fn wrap<T>(self, msg: T) -> Wrap<Self, T> #[inline] fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
where T: fmt::Display
{ {
Wrap(self, msg) self.0.long_message(f)
} }
} }
pub trait WrapErrExt<T,E>: Sized impl<E> Reporter for Box<E>
where E: error::Error + ?Sized
{ {
/// Wrap the error from this result with a nicer error message type Error = E;
fn wrap_err<U>(self, msg: U) -> Result<T,Wrap<E, U>>
where U: fmt::Display;
}
impl<T,E> WrapErrExt<T,E> for Result<T,E> fn source(&self) -> &Self::Error
{
fn wrap_err<U>(self, msg: U) -> Result<T,Wrap<E, U>>
where U: fmt::Display
{ {
self.map_err(|e| Wrap(e, msg)) &self
} }
}
fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
impl<E,D> fmt::Debug for Wrap<E,D>
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{}", std::any::type_name::<Self>()) write!(f, "{}", &self)
} }
} }
impl<E,D> fmt::Display for Wrap<E,D> impl<'a, R> Reporter for &'a R
where D: fmt::Display where R: Reporter
{ {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result type Error = <R as Reporter>::Error;
#[inline] fn context(&self) -> Option<&dyn Context>
{ {
write!(f, "{}", self.1) (*self).context()
} }
} fn source(&self) -> &Self::Error
impl<E,D> error::Error for Wrap<E,D>
where E: error::Error + 'static,
D: fmt::Display
{
fn source(&self) -> Option<&(dyn error::Error + 'static)>
{ {
Some(&self.0) (*self).source()
} }
}
fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
/// An error with a context
#[derive(Debug)]
pub struct ContextError<E>(pub E, pub Context)
where E: fmt::Debug;
impl<E> WrappedError for ContextError<E>
where E: fmt::Debug + fmt::Display
{
fn long_msg(&self,f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{}", self) (*self).short_message(f)
} }
fn short_msg(&self,f: &mut fmt::Formatter<'_>) -> fmt::Result #[inline] fn long_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{}", self.0) (*self).long_message(f)
} }
} }
impl<E> fmt::Display for ContextError<E> pub struct Wrap<E,M>(E, M);
where E: fmt::Display + fmt::Debug
impl<E,M> Reporter for Wrap<E,M>
where E: Reporter,
M: fmt::Display
{ {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result type Error = E::Error;
#[inline] fn context(&self) -> Option<&dyn Context>
{ {
write!(f, "{}: {}", self.1, self.0) self.0.context()
} }
} fn source(&self) -> &Self::Error
impl<E> error::Error for ContextError<E>
where E: fmt::Display + fmt::Debug{}
#[macro_export]
macro_rules! context {
() => ($crate::error::Context::new(Some(line!()), Some(column!()), file!()));
}
/// Construct a new error with context
#[macro_export] macro_rules! error {
($err:expr) => ($crate::error::ContextError($err, $crate::context!()));
($fmt:literal $($tt:tt)*) => ($crate::error!(format!($fmt $($tt)*)));
}
pub struct ErrorStack(Box<dyn WrappedError>);
impl<T: WrappedError + 'static> From<T> for ErrorStack
{
fn from(from: T) -> Self
{ {
Self(Box::new(from)) self.0.source()
} }
}
fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
impl fmt::Display for ErrorStack
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
self.0.long_msg(f) write!(f, "{}", self.1)
} }
} #[inline] fn long_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
impl fmt::Debug for ErrorStack
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f,"<opaque stack>") write!(f, "{}", self.1)?;
self.0.long_message(f) //ehh, also failed
} }
} }
impl error::Error for ErrorStack{} #[cfg(test)]
mod tests {
impl ErrorStack use super::*;
{ fn test() -> Box<dyn error::Error>
/// Get the short message from this stack
pub fn short_message(&self) -> impl fmt::Display + '_
{ {
pub struct ShortMessage<'a>(&'a ErrorStack); Box::new(std::io::Error::new(std::io::ErrorKind::UnexpectedEof, "uhhhhh"))
}
impl<'a> fmt::Display for ShortMessage<'a> #[test]
{ fn atest()
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
{ let bx = test();
self.0.0.short_msg(f) assert_eq!(format!("{}", LongMessage(&bx)), format!("{}", bx));
}
}
ShortMessage(self)
} }
} }

@ -7,10 +7,6 @@ mod ext;
use ext::*; use ext::*;
mod error; mod error;
use error::{
WrapErrExt as _,
WrapExt as _,
};
mod consts; mod consts;
mod util; mod util;
mod hash; mod hash;
@ -18,21 +14,8 @@ mod metadata;
mod resolve; mod resolve;
mod database; mod database;
fn inner() -> Result<(), error::ErrorStack> async fn begin() -> Result<i32, Box<dyn std::error::Error>>
{ {
Err(error!("constructing an error"))?;
Ok(())
}
fn outer() -> Result<(), error::ErrorStack>
{
inner().wrap_err("propagating an error")?;
Ok(())
}
async fn begin() -> Result<i32, error::ErrorStack>
{
outer().wrap_err("outer failed").wrap_err("outer failed twice!")?;
Ok(0) Ok(0)
} }
@ -41,8 +24,7 @@ async fn main() {
std::process::exit(match begin().await { std::process::exit(match begin().await {
Ok(0) => return, Ok(0) => return,
Err(err) => { Err(err) => {
eprintln!("\nexited with error: {}", err.short_message()); eprintln!("\nexited with error: {}", err);
println!("\n{}", err);
1 1
}, },
Ok(v) => v Ok(v) => v

Loading…
Cancel
Save