refactor(experimental): graphql: token-2022 extensions account state:…

… non transferable This PR adds Token-2022 extension parsed account state support for `NonTransferable`. Ref #2644.
solana-labs · May 6, 2024 · 2614fa2 · 2614fa2
1 parent 5ab9e90
commit 2614fa2
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 0 deletions.
diff --git a/packages/rpc-graphql/src/__tests__/account-test.ts b/packages/rpc-graphql/src/__tests__/account-test.ts
@@ -1395,6 +1395,34 @@ describe('account', () => {
                     },
                 });
             });
+            it('non-transferable', async () => {
+                expect.assertions(1);
+                const source = /* GraphQL */ `
+                    query testQuery($address: Address!) {
+                        account(address: $address) {
+                            ... on MintAccount {
+                                extensions {
+                                    ... on SplToken2022ExtensionNonTransferable {
+                                        extension # Interface field
+                                    }
+                                }
+                            }
+                        }
+                    }
+                `;
+                const result = await rpcGraphQL.query(source, { address: megaMintAddress });
+                expect(result).toMatchObject({
+                    data: {
+                        account: {
+                            extensions: expect.arrayContaining([
+                                {
+                                    extension: 'nonTransferable',
+                                },
+                            ]),
+                        },
+                    },
+                });
+            });
         });
     });
 });
diff --git a/packages/rpc-graphql/src/resolvers/account.ts b/packages/rpc-graphql/src/resolvers/account.ts
@@ -214,6 +214,9 @@ function resolveToken2022ExtensionType(extensionResult: Token2022ExtensionResult
     if (extensionResult.extension === 'mintCloseAuthority') {
         return 'SplToken2022ExtensionMintCloseAuthority';
     }
+    if (extensionResult.extension === 'nonTransferable') {
+        return 'SplToken2022ExtensionNonTransferable';
+    }
     if (extensionResult.extension === 'permanentDelegate') {
         return 'SplToken2022ExtensionPermanentDelegate';
     }

diff --git a/packages/rpc-graphql/src/schema/account.ts b/packages/rpc-graphql/src/schema/account.ts
@@ -34,6 +34,13 @@ export const accountTypeDefs = /* GraphQL */ `
         rateAuthority: Account
     }
 
+    """
+    Token-2022 Extension: Non-Transferable
+    """
+    type SplToken2022ExtensionNonTransferable implements SplToken2022Extension {
+        extension: String
+    }
+
     """
     Account interface
     """