Merge pull request #152 from m0-foundation/kurtis/add-set-mint-ix

Set portal mint instruction

Author: Oighty

programs/portal/src/instructions/admin.rs

@@ -1,5 +1,8 @@
 use anchor_lang::prelude::*;
-use anchor_spl::{token_2022::spl_token_2022::instruction::AuthorityType, token_interface};
+use anchor_spl::{
+    token_2022::{spl_token_2022::instruction::AuthorityType, Token2022},
+    token_interface,
+};
 
 use crate::{
     config::Config,
@@ -9,6 +12,7 @@ use crate::{
     pending_token_authority::PendingTokenAuthority,
     queue::{inbox::InboxRateLimit, outbox::OutboxRateLimit, rate_limit::RateLimitState},
     registered_transceiver::RegisteredTransceiver,
+    TOKEN_AUTHORITY_SEED,
 };
 
 // * Transfer ownership
@@ -625,3 +629,38 @@ pub fn set_threshold(ctx: Context<SetThreshold>, threshold: u8) -> Result<()> {
     ctx.accounts.config.threshold = threshold;
     Ok(())
 }
+
+// * Set Mint
+#[derive(Accounts)]
+pub struct SetMint<'info> {
+    pub owner: Signer<'info>,
+
+    #[account(
+        mut,
+        has_one = owner,
+        constraint = config.paused @ NTTError::NotPaused,
+    )]
+    pub config: Account<'info, Config>,
+
+    pub mint: InterfaceAccount<'info, token_interface::Mint>,
+
+    /// CHECK: This account is validated by its seeds
+    #[account(
+        seeds = [TOKEN_AUTHORITY_SEED],
+        bump,
+    )]
+    pub token_authority: UncheckedAccount<'info>,
+
+    #[account(
+        associated_token::mint = mint,
+        associated_token::authority = token_authority,
+        associated_token::token_program = Token2022::id(),
+    )]
+    pub custody: InterfaceAccount<'info, token_interface::TokenAccount>,
+}
+
+pub fn set_mint(ctx: Context<SetMint>) -> Result<()> {
+    ctx.accounts.config.mint = ctx.accounts.mint.key();
+    ctx.accounts.config.custody = ctx.accounts.custody.key();
+    Ok(())
+}

programs/portal/src/lib.rs

@@ -172,6 +172,10 @@ pub mod portal {
         instructions::claim_token_authority(ctx)
     }
 
+    pub fn set_mint(ctx: Context<SetMint>) -> Result<()> {
+        instructions::set_mint(ctx)
+    }
+
     pub fn set_paused(ctx: Context<SetPaused>, pause: bool) -> Result<()> {
         instructions::set_paused(ctx, pause)
     }

services/cli/main.ts

@@ -45,7 +45,7 @@ import {
 } from '../../sdk/src';
 import { createInitializeConfidentialTransferMintInstruction } from './confidential-transfers';
 import { Program, BN } from '@coral-xyz/anchor';
-import { anchorProvider, keysFromEnv, NttManager } from './utils';
+import { anchorProvider, keysFromEnv, NttManager, updatePortalMint } from './utils';
 import { MerkleTree } from '../../sdk/src/merkle';
 import { bs58 } from '@coral-xyz/anchor/dist/cjs/utils/bytes';
 import { SolanaUnsignedTransaction } from '@wormhole-foundation/sdk-solana/dist/cjs';
@@ -78,7 +78,7 @@ async function main() {
     .command('print-addresses')
     .description('Print the addresses of all the relevant programs and accounts')
     .action(() => {
-      const [mMint, wmMint, multisig] = keysFromEnv(['M_MINT_KEYPAIR', 'WM_MINT_KEYPAIR', 'M_MINT_MULTISIG_KEYPAIR']);
+      const [mMint, wmMint] = keysFromEnv(['M_MINT_KEYPAIR', 'WM_MINT_KEYPAIR']);
       const [portalTokenAuthPda] = PublicKey.findProgramAddressSync([Buffer.from('token_authority')], PROGRAMS.portal);
       const [earnTokenAuthPda] = PublicKey.findProgramAddressSync([Buffer.from('token_authority')], PROGRAMS.earn);
       const [portalEmitter] = PublicKey.findProgramAddressSync([Buffer.from('emitter')], PROGRAMS.portal);
@@ -94,7 +94,6 @@ async function main() {
         'Earn Program': PROGRAMS.earn,
         'Swap Program': PROGRAMS.swap,
         'M Mint': mMint.publicKey,
-        'M Mint Multisig': multisig.publicKey,
         'Portal Token Authority': portalTokenAuthPda,
         'Earn Token Authority': earnTokenAuthPda,
         'wM Mint': wmMint.publicKey,
@@ -246,7 +245,26 @@ async function main() {
     });
 
   program.command('update-portal-mint').action(async () => {
-    const [owner] = keysFromEnv(['PAYER_KEYPAIR']);
+    const [payer, mint] = keysFromEnv(['PAYER_KEYPAIR', 'M_MINT_KEYPAIR']);
+    const { ntt } = NttManager(connection, payer, mint.publicKey);
+
+    let owner = payer.publicKey;
+    if (process.env.SQUADS_VAULT) {
+      owner = new PublicKey(process.env.SQUADS_VAULT);
+    }
+
+    const tx = new Transaction().add(updatePortalMint(owner, ntt.pdas.configAccount(), mint.publicKey));
+
+    if (process.env.SQUADS_VAULT) {
+      const b = tx.serialize({ verifySignatures: false });
+      console.log('Transaction:', {
+        b64: b.toString('base64'),
+        b58: bs58.encode(b),
+      });
+    } else {
+      const sig = await connection.sendTransaction(tx, [payer]);
+      console.log(`Paused: ${sig}`);
+    }
   });
 
   program
@@ -366,7 +384,7 @@ async function main() {
     const pauseTxn = (await ntt.pause(sender).next()).value as SolanaUnsignedTransaction<'Mainnet', 'Solana'>;
     const tx = pauseTxn.transaction.transaction as Transaction;
 
-    if (process.env.SQUADS_MULTISIG) {
+    if (process.env.SQUADS_VAULT) {
       const b = tx.serialize({ verifySignatures: false });
       console.log('Transaction:', {
         b64: b.toString('base64'),
@@ -390,7 +408,7 @@ async function main() {
       const pauseTxn = (await cmd(sender).next()).value as SolanaUnsignedTransaction<'Mainnet', 'Solana'>;
       const tx = pauseTxn.transaction.transaction as Transaction;
 
-      if (process.env.SQUADS_MULTISIG) {
+      if (process.env.SQUADS_VAULT) {
         const b = tx.serialize({ verifySignatures: false });
         console.log('Transaction:', {
           b64: b.toString('base64'),

services/cli/utils.ts

@@ -1,8 +1,9 @@
-import { Keypair, Connection, PublicKey } from '@solana/web3.js';
+import { Keypair, Connection, PublicKey, TransactionInstruction } from '@solana/web3.js';
 import { SolanaNtt } from '@wormhole-foundation/sdk-solana-ntt';
 import { SolanaPlatform, SolanaSendSigner } from '@wormhole-foundation/sdk-solana';
-import { AccountAddress, Wormhole } from '@wormhole-foundation/sdk';
+import { AccountAddress, sha256, Wormhole } from '@wormhole-foundation/sdk';
 import { AnchorProvider, Wallet } from '@coral-xyz/anchor';
+import { getAssociatedTokenAddressSync, TOKEN_2022_PROGRAM_ID } from '@solana/spl-token';
 
 const PORTAL = new PublicKey('mzp1q2j5Hr1QuLC3KFBCAUz5aUckT6qyuZKZ3WJnMmY');
 
@@ -45,3 +46,42 @@ export function anchorProvider(connection: Connection, owner: Keypair) {
     skipPreflight: false,
   });
 }
+
+export function updatePortalMint(owner: PublicKey, config: PublicKey, mMint: PublicKey): TransactionInstruction {
+  return new TransactionInstruction({
+    programId: PORTAL,
+    keys: [
+      {
+        pubkey: owner,
+        isSigner: true,
+        isWritable: false,
+      },
+      {
+        pubkey: config,
+        isSigner: false,
+        isWritable: true,
+      },
+      {
+        pubkey: mMint,
+        isSigner: false,
+        isWritable: false,
+      },
+      {
+        pubkey: PublicKey.findProgramAddressSync([Buffer.from('token_authority')], PORTAL)[0],
+        isSigner: false,
+        isWritable: false,
+      },
+      {
+        pubkey: getAssociatedTokenAddressSync(
+          mMint,
+          PublicKey.findProgramAddressSync([Buffer.from('token_authority')], PORTAL)[0],
+          true,
+          TOKEN_2022_PROGRAM_ID,
+        ),
+        isSigner: false,
+        isWritable: false,
+      },
+    ],
+    data: Buffer.concat([Buffer.from(sha256('global:set_mint').subarray(0, 8))]),
+  });
+}