1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

//
// Wildland Project
//
// Copyright © 2022 Golem Foundation
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License version 3 as published by
// the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

use std::fmt::Display;

use thiserror::Error;
use wildland_crypto::error::CryptoError;

use crate::catlib_service::error::CatlibError;
use crate::sfo::api::SfoError;
use crate::LssError;

pub trait ErrContext<E, T> {
    fn context(self, ctx: impl Display) -> Result<T, E>;
    fn format(err: impl Display, ctx: impl Display) -> String {
        format!("{ctx}: {err}")
    }
}

#[derive(Error, Debug, Clone, PartialEq, Eq)]
#[repr(C)]
pub enum ForestRetrievalError {
    #[error(transparent)]
    LssError(#[from] LssError),
    #[error("Could not create keypair from bytes retrieved from LSS: {0}")]
    KeypairParseError(CryptoError),
}

#[derive(Error, Debug, PartialEq, Eq, Clone)]
#[repr(C)]
pub enum CoreXError {
    #[error("Cannot create forest identity: {0}")]
    CannotCreateForestIdentityError(String),
    #[error("Identity read error: {0}")]
    IdentityReadError(String),
    #[error("LSS Error: {0}: {1}")]
    LSSErr(String, LssError),
    #[error("Catlib Error: {0}: {1}")]
    CatlibErr(String, CatlibError),
    #[error("Crypto Error: {0}: {1}")]
    CryptoErr(String, CryptoError),
    #[error("CoreX error: {0}")]
    Generic(String),
    #[error("SFO Error: {0}: {1}")]
    SfoErr(String, SfoError),
}

impl<T> ErrContext<CoreXError, T> for Result<T, CoreXError> {
    fn context(mut self, ctx: impl Display) -> Result<T, CoreXError> {
        if let Err(
            CoreXError::CannotCreateForestIdentityError(s)
            | CoreXError::IdentityReadError(s)
            | CoreXError::LSSErr(s, _)
            | CoreXError::CatlibErr(s, _)
            | CoreXError::CryptoErr(s, _)
            | CoreXError::Generic(s)
            | CoreXError::SfoErr(s, _),
        ) = &mut self
        {
            *s = format!("{ctx}: {s}")
        }
        self
    }
}

impl<T> ErrContext<CoreXError, T> for Result<T, CatlibError> {
    fn context(self, ctx: impl Display) -> Result<T, CoreXError> {
        self.map_err(|e| CoreXError::CatlibErr(ctx.to_string(), e))
    }
}

impl<T> ErrContext<CoreXError, T> for Result<T, CryptoError> {
    fn context(self, ctx: impl Display) -> Result<T, CoreXError> {
        self.map_err(|e| CoreXError::CryptoErr(ctx.to_string(), e))
    }
}

impl<T> ErrContext<CoreXError, T> for Result<T, LssError> {
    fn context(self, ctx: impl Display) -> Result<T, CoreXError> {
        self.map_err(|e| CoreXError::LSSErr(ctx.to_string(), e))
    }
}