// SPDX-License-Identifier: Apache-2.0

// SPDX-FileCopyrightText: Copyright the Vortex contributors

use pyo3::CastError;

use pyo3::PyErr;

use pyo3::exceptions::PyRuntimeError;

use vortex::error::VortexError;

/// Error type to merge [`VortexError`] and [`PyErr`].

pub enum PyVortexError {

Py(PyErr),

Vortex(VortexError),

}

/// A [`Result`] alias where the error is [`PyVortexError`].

pub type PyVortexResult<T> = Result<T, PyVortexError>;

impl From<PyErr> for PyVortexError {

fn from(value: PyErr) -> Self {

Self::Py(value)

}

}

impl From<VortexError> for PyVortexError {

fn from(value: VortexError) -> Self {

Self::Vortex(value)

}

}

impl<'py, 'a> From<CastError<'py, 'a>> for PyVortexError {

fn from(value: CastError<'py, 'a>) -> Self {

Self::Py(value.into())

}

}

impl From<PyVortexError> for PyErr {

fn from(value: PyVortexError) -> Self {

match value {

PyVortexError::Py(py) => py,

PyVortexError::Vortex(vx) => PyRuntimeError::new_err(vx.to_string()),

}

}

}