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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

use anchor_lang::{prelude::*, solana_program::instruction::Instruction, InstructionData};
use sablier_utils::thread::ThreadResponse;

use crate::{constants::*, state::*};

#[derive(Accounts)]
pub struct DeleteSnapshotProcessEntry<'info> {
    #[account(address = Config::pubkey())]
    pub config: AccountLoader<'info, Config>,

    #[account(
        address = Registry::pubkey(),
        constraint = !registry.locked
    )]
    pub registry: Account<'info, Registry>,

    #[account(
        mut,
        seeds = [
            SEED_SNAPSHOT,
            snapshot.id.to_be_bytes().as_ref(),
        ],
        bump,
        constraint = snapshot.id < registry.current_epoch
    )]
    pub snapshot: Account<'info, Snapshot>,

    #[account(
        mut,
        seeds = [
            SEED_SNAPSHOT_ENTRY,
            snapshot_entry.snapshot_frame.as_ref(),
            snapshot_entry.id.to_be_bytes().as_ref(),
        ],
        bump,
        has_one = snapshot_frame
    )]
    pub snapshot_entry: Account<'info, SnapshotEntry>,

    #[account(
        mut,
        seeds = [
            SEED_SNAPSHOT_FRAME,
            snapshot_frame.snapshot.as_ref(),
            snapshot_frame.id.to_be_bytes().as_ref(),
        ],
        bump,
        has_one = snapshot,
    )]
    pub snapshot_frame: Account<'info, SnapshotFrame>,

    #[account(
        mut,
        address = config.load()?.epoch_thread
    )]
    pub thread: Signer<'info>,
}

pub fn handler(ctx: Context<DeleteSnapshotProcessEntry>) -> Result<ThreadResponse> {
    // Get accounts
    let config = &ctx.accounts.config;
    let registry = &ctx.accounts.registry;
    let snapshot = &mut ctx.accounts.snapshot;
    let snapshot_entry = &mut ctx.accounts.snapshot_entry;
    let snapshot_frame = &mut ctx.accounts.snapshot_frame;
    let thread = &mut ctx.accounts.thread;

    // Close the snapshot entry account.
    let snapshot_entry_lamports = snapshot_entry.get_lamports();
    snapshot_entry.sub_lamports(snapshot_entry_lamports)?;
    thread.add_lamports(snapshot_entry_lamports)?;

    // If this frame has no more entries, then close the frame account.
    if (snapshot_entry.id + 1) == snapshot_frame.total_entries {
        let snapshot_frame_lamports = snapshot_frame.get_lamports();
        snapshot_frame.sub_lamports(snapshot_frame_lamports)?;
        thread.add_lamports(snapshot_frame_lamports)?;

        // If this is also the last frame in the snapshot, then close the snapshot account.
        if (snapshot_frame.id + 1) == snapshot.total_frames {
            let snapshot_lamports = snapshot.get_lamports();
            snapshot.sub_lamports(snapshot_lamports)?;
            thread.add_lamports(snapshot_frame_lamports)?;
        }
    }

    // Build the next instruction
    let dynamic_instruction = if (snapshot_entry.id + 1) < snapshot_frame.total_entries {
        // Move on to the next entry.
        Some(
            Instruction {
                program_id: crate::ID,
                accounts: crate::accounts::DeleteSnapshotProcessEntry {
                    config: config.key(),
                    registry: registry.key(),
                    snapshot: snapshot.key(),
                    snapshot_entry: SnapshotEntry::pubkey(
                        snapshot_frame.key(),
                        snapshot_entry.id + 1,
                    ),
                    snapshot_frame: snapshot_frame.key(),
                    thread: thread.key(),
                }
                .to_account_metas(Some(true)),
                data: crate::instruction::DeleteSnapshotProcessEntry {}.data(),
            }
            .into(),
        )
    } else if (snapshot_frame.id + 1) < snapshot.total_frames {
        // This frame has no more entries. Move onto the next frame.
        Some(
            Instruction {
                program_id: crate::ID,
                accounts: crate::accounts::DeleteSnapshotProcessFrame {
                    config: config.key(),
                    registry: registry.key(),
                    snapshot: snapshot.key(),
                    snapshot_frame: SnapshotFrame::pubkey(snapshot.key(), snapshot_frame.id + 1),
                    thread: thread.key(),
                }
                .to_account_metas(Some(true)),
                data: crate::instruction::DeleteSnapshotProcessFrame {}.data(),
            }
            .into(),
        )
    } else {
        // This frame as no more entires and it was the last frame in the snapshot. We are done!
        None
    };

    Ok(ThreadResponse {
        dynamic_instruction,
        close_to: None,
        trigger: None,
    })
}