1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use crate::errors::ErrorCode;
use crate::state::*;
use anchor_lang::prelude::*;
use anchor_spl::token::Mint;
use anchor_spl::token::TokenAccount;

#[derive(AnchorSerialize, AnchorDeserialize)]
pub struct InitIx {
    pub amount: u64,
    pub kind: u8,
    pub invalidation_type: u8,
    pub num_invalidators: u8,
}

#[derive(Accounts)]
#[instruction(num_invalidators: u8)]
pub struct InitCtx<'info> {
    #[account(
        init_if_needed,
        payer = payer,
        seeds = [TOKEN_MANAGER_SEED.as_bytes(), mint.key().as_ref()], bump,
        space = token_manager_size(num_invalidators as usize),
    )]
    token_manager: Box<Account<'info, TokenManager>>,

    #[account(
        init_if_needed,
        payer = payer,
        seeds = [MINT_COUNTER_SEED.as_bytes(), mint.key().as_ref()], bump,
        space = MINT_COUNTER_SIZE,
    )]
    mint_counter: Box<Account<'info, MintCounter>>,
    mint: Box<Account<'info, Mint>>,

    #[account(mut)]
    issuer: Signer<'info>,
    #[account(mut)]
    payer: Signer<'info>,
    #[account(mut, constraint =
        issuer_token_account.owner == issuer.key()
        && issuer_token_account.mint == mint.key()
        && issuer_token_account.amount >= 1
        @ ErrorCode::InvalidIssuerTokenAccount
    )]
    issuer_token_account: Box<Account<'info, TokenAccount>>,
    system_program: Program<'info, System>,
}

pub fn handler(ctx: Context<InitCtx>, ix: InitIx) -> Result<()> {
    if ix.num_invalidators > MAX_INVALIDATORS {
        return Err(error!(ErrorCode::MaximumInvalidatorsReached));
    }
    let token_manager = &mut ctx.accounts.token_manager;
    if token_manager.state != TokenManagerState::Initialized as u8 {
        return Err(error!(ErrorCode::InvalidTokenManagerState));
    }
    if token_manager.num_invalidators != 0 && ix.num_invalidators >= token_manager.num_invalidators {
        return Err(error!(ErrorCode::InvalidNumInvalidators));
    }
    if ctx.accounts.mint.supply > 1 {
        return Err(error!(ErrorCode::InvalidMintSupply));
    }
    if ix.kind != TokenManagerKind::Managed as u8
        && ix.kind != TokenManagerKind::Unmanaged as u8
        && ix.kind != TokenManagerKind::Edition as u8
        && ix.kind != TokenManagerKind::Permissioned as u8
        && ix.kind != TokenManagerKind::Programmable as u8
    {
        return Err(error!(ErrorCode::InvalidTokenManagerKind));
    }
    if ix.invalidation_type != InvalidationType::Return as u8
        && ix.invalidation_type != InvalidationType::Invalidate as u8
        && ix.invalidation_type != InvalidationType::Release as u8
        && ix.invalidation_type != InvalidationType::Reissue as u8
        && ix.invalidation_type != InvalidationType::Vest as u8
    {
        return Err(error!(ErrorCode::InvalidInvalidationType));
    }

    if ix.kind == TokenManagerKind::Permissioned as u8 {
        let mint = &ctx.accounts.mint;
        let mint_key = mint.key();
        let path = &[MINT_MANAGER_SEED.as_bytes(), mint_key.as_ref()];
        let (mint_manager_key, _bump) = Pubkey::find_program_address(path, ctx.program_id);
        if mint.mint_authority.is_none() || mint.mint_authority.unwrap() != mint_manager_key {
            return Err(error!(ErrorCode::InvalidMintAuthority));
        }
        if mint.freeze_authority.is_none() || mint.freeze_authority.unwrap() != mint_manager_key {
            return Err(error!(ErrorCode::InvalidFreezeAuthority));
        }
    }

    // Unamanged must use invalidate
    if ix.kind == TokenManagerKind::Unmanaged as u8 && ix.invalidation_type != InvalidationType::Invalidate as u8 {
        return Err(error!(ErrorCode::InvalidInvalidationType));
    }

    let mint_counter = &mut ctx.accounts.mint_counter;
    mint_counter.bump = *ctx.bumps.get("mint_counter").unwrap();
    mint_counter.count = mint_counter.count.checked_add(1).expect("Addition error");
    mint_counter.mint = ctx.accounts.mint.key();

    token_manager.bump = *ctx.bumps.get("token_manager").unwrap();
    token_manager.count = mint_counter.count;
    token_manager.num_invalidators = ix.num_invalidators;
    token_manager.issuer = ctx.accounts.issuer.key();
    token_manager.mint = ctx.accounts.mint.key();
    token_manager.state = TokenManagerState::Initialized as u8;
    token_manager.state_changed_at = Clock::get().unwrap().unix_timestamp;
    token_manager.claim_approver = None;
    token_manager.invalidators = Vec::new();
    token_manager.amount = ix.amount;
    token_manager.kind = ix.kind;
    token_manager.invalidation_type = ix.invalidation_type;

    // default to itself to avoid someone not setting it
    token_manager.transfer_authority = Some(token_manager.key());
    Ok(())
}