safe_zk_token_sdk/
zk_token_proof_instruction.rs

1///! Instructions provided by the ZkToken Proof program
2pub use crate::instruction::*;
3use {
4    bytemuck::{bytes_of, Pod},
5    num_derive::{FromPrimitive, ToPrimitive},
6    num_traits::{FromPrimitive, ToPrimitive},
7    solana_program::instruction::Instruction,
8};
9
10#[derive(Clone, Copy, Debug, FromPrimitive, ToPrimitive, PartialEq, Eq)]
11#[repr(u8)]
12pub enum ProofInstruction {
13    /// Verify a `CloseAccountData` struct
14    ///
15    /// Accounts expected by this instruction:
16    ///   None
17    ///
18    /// Data expected by this instruction:
19    ///   `CloseAccountData`
20    ///
21    VerifyCloseAccount,
22
23    /// Verify a `WithdrawData` struct
24    ///
25    /// Accounts expected by this instruction:
26    ///   None
27    ///
28    /// Data expected by this instruction:
29    ///   `WithdrawData`
30    ///
31    VerifyWithdraw,
32
33    /// Verify a `WithdrawWithheldTokensData` struct
34    ///
35    /// Accounts expected by this instruction:
36    ///   None
37    ///
38    /// Data expected by this instruction:
39    ///   `WithdrawWithheldTokensData`
40    ///
41    VerifyWithdrawWithheldTokens,
42
43    /// Verify a `TransferData` struct
44    ///
45    /// Accounts expected by this instruction:
46    ///   None
47    ///
48    /// Data expected by this instruction:
49    ///   `TransferData`
50    ///
51    VerifyTransfer,
52
53    /// Verify a `TransferWithFeeData` struct
54    ///
55    /// Accounts expected by this instruction:
56    ///   None
57    ///
58    /// Data expected by this instruction:
59    ///   `TransferWithFeeData`
60    ///
61    VerifyTransferWithFee,
62
63    /// Verify a `PubkeyValidityData` struct
64    ///
65    /// Accounts expected by this instruction:
66    ///   None
67    ///
68    /// Data expected by this instruction:
69    ///   `PubkeyValidityData`
70    ///
71    VerifyPubkeyValidity,
72}
73
74impl ProofInstruction {
75    pub fn encode<T: Pod>(&self, proof: &T) -> Instruction {
76        let mut data = vec![ToPrimitive::to_u8(self).unwrap()];
77        data.extend_from_slice(bytes_of(proof));
78        Instruction {
79            program_id: crate::zk_token_proof_program::id(),
80            accounts: vec![],
81            data,
82        }
83    }
84
85    pub fn decode_type(input: &[u8]) -> Option<Self> {
86        input.first().and_then(|x| FromPrimitive::from_u8(*x))
87    }
88
89    pub fn decode_data<T: Pod>(input: &[u8]) -> Option<&T> {
90        if input.is_empty() {
91            None
92        } else {
93            bytemuck::try_from_bytes(&input[1..]).ok()
94        }
95    }
96}
97
98pub fn verify_close_account(proof_data: &CloseAccountData) -> Instruction {
99    ProofInstruction::VerifyCloseAccount.encode(proof_data)
100}
101
102pub fn verify_withdraw(proof_data: &WithdrawData) -> Instruction {
103    ProofInstruction::VerifyWithdraw.encode(proof_data)
104}
105
106pub fn verify_withdraw_withheld_tokens(proof_data: &WithdrawWithheldTokensData) -> Instruction {
107    ProofInstruction::VerifyWithdrawWithheldTokens.encode(proof_data)
108}
109
110pub fn verify_transfer(proof_data: &TransferData) -> Instruction {
111    ProofInstruction::VerifyTransfer.encode(proof_data)
112}
113
114pub fn verify_transfer_with_fee(proof_data: &TransferWithFeeData) -> Instruction {
115    ProofInstruction::VerifyTransferWithFee.encode(proof_data)
116}
117
118pub fn verify_pubkey_validity(proof_data: &PubkeyValidityData) -> Instruction {
119    ProofInstruction::VerifyPubkeyValidity.encode(proof_data)
120}
safe_zk_token_sdk/zk_token_proof_instruction.rs

safe_zk_token_sdk/
zk_token_proof_instruction.rs
