// Copyright (C) 2023 Parity Technologies (UK) Ltd. (admin@parity.io)
// This file is a part of the scale-decode crate.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//         http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//! An error that is emitted whenever some decoding fails.
mod context;

pub use context::{Context, Location};

use crate::visitor::DecodeError;
use alloc::{borrow::Cow, boxed::Box, string::String, vec::Vec};
use core::fmt::Display;

/// An error produced while attempting to decode some type.
#[derive(Debug)]
pub struct Error {
    context: Context,
    kind: ErrorKind,
}

impl Error {
    /// Construct a new error given an error kind.
    pub fn new(kind: ErrorKind) -> Error {
        Error { context: Context::new(), kind }
    }
    /// Construct a new, custom error.
    pub fn custom(error: impl Into<CustomError>) -> Error {
        Error::new(ErrorKind::Custom(error.into()))
    }
    /// Retrieve more information about what went wrong.
    pub fn kind(&self) -> &ErrorKind {
        &self.kind
    }
    /// Retrieve details about where the error occurred.
    pub fn context(&self) -> &Context {
        &self.context
    }
    /// Give some context to the error.
    pub fn at(mut self, loc: Location) -> Self {
        self.context.push(loc);
        Error { context: self.context, kind: self.kind }
    }
    /// Note which sequence index the error occurred in.
    pub fn at_idx(mut self, idx: usize) -> Self {
        self.context.push(Location::idx(idx));
        Error { context: self.context, kind: self.kind }
    }
    /// Note which field the error occurred in.
    pub fn at_field(mut self, field: impl Into<Cow<'static, str>>) -> Self {
        self.context.push(Location::field(field));
        Error { context: self.context, kind: self.kind }
    }
    /// Note which variant the error occurred in.
    pub fn at_variant(mut self, variant: impl Into<Cow<'static, str>>) -> Self {
        self.context.push(Location::variant(variant));
        Error { context: self.context, kind: self.kind }
    }
}

impl Display for Error {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        let path = self.context.path();
        let kind = &self.kind;
        write!(f, "Error at {path}: {kind}")
    }
}

impl From<DecodeError> for Error {
    fn from(err: DecodeError) -> Error {
        Error::new(err.into())
    }
}

/// The underlying nature of the error.
#[derive(Debug)]
pub enum ErrorKind {
    /// Something went wrong decoding the bytes based on the type
    /// and type registry provided.
    VisitorDecodeError(DecodeError),
    /// We cannot decode the number seen into the target type; it's out of range.
    NumberOutOfRange {
        /// A string representation of the numeric value that was out of range.
        value: String,
    },
    /// We cannot find the variant we're trying to decode from in the target type.
    CannotFindVariant {
        /// The variant that we are given back from the encoded bytes.
        got: String,
        /// The possible variants that we can decode into.
        expected: Vec<&'static str>,
    },
    /// The types line up, but the expected length of the target type is different from the length of the input value.
    WrongLength {
        /// Length of the type we are trying to decode from
        actual_len: usize,
        /// Length fo the type we're trying to decode into
        expected_len: usize,
    },
    /// Cannot find a field that we need to decode to our target type
    CannotFindField {
        /// Name of the field which was not provided.
        name: String,
    },
    /// A custom error.
    Custom(CustomError),
}

impl From<DecodeError> for ErrorKind {
    fn from(err: DecodeError) -> ErrorKind {
        ErrorKind::VisitorDecodeError(err)
    }
}

impl From<CustomError> for ErrorKind {
    fn from(err: CustomError) -> ErrorKind {
        ErrorKind::Custom(err)
    }
}

impl Display for ErrorKind {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        match self {
            ErrorKind::VisitorDecodeError(decode_error) => {
                write!(f, "Error decoding bytes given the type ID and registry provided: {decode_error:?}")
            }
            ErrorKind::NumberOutOfRange { value } => {
                write!(f, "Number {value} is out of range")
            }
            ErrorKind::CannotFindVariant { got, expected } => {
                write!(f, "Cannot find variant {got}; expects one of {expected:?}")
            }
            ErrorKind::WrongLength { actual_len, expected_len } => {
                write!(f, "Cannot decode from type; expected length {expected_len} but got length {actual_len}")
            }
            ErrorKind::CannotFindField { name } => {
                write!(f, "Field {name} does not exist in our encoded data")
            }
            ErrorKind::Custom(custom_error) => {
                write!(f, "Custom error: {custom_error:?}")
            }
        }
    }
}

#[cfg(feature = "std")]
type CustomError = Box<dyn std::error::Error + Send + Sync + 'static>;

#[cfg(not(feature = "std"))]
type CustomError = Box<dyn core::fmt::Debug + Send + Sync + 'static>;

#[cfg(test)]
mod test {
    use super::*;

    #[derive(Debug)]
    enum MyError {
        Foo,
    }

    impl Display for MyError {
        fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
            write!(f, "{self:?}")
        }
    }

    #[cfg(feature = "std")]
    impl std::error::Error for MyError {}

    #[cfg(not(feature = "std"))]
    impl Into<CustomError> for MyError {
        fn into(self) -> CustomError {
            Box::new(self)
        }
    }

    #[test]
    fn custom_error() {
        // Just a compile-time check that we can ergonomically provide an arbitrary custom error:
        Error::custom(MyError::Foo);
    }
}