1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
// // imag - the personal information management suite for the commandline // Copyright (C) 2015-2020 the imag 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::io::ErrorKind; use crate::exit::ExitCode; pub enum Settings { Ignore(ErrorKind), IgnoreAny(Vec<ErrorKind>), } pub trait ToExitCode<T> { fn to_exit_code(self) -> Result<T, ExitCode>; fn to_exit_code_with(self, _: Settings) -> Result<T, ExitCode>; } impl<T> ToExitCode<T> for Result<T, ::std::io::Error> { /// Returns an exit code of 0 if the error was a broken pipe, else 1 fn to_exit_code(self) -> Result<T, ExitCode> { self.to_exit_code_with(Settings::Ignore(ErrorKind::BrokenPipe)) } /// Returns an exit code depending on the settings /// /// Via the settings, errors can be ignores (translates to exit code zero). All other errors /// are translated into exit code 1 /// fn to_exit_code_with(self, settings: Settings) -> Result<T, ExitCode> { self.map_err(move |e| match settings { Settings::Ignore(kind) => if e.kind() == kind { 0 } else { 1 }, Settings::IgnoreAny(v) => if v.iter().any(|el| e.kind() == *el) { 0 } else { 1 }, }) .map_err(ExitCode::from) } }