93 lines
2.8 KiB
Rust
93 lines
2.8 KiB
Rust
//
|
|
// imag - the personal information management suite for the commandline
|
|
// Copyright (C) 2015-2018 Matthias Beyer <mail@beyermatthias.de> and contributors
|
|
//
|
|
// This library is free software; you can redistribute it and/or
|
|
// modify it under the terms of the GNU Lesser General Public
|
|
// License as published by the Free Software Foundation; version
|
|
// 2.1 of the License.
|
|
//
|
|
// This library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
// Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public
|
|
// License along with this library; if not, write to the Free Software
|
|
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
//
|
|
|
|
use std::process::exit;
|
|
use std::fmt::Display;
|
|
use std::fmt::Formatter;
|
|
use std::fmt::Result as FmtResult;
|
|
use error_chain::ChainedError;
|
|
use ansi_term::Colour::Red;
|
|
|
|
struct ImagTrace<'a, T: 'a + ?Sized>(&'a T);
|
|
|
|
impl<'a, T: 'a + ?Sized> ImagTrace<'a, T> {
|
|
fn new(d: &'a T) -> ImagTrace<'a, T> {
|
|
ImagTrace(d)
|
|
}
|
|
}
|
|
|
|
impl<'a, T> Display for ImagTrace<'a, T>
|
|
where T: ChainedError
|
|
{
|
|
fn fmt(&self, fmt: &mut Formatter) -> FmtResult {
|
|
try!(writeln!(fmt, "{}: {}", Red.blink().paint("ERROR[ 0]"), self.0));
|
|
|
|
for (i, e) in self.0.iter().enumerate().skip(1) {
|
|
try!(writeln!(fmt, "{}: {}", Red.blink().paint(format!("ERROR[{:>4}]", i)), e));
|
|
}
|
|
|
|
if let Some(backtrace) = self.0.backtrace() {
|
|
try!(writeln!(fmt, "{}", Red.paint("--- BACKTRACE ---")));
|
|
try!(writeln!(fmt, "{:?}", backtrace));
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
|
|
pub fn trace_error<K, C: ChainedError<ErrorKind = K>>(e: &C) {
|
|
eprintln!("{}", ImagTrace::new(e));
|
|
}
|
|
|
|
pub fn trace_error_dbg<K, C: ChainedError<ErrorKind = K>>(e: &C) {
|
|
debug!("{}", e.display_chain());
|
|
}
|
|
|
|
/// Helper functions for `Result<T, E>` types to reduce overhead in the following situations:
|
|
///
|
|
/// ```ignore
|
|
/// function().map_err(|e| { trace_error(&e); e })
|
|
/// ```
|
|
///
|
|
/// and variants
|
|
pub trait MapErrTrace {
|
|
type Output;
|
|
|
|
fn map_err_trace(self) -> Self;
|
|
fn map_err_trace_exit_unwrap(self, code: i32) -> Self::Output;
|
|
}
|
|
|
|
impl<U, K, E: ChainedError<ErrorKind = K>> MapErrTrace for Result<U, E> {
|
|
type Output = U;
|
|
|
|
/// Simply call `trace_error()` on the Err (if there is one) and return the error.
|
|
///
|
|
/// This does nothing besides the side effect of printing the error trace
|
|
fn map_err_trace(self) -> Self {
|
|
self.map_err(|e| { trace_error(&e); e })
|
|
}
|
|
|
|
/// Trace the error and exit or unwrap the Ok(_).
|
|
fn map_err_trace_exit_unwrap(self, code: i32) -> Self::Output {
|
|
self.map_err(|e| { trace_error(&e); exit(code) }).unwrap()
|
|
}
|
|
|
|
}
|
|
|