nucleo-picker 0.7.0-alpha.2

//! # Errors during interactive picker usage
//! This module contains the custom error type [`PickError`] returned by the
//! [`Picker::pick`](crate::Picker::pick) method, and siblings `Picker::pick_*`. The error type is
//! comprehensive and the individual picker method used may or may not result in the corresponding
//! errors.
//!
//! See the [`PickError`] documentation for more detail.
//!
//! ## Example
//! Convert a [`PickError::UserInterrupted`] silently into no choice, propagating any other error as an IO
//! error. Use with `picker.pick().or_else(suppress_abort)`.
//! ```
//! # use nucleo_picker::error::PickError;
//! # use std::io;
//! fn suppress_abort<D: Default>(err: PickError) -> Result<D, io::Error> {
//!     match err {
//!         PickError::UserInterrupted => Ok(D::default()),
//!         e => Err(e.into()),
//!     }
//! }
//! ```

use std::{convert::Infallible, error::Error as StdError, fmt, io};

/// An error which may be returned while running the picker interactively.
///
/// This is marked non-exhaustive since more variants may be added in the future. It is recommended
/// to handle the errors that are relevant to your application and propagate any remaining errors
/// as an [`io::Error`].
///
/// ## Type parameter for `Aborted` variant
/// The [`PickError::Aborted`] variant can be used by the application to propagate errors to the
/// picker; the application-defined error type is the type parameter `A`. By default, `A = !`
/// which means this type of abort will *never occur* and can be ignored during pattern matching.
///
/// This library will never generate an abort error directly. In order to pass errors downstream to
/// the picker, the application can define an abort error type using the
/// [`EventSource::AbortErr`](crate::EventSource::AbortErr) associated type. This associated type
/// is the same as the type parameter here when used in
/// [`Picker::pick_with_io`](crate::Picker::pick_with_io).
///
/// ## Relationship to `io::Error`
/// This error type with the default type parameter is (in spirit) an [`io::Error`], but with
/// more precise variants not present in the default [`io::Error`]. For convenience and
/// (partial) backwards compatibility, there is a `From<PickError> for io::Error` implementation;
/// this propagates the underlying IO error and converts any other error message to an
/// [`io::Error`] using [`io::Error::other`].
///
/// There is also a `From<PickError<io::Error>> for io::Error` to handle the common use-case that
/// the only error type which may occur during standard operation of your application is an IO
/// error; in this case, the conversion maps both the `Aborted(io::Error)` and `IO(io::Error)`
/// versions directly to an `io::Error`.
///
/// Any other abort error type `A` requires manual handling. The [`PickError::factor`] method
/// can be used to unwind non-aborted variants into an `io::Error` and extract the
/// error present in the `Aborted` variant.
#[derive(Debug)]
#[non_exhaustive]
pub enum PickError<A = Infallible> {
    /// A read or write resulted in an IO error.
    IO(io::Error),
    /// The event stream disconnected while the picker was still running.
    Disconnected,
    /// The picker quit at the user's request.
    UserInterrupted,
    /// The picker could not be started since the writer is not interactive.
    NotInteractive,
    /// The picker was aborted because of an upstream error.
    Aborted(A),
}

impl<A> PickError<A> {
    /// Convert a `PickError<A>` into either an `Ok(A)` or `Err(PickError<!>)`, for
    /// convenience of error propogation.
    ///
    /// # Example
    /// Use `factor` to simplify processing of custom [`PickError<A>`]s when you mainly care about
    /// your application error.
    /// ```
    /// # use nucleo_picker::error::PickError;
    /// use std::{error, io};
    ///
    /// // Even though `PickError<A>` need not satisfy `Into<io::Error>`, `PickError<!>`
    /// // always does.
    /// fn print_or_propogate<A: error::Error>(pick_err: PickError<A>) -> Result<(), io::Error> {
    ///     let app_err = pick_err.factor()?;
    ///     eprintln!("{app_err}");
    ///     Ok(())
    /// }
    /// ```
    pub fn factor(self) -> Result<A, PickError> {
        match self {
            PickError::IO(error) => Err(PickError::IO(error)),
            PickError::Disconnected => Err(PickError::Disconnected),
            PickError::UserInterrupted => Err(PickError::UserInterrupted),
            PickError::NotInteractive => Err(PickError::NotInteractive),
            PickError::Aborted(a) => Ok(a),
        }
    }
}

impl<A: fmt::Display> fmt::Display for PickError<A> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            PickError::IO(error) => error.fmt(f),
            PickError::Disconnected => {
                f.write_str("event source disconnected while picker was still active")
            }
            PickError::Aborted(err) => write!(f, "received abort: {err}"),
            PickError::UserInterrupted => f.write_str("keyboard interrupt"),
            PickError::NotInteractive => {
                f.write_str("picker could not start since the screen is not interactive")
            }
        }
    }
}

impl<A: StdError> StdError for PickError<A> {}

impl<A> From<io::Error> for PickError<A> {
    fn from(err: io::Error) -> Self {
        Self::IO(err)
    }
}

// ideally we would like to replace these two conversions with a blanket implementation
// `impl<A: Into<io::Error>> From<PickError<A>> for io::Error`; however, currently there is no
// implementation of `From<!> for T` for a variety of reasons; so we are stuck with doing this for
// maximal compabitility since in the vast majority of cases, `A = !`.
impl From<PickError> for io::Error {
    fn from(err: PickError) -> Self {
        match err {
            PickError::IO(io_err) => io_err,
            _ => io::Error::other(err),
        }
    }
}

impl From<PickError<io::Error>> for io::Error {
    fn from(err: PickError<io::Error>) -> Self {
        match err.factor() {
            Ok(io_err) => io_err,
            Err(pick_err) => pick_err.into(),
        }
    }
}