// Copyright 2022 Google LLC
//
// 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.

use std::fmt::{Debug, Formatter};

use thiserror::Error;
use tracing_error::SpanTrace;

pub use backend::YubikeyPivBackend;
use native_pkcs11_traits::SignatureAlgorithm;

mod backend;
pub mod certificate;
pub mod key;
mod pinentry;
mod util;
mod slot;

pub type Result<T> = std::result::Result<T, Error>;

pub struct Error {
    error: ErrorKind,
    context: SpanTrace,
}

impl Debug for Error {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        writeln!(f, "{:?}", self.error)?;
        self.context.fmt(f)
    }
}

impl std::error::Error for Error {}

impl<E: Into<ErrorKind>> From<E> for Error {
    fn from(e: E) -> Self {
        Error {
            error: e.into(),
            context: SpanTrace::capture(),
        }
    }
}

impl std::fmt::Display for Error {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        writeln!(f, "{}", self.error)?;
        std::fmt::Display::fmt(&self.context, f)
    }
}

#[derive(Error, Debug)]
pub enum ErrorKind {
    #[error("GenericError {0}")]
    Generic(String),

    #[error("{0}")]
    Der(#[from] x509_cert::der::Error),

    #[error("{0}")]
    Spki(#[from] x509_cert::spki::Error),

    #[error("{0}")]
    P256(#[from] p256::elliptic_curve::Error),

    #[error("{0}")]
    Pkcs1(#[from] rsa::pkcs1::Error),

    #[error("{0:?}")]
    UnsupportedSignatureAlgorithm(SignatureAlgorithm),
}

impl From<&str> for ErrorKind {
    fn from(s: &str) -> Self {
        ErrorKind::Generic(s.to_string())
    }
}

impl From<String> for ErrorKind {
    fn from(s: String) -> Self {
        ErrorKind::Generic(s)
    }
}