pinoc 0.2.2

A CLI tool for setting up pinocchio program project
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

use super::types::render_type_body;
use anyhow::Result;
use heck::{ToPascalCase, ToSnakeCase};
use shank_idl::idl_type_definition::IdlTypeDefinition;

pub fn account_rs(account: &IdlTypeDefinition) -> Result<String> {
    let name = account.name.to_pascal_case();
    let snake_name = account.name.to_snake_case();
    let (body, imports) = render_type_body(&name, &account.ty)?;

    Ok(format!(
        r#"//! Autogenerated by `pinoc client generate`. Do not edit by hand.

use borsh::{{BorshDeserialize, BorshSerialize}};
{imports}
#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, PartialEq)]
{body}
impl {name} {{
    pub fn from_bytes(data: &[u8]) -> std::io::Result<Self> {{
        let mut data = data;
        <Self as BorshDeserialize>::deserialize(&mut data)
    }}

    pub fn to_bytes(&self) -> std::io::Result<Vec<u8>> {{
        borsh::to_vec(self)
    }}
}}

#[cfg(feature = "fetch")]
pub fn fetch_{snake_name}(
    rpc: &solana_rpc_client::rpc_client::RpcClient,
    address: &solana_address::Address,
) -> Result<crate::shared::DecodedAccount<{name}>, std::io::Error> {{
    let accounts = fetch_all_{snake_name}(rpc, &[*address])?;
    Ok(accounts[0].clone())
}}

#[cfg(feature = "fetch")]
pub fn fetch_all_{snake_name}(
    rpc: &solana_rpc_client::rpc_client::RpcClient,
    addresses: &[solana_address::Address],
) -> Result<Vec<crate::shared::DecodedAccount<{name}>>, std::io::Error> {{
    let accounts = rpc
        .get_multiple_accounts(addresses)
        .map_err(|e| std::io::Error::other(e.to_string()))?;
    let mut decoded_accounts = Vec::new();
    for (i, address) in addresses.iter().enumerate() {{
        let account = accounts[i]
            .as_ref()
            .ok_or_else(|| std::io::Error::other(format!("Account not found: {{address}}")))?;
        let data = {name}::from_bytes(&account.data)?;
        decoded_accounts.push(crate::shared::DecodedAccount {{
            address: *address,
            account: account.clone(),
            data,
        }});
    }}
    Ok(decoded_accounts)
}}

#[cfg(feature = "fetch")]
pub fn fetch_maybe_{snake_name}(
    rpc: &solana_rpc_client::rpc_client::RpcClient,
    address: &solana_address::Address,
) -> Result<crate::shared::MaybeAccount<{name}>, std::io::Error> {{
    let accounts = fetch_all_maybe_{snake_name}(rpc, &[*address])?;
    Ok(accounts[0].clone())
}}

#[cfg(feature = "fetch")]
pub fn fetch_all_maybe_{snake_name}(
    rpc: &solana_rpc_client::rpc_client::RpcClient,
    addresses: &[solana_address::Address],
) -> Result<Vec<crate::shared::MaybeAccount<{name}>>, std::io::Error> {{
    let accounts = rpc
        .get_multiple_accounts(addresses)
        .map_err(|e| std::io::Error::other(e.to_string()))?;
    let mut decoded_accounts = Vec::new();
    for (i, address) in addresses.iter().enumerate() {{
        decoded_accounts.push(match accounts[i].as_ref() {{
            Some(account) => {{
                let data = {name}::from_bytes(&account.data)?;
                crate::shared::MaybeAccount::Exists(crate::shared::DecodedAccount {{
                    address: *address,
                    account: account.clone(),
                    data,
                }})
            }}
            None => crate::shared::MaybeAccount::NotFound(*address),
        }});
    }}
    Ok(decoded_accounts)
}}
"#
    ))
}