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

//! Registry to help the frontend quickly locate all active quarries.

use anchor_lang::prelude::*;
use anchor_lang::Key;
use quarry_mine::Quarry;
use quarry_mine::Rewarder;

mod account_validators;

solana_program::declare_id!("QREGBnEj9Sa5uR91AV8u3FxThgP5ZCvdZUW2bHAkfNc");

#[program]
pub mod quarry_registry {
    use vipers::validate::Validate;

    use super::*;

    /// Provisions a new registry for a [Rewarder].
    ///
    /// # Arguments
    ///
    /// * `max_quarries` - The maximum number of quarries that can be held in the registry.
    /// * `space` - The amount of space to provision for the [Registry], in bytes. Since a [Pubkey] is 32 bytes, this should be about `8 + 1 + 32 + 32 * max_quarries`.
    /// * `bump` - Bump seed.
    pub fn new_registry(
        ctx: Context<NewRegistry>,
        max_quarries: u16,
        _space: u16,
        bump: u8,
    ) -> ProgramResult {
        ctx.accounts.validate()?;
        let registry = &mut ctx.accounts.registry;
        registry.bump = bump;
        registry.rewarder = ctx.accounts.rewarder.key();
        registry
            .tokens
            .resize(max_quarries as usize, Pubkey::default());
        Ok(())
    }

    /// Synchronizes a [Quarry]'s token mint with the registry of its [Rewarder].
    pub fn sync_quarry(ctx: Context<SyncQuarry>) -> ProgramResult {
        ctx.accounts.validate()?;
        let quarry = &ctx.accounts.quarry;
        let registry = &mut ctx.accounts.registry;
        registry.tokens[quarry.index as usize] = quarry.token_mint_key;
        Ok(())
    }
}

/// Accounts for [quarry_registry::new_registry].
#[derive(Accounts)]
#[instruction(space: u16, bump: u8)]
pub struct NewRegistry<'info> {
    /// [Rewarder].
    pub rewarder: CpiAccount<'info, Rewarder>,

    /// [Rewarder] of mines.
    #[account(
        init,
        seeds = [
            b"QuarryRegistry".as_ref(),
            rewarder.key().to_bytes().as_ref(),
            &[bump],
        ],
        payer = payer,
        space = space as usize
    )]
    pub registry: ProgramAccount<'info, Registry>,

    /// Payer of the [Registry] initialization.
    #[account(signer)]
    pub payer: AccountInfo<'info>,

    /// System program.
    pub system_program: AccountInfo<'info>,
}

/// Accounts for [quarry_registry::sync_quarry].
#[derive(Accounts)]
#[instruction(bump: u8)]
pub struct SyncQuarry<'info> {
    /// [Quarry] to sync.
    pub quarry: CpiAccount<'info, Quarry>,
    /// [Registry] to write to.
    #[account(mut)]
    pub registry: ProgramAccount<'info, Registry>,
}

/// The [Registry] of all token mints associated with a [Rewarder].
#[account]
#[derive(Default, Debug)]
pub struct Registry {
    /// Bump seed
    pub bump: u8,
    /// Rewarder
    pub rewarder: Pubkey,
    /// Tokens
    pub tokens: Vec<Pubkey>,
}