primitivo-macro 0.0.1

Shared ownership and pausable helpers for Primitivo programs
Documentation 
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 anchor_lang::prelude::*;

#[error_code]
pub enum PausableError {
    #[msg("Only owner can perform this action")]
    NotOwner,
    #[msg("Program is paused")]
    ProgramPaused,
    #[msg("Program is already paused")]
    AlreadyPaused,
    #[msg("Program is already unpaused")]
    AlreadyUnpaused,
}

#[derive(AnchorSerialize, AnchorDeserialize, Clone, InitSpace)]
pub struct Pausable {
    pub paused: bool,
}

impl Pausable {
    pub fn new() -> Self {
        Self { paused: false }
    }

    pub fn pause(&mut self, owner: Pubkey, signer: Pubkey) -> Result<()> {
        require!(owner == signer, PausableError::NotOwner);
        require!(!self.paused, PausableError::AlreadyPaused);
        self.paused = true;
        Ok(())
    }

    pub fn unpause(&mut self, owner: Pubkey, signer: Pubkey) -> Result<()> {
        require!(owner == signer, PausableError::NotOwner);
        require!(self.paused, PausableError::AlreadyUnpaused);
        self.paused = false;
        Ok(())
    }

    pub fn require_not_paused(&self) -> Result<()> {
        require!(!self.paused, PausableError::ProgramPaused);
        Ok(())
    }
}