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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

//! Allows getting the reference of the key of an account or [Pubkey].

use anchor_lang::prelude::*;
use anchor_lang::ZeroCopy;

/// Defines the Pubkey of an account, fetching it as a reference.
pub trait AsKeyRef {
    /// Returns the [Pubkey] of the account or key as a reference.
    fn as_key_ref(&self) -> &Pubkey;
}

impl AsKeyRef for Pubkey {
    fn as_key_ref(&self) -> &Pubkey {
        self
    }
}

impl<'info, T: AccountSerialize + AccountDeserialize + Owner + Clone> AsKeyRef
    for Box<Account<'info, T>>
{
    fn as_key_ref(&self) -> &Pubkey {
        self.as_ref().as_key_ref()
    }
}

impl<'info, T: AccountSerialize + AccountDeserialize + Owner + Clone> AsKeyRef
    for Account<'info, T>
{
    fn as_key_ref(&self) -> &Pubkey {
        let my_ref: &AccountInfo<'info> = self.as_ref();
        my_ref.key
    }
}

impl<'info> AsKeyRef for AccountInfo<'info> {
    fn as_key_ref(&self) -> &Pubkey {
        self.key
    }
}

impl<'info, T: ZeroCopy + Owner> AsKeyRef for AccountLoader<'info, T> {
    fn as_key_ref(&self) -> &Pubkey {
        self.as_ref().key
    }
}

impl<'info> AsKeyRef for Signer<'info> {
    fn as_key_ref(&self) -> &Pubkey {
        self.as_ref().key
    }
}

impl<'info> AsKeyRef for SystemAccount<'info> {
    fn as_key_ref(&self) -> &Pubkey {
        self.as_ref().key
    }
}

impl<'info, T: anchor_lang::solana_program::sysvar::Sysvar> AsKeyRef for Sysvar<'info, T> {
    fn as_key_ref(&self) -> &Pubkey {
        self.as_ref().key
    }
}

impl<'info> AsKeyRef for UncheckedAccount<'info> {
    fn as_key_ref(&self) -> &Pubkey {
        self.as_ref().key
    }
}

#[allow(deprecated)]
mod deprecated {
    use super::*;
    use anchor_lang::accounts::{
        cpi_account::CpiAccount, cpi_state::CpiState, loader::Loader,
        program_account::ProgramAccount, state::ProgramState,
    };

    #[allow(deprecated)]
    impl<'info, T: AccountDeserialize + Clone> AsKeyRef for CpiAccount<'info, T> {
        fn as_key_ref(&self) -> &Pubkey {
            self.as_ref().key
        }
    }

    #[allow(deprecated)]
    impl<'info, T: AccountSerialize + AccountDeserialize + Clone> AsKeyRef for CpiState<'info, T> {
        fn as_key_ref(&self) -> &Pubkey {
            self.as_ref().key
        }
    }

    #[allow(deprecated)]
    impl<'info, T: ZeroCopy> AsKeyRef for Loader<'info, T> {
        fn as_key_ref(&self) -> &Pubkey {
            self.as_ref().key
        }
    }

    #[allow(deprecated)]
    impl<'info, T: AccountSerialize + AccountDeserialize + Clone> AsKeyRef
        for ProgramAccount<'info, T>
    {
        fn as_key_ref(&self) -> &Pubkey {
            self.as_ref().key
        }
    }

    #[allow(deprecated)]
    impl<'info, T: AccountSerialize + AccountDeserialize + Clone> AsKeyRef for ProgramState<'info, T> {
        fn as_key_ref(&self) -> &Pubkey {
            self.as_ref().key
        }
    }
}