Add fetch function helpers to accounts page

Author: wjthieme

packages/renderers-rust/public/templates/accountsMod.njk

@@ -2,6 +2,21 @@
 
 {% block main %}
 
+#[cfg(feature = "fetch")]
+#[derive(Debug, Clone)]
+pub struct DecodedAccount<T> {
+  pub address: solana_program::pubkey::Pubkey,
+  pub account: solana_sdk::account::Account,
+  pub data: T,
+}
+
+#[cfg(feature = "fetch")]
+#[derive(Debug, Clone)]
+pub enum MaybeAccount<T> {
+  Exists(DecodedAccount<T>),
+  NotFound(solana_program::pubkey::Pubkey),
+}
+
 {% for account in accountsToExport | sort(false, false, 'name') %}
   pub(crate) mod r#{{ account.name | snakeCase }};
 {% endfor %}

packages/renderers-rust/public/templates/accountsPage.njk

@@ -128,6 +128,59 @@ impl<'a> TryFrom<&solana_program::account_info::AccountInfo<'a>> for {{ account.
   }
 }
 
+#[cfg(feature = "fetch")]
+pub fn fetch_{{ account.name | snakeCase }}(
+  rpc: &solana_client::rpc_client::RpcClient,
+  address: &Pubkey,
+) -> Result<super::DecodedAccount<{{ account.name | pascalCase }}>, Error> {
+  let accounts = fetch_all_{{ account.name | snakeCase }}(rpc, vec![address])?;
+  Ok(accounts[0].clone())
+}
+
+#[cfg(feature = "fetch")]
+pub fn fetch_all_{{ account.name | snakeCase }}(
+  rpc: &solana_client::rpc_client::RpcClient,
+  addresses: Vec<Pubkey>,
+) -> Result<Vec<super::DecodedAccount<{{ account.name | pascalCase }}>>, Error> {
+    let accounts = rpc.get_multiple_accounts(&addresses)?;
+    let mut decoded_accounts: Vec<super::DecodedAccount<{{ account.name | pascalCase }}>> = Vec::new();
+    for i in 0..addresses.len() {
+      let address = addresses[i];
+      let account = accounts[i].as_ref().ok_or(format!("Account not found: {}", address))?;
+      let data = {{ account.name | pascalCase }}::from_bytes(&account.data)?;
+      decoded_accounts.push(super::DecodedAccount { address, account: account.clone(), data });
+    }
+    Ok(decoded_accounts)
+}
+
+#[cfg(feature = "fetch")]
+pub fn fetch_maybe_{{ account.name | snakeCase }}(
+  rpc: &solana_client::rpc_client::RpcClient,
+  address: &Pubkey,
+) -> Result<super::MaybeAccount<{{ account.name | pascalCase }}>, Error> {
+    let accounts = fetch_all_maybe_{{ account.name | snakeCase }}(rpc, vec![address])?;
+    Ok(accounts[0].clone())
+}
+
+#[cfg(feature = "fetch")]
+pub fn fetch_all_maybe_{{ account.name | snakeCase }}(
+  rpc: &solana_client::rpc_client::RpcClient,
+  addresses: Vec<Pubkey>,
+) -> Result<Vec<super::MaybeAccount<{{ account.name | pascalCase }}>>, Error> {
+    let accounts = rpc.get_multiple_accounts(&addresses)?;
+    let mut decoded_accounts: Vec<super::MaybeAccount<{{ account.name | pascalCase }}>> = Vec::new();
+    for i in 0..addresses.len() {
+      let address = addresses[i];
+      if let Some(account) = accounts[i].as_ref() {
+        let data = {{ account.name | pascalCase }}::from_bytes(&account.data)?;
+        decoded_accounts.push(super::MaybeAccount::Exists(super::DecodedAccount { address, account: account.clone(), data }));
+      } else {
+        decoded_accounts.push(super::MaybeAccount::NotFound(address));
+      }
+    }
+  Ok(decoded_accounts)
+}
+
 #[cfg(feature = "anchor")]
 impl anchor_lang::AccountDeserialize for {{ account.name | pascalCase }} {
     fn try_deserialize_unchecked(buf: &mut &[u8]) -> anchor_lang::Result<Self> {

packages/renderers-rust/test/accountsPage.test.ts

@@ -134,3 +134,35 @@ test('it renders anchor traits impl', () => {
         'impl anchor_lang::Owner for TestAccount',
     ]);
 });
+
+test('it renders fetch functions', () => {
+    // Given the following account.
+    const node = programNode({
+        accounts: [
+            accountNode({
+                discriminators: [
+                    {
+                        kind: 'fieldDiscriminatorNode',
+                        name: camelCase('discriminator'),
+                        offset: 0,
+                    },
+                ],
+                name: 'testAccount',
+                pda: pdaLinkNode('testPda'),
+            }),
+        ],
+        name: 'myProgram',
+        publicKey: '1111',
+    });
+
+    // When we render it.
+    const renderMap = visit(node, getRenderMapVisitor());
+
+    // Then we expect the following fetch functions to be rendered.
+    codeContains(renderMap.get('accounts/test_account.rs'), [
+        'pub fn fetch_test_account',
+        'pub fn fetch_maybe_test_account',
+        'pub fn fetch_all_test_account',
+        'pub fn fetch_all_maybe_test_account',
+    ]);
+});