use mpl_token_metadata::instructions::TransferV1;
use mpl_token_metadata::instructions::TransferV1InstructionArgs;
use crate::errors::ErrorCode;
use crate::state::*;
use anchor_lang::prelude::*;
use anchor_spl::token::Token;
use anchor_spl::token::TokenAccount;
use anchor_spl::token::Transfer;
use anchor_spl::token::{self};
use solana_program::program::invoke;
use solana_program::system_instruction;
#[derive(Accounts)]
pub struct IssueCtx<'info> {
#[account(mut, constraint = token_manager.state == TokenManagerState::Initialized as u8 @ ErrorCode::InvalidTokenManagerState)]
token_manager: Box<Account<'info, TokenManager>>,
#[account(mut, constraint = token_manager_token_account.owner == token_manager.key() @ ErrorCode::InvalidTokenManagerTokenAccount)]
token_manager_token_account: Box<Account<'info, TokenAccount>>,
#[account(constraint = issuer.key() == token_manager.issuer @ ErrorCode::InvalidIssuer)]
issuer: Signer<'info>,
#[account(mut, constraint = issuer_token_account.mint == token_manager.mint && issuer_token_account.owner == issuer.key() @ ErrorCode::InvalidIssuerTokenAccount)]
issuer_token_account: Box<Account<'info, TokenAccount>>,
#[account(mut)]
payer: Signer<'info>,
token_program: Program<'info, Token>,
system_program: Program<'info, System>,
}
pub fn handler<'key, 'accounts, 'remaining, 'info>(ctx: Context<'key, 'accounts, 'remaining, 'info, IssueCtx<'info>>) -> Result<()> {
let token_manager = &mut ctx.accounts.token_manager;
if token_manager.invalidation_type == InvalidationType::Vest as u8 && token_manager.claim_approver.is_none() {
return Err(error!(ErrorCode::ClaimApproverMustBeSet));
}
if token_manager.kind == TokenManagerKind::Permissioned as u8 && token_manager.invalidation_type != InvalidationType::Release as u8 {
return Err(error!(ErrorCode::InvalidInvalidationTypeKindMatch));
}
let remaining_accs = &mut ctx.remaining_accounts.iter();
if token_manager.kind == TokenManagerKind::Permissioned as u8 {
let permisisoned_reward_info = next_account_info(remaining_accs)?;
if permisisoned_reward_info.key().to_string() != PERMISSIONED_REWARD_ADDRESS {
return Err(error!(ErrorCode::InvalidPermissionedRewardAddress));
}
invoke(
&system_instruction::transfer(&ctx.accounts.issuer.key(), &permisisoned_reward_info.key(), PERMISSIONED_REWARD_LAMPORTS),
&[
ctx.accounts.issuer.to_account_info(),
permisisoned_reward_info.to_account_info(),
ctx.accounts.system_program.to_account_info(),
],
)?;
}
token_manager.issuer = ctx.accounts.issuer.key();
token_manager.recipient_token_account = ctx.accounts.token_manager_token_account.key();
token_manager.state = TokenManagerState::Issued as u8;
token_manager.state_changed_at = Clock::get().unwrap().unix_timestamp;
match token_manager.kind {
k if k == TokenManagerKind::Programmable as u8 => {
let mint_info = next_account_info(remaining_accs)?;
let mint_metadata_info = next_account_info(remaining_accs)?;
let mint_edition_info = next_account_info(remaining_accs)?;
let issuer_token_record_info = next_account_info(remaining_accs)?;
let token_manager_token_record_info = next_account_info(remaining_accs)?;
let sysvar_instructions_info = next_account_info(remaining_accs)?;
let associated_token_program_info = next_account_info(remaining_accs)?;
let authorization_rules_program_info = next_account_info(remaining_accs)?;
let authorization_rules_info = next_account_info(remaining_accs)?;
invoke(
&TransferV1 {
token: ctx.accounts.issuer_token_account.key(),
token_owner: ctx.accounts.issuer_token_account.owner.key(),
destination_token: ctx.accounts.token_manager_token_account.key(),
destination_owner: token_manager.key(),
mint: mint_info.key(),
metadata: mint_metadata_info.key(),
edition: Some(mint_edition_info.key()),
token_record: Some(issuer_token_record_info.key()),
destination_token_record: Some(token_manager_token_record_info.key()),
authority: ctx.accounts.issuer.key(),
payer: ctx.accounts.payer.key(),
system_program: ctx.accounts.system_program.key(),
sysvar_instructions: sysvar_instructions_info.key(),
spl_token_program: ctx.accounts.token_program.key(),
spl_ata_program: associated_token_program_info.key(),
authorization_rules_program: Some(authorization_rules_program_info.key()),
authorization_rules: Some(authorization_rules_info.key()),
}
.instruction(TransferV1InstructionArgs {
amount: token_manager.amount,
authorization_data: None,
}),
&[
ctx.accounts.issuer_token_account.to_account_info(),
ctx.accounts.issuer.to_account_info(),
ctx.accounts.token_manager_token_account.to_account_info(),
ctx.accounts.token_manager.to_account_info(),
mint_info.to_account_info(),
mint_metadata_info.to_account_info(),
mint_edition_info.to_account_info(),
issuer_token_record_info.to_account_info(),
token_manager_token_record_info.to_account_info(),
ctx.accounts.issuer.to_account_info(),
ctx.accounts.payer.to_account_info(),
ctx.accounts.system_program.to_account_info(),
sysvar_instructions_info.to_account_info(),
ctx.accounts.token_program.to_account_info(),
associated_token_program_info.to_account_info(),
authorization_rules_program_info.to_account_info(),
authorization_rules_info.to_account_info(),
],
)?;
}
_ => {
let cpi_accounts = Transfer {
from: ctx.accounts.issuer_token_account.to_account_info(),
to: ctx.accounts.token_manager_token_account.to_account_info(),
authority: ctx.accounts.issuer.to_account_info(),
};
let cpi_program = ctx.accounts.token_program.to_account_info();
let cpi_context = CpiContext::new(cpi_program, cpi_accounts);
token::transfer(cpi_context, token_manager.amount)?;
}
}
Ok(())
}