database_error.rs

use diesel::result;

use std::convert::From;
use std::error::Error;
use std::path::PathBuf;
use std::{fmt, io};

use self::DatabaseError::*;

pub type DatabaseResult<T> = Result<T, DatabaseError>;

#[derive(Debug)]
pub enum DatabaseError {
    ProjectRootNotFound(PathBuf),
    DatabaseUrlMissing,
    IoError(io::Error),
    QueryError(result::Error),
    ConnectionError(result::ConnectionError),
    MigrationError(Box<dyn Error + Send + Sync + 'static>),
}

impl From<io::Error> for DatabaseError {
    fn from(e: io::Error) -> Self {
        IoError(e)
    }
}

impl From<result::Error> for DatabaseError {
    fn from(e: result::Error) -> Self {
        QueryError(e)
    }
}

impl From<result::ConnectionError> for DatabaseError {
    fn from(e: result::ConnectionError) -> Self {
        ConnectionError(e)
    }
}

impl From<Box<dyn Error + Send + Sync + 'static>> for DatabaseError {
    fn from(e: Box<dyn Error + Send + Sync + 'static>) -> Self {
        MigrationError(e)
    }
}

impl Error for DatabaseError {}

impl fmt::Display for DatabaseError {
    fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
        match *self {
            ProjectRootNotFound(ref p) => {
                write!(f, "Unable to find diesel.toml or Cargo.toml in {:?} or any parent directories.", p)
            }
            DatabaseUrlMissing => {
                f.write_str("The --database-url argument must be passed, or the DATABASE_URL environment variable must be set.")
            }
            IoError(ref error) => f.write_str(&error
                .source()
                .map(ToString::to_string)
                .unwrap_or_else(|| error.to_string())),
            QueryError(ref error) => f.write_str(&error
                .source()
                .map(ToString::to_string)
                .unwrap_or_else(|| error.to_string())),
            ConnectionError(ref error) => f.write_str(&error
                .source()
                .map(ToString::to_string)
                .unwrap_or_else(|| error.to_string())),
            MigrationError(ref error) => {
                write!(f, "Failed to run migrations: {}", error)
            }
        }
    }
}

impl PartialEq for DatabaseError {
    fn eq(&self, other: &Self) -> bool {
        matches!(
            (self, other),
            (&ProjectRootNotFound(_), &ProjectRootNotFound(_))
        )
    }
}