Added cpi guard instruction extension

solana-program · Nov 13, 2024 · afb5d2c · afb5d2c
1 parent 0533751
commit afb5d2c
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/clients/js/src/getInitializeInstructionsForExtensions.ts b/clients/js/src/getInitializeInstructionsForExtensions.ts
@@ -10,14 +10,16 @@ import {
   ExtensionArgs,
   getDisableMemoTransfersInstruction,
   getEnableMemoTransfersInstruction,
+  getEnableCpiGuardInstruction,
+  getDisableCpiGuardInstruction,
   getInitializeConfidentialTransferMintInstruction,
   getInitializeDefaultAccountStateInstruction,
   getInitializeGroupMemberPointerInstruction,
   getInitializeGroupPointerInstruction,
   getInitializeMetadataPointerInstruction,
   getInitializeTokenGroupInstruction,
   getInitializeTokenMetadataInstruction,
-  getInitializeTransferFeeConfigInstruction,
+  getInitializeTransferFeeConfigInstruction
 } from './generated';
 
 /**
@@ -153,10 +155,20 @@ export function getPostInitializeInstructionsForTokenExtensions(
           extension.requireIncomingTransferMemos
             ? getEnableMemoTransfersInstruction({ owner, token, multiSigners })
             : getDisableMemoTransfersInstruction({
-                owner,
-                token,
-                multiSigners,
-              }),
+              owner,
+              token,
+              multiSigners,
+            }),
+        ];
+      case 'CpiGuard':
+        return [
+          extension.lockCpi
+            ? getEnableCpiGuardInstruction({ owner, token, multiSigners })
+            : getDisableCpiGuardInstruction({
+              owner,
+              token,
+              multiSigners,
+            }),
         ];
       default:
         return [];