light_account_checks/
error.rs

1use thiserror::Error;
2
3#[derive(Debug, Clone, Copy, Error, PartialEq)]
4pub enum AccountError {
5    #[error("Account owned by wrong program.")]
6    AccountOwnedByWrongProgram,
7    #[error("Account not mutable.")]
8    AccountNotMutable,
9    #[error("Invalid Discriminator.")]
10    InvalidDiscriminator,
11    #[error("Borrow account data failed.")]
12    BorrowAccountDataFailed,
13    #[error("Account is already initialized.")]
14    AlreadyInitialized,
15    #[error("Invalid Account size.")]
16    InvalidAccountSize,
17    #[error("Account is mutable.")]
18    AccountMutable,
19    #[error("Invalid account balance.")]
20    InvalidAccountBalance,
21    #[error("Failed to borrow rent sysvar.")]
22    FailedBorrowRentSysvar,
23    #[error("Invalid Signer")]
24    InvalidSigner,
25    #[error("Invalid Seeds")]
26    InvalidSeeds,
27    #[error("Invalid Program Id")]
28    InvalidProgramId,
29    #[error("Program not executable.")]
30    ProgramNotExecutable,
31    #[error("Account not zeroed.")]
32    AccountNotZeroed,
33    #[error("Not enough account keys provided.")]
34    NotEnoughAccountKeys,
35    #[error("Invalid Account.")]
36    InvalidAccount,
37    #[error("Pinocchio program error with code: {0}")]
38    PinocchioProgramError(u32),
39}
40
41impl From<AccountError> for u32 {
42    fn from(e: AccountError) -> u32 {
43        match e {
44            AccountError::InvalidDiscriminator => 20000,
45            AccountError::AccountOwnedByWrongProgram => 20001,
46            AccountError::AccountNotMutable => 20002,
47            AccountError::BorrowAccountDataFailed => 20003,
48            AccountError::InvalidAccountSize => 20004,
49            AccountError::AccountMutable => 20005,
50            AccountError::AlreadyInitialized => 20006,
51            AccountError::InvalidAccountBalance => 20007,
52            AccountError::FailedBorrowRentSysvar => 20008,
53            AccountError::InvalidSigner => 20009,
54            AccountError::InvalidSeeds => 20010,
55            AccountError::InvalidProgramId => 20011,
56            AccountError::ProgramNotExecutable => 20012,
57            AccountError::AccountNotZeroed => 20013,
58            AccountError::NotEnoughAccountKeys => 20014,
59            AccountError::InvalidAccount => 20015,
60            AccountError::PinocchioProgramError(code) => code,
61        }
62    }
63}
64
65#[cfg(feature = "pinocchio")]
66impl From<AccountError> for pinocchio::program_error::ProgramError {
67    fn from(e: AccountError) -> Self {
68        pinocchio::program_error::ProgramError::Custom(e.into())
69    }
70}
71
72#[cfg(feature = "solana")]
73impl From<AccountError> for solana_program_error::ProgramError {
74    fn from(e: AccountError) -> Self {
75        solana_program_error::ProgramError::Custom(e.into())
76    }
77}
78
79#[cfg(feature = "pinocchio")]
80impl From<pinocchio::program_error::ProgramError> for AccountError {
81    fn from(error: pinocchio::program_error::ProgramError) -> Self {
82        match error {
83            pinocchio::program_error::ProgramError::Custom(code) => {
84                AccountError::PinocchioProgramError(code)
85            }
86            _ => {
87                // Convert other ProgramError variants to error codes
88                let error_code = match error {
89                    pinocchio::program_error::ProgramError::InvalidArgument => 1,
90                    pinocchio::program_error::ProgramError::InvalidInstructionData => 2,
91                    pinocchio::program_error::ProgramError::InvalidAccountData => 3,
92                    pinocchio::program_error::ProgramError::AccountDataTooSmall => 4,
93                    pinocchio::program_error::ProgramError::InsufficientFunds => 5,
94                    pinocchio::program_error::ProgramError::IncorrectProgramId => 6,
95                    pinocchio::program_error::ProgramError::MissingRequiredSignature => 7,
96                    pinocchio::program_error::ProgramError::AccountAlreadyInitialized => 8,
97                    pinocchio::program_error::ProgramError::UninitializedAccount => 9,
98                    pinocchio::program_error::ProgramError::NotEnoughAccountKeys => 10,
99                    pinocchio::program_error::ProgramError::AccountBorrowFailed => 11,
100                    _ => 0, // Unknown error
101                };
102                AccountError::PinocchioProgramError(error_code)
103            }
104        }
105    }
106}
107
108#[cfg(feature = "solana")]
109impl From<core::cell::BorrowError> for AccountError {
110    fn from(_: core::cell::BorrowError) -> Self {
111        AccountError::BorrowAccountDataFailed
112    }
113}
114
115#[cfg(feature = "solana")]
116impl From<core::cell::BorrowMutError> for AccountError {
117    fn from(_: core::cell::BorrowMutError) -> Self {
118        AccountError::BorrowAccountDataFailed
119    }
120}
light_account_checks/error.rs

light_account_checks/
error.rs
