Fix PDA prefix in Rust renderer (#25)

Author: lorisleiva

.changeset/breezy-lobsters-begin.md

@@ -0,0 +1,5 @@
+---
+"@kinobi-so/renderers-rust": patch
+---
+
+Fix generated PDA prefix

packages/renderers-rust/src/getRenderMapVisitor.ts

@@ -74,7 +74,7 @@ export function getRenderMapVisitor(options: GetRenderMapOptions = {}) {
                         return { ...seed, resolvedType, typeManifest: seedManifest, valueManifest };
                     });
                     const hasVariableSeeds = pdaSeeds.filter(isNodeFilter('variablePdaSeedNode')).length > 0;
-                    const constantSeeds = pdaSeeds
+                    const constantSeeds = seeds
                         .filter(isNodeFilter('constantPdaSeedNode'))
                         .filter(seed => !isNode(seed.value, 'programIdValueNode'));
 

packages/renderers-rust/test/_setup.ts

@@ -1,6 +1,6 @@
 import { expect } from 'vitest';
 
-export function codeContains(actual: string, expected: RegExp | RegExp[] | string[] | string) {
+export function codeContains(actual: string, expected: (RegExp | string)[] | RegExp | string) {
     const expectedArray = Array.isArray(expected) ? expected : [expected];
     expectedArray.forEach(e => {
         if (typeof e === 'string') {

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

@@ -1,10 +1,15 @@
 import {
     accountNode,
     bytesTypeNode,
+    constantPdaSeedNode,
+    constantPdaSeedNodeFromString,
     fixedSizeTypeNode,
+    numberTypeNode,
+    numberValueNode,
     pdaLinkNode,
     pdaNode,
     programNode,
+    publicKeyTypeNode,
     variablePdaSeedNode,
 } from '@kinobi-so/nodes';
 import { visit } from '@kinobi-so/visitors-core';
@@ -41,7 +46,7 @@ test('it renders a byte array seed used on an account', () => {
     codeContains(renderMap.get('accounts/test_account.rs'), [`byte_array_seed: [u8; 32],`, `&byte_array_seed,`]);
 });
 
-test('it renders an empty array seed used on an account', () => {
+test('it renders an empty array of seeds for seedless PDAs', () => {
     // Given the following program with 1 account and 1 pda with empty seeds.
     const node = programNode({
         accounts: [
@@ -66,3 +71,33 @@ test('it renders an empty array seed used on an account', () => {
     // as a parameters to be rendered.
     codeContains(renderMap.get('accounts/test_account.rs'), [/pub fn find_pda\(/, /&\[\s*\]/]);
 });
+
+test('it renders constant PDA seeds as prefix consts', () => {
+    // Given the following PDA node attached to an account.
+    const node = programNode({
+        accounts: [accountNode({ discriminators: [], name: 'testAccount', pda: pdaLinkNode('testPda') })],
+        name: 'myProgram',
+        pdas: [
+            pdaNode({
+                name: 'testPda',
+                seeds: [
+                    constantPdaSeedNodeFromString('utf8', 'myPrefix'),
+                    variablePdaSeedNode('myAccount', publicKeyTypeNode()),
+                    constantPdaSeedNode(numberTypeNode('u64'), numberValueNode(42)),
+                ],
+            }),
+        ],
+        publicKey: '1111',
+    });
+
+    // When we render it.
+    const renderMap = visit(node, getRenderMapVisitor());
+
+    // Then we expect the following const helpers for constant seeds.
+    codeContains(renderMap.get('accounts/test_account.rs'), [
+        '///   0. `TestAccount::PREFIX.0`',
+        '///   1. my_account (`Pubkey`)',
+        '///   2. `TestAccount::PREFIX.1`',
+        /pub const PREFIX: \(\s*&'static \[u8\],\s*&'static \[u8\],\s*\) = \(\s*"myPrefix"\.as_bytes\(\),\s*42\.as_bytes\(\),\s*\)/,
+    ]);
+});