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

use super::*;

pub const ESCROW_POSTFIX: &str = "escrow";

#[repr(C)]
#[cfg_attr(feature = "serde-feature", derive(Serialize, Deserialize))]
#[derive(BorshSerialize, BorshDeserialize, PartialEq, Eq, Debug, Clone, Copy)]
pub enum EscrowAuthority {
    TokenOwner,
    Creator(Pubkey),
}

impl EscrowAuthority {
    pub fn to_seeds(&self) -> Vec<&[u8]> {
        match self {
            EscrowAuthority::TokenOwner => vec![&[0]],
            EscrowAuthority::Creator(creator) => vec![&[1], creator.as_ref()],
        }
    }
}

#[repr(C)]
#[cfg_attr(feature = "serde-feature", derive(Serialize, Deserialize))]
#[derive(BorshSerialize, BorshDeserialize, PartialEq, Eq, Debug, Clone, ShankAccount)]
pub struct TokenOwnedEscrow {
    pub key: Key,
    pub base_token: Pubkey,
    pub authority: EscrowAuthority,
    pub bump: u8,
}

impl TokenMetadataAccount for TokenOwnedEscrow {
    fn key() -> Key {
        Key::TokenOwnedEscrow
    }

    fn size() -> usize {
        0
    }

    fn is_correct_account_type(data: &[u8], data_type: Key, _data_size: usize) -> bool {
        let key: Option<Key> = Key::from_u8(data[0]);
        match key {
            Some(key) => key == data_type || key == Key::Uninitialized,
            None => false,
        }
    }
}