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

use std::cell::RefMut;

use anchor_lang::prelude::*;
use light_macros::pubkey;
use light_merkle_tree::{
    config::MerkleTreeConfig,
    constants::{sha256::ZERO_BYTES, ZeroBytes},
    hasher::Sha256,
    HashFunction, MerkleTree,
};

use crate::{impl_indexed_merkle_tree, utils::config::MERKLE_TREE_HEIGHT, MerkleTreeAuthority};

#[derive(Clone, Copy)]
pub struct EventMerkleTreeConfig {}

impl MerkleTreeConfig for EventMerkleTreeConfig {
    const ZERO_BYTES: ZeroBytes = ZERO_BYTES;
    const PROGRAM_ID: Pubkey = pubkey!("JA5cjkRJ1euVi9xLWsCJVzsRzEkT8vcC4rqw9sVAo5d6");
}

#[account(zero_copy)]
pub struct EventMerkleTree {
    pub merkle_tree_nr: u64,
    pub newest: u8,
    _padding: [u8; 7],
    pub merkle_tree: MerkleTree<Sha256, EventMerkleTreeConfig>,
}

impl_indexed_merkle_tree!(EventMerkleTree);

pub fn process_initialize_new_event_merkle_tree(
    merkle_tree: &mut RefMut<'_, EventMerkleTree>,
    merkle_tree_authority: &mut Account<'_, MerkleTreeAuthority>,
) {
    merkle_tree
        .merkle_tree
        .init(MERKLE_TREE_HEIGHT, HashFunction::Sha256);
    merkle_tree.merkle_tree_nr = merkle_tree_authority.event_merkle_tree_index;
    merkle_tree.newest = 1;

    merkle_tree_authority.event_merkle_tree_index += 1;
}