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

use crate::errors::ErrorCode;
use crate::state::*;
use anchor_lang::prelude::*;
use cardinal_token_manager::state::TokenManager;
use cardinal_token_manager::state::TokenManagerState;

#[derive(AnchorSerialize, AnchorDeserialize)]
pub struct InitIx {
    pub collector: Pubkey,
    pub payment_manager: Pubkey,
    pub duration_seconds: Option<i64>,
    pub extension_payment_amount: Option<u64>,
    pub extension_duration_seconds: Option<u64>,
    pub extension_payment_mint: Option<Pubkey>,
    pub max_expiration: Option<i64>,
    pub disable_partial_extension: Option<bool>,
}

#[derive(Accounts)]
pub struct InitCtx<'info> {
    #[account(constraint = token_manager.state == TokenManagerState::Initialized as u8 @ ErrorCode::InvalidTokenManager)]
    token_manager: Box<Account<'info, TokenManager>>,

    #[account(
        init_if_needed,
        payer = payer,
        space = TIME_INVALIDATOR_SIZE,
        seeds = [TIME_INVALIDATOR_SEED.as_bytes(), token_manager.key().as_ref()], bump,
    )]
    time_invalidator: Box<Account<'info, TimeInvalidator>>,

    #[account(mut, constraint = issuer.key() == token_manager.issuer @ ErrorCode::InvalidIssuer)]
    issuer: Signer<'info>,
    #[account(mut)]
    payer: Signer<'info>,
    system_program: Program<'info, System>,
}

pub fn handler(ctx: Context<InitCtx>, ix: InitIx) -> Result<()> {
    if ix.duration_seconds.is_none() && ix.max_expiration.is_none() {
        return Err(error!(ErrorCode::InvalidInstruction));
    } else if (ix.extension_payment_amount.is_none() && ix.extension_duration_seconds.is_some()) || (ix.extension_payment_amount.is_some() && ix.extension_duration_seconds.is_none()) {
        return Err(error!(ErrorCode::InvalidInstruction));
    } else if ix.extension_payment_amount.is_some() && ix.extension_payment_mint.is_none() {
        return Err(error!(ErrorCode::InvalidInstruction));
    }
    // discriminator check
    let time_invalidator = &mut ctx.accounts.time_invalidator;
    time_invalidator.bump = *ctx.bumps.get("time_invalidator").unwrap();
    time_invalidator.token_manager = ctx.accounts.token_manager.key();
    time_invalidator.collector = ix.collector;
    time_invalidator.payment_manager = ix.payment_manager;
    time_invalidator.expiration = None;
    time_invalidator.duration_seconds = ix.duration_seconds;
    time_invalidator.extension_payment_amount = ix.extension_payment_amount;
    time_invalidator.extension_duration_seconds = ix.extension_duration_seconds;
    time_invalidator.extension_payment_mint = ix.extension_payment_mint;
    time_invalidator.max_expiration = ix.max_expiration;
    time_invalidator.disable_partial_extension = ix.disable_partial_extension;
    Ok(())
}