Preventing close when AIA's remain.

metaplex-foundation · Jan 15, 2024 · ad99e9e · ad99e9e
1 parent ca6f856
commit ad99e9e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/programs/mpl-inscription/src/error.rs b/programs/mpl-inscription/src/error.rs
@@ -75,6 +75,10 @@ pub enum MplInscriptionError {
     /// 16 - Authority Already Exists
     #[error("The authority already exists.")]
     AuthorityAlreadyExists,
+
+    /// 17 - Remaining Associated Inscription Accounts
+    #[error("Cannot close Inscription accounts until all Associated Inscriptions are closed.")]
+    RemainingAssociatedInscriptionAccounts,
 }
 
 impl PrintProgramError for MplInscriptionError {

diff --git a/programs/mpl-inscription/src/processor/close.rs b/programs/mpl-inscription/src/processor/close.rs
@@ -120,6 +120,10 @@ pub(crate) fn process_close<'a>(accounts: &'a [AccountInfo<'a>], args: CloseArgs
             close_account_raw(ctx.accounts.payer, ctx.accounts.inscription_account)?;
         }
         None => {
+            if !inscription_metadata.associated_inscriptions.is_empty() {
+                return Err(MplInscriptionError::RemainingAssociatedInscriptionAccounts.into());
+            }
+
             let bump = assert_derivation(
                 &crate::ID,
                 ctx.accounts.inscription_metadata_account,