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

use {
    anchor_lang::{prelude::*, AnchorDeserialize},
    std::convert::TryFrom,
};

const SEED_CONFIG: &[u8] = b"config";

static DEFAULT_CRANK_FEE: u64 = 1_000;

/// The config object, recording the config values of a specific Clockwork queue program deployment.
#[account]
#[derive(Debug)]
pub struct Config {
    /// The admin of the deployed program. This value is initialized to whomever calls `initialize`
    /// first after deployment. The admin may redelegate authority to another account.
    pub admin: Pubkey,

    /// The fee paid out to workers by users per successful crank.
    pub crank_fee: u64,

    /// The public address of the worker pool.
    pub worker_pool: Pubkey,
}

impl Config {
    /// Derive the pubkey of the config account.
    pub fn pubkey() -> Pubkey {
        Pubkey::find_program_address(&[SEED_CONFIG], &crate::ID).0
    }
}

impl TryFrom<Vec<u8>> for Config {
    type Error = Error;
    fn try_from(data: Vec<u8>) -> std::result::Result<Self, Self::Error> {
        Config::try_deserialize(&mut data.as_slice())
    }
}

/// The mutable config settings.
#[derive(AnchorSerialize, AnchorDeserialize)]
pub struct ConfigSettings {
    /// The admin authority of the deployed program.
    pub admin: Pubkey,
    /// The fee paid out to workers by users per successful crank.
    pub crank_fee: u64,
    /// The public address of the worker pool.
    pub worker_pool: Pubkey,
}

/// Trait for reading and writing to the config account.
pub trait ConfigAccount {
    /// Initialize the account to hold config object.
    fn init(&mut self, admin: Pubkey, worker_pool: Pubkey) -> Result<()>;

    /// Updates the config object.
    fn update(&mut self, settings: ConfigSettings) -> Result<()>;
}

impl ConfigAccount for Account<'_, Config> {
    fn init(&mut self, admin: Pubkey, worker_pool: Pubkey) -> Result<()> {
        self.admin = admin;
        self.crank_fee = DEFAULT_CRANK_FEE;
        self.worker_pool = worker_pool;
        Ok(())
    }

    fn update(&mut self, settings: ConfigSettings) -> Result<()> {
        self.admin = settings.admin;
        self.crank_fee = settings.crank_fee;
        self.worker_pool;
        Ok(())
    }
}