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

use crate::account::auth_address;
use crate::error::TransactionError;
use crate::transaction::{SignedLogic, SignedTransaction, Transaction, TransactionSignature};
use algonaut_core::{Address, CompiledTeal, LogicSignature};
use serde::{Deserialize, Serialize};

/// Convenience CompiledTeal "view", used to sign as contract account.
/// The program hash is interpreted as an address.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ContractAccount {
    address: Address,
    pub program: CompiledTeal,
}

impl ContractAccount {
    pub fn new(program: CompiledTeal) -> ContractAccount {
        ContractAccount {
            address: program.hash().into(),
            program,
        }
    }

    pub fn address(&self) -> &Address {
        &self.address
    }

    pub fn sign(
        &self,
        transaction: Transaction,
        args: Vec<Vec<u8>>,
    ) -> Result<SignedTransaction, TransactionError> {
        let auth_address = auth_address(&transaction, &self.address);
        Ok(SignedTransaction {
            transaction_id: transaction.id()?,
            transaction,
            sig: TransactionSignature::Logic(SignedLogic {
                logic: self.program.clone(),
                args,
                sig: LogicSignature::ContractAccount,
            }),
            auth_address,
        })
    }
}