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>, fn line(&self) -> Option<u32>;
column: Option<u32>, fn column(&self) -> Option<u32>;
file: &'static str, fn file(&self) -> Option<&str>;
} }
impl Context pub trait Reporter
{ {
pub fn new(line: Option<u32>, column: Option<u32>, file: &'static str) -> Self type Error: error::Error + ?Sized;
#[inline] fn context(&self) -> Option<&dyn Context>
{ {
Self { None
line,column,file
}
}
} }
fn source(&self) -> &Self::Error;
impl fmt::Display for Context fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result;
{ #[inline] fn long_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{} ", self.file)?; self.short_message(f)
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 struct ShortMessage<R>(R)
/// where R: Reporter;
/// Like doushio's `Muggle`.
pub struct Wrap<E,D>(E, D);
pub trait WrappedError impl<R> fmt::Display for ShortMessage<R>
where R: Reporter
{ {
fn long_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result; #[inline] fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
fn short_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result;
}
impl<E,D> WrappedError for Wrap<E,D>
where D: fmt::Display,
E: error::Error + 'static
{ {
fn long_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result self.0.short_message(f)
{
write!(f, "{}", self.long_message())
}
fn short_msg(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{
write!(f, "{}", self)
}
} }
impl<E,D> Wrap<E,D>
where D: fmt::Display,
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 struct LongMessage<R>(R)
pub fn into_inner(self) -> E where R: Reporter;
{
self.0
}
/// Get the full and long error message impl<R> fmt::Display for LongMessage<R>
pub fn long_message(&self) -> impl fmt::Display + '_ where R: Reporter
{
struct LongOutput<'a, E,D>(&'a Wrap<E,D>);
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)?; #[inline] fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
use std::error::Error;
for (i, spec) in (1..).zip(std::iter::successors(self.0.source(), |x| x.source()))
{ {
writeln!(f, " [{}] -> {}", i, spec)?; self.0.long_message(f)
}
Ok(())
} }
} }
LongOutput(&self) impl<E> Reporter for Box<E>
} where E: error::Error + ?Sized
} {
type Error = E;
pub trait WrapExt: Sized fn source(&self) -> &Self::Error
{ {
/// Wrap this error in another &self
fn wrap<T>(self, msg: T) -> Wrap<Self, T>
where T: fmt::Display;
} }
impl<E> WrapExt for E where E: error::Error fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{
fn wrap<T>(self, msg: T) -> Wrap<Self, T>
where T: fmt::Display
{ {
Wrap(self, msg) write!(f, "{}", &self)
} }
} }
pub trait WrapErrExt<T,E>: Sized impl<'a, R> Reporter for &'a R
where R: Reporter
{ {
/// Wrap the error from this result with a nicer error message type Error = <R as Reporter>::Error;
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> #[inline] fn context(&self) -> Option<&dyn Context>
{
fn wrap_err<U>(self, msg: U) -> Result<T,Wrap<E, U>>
where U: fmt::Display
{ {
self.map_err(|e| Wrap(e, msg)) (*self).context()
} }
} fn source(&self) -> &Self::Error
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>()) (*self).source()
}
} }
impl<E,D> fmt::Display for Wrap<E,D> fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
where D: fmt::Display
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{}", self.1) (*self).short_message(f)
}
} }
#[inline] fn long_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
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)> (*self).long_message(f)
{
Some(&self.0)
} }
} }
/// An error with a context pub struct Wrap<E,M>(E, M);
#[derive(Debug)]
pub struct ContextError<E>(pub E, pub Context)
where E: fmt::Debug;
impl<E> WrappedError for ContextError<E> impl<E,M> Reporter for Wrap<E,M>
where E: fmt::Debug + fmt::Display where E: Reporter,
{ M: fmt::Display
fn long_msg(&self,f: &mut fmt::Formatter<'_>) -> fmt::Result
{ {
write!(f, "{}", self) type Error = E::Error;
}
fn short_msg(&self,f: &mut fmt::Formatter<'_>) -> fmt::Result
{
write!(f, "{}", self.0)
}
}
impl<E> fmt::Display for ContextError<E> #[inline] fn context(&self) -> Option<&dyn Context>
where E: fmt::Display + fmt::Debug
{ {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result self.0.context()
{
write!(f, "{}: {}", self.1, self.0)
}
}
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)*)));
} }
fn source(&self) -> &Self::Error
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()
}
} }
impl fmt::Display for ErrorStack fn short_message(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
{
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);
impl<'a> fmt::Display for ShortMessage<'a>
{ {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result Box::new(std::io::Error::new(std::io::ErrorKind::UnexpectedEof, "uhhhhh"))
{
self.0.0.short_msg(f)
}
} }
#[test]
ShortMessage(self) fn atest()
{
let bx = test();
assert_eq!(format!("{}", LongMessage(&bx)), format!("{}", bx));
} }
} }

@ -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