token-2022: Add PermanentDelegate extension (#3725)

* token-2022: Add PermanentDelegate extension

* Address feedback

* Refactor getting permanent delegate

* Rename function

* More cleanup

* Fix ATA

Author: joncinque

associated-token-account/program-test/tests/extended_mint.rs

@@ -17,7 +17,9 @@ use {
     },
     spl_token_2022::{
         error::TokenError,
-        extension::{transfer_fee, ExtensionType, StateWithExtensionsOwned},
+        extension::{
+            transfer_fee, BaseStateWithExtensions, ExtensionType, StateWithExtensionsOwned,
+        },
         state::{Account, Mint},
     },
 };

token-swap/program/src/processor.rs

@@ -33,7 +33,7 @@ use spl_token_2022::{
     error::TokenError,
     extension::{
         mint_close_authority::MintCloseAuthority, transfer_fee::TransferFeeConfig,
-        StateWithExtensions,
+        BaseStateWithExtensions, StateWithExtensions,
     },
     state::{Account, Mint},
 };

token/cli/src/main.rs

@@ -43,7 +43,7 @@ use spl_token_2022::{
         memo_transfer::MemoTransfer,
         mint_close_authority::MintCloseAuthority,
         transfer_fee::{TransferFeeAmount, TransferFeeConfig},
-        ExtensionType, StateWithExtensionsOwned,
+        BaseStateWithExtensions, ExtensionType, StateWithExtensionsOwned,
     },
     instruction::*,
     state::{Account, AccountState, Mint},
@@ -744,6 +744,7 @@ async fn command_authorize(
         AuthorityType::TransferFeeConfig => "transfer fee authority",
         AuthorityType::WithheldWithdraw => "withdraw withheld authority",
         AuthorityType::InterestRate => "interest rate authority",
+        AuthorityType::PermanentDelegate => "permanent delegate",
     };
 
     let (mint_pubkey, previous_authority) = if !config.sign_only {
@@ -781,6 +782,7 @@ async fn command_authorize(
                         Err(format!("Mint `{}` is not interest-bearing", account))
                     }
                 }
+                AuthorityType::PermanentDelegate => unimplemented!(),
             }?;
 
             Ok((account, previous_authority))
@@ -813,7 +815,8 @@ async fn command_authorize(
                 | AuthorityType::CloseMint
                 | AuthorityType::TransferFeeConfig
                 | AuthorityType::WithheldWithdraw
-                | AuthorityType::InterestRate => Err(format!(
+                | AuthorityType::InterestRate
+                | AuthorityType::PermanentDelegate => Err(format!(
                     "Authority type `{}` not supported for SPL Token accounts",
                     auth_str
                 )),
@@ -3662,6 +3665,7 @@ async fn process_command<'a>(
                 "transfer-fee-config" => AuthorityType::TransferFeeConfig,
                 "withheld-withdraw" => AuthorityType::WithheldWithdraw,
                 "interest-rate" => AuthorityType::InterestRate,
+                "permanent-delegate" => AuthorityType::PermanentDelegate,
                 _ => unreachable!(),
             };
 

token/client/src/token.rs

@@ -21,7 +21,8 @@ use {
     spl_token_2022::{
         extension::{
             confidential_transfer, cpi_guard, default_account_state, interest_bearing_mint,
-            memo_transfer, transfer_fee, ExtensionType, StateWithExtensionsOwned,
+            memo_transfer, transfer_fee, BaseStateWithExtensions, ExtensionType,
+            StateWithExtensionsOwned,
         },
         instruction,
         solana_zk_token_sdk::{
@@ -122,6 +123,9 @@ pub enum ExtensionInitializationParams {
         rate: i16,
     },
     NonTransferable,
+    PermanentDelegate {
+        delegate: Pubkey,
+    },
 }
 impl ExtensionInitializationParams {
     /// Get the extension type associated with the init params
@@ -133,6 +137,7 @@ impl ExtensionInitializationParams {
             Self::TransferFeeConfig { .. } => ExtensionType::TransferFeeConfig,
             Self::InterestBearingConfig { .. } => ExtensionType::InterestBearingConfig,
             Self::NonTransferable => ExtensionType::NonTransferable,
+            Self::PermanentDelegate { .. } => ExtensionType::PermanentDelegate,
         }
     }
     /// Generate an appropriate initialization instruction for the given mint
@@ -188,6 +193,9 @@ impl ExtensionInitializationParams {
             Self::NonTransferable => {
                 instruction::initialize_non_transferable_mint(token_program_id, mint)
             }
+            Self::PermanentDelegate { delegate } => {
+                instruction::initialize_permanent_delegate(token_program_id, mint, &delegate)
+            }
         }
     }
 }

token/program-2022-test/tests/confidential_transfer.rs

@@ -15,7 +15,7 @@ use {
             confidential_transfer::{
                 ConfidentialTransferAccount, ConfidentialTransferMint, EncryptedWithheldAmount,
             },
-            ExtensionType,
+            BaseStateWithExtensions, ExtensionType,
         },
         solana_zk_token_sdk::{
             encryption::{auth_encryption::*, elgamal::*},

token/program-2022-test/tests/cpi_guard.rs

@@ -17,7 +17,7 @@ use {
         error::TokenError,
         extension::{
             cpi_guard::{self, CpiGuard},
-            ExtensionType,
+            BaseStateWithExtensions, ExtensionType,
         },
         instruction::{self, AuthorityType},
         processor::Processor as SplToken2022Processor,

token/program-2022-test/tests/default_account_state.rs

@@ -9,8 +9,10 @@ use {
         signer::keypair::Keypair, transaction::TransactionError, transport::TransportError,
     },
     spl_token_2022::{
-        error::TokenError, extension::default_account_state::DefaultAccountState,
-        instruction::AuthorityType, state::AccountState,
+        error::TokenError,
+        extension::{default_account_state::DefaultAccountState, BaseStateWithExtensions},
+        instruction::AuthorityType,
+        state::AccountState,
     },
     spl_token_client::token::{ExtensionInitializationParams, TokenError as TokenClientError},
     std::convert::TryFrom,

token/program-2022-test/tests/initialize_account.rs

@@ -17,7 +17,7 @@ use {
         error::TokenError,
         extension::{
             transfer_fee::{self, TransferFeeAmount},
-            ExtensionType, StateWithExtensions,
+            BaseStateWithExtensions, ExtensionType, StateWithExtensions,
         },
         instruction,
         state::{Account, Mint},

token/program-2022-test/tests/initialize_mint.rs

@@ -16,7 +16,10 @@ use {
     },
     spl_token_2022::{
         error::TokenError,
-        extension::{mint_close_authority::MintCloseAuthority, transfer_fee, ExtensionType},
+        extension::{
+            mint_close_authority::MintCloseAuthority, transfer_fee, BaseStateWithExtensions,
+            ExtensionType,
+        },
         instruction, native_mint,
         state::Mint,
     },

token/program-2022-test/tests/interest_bearing_mint.rs

@@ -23,7 +23,7 @@ use {
     },
     spl_token_2022::{
         error::TokenError,
-        extension::interest_bearing_mint::InterestBearingConfig,
+        extension::{interest_bearing_mint::InterestBearingConfig, BaseStateWithExtensions},
         instruction::{amount_to_ui_amount, ui_amount_to_amount, AuthorityType},
         processor::Processor,
     },